Accelerates the computation of the transport for boxes in regions

pycif/plugins/obsoperators/fp/obsoper.py

@@ -210,14 +210,14 @@ def obsoper(self, inputs, mode,
             
             
             # Transport for boxes in regions
-            hbox = np.zeros((nbox*ngrid), np.float64)
-
-            for n in range(ngrid):
-                for jy in range(fp_header_nest.numy):
-                    for ix in range(fp_header_nest.numx):
-                        if tracer.regions[jy, ix] > 0:
-                            hbox[n*nbox + tracer.regions[jy, ix] - 1] += \
-                            grid_nest[ix, jy, n]
+            hbox = np.zeros((ngrid,nbox), np.float64)
+            for jy in range(fp_header_nest.numy):
+                for ix in range(fp_header_nest.numx):
+                    if tracer.regions[jy, ix] > 0:
+                        hbox[:,tracer.regions[jy, ix] - 1] += \
+                            grid_nest[ix, jy, :ngrid]
+            hbox = hbox.flatten()
+
 
             # TODO: account for possible difference nested grid/inversion domain
             hnest = grid_nest