fix: adding option to not store vertices for 2d unstructured supports to save memory

Author: lachlangrose

LoopStructural/interpolators/supports/_2d_base_unstructured.py

@@ -234,7 +234,12 @@ def evaluate_gradient(self, evaluation_points, property_array):
         return values
 
     def get_element_for_location(
-        self, points: np.ndarray
+        self,
+        points: np.ndarray,
+        return_verts=True,
+        return_bc=True,
+        return_inside=True,
+        return_tri=True,
     ) -> Tuple[np.ndarray, np.ndarray, np.ndarray, np.ndarray]:
         """
         Determine the elements from a numpy array of points
@@ -249,14 +254,18 @@ def get_element_for_location(
         -------
 
         """
-        verts = np.zeros((points.shape[0], self.dimension + 1, self.dimension))
+        if return_verts:
+            verts = np.zeros((points.shape[0], self.dimension + 1, self.dimension))
+        else:
+            verts = np.zeros((0, 0, 0))
         bc = np.zeros((points.shape[0], self.dimension + 1))
         tetras = np.zeros(points.shape[0], dtype="int64")
         inside = np.zeros(points.shape[0], dtype=bool)
         npts = 0
         npts_step = int(1e4)
         # break into blocks of 10k points
         while npts < points.shape[0]:
+            print(npts, npts_step, points.shape[0])
             cell_index, inside = self.aabb_grid.position_to_cell_index(
                 points[: npts + npts_step, :]
             )
@@ -266,6 +275,7 @@ def get_element_for_location(
             row = tetra_indices.row
             col = tetra_indices.col
             # using returned indexes calculate barycentric coords to determine which tetra the points are in
+
             vertices = self.nodes[self.elements[col, : self.dimension + 1]]
             pos = points[row, : self.dimension]
             row = tetra_indices.row
@@ -286,8 +296,8 @@ def get_element_for_location(
             c[:, 2] = 1.0 - c[:, 0] - c[:, 1]
 
             mask = np.all(c >= 0, axis=1)
-
-            verts[: npts + npts_step, :, :][row[mask], :, :] = vertices[mask, :, :]
+            if return_verts:
+                verts[: npts + npts_step, :, :][row[mask], :, :] = vertices[mask, :, :]
             bc[: npts + npts_step, :][row[mask], :] = c[mask, :]
             tetras[: npts + npts_step][row[mask]] = col[mask]
             inside[: npts + npts_step][row[mask]] = True
@@ -312,5 +322,5 @@ def get_element_gradient_for_location(
         -------
 
         """
-        verts, c, tri, inside = self.get_element_for_location(pos)
+        verts, c, tri, inside = self.get_element_for_location(pos, return_verts=False)
         return self.evaluate_shape_derivatives(pos, tri)

LoopStructural/interpolators/supports/_2d_p1_unstructured.py

@@ -61,7 +61,7 @@ def evaluate_shape_derivatives(self, locations, elements=None):
 
     def evaluate_shape(self, locations):
         locations = np.array(locations)
-        vertices, c, tri, inside = self.get_element_for_location(locations)
+        vertices, c, tri, inside = self.get_element_for_location(locations, return_verts=False)
         # c = np.dot(np.array([1,x,y]),np.linalg.inv(M)) # convert to barycentric coordinates
         # order of bary coord is (1-s-t,s,t)
         N = c  # np.zeros((c.shape[0],3)) #evaluate shape functions at barycentric coordinates