fix: implementing model.save method

Author: lachlangrose

LoopStructural/modelling/core/geological_model.py

@@ -111,7 +111,6 @@ def __init__(
 
 
         """
-        # print('tet')
         if logfile:
             self.logfile = logfile
             log_to_file(logfile, level=loglevel)
@@ -1800,18 +1799,18 @@ def get_stratigraphic_surfaces(self, units: List[str] = [], bottoms: bool = True
 
         return surfaces
 
-    def get_block_model(self):
-        grid = self.bounding_box.vtk()
+    def get_block_model(self, name='block model'):
+        grid = self.bounding_box.structured_grid(name=name)
 
-        grid.cell_data['stratigraphy'] = self.evaluate_model(
+        grid.properties_cell['stratigraphy'] = self.evaluate_model(
             self.bounding_box.cell_centers(), scale=False
         )
         return grid, self.stratigraphic_ids()
 
     def save(
         self,
         filename: str,
-        block_model: bool = False,
+        block_model: bool = True,
         stratigraphic_surfaces=True,
         fault_surfaces=True,
         stratigraphic_data=True,
@@ -1844,14 +1843,16 @@ def save(
             for group in self.stratigraphic_column.keys():
                 if group == "faults":
                     continue
-                for series in self.stratigraphic_column[group].keys():
+
+                for d in self.__getitem__(group).get_data():
                     if extension == ".geoh5":
-                        self.__getitem__(series).save(filename)
+                        d.save(filename)
                     else:
-                        self.__getitem__(series).save(f'{name}_{series}.{extension}')
+                        d.save(f'{name}_{group}.{extension}')
         if fault_data:
             for f in self.fault_names():
-                if extension == ".geoh5":
-                    self.__getitem__(f).save(filename)
-                else:
-                    self.__getitem__(f).save(f'{name}_{f}.{extension}')
+                for d in self.__getitem__(f).get_data():
+                    if extension == ".geoh5":
+                        d.save(filename)
+                    else:
+                        d.save(f'{name}_{group}.{extension}')