fix: adding groupname to save option to geoh5

Author: lachlangrose

LoopStructural/datatypes/_point.py

@@ -49,7 +49,7 @@ def plot(self, pyvista_kwargs={}):
         except ImportError:
             logger.error("pyvista is required for vtk")
 
-    def save(self, filename: Union[str, io.StringIO], ext=None):
+    def save(self, filename: Union[str, io.StringIO], *, group='Loop',ext=None):
         if isinstance(filename, io.StringIO):
             if ext is None:
                 raise ValueError('Please provide an extension for StringIO')
@@ -68,7 +68,7 @@ def save(self, filename: Union[str, io.StringIO], ext=None):
         elif ext == 'geoh5':
             from LoopStructural.export.geoh5 import add_points_to_geoh5
 
-            add_points_to_geoh5(filename, self)
+            add_points_to_geoh5(filename, self, groupname=group)
         elif ext == 'pkl':
             import pickle
 
@@ -189,7 +189,7 @@ def plot(self, pyvista_kwargs={}):
         except ImportError:
             logger.error("pyvista is required for vtk")
 
-    def save(self, filename):
+    def save(self, filename,*, group='Loop'):
         filename = str(filename)
         ext = filename.split('.')[-1]
         if ext == 'json':
@@ -203,7 +203,7 @@ def save(self, filename):
         elif ext == 'geoh5':
             from LoopStructural.export.geoh5 import add_points_to_geoh5
 
-            add_points_to_geoh5(filename, self)
+            add_points_to_geoh5(filename, self, groupname=group)
         elif ext == 'pkl':
             import pickle
 

LoopStructural/datatypes/_structured_grid.py

@@ -104,7 +104,7 @@ def merge(self, other):
         for name, data in other.properties.items():
             self.properties[name] = data
 
-    def save(self, filename):
+    def save(self, filename, *,group='Loop'):
         filename = str(filename)
         ext = filename.split('.')[-1]
         if ext == 'json':
@@ -118,7 +118,7 @@ def save(self, filename):
         elif ext == 'geoh5':
             from LoopStructural.export.geoh5 import add_structured_grid_to_geoh5
 
-            add_structured_grid_to_geoh5(filename, self)
+            add_structured_grid_to_geoh5(filename, self, groupname=group)
         elif ext == 'pkl':
             import pickle
 

LoopStructural/datatypes/_surface.py

@@ -192,7 +192,7 @@ def from_dict(cls, d, flatten=False):
             d.get('cell_properties', None),
         )
 
-    def save(self, filename, replace_spaces=True, ext=None):
+    def save(self, filename, *, group='Loop',replace_spaces=True, ext=None):
         filename = filename.replace(' ', '_') if replace_spaces else filename
         if isinstance(filename, (io.StringIO, io.BytesIO)):
             if ext is None:
@@ -225,7 +225,7 @@ def save(self, filename, replace_spaces=True, ext=None):
         elif ext == 'geoh5':
             from LoopStructural.export.geoh5 import add_surface_to_geoh5
 
-            add_surface_to_geoh5(filename, self)
+            add_surface_to_geoh5(filename, self, groupname=group)
 
         elif ext == 'pkl':
             import pickle