3D case, removing meshgrid and using mgrid instead

Author: ccluri

utility_functions.py

@@ -84,10 +84,8 @@ def distribute_srcs_2D(X, Y, n_src, ext_x, ext_y, R_init):
     [nx, ny, Lx_nn, Ly_nn, ds] = get_src_params_2D(Lx_n, Ly_n, n_src)
     ext_x_n = (Lx_nn - Lx)/2
     ext_y_n = (Ly_nn - Ly)/2
-
     X_src, Y_src = np.mgrid[(np.min(X) - ext_x_n):(np.max(X) + ext_x_n):np.complex(0,nx),
                             (np.min(Y) - ext_y_n):(np.max(Y) + ext_y_n):np.complex(0,ny)]
-
     d = round(R_init/ds)
     R = d * ds
     return X_src, Y_src, R
@@ -164,10 +162,9 @@ def distribute_srcs_3D(X, Y, Z, n_src, ext_x, ext_y, ext_z, R_init):
     ext_x_n = (Lx_nn - Lx)/2
     ext_y_n = (Ly_nn - Ly)/2
     ext_z_n = (Lz_nn - Lz)/2
-    lin_x = np.linspace(np.min(X) - ext_x_n, np.max(X) + ext_x_n, nx)
-    lin_y = np.linspace(np.min(Y) - ext_y_n, np.max(Y) + ext_y_n, ny)
-    lin_z = np.linspace(np.min(Z) - ext_z_n, np.max(Z) + ext_z_n, nz)
-    (X_src, Y_src, Z_src) = np.meshgrid(lin_x, lin_y, lin_z)
+    X_src, Y_src, Z_src = np.mgrid[(np.min(X) - ext_x_n):(np.max(X) + ext_x_n):np.complex(0,nx),
+                                   (np.min(Y) - ext_y_n):(np.max(Y) + ext_y_n):np.complex(0,ny),
+                                   (np.min(Z) - ext_z_n):(np.max(Z) + ext_z_n):np.complex(0,nz)]
     d = np.round(R_init/ds)
     R = d * ds
     return (X_src, Y_src, Z_src, R)