resqpy 5.1.5__py3-none-any.whl → 5.1.6__py3-none-any.whl

resqpy/__init__.py

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

resqpy/grid/_extract_functions.py

@@ -445,7 +445,12 @@ def extract_stratigraphy(grid):
                                 object = grid,
                                 array_attribute = 'stratigraphic_units',
                                 dtype = 'int')
-    assert len(grid.stratigraphic_units) == grid.nk_plus_k_gaps
+    if (len(grid.stratigraphic_units) != grid.nk_plus_k_gaps):
+        log.error("Unable to load Stratigraphy," +
+                  f" the number of stratigraphic units {len(grid.stratigraphic_units)}" +
+                  f" does not equal the number of layers plus gaps {grid.nk_plus_k_gaps}")
+        grid.stratigraphic_column_rank_uuid = None
+        grid.stratigraphic_units = None
 
 
 def extract_children(grid):

resqpy/grid_surface/_find_faces.py

@@ -1167,68 +1167,6 @@ def find_faces_to_represent_surface_regular_optimised(grid,
         else:
             return None
 
-    log.debug("converting face sets into grid connection set")
-    # NB: kji0 arrays in internal face protocol: used as cell_kji0 with polarity of 1
-    # property lists have elements replaced with sorted and filtered equivalents
-    gcs = rqf.GridConnectionSet.from_faces_indices(grid = grid,
-                                                   k_faces_kji0 = k_faces_kji0,
-                                                   j_faces_kji0 = j_faces_kji0,
-                                                   i_faces_kji0 = i_faces_kji0,
-                                                   remove_duplicates = not patchwork,
-                                                   k_properties = k_props,
-                                                   j_properties = j_props,
-                                                   i_properties = i_props,
-                                                   feature_name = name,
-                                                   feature_type = feature_type,
-                                                   create_organizing_objects_where_needed = True,
-                                                   title = title)
-    # log.debug('finished coversion to gcs')
-
-    # NB. following assumes faces have been added to gcs in a particular order!
-    all_tris = None
-    if return_triangles:
-        # log.debug('preparing triangles array')
-        k_triangles = np.empty((0,), dtype = np.int32) if k_props is None else k_props.pop(0)
-        j_triangles = np.empty((0,), dtype = np.int32) if j_props is None else j_props.pop(0)
-        i_triangles = np.empty((0,), dtype = np.int32) if i_props is None else i_props.pop(0)
-        all_tris = np.concatenate((k_triangles, j_triangles, i_triangles), axis = 0)
-        # log.debug(f'gcs count: {gcs.count}; all triangles shape: {all_tris.shape}')
-        assert all_tris.shape == (gcs.count,)
-
-    # NB. following assumes faces have been added to gcs in a particular order!
-    all_depths = None
-    if return_depths:
-        # log.debug('preparing depths array')
-        k_depths = np.empty((0,), dtype = np.float64) if k_props is None else k_props.pop(0)
-        j_depths = np.empty((0,), dtype = np.float64) if j_props is None else j_props.pop(0)
-        i_depths = np.empty((0,), dtype = np.float64) if i_props is None else i_props.pop(0)
-        all_depths = np.concatenate((k_depths, j_depths, i_depths), axis = 0)
-        # log.debug(f'gcs count: {gcs.count}; all depths shape: {all_depths.shape}')
-        assert all_depths.shape == (gcs.count,)
-
-    # NB. following assumes faces have been added to gcs in a particular order!
-    all_offsets = None
-    if return_offsets:
-        # log.debug('preparing offsets array')
-        k_offsets = np.empty((0,), dtype = np.float64) if k_props is None else k_props[0]
-        j_offsets = np.empty((0,), dtype = np.float64) if j_props is None else j_props[0]
-        i_offsets = np.empty((0,), dtype = np.float64) if i_props is None else i_props[0]
-        all_offsets = _all_offsets(grid.crs, k_offsets, j_offsets, i_offsets)
-        # log.debug(f'gcs count: {gcs.count}; all offsets shape: {all_offsets.shape}')
-        assert all_offsets.shape == (gcs.count,)
-
-    all_flange = None
-    if return_flange_bool:
-        # log.debug('preparing flange array')
-        flange_bool_uuid = surface.model.uuid(title = "flange bool",
-                                              obj_type = "DiscreteProperty",
-                                              related_uuid = surface.uuid)
-        assert (flange_bool_uuid is not None), f"No flange bool property found for surface: {surface.title}"
-        flange_bool = rqp.Property(surface.model, uuid = flange_bool_uuid)
-        flange_array = flange_bool.array_ref(dtype = bool)
-        all_flange = np.take(flange_array, all_tris)
-        assert all_flange.shape == (gcs.count,)
-
     # note: following is a grid cells property, not a gcs property
     bisector = None
     if return_bisector:
@@ -1245,6 +1183,15 @@ def find_faces_to_represent_surface_regular_optimised(grid,
             bisector = column_bisector_from_face_indices((grid.nj, grid.ni), j_faces_ji0, i_faces_ji0)
             # log.debug('finished preparing columns bisector')
         elif patchwork:
+            # NB. following assumes faces have been added to gcs in a particular order!
+            all_tris = None
+            assert return_triangles
+            # log.debug('preparing triangles array')
+            k_triangles = np.empty((0,), dtype = np.int32) if k_props is None else k_props[0]
+            j_triangles = np.empty((0,), dtype = np.int32) if j_props is None else j_props[0]
+            i_triangles = np.empty((0,), dtype = np.int32) if i_props is None else i_props[0]
+            all_tris = np.concatenate((k_triangles, j_triangles, i_triangles), axis = 0)
+            # log.debug(f'gcs count: {gcs.count}; all triangles shape: {all_tris.shape}')
             n_patches = surface.number_of_patches()
             log.info(f'preparing composite cells bisector for surface: {surface.title}; number of patches: {n_patches}')
             nkf = 0 if k_faces_kji0 is None else len(k_faces_kji0)
@@ -1328,6 +1275,98 @@ def find_faces_to_represent_surface_regular_optimised(grid,
                 if is_curtain:
                     bisector = bisector[0]  # reduce to a columns property
 
+    # if using patchwork, filter all faces (and properties) to those within (or on boundary of) volume corresponding to patch
+    if patchwork:
+        if k_faces_kji0 is not None:
+            selection = filter_faces(k_faces_kji0, patch_indices_k, patch_indices, 0)
+            if np.any(selection):
+                k_faces_kji0 = k_faces_kji0[selection]
+                if k_props is not None:
+                    k_props = [prop[selection] for prop in k_props]
+            else:
+                k_faces_kji0 = None
+                k_props = None
+        if j_faces_kji0 is not None:
+            selection = filter_faces(j_faces_kji0, patch_indices_j, patch_indices, 1)
+            if np.any(selection):
+                j_faces_kji0 = j_faces_kji0[selection]
+                if j_props is not None:
+                    j_props = [prop[selection] for prop in j_props]
+            else:
+                j_faces_kji0 = None
+                j_props = None
+        if i_faces_kji0 is not None:
+            selection = filter_faces(i_faces_kji0, patch_indices_i, patch_indices, 2)
+            if np.any(selection):
+                i_faces_kji0 = i_faces_kji0[selection]
+                if i_props is not None:
+                    i_props = [prop[selection] for prop in i_props]
+            else:
+                i_faces_kji0 = None
+                i_props = None
+
+    log.debug("converting face sets into grid connection set")
+    # NB: kji0 arrays in internal face protocol: used as cell_kji0 with polarity of 1
+    # property lists have elements replaced with sorted and filtered equivalents
+    gcs = rqf.GridConnectionSet.from_faces_indices(grid = grid,
+                                                   k_faces_kji0 = k_faces_kji0,
+                                                   j_faces_kji0 = j_faces_kji0,
+                                                   i_faces_kji0 = i_faces_kji0,
+                                                   remove_duplicates = not patchwork,
+                                                   k_properties = k_props,
+                                                   j_properties = j_props,
+                                                   i_properties = i_props,
+                                                   feature_name = name,
+                                                   feature_type = feature_type,
+                                                   create_organizing_objects_where_needed = True,
+                                                   title = title)
+    # log.debug('finished coversion to gcs')
+
+    # NB. following assumes faces have been added to gcs in a particular order!
+    all_tris = None
+    if return_triangles:
+        # log.debug('preparing triangles array')
+        k_triangles = np.empty((0,), dtype = np.int32) if k_props is None else k_props.pop(0)
+        j_triangles = np.empty((0,), dtype = np.int32) if j_props is None else j_props.pop(0)
+        i_triangles = np.empty((0,), dtype = np.int32) if i_props is None else i_props.pop(0)
+        all_tris = np.concatenate((k_triangles, j_triangles, i_triangles), axis = 0)
+        # log.debug(f'gcs count: {gcs.count}; all triangles shape: {all_tris.shape}')
+        assert all_tris.shape == (gcs.count,)
+
+    # NB. following assumes faces have been added to gcs in a particular order!
+    all_depths = None
+    if return_depths:
+        # log.debug('preparing depths array')
+        k_depths = np.empty((0,), dtype = np.float64) if k_props is None else k_props.pop(0)
+        j_depths = np.empty((0,), dtype = np.float64) if j_props is None else j_props.pop(0)
+        i_depths = np.empty((0,), dtype = np.float64) if i_props is None else i_props.pop(0)
+        all_depths = np.concatenate((k_depths, j_depths, i_depths), axis = 0)
+        # log.debug(f'gcs count: {gcs.count}; all depths shape: {all_depths.shape}')
+        assert all_depths.shape == (gcs.count,)
+
+    # NB. following assumes faces have been added to gcs in a particular order!
+    all_offsets = None
+    if return_offsets:
+        # log.debug('preparing offsets array')
+        k_offsets = np.empty((0,), dtype = np.float64) if k_props is None else k_props[0]
+        j_offsets = np.empty((0,), dtype = np.float64) if j_props is None else j_props[0]
+        i_offsets = np.empty((0,), dtype = np.float64) if i_props is None else i_props[0]
+        all_offsets = _all_offsets(grid.crs, k_offsets, j_offsets, i_offsets)
+        # log.debug(f'gcs count: {gcs.count}; all offsets shape: {all_offsets.shape}')
+        assert all_offsets.shape == (gcs.count,)
+
+    all_flange = None
+    if return_flange_bool:
+        # log.debug('preparing flange array')
+        flange_bool_uuid = surface.model.uuid(title = "flange bool",
+                                              obj_type = "DiscreteProperty",
+                                              related_uuid = surface.uuid)
+        assert (flange_bool_uuid is not None), f"No flange bool property found for surface: {surface.title}"
+        flange_bool = rqp.Property(surface.model, uuid = flange_bool_uuid)
+        flange_array = flange_bool.array_ref(dtype = bool)
+        all_flange = np.take(flange_array, all_tris)
+        assert all_flange.shape == (gcs.count,)
+
     # note: following is a grid cells property, not a gcs property
     shadow = None
     if return_shadow:
@@ -2475,7 +2514,7 @@ def _bitwise_count_njit(a: np.ndarray) -> np.int64:
     return c
 
 
-@njit
+@njit  # pragma: no cover
 def box_intersection(box_a: np.ndarray, box_b: np.ndarray) -> np.ndarray:
     """Return a box which is the intersection of two boxes, python protocol; all zeros if no intersection."""
     box = np.zeros((2, 3), dtype = np.int32)
@@ -2486,8 +2525,8 @@ def box_intersection(box_a: np.ndarray, box_b: np.ndarray) -> np.ndarray:
     return box
 
 
-@njit
-def get_box(mask: np.ndarray) -> Tuple[np.ndarray, int]:  # pragma: no cover
+@njit  # pragma: no cover
+def get_box(mask: np.ndarray) -> Tuple[np.ndarray, int]:
     """Returns a python protocol box enclosing True elements of 3D boolean mask, and count which is zero if all False."""
     box = np.full((2, 3), -1, dtype = np.int32)
     count = 0
@@ -2517,3 +2556,28 @@ def get_box(mask: np.ndarray) -> Tuple[np.ndarray, int]:  # pragma: no cover
                             box[1, 2] = i + 1
                     count += 1
     return box, count
+
+
+@njit  # pragma: no cover
+def filter_faces(faces_kji0: np.ndarray, face_patches: np.ndarray, cell_patches: np.ndarray, axis: int) -> np.ndarray:
+    """Return 1D boolean selection array indicating subset of faces that are applicable to cells with matching patch."""
+    n: int = len(faces_kji0)
+    assert len(face_patches) == n
+    selection = np.zeros(n, dtype = np.bool_)
+    for f in range(n):
+        k: int = faces_kji0[f, 0]
+        j: int = faces_kji0[f, 1]
+        i: int = faces_kji0[f, 2]
+        if face_patches[f] == cell_patches[k, j, i]:
+            selection[f] = True
+        else:
+            if axis == 0:
+                if face_patches[f] == cell_patches[k + 1, j, i]:
+                    selection[f] = True
+            elif axis == 1:
+                if face_patches[f] == cell_patches[k, j + 1, i]:
+                    selection[f] = True
+            else:
+                if face_patches[f] == cell_patches[k, j, i + 1]:
+                    selection[f] = True
+    return selection

{resqpy-5.1.5.dist-info → resqpy-5.1.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: resqpy
-Version: 5.1.5
+Version: 5.1.6
 Summary: Python API for working with RESQML models
 License: MIT
 Keywords: RESQML

{resqpy-5.1.5.dist-info → resqpy-5.1.6.dist-info}/RECORD

@@ -1,4 +1,4 @@
-resqpy/__init__.py,sha256=9OUOfmvTwC90nqrnAXAqmHSuL_COKkd2jhWcHGcMyLs,555
+resqpy/__init__.py,sha256=cnR7ZBD-Sv8zdt5JWWw7n0iqZIaIhsYS7akdPjFzG10,555
 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
@@ -31,7 +31,7 @@ resqpy/grid/_cell_properties.py,sha256=pbyAK2eV9n4teOxm2q5hyBinohEbevFPrCfMcpGiq
 resqpy/grid/_connection_sets.py,sha256=-277bh9pMoeESSzy9oZehL-vc82aMGZuSLQs2KJ4Wfg,10120
 resqpy/grid/_create_grid_xml.py,sha256=p-jKo1BZ-DlhEYixVzadtY-QsqA8ygcXvMYS_TvQCjg,22837
 resqpy/grid/_defined_geometry.py,sha256=QYQ3wLbPrlPobgUi9R1izTD4JD9qMGf5eyqbM68Hg-0,32297
-resqpy/grid/_extract_functions.py,sha256=pOVC2vv4dJ9G7e6p3uh2mz4QPb6W7_pKiaq8HAWeyzg,28263
+resqpy/grid/_extract_functions.py,sha256=zfB8rLnt1qdz627BOiM4igXQKbCnsM8fV0HMbnLusTk,28585
 resqpy/grid/_face_functions.py,sha256=0I7O6DDz7nJWczi_G2bE3L2XUr4acxREwKygXWEp6F4,16516
 resqpy/grid/_faults.py,sha256=OmukVoLpdrndqDxwE6Rj7Ul5tj3FUQVPhE0raH2FHpg,12236
 resqpy/grid/_grid.py,sha256=-OhJpi9x8r74_7_CBrQsanqhFiZQcCiJkvbnhPRq2qQ,135758
@@ -47,7 +47,7 @@ resqpy/grid/_write_nexus_corp.py,sha256=yEVfiObsedEAXX6UG6ZTf56kZnQVkd3lLqE2NpL-
 resqpy/grid/_xyz.py,sha256=RLQWOdM_DRoCj4JypwB5gUJ78HTdk5JnZHSeAzuU634,13087
 resqpy/grid_surface/__init__.py,sha256=zSbyDwUH_p0UXd9PAcAXvcS1wrmRVnZvXXa1zQSQBlY,2925
 resqpy/grid_surface/_blocked_well_populate.py,sha256=Lme1AR-nLWOUlNnmHMVThk6jEg_lAZxWWtL82Yksppw,35867
-resqpy/grid_surface/_find_faces.py,sha256=cvxtMaHM5NENdMFdyeHzMEr_WOLWxh1YeVEtaLAla1Q,117517
+resqpy/grid_surface/_find_faces.py,sha256=sBzy9mWQZNU41kE5v3taVNDKnqzTcvnOT2FVAXABP0c,120573
 resqpy/grid_surface/_grid_skin.py,sha256=D0cjHkcuT5KCKb-8EZfXgh0GgJj3kzOBS2wVNXg4bfY,26056
 resqpy/grid_surface/_grid_surface.py,sha256=l2NJo7Kiucolbb_TlLPC7NGdksg_JahkihfsrJVq99w,14379
 resqpy/grid_surface/_trajectory_intersects.py,sha256=Och9cZYU9Y7ofovhPzsLyIblRUl2xj9_5nHH3fMZp-A,22498
@@ -193,7 +193,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=1O4EVPuTn-kN3uT_V4TbSwehnMUMY0TX36XOUgasTcc,24689
 resqpy/well/well_utils.py,sha256=-g_pg2v5XD9g4SQz9sk7KK-x2xEQZHzWehCQqiEGo6M,7627
-resqpy-5.1.5.dist-info/LICENSE,sha256=2duHPIkKQyESMdQ4hKjL8CYEsYRHXaYxt0YQkzsUYE4,1059
-resqpy-5.1.5.dist-info/METADATA,sha256=q-3olpPpXHhKu34E49vbo7WTdmBh9MmDgfsvWHZcDBw,3986
-resqpy-5.1.5.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-resqpy-5.1.5.dist-info/RECORD,,
+resqpy-5.1.6.dist-info/LICENSE,sha256=2duHPIkKQyESMdQ4hKjL8CYEsYRHXaYxt0YQkzsUYE4,1059
+resqpy-5.1.6.dist-info/METADATA,sha256=ixIzugCD9ZynPEJP_6VZiiUrn042oJmOx8HoilUgePw,3986
+resqpy-5.1.6.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+resqpy-5.1.6.dist-info/RECORD,,