resqpy 4.18.7__py3-none-any.whl → 4.18.9__py3-none-any.whl

resqpy/__init__.py

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

resqpy/grid/_grid.py

@@ -1498,7 +1498,7 @@ class Grid(BaseResqpy):
         """
         return z_corner_point_depths(self, order = order)
 
-    def frontier(self, set_z_zero = True, title = None, add_as_part = True):
+    def frontier(self, set_z_zero = True, title = None, add_as_part = True, mode = 'mean'):
         """Returns a frontier polygon (closed polyline) based on midpoints of edge coordinate lines, with z set to zero.
         
         arguments:
@@ -1507,35 +1507,59 @@ class Grid(BaseResqpy):
             - title (str, optional): the citation title for the polyline; if None, one is generated using the grid title
             - add_as_part (bool default True): if True, the xml is created for the polyline and it is added as a part to
               the model; if False, the create_xml() method is not called fot the polyline
-              
+            - mode (str, default 'mean'): one of 'mean', 'top', 'base', 'inner', 'outer' determining how a non-vertical
+              edge coordinate line is converted to a point to include in the frontier
+
         returns:
             - closed Polyline representing the frontier of the grid in plan view
         """
+
+        def get_point(clep, mode, centrum):
+            if mode == 'mean':
+                return np.nanmean(clep, axis = 0)
+            if mode == 'top':
+                return clep[0]
+            if mode == 'base':
+                return clep[-1]
+            assert mode in ['inner', 'outer'], f'unrecognised frontier mode: {mode}'
+            vp = clep - np.expand_dims(centrum, axis = 0)
+            dp = np.sum(vp * vp, axis = -1)
+            if (dp[0] < dp[1] and mode == 'inner') or (dp[0] > dp[1] and mode == 'outer'):
+                return clep[0]
+            return clep[-1]
+
+        assert mode in ['mean', 'top', 'base', 'inner', 'outer']
         coords = self.coordinate_line_end_points()
+        centrum = None
+        if mode in ['inner', 'outer']:
+            centrum = np.nanmean(np.concatenate((coords[0, 0], coords[0, -1], coords[-1, 0], coords[-1, -1]), axis = 0),
+                                 axis = 0)
+            assert not np.any(
+                np.isnan(centrum)), f'trying to make frontier polygon in mode {mode} when corners of grid not defined'
         nj = self.nj
         ni = self.ni
         frontier = np.zeros((2 * (nj + ni), 3), dtype = float)
         f_i = 0
         for i in range(self.ni):
-            c = np.nanmean(coords[0, i], axis = 0)
+            c = get_point(coords[0, i], mode, centrum)
             if np.any(np.isnan(c)):
                 continue
             frontier[f_i] = c
             f_i += 1
         for j in range(self.nj):
-            c = np.nanmean(coords[j, ni], axis = 0)
+            c = get_point(coords[j, ni], mode, centrum)
             if np.any(np.isnan(c)):
                 continue
             frontier[f_i] = c
             f_i += 1
         for i in range(self.ni, 0, -1):
-            c = np.nanmean(coords[nj, i], axis = 0)
+            c = get_point(coords[nj, i], mode, centrum)
             if np.any(np.isnan(c)):
                 continue
             frontier[f_i] = c
             f_i += 1
         for j in range(self.nj, 0, -1):
-            c = np.nanmean(coords[j, 0], axis = 0)
+            c = get_point(coords[j, 0], mode, centrum)
             if np.any(np.isnan(c)):
                 continue
             frontier[f_i] = c

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.
 

{resqpy-4.18.7.dist-info → resqpy-4.18.9.dist-info}/METADATA

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

{resqpy-4.18.7.dist-info → resqpy-4.18.9.dist-info}/RECORD

@@ -1,4 +1,4 @@
-resqpy/__init__.py,sha256=8ToDgLcOS7bQ6zpqngyQ90B4ZH_HoYCRRMhZRyyblYs,556
+resqpy/__init__.py,sha256=lB-7ZcQp9dDxWaqMRVMU8EvWJuaaUZH9Pp6m7NuwNiQ,556
 resqpy/crs.py,sha256=R7DfcTP5xGv5pu9Y8RHA2WVM9DjBCSVMoHcz4RmQ7Yw,27646
 resqpy/derived_model/__init__.py,sha256=NFvMSOKI3cxmH7lAbddV43JjoUj-r2G7ExEfOqinD1I,1982
 resqpy/derived_model/_add_edges_per_column_property_array.py,sha256=cpW3gwp6MSYIrtvFmCjoJXcyUsgGuCDbgmwlJCJebUs,6410
@@ -34,7 +34,7 @@ resqpy/grid/_defined_geometry.py,sha256=QYQ3wLbPrlPobgUi9R1izTD4JD9qMGf5eyqbM68H
 resqpy/grid/_extract_functions.py,sha256=pOVC2vv4dJ9G7e6p3uh2mz4QPb6W7_pKiaq8HAWeyzg,28263
 resqpy/grid/_face_functions.py,sha256=0I7O6DDz7nJWczi_G2bE3L2XUr4acxREwKygXWEp6F4,16516
 resqpy/grid/_faults.py,sha256=OmukVoLpdrndqDxwE6Rj7Ul5tj3FUQVPhE0raH2FHpg,12236
-resqpy/grid/_grid.py,sha256=vjLZJoHfvlCi8vFY09_1xQEBMRnCR6AbFFs88leKrxA,134892
+resqpy/grid/_grid.py,sha256=d4hSTmtdfmuEiQIjOhSAAlJZNlYBf0Fui4PPGJE5I4U,136144
 resqpy/grid/_grid_types.py,sha256=YAmRrrm-J2HKuuwCRCAkE3ppcQLiwc4EEc6dc-sy9Fc,4362
 resqpy/grid/_intervals_info.py,sha256=ODjDz22n8U9pSpO0Muj8mJr2hYWauFDzgcVQ0RM3csQ,338
 resqpy/grid/_moved_functions.py,sha256=XboxA0pE55j-P_x5g051WheVamxkAatQGbU5aq2GkaE,604
@@ -163,7 +163,7 @@ resqpy/surface/_base_surface.py,sha256=LsWrDrbuuaEVRgf2Dlbc-6ZvGQpjtrKuxF7Jjebvl
 resqpy/surface/_combined_surface.py,sha256=8TnNbSywjej6tW_vRr5zoVgBbvnadCaqWk6WyHWHTYQ,3082
 resqpy/surface/_mesh.py,sha256=yEFldNWT2g8MCGcU4mTeWzDrLHHGLLGLIle1gAjJ_lg,42352
 resqpy/surface/_pointset.py,sha256=niTkBik9hAvqrY8340K1TRG7mg4FMQbbp12WZiiXPMs,27416
-resqpy/surface/_surface.py,sha256=yDU-5MdVtrRUeRU2X1AFn4dYqpO-D1ydbX49HIBVKIo,71697
+resqpy/surface/_surface.py,sha256=pGVpZiIoaZm7UA26VnEEthMdKr3XtNIIhWFV8mOVDXY,73491
 resqpy/surface/_tri_mesh.py,sha256=EmV4FhyjuusQFruW1SseufbnHF5YFoJ6Uvb07UJbH6s,26609
 resqpy/surface/_tri_mesh_stencil.py,sha256=eXt_HIKvsXGsjQ7nm_NbozR6ProQxPbeO52r79j80ig,16087
 resqpy/surface/_triangulated_patch.py,sha256=FKn_Irzp4aLFkkN_-tx1MLMKjEAiOLE8636sOA481TQ,26802
@@ -194,7 +194,7 @@ resqpy/well/_wellbore_marker_frame.py,sha256=xvYH2_2Ie3a18LReFymbUrZboOx7Rhv5DOD
 resqpy/well/blocked_well_frame.py,sha256=Rx8jwkCjchseDZaTttPkA1-f6l7W6vRGrxWtDHlEPx8,22560
 resqpy/well/well_object_funcs.py,sha256=tWufc8wahihzMEO-Ou1dncIttrf4bNo1qmLgh3I2pOM,24717
 resqpy/well/well_utils.py,sha256=zwpYjT85nXAwWBhYB1Pygu2SgouZ-44k6hEOnpoMfBI,5969
-resqpy-4.18.7.dist-info/LICENSE,sha256=2duHPIkKQyESMdQ4hKjL8CYEsYRHXaYxt0YQkzsUYE4,1059
-resqpy-4.18.7.dist-info/METADATA,sha256=0jTMGWkS-S4La3Q1k-uxQds81bNPDHmgzk1mgbVnM-A,4028
-resqpy-4.18.7.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-resqpy-4.18.7.dist-info/RECORD,,
+resqpy-4.18.9.dist-info/LICENSE,sha256=2duHPIkKQyESMdQ4hKjL8CYEsYRHXaYxt0YQkzsUYE4,1059
+resqpy-4.18.9.dist-info/METADATA,sha256=cgQ2tIjpkdQz_Eu2ccGfKDuF_zcmf6K2QIZmL4R8oIs,4028
+resqpy-4.18.9.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+resqpy-4.18.9.dist-info/RECORD,,