fix: adding option to define custom mesh builder for pli

Loop3D · Oct 11, 2021 · b21fa8c · b21fa8c
1 parent f5a5f9b
commit b21fa8c
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/LoopStructural/modelling/core/geological_model.py b/LoopStructural/modelling/core/geological_model.py
@@ -588,7 +588,10 @@ def get_interpolator(self, interpolatortype='FDI', nelements=1e4,
                 if mesh_id not in self.support:
                     self.support[mesh_id] = mesh
             else:
-                mesh = TetMesh(origin=bb[0, :], nsteps=nsteps, step_vector=step_vector)
+                if 'meshbuilder' in kwargs:
+                    mesh = kwargs['meshbuilder'](bb,nelements)
+                else:
+                    mesh = TetMesh(origin=bb[0, :], nsteps=nsteps, step_vector=step_vector)
             logger.info("Creating regular tetrahedron mesh with %i elements \n"
                         "for modelling using PLI" % (mesh.ntetra))
 
@@ -635,8 +638,11 @@ def get_interpolator(self, interpolatortype='FDI', nelements=1e4,
             # number of steps is the length of the box / step vector
             nsteps = np.ceil((bb[1, :] - bb[0, :]) / step_vector).astype(int)
             # create a structured grid using the origin and number of steps
-            mesh = kwargs.get('mesh', TetMesh(origin=bb[0, :], nsteps=nsteps,
-                                              step_vector=step_vector))
+            if 'meshbuilder' in kwargs:
+                    mesh = kwargs['meshbuilder'].build(bb,nelements)
+            else:
+                mesh = kwargs.get('mesh', TetMesh(origin=bb[0, :], nsteps=nsteps,
+                                                step_vector=step_vector))
             logger.info("Creating regular tetrahedron mesh with %i elements \n"
                         "for modelling using DFI" % mesh.ntetra)
             return DFI(mesh, kwargs['fold'])