fix: added function for plotting structural frames

plots data and -1,0,1 isovalues for each surface
Loop3D · Feb 7, 2022 · 69dbc6d · 69dbc6d
1 parent 9706f25
commit 69dbc6d
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/LoopStructural/visualisation/model_plotter.py b/LoopStructural/visualisation/model_plotter.py
@@ -535,6 +535,12 @@ def add_fault(self, fault, step=100):
         self.add_isosurface(fault, value=0, name=fault.name)
         self.add_vector_field(fault, locations=self.model.regular_grid()[::step])
 
+    def add_structural_frame(self, frame, step=100, data=True, **kwargs):
+        for i in range(3):
+            self.add_isosurface(frame[i], slices=[-1,0,1],**kwargs)
+            if data:
+               self.add_data(frame[i])
+
     def unfault_grid(self, feature, grid=None):
         if grid is None:
             grid = self.model.regular_grid()