From 3db6ac0181b9fd086f9480a848dd61d0d9cf9f2a Mon Sep 17 00:00:00 2001
From: nmannall <n.mannall@epcc.ed.ac.uk>
Date: Fri, 17 May 2024 10:55:29 +0100
Subject: [PATCH] Refactor Rx UserObject build process

---
 gprMax/cmds_multiuse.py | 9 +++++----
 gprMax/receivers.py     | 2 +-
 2 files changed, 6 insertions(+), 5 deletions(-)

diff --git a/gprMax/cmds_multiuse.py b/gprMax/cmds_multiuse.py
index b3607d1d..e4af7acb 100644
--- a/gprMax/cmds_multiuse.py
+++ b/gprMax/cmds_multiuse.py
@@ -79,7 +79,7 @@ class UserObjectMulti(ABC):
         """Creates object and adds it to model."""
         pass
 
-    # TODO: Check if this is actually needed
+    # TODO: Make _do_rotate not use a grid object
     def rotate(self, axis, angle, origin=None):
         """Rotates object (specialised for each object)."""
         pass
@@ -942,13 +942,14 @@ class Rx(UserObjectMulti):
         except KeyError:
             pass
 
-    def build(self, grid, uip):
+    def build(self, model, uip):
         try:
             p1 = self.kwargs["p1"]
         except KeyError:
             logger.exception(self.params_str())
             raise
 
+        grid = uip.grid
         if self.do_rotate:
             self._do_rotate(grid)
 
@@ -967,7 +968,7 @@ class Rx(UserObjectMulti):
             r.ID = f"{r.__class__.__name__}({str(r.xcoord)},{str(r.ycoord)},{str(r.zcoord)})"
             for key in RxUser.defaultoutputs:
                 r.outputs[key] = np.zeros(
-                    grid.iterations, dtype=config.sim_config.dtypes["float_or_double"]
+                    model.iterations, dtype=config.sim_config.dtypes["float_or_double"]
                 )
         else:
             outputs.sort()
@@ -980,7 +981,7 @@ class Rx(UserObjectMulti):
             for field in outputs:
                 if field in allowableoutputs:
                     r.outputs[field] = np.zeros(
-                        grid.iterations, dtype=config.sim_config.dtypes["float_or_double"]
+                        model.iterations, dtype=config.sim_config.dtypes["float_or_double"]
                     )
                 else:
                     logger.exception(
diff --git a/gprMax/receivers.py b/gprMax/receivers.py
index de3ed87e..e78344d7 100644
--- a/gprMax/receivers.py
+++ b/gprMax/receivers.py
@@ -29,7 +29,7 @@ class Rx:
     allowableoutputs_dev = allowableoutputs[:-3]
 
     def __init__(self):
-        self.ID = None
+        self.ID: str
         self.outputs = {}
         self.xcoord: int
         self.ycoord: int