halfedge strip

Author: tomvanmele

CHANGELOG.md

@@ -15,6 +15,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 * Added `redraw` flag to the `compas_rhino` methods `delete_object`, `delete_objects` and `purge_objects`.
 * Added the `__eq__` method for `compas.geometry.Circle` and `compas.geometry.Line`.
 * Added support for Pylance through static API definitions.
+* Added `halfedge_strip` method to `compas.datastructures.HalfEdge`.
 
 ### Changed
 

src/compas/datastructures/mesh/core/halfedge.py

@@ -2242,6 +2242,34 @@ def edge_strip(self, edge):
             edges.append((u, v))
         return edges
 
+    def halfedge_strip(self, edge):
+        """Find all edges on the same strip as a given halfedge.
+
+        Parameters
+        ----------
+        edge : tuple of int
+            The identifier of the starting edge.
+
+        Returns
+        -------
+        list of tuple of int
+            The edges on the same strip as the given halfedge.
+        """
+        u, v = edge
+        edges = [(u, v)]
+        while True:
+            face = self.halfedge[u][v]
+            if face is None:
+                break
+            vertices = self.face_vertices(face)
+            if len(vertices) != 4:
+                break
+            i = vertices.index(u)
+            u = vertices[i - 1]
+            v = vertices[i - 2]
+            edges.append((u, v))
+        return edges
+
     # --------------------------------------------------------------------------
     # face topology
     # --------------------------------------------------------------------------