resqpy 4.18.8__tar.gz → 4.18.9__tar.gz

{resqpy-4.18.8 → resqpy-4.18.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: resqpy
-Version: 4.18.8
+Version: 4.18.9
 Summary: Python API for working with RESQML models
 Home-page: https://github.com/bp/resqpy
 License: MIT

{resqpy-4.18.8 → resqpy-4.18.9}/pyproject.toml

@@ -9,7 +9,7 @@ build-backend = "poetry.masonry.api"
 
 [tool.poetry]
 name = "resqpy"
-version = "4.18.8" # Set at build time
+version = "4.18.9" # Set at build time
 description = "Python API for working with RESQML models"
 authors = ["BP"]
 license = "MIT"

{resqpy-4.18.8 → resqpy-4.18.9}/resqpy/__init__.py

@@ -28,6 +28,6 @@
 
 import logging
 
-__version__ = "4.18.8"  # Set at build time
+__version__ = "4.18.9"  # Set at build time
 log = logging.getLogger(__name__)
 log.info(f"Imported resqpy version {__version__}")

{resqpy-4.18.8 → resqpy-4.18.9}/resqpy/surface/_surface.py

@@ -1169,6 +1169,33 @@ class Surface(rqsb.BaseSurface):
 
         return resampled
 
+    def resample_surface_unique_edges(self):
+        """Returns a new surface, with the same model, title and crs as the original surface, but with additional refined points along original surface tears and edges. 
+        
+        Each edge forming a tear or outer edge in the surface will have 3 additional points added, with 2 additional points on each edge of the original triangle. The output surface is re-triangulated using these new points (tears will be filled)
+
+        returns: 
+            resqpy.surface.Surface object with extra_metadata ('unique edges resampled from surface': uuid), where uuid is for the original surface uuid
+        """
+        _, op = self.triangles_and_points()
+        ref = self.resampled_surface()  # resample the original surface
+        rt, rp = ref.triangles_and_points()
+        de, dc = ref.distinct_edges_and_counts()  # find the distinct edges and counts for the resampled surface
+        de_edge = de[dc == 1]  # find edges that only appear once - tears or surface edges
+        edge_tri_index = np.sum(np.isin(rt, de_edge), axis = 1) == 2
+        edge_tris = rp[rt[edge_tri_index]]
+        mid = np.mean(rp[de_edge], axis = 1)  # get the midpoint of each surface edge
+        edge_ref_points = np.unique(np.concatenate([op, edge_tris.reshape(-1, 3), mid]), axis = 0)  # combine all points
+
+        points = rqs.PointSet(self.model, points_array = edge_ref_points, title = self.title,
+                              crs_uuid = self.crs_uuid)  # generate a pointset from these points
+
+        output = Surface(self.model, point_set = points,
+                         extra_metadata = {'resampled from surface': str(self.uuid)
+                                          })  # return a surface with generated from these points
+
+        return output
+
     def write_hdf5(self, file_name = None, mode = 'a'):
         """Create or append to an hdf5 file, writing datasets for the triangulated patches after caching arrays.