get sources def

gprMax/grid.py

@@ -283,6 +283,9 @@ class FDTDGrid:
         # binary representation of floating point number.
         self.dt = round_value(self.dt, decimalplaces=d.getcontext().prec - 1)
     
+    def get_srcs(self):
+        return self.hertziandipoles + self.magneticdipoles + self.voltagesources + self.transmissionlines
+
 
 class CUDAGrid(FDTDGrid):
     """Additional grid methods for solving on GPU using CUDA."""