fixes boundary generation

Author: EliHei2

src/segger/prediction/boundary.py

@@ -223,8 +223,7 @@ def calculate_part_2(self, plot=True):
                 simplex_id = list(edges[current_edge]["simplices"].keys())[0]
                 simplex = d.simplices[simplex_id]
                 if (
-                    edges[current_edge]["length"] > 1.5 * d_max
-                    and edges[current_edge]["simplices"][simplex_id] > 90
+                    edges[current_edge]["length"] > 1.5 * d_max and edges[current_edge]["simplices"][simplex_id] > 90
                 ) or edges[current_edge]["simplices"][simplex_id] > 180 - 180 / 16:
 
                     # delete edge and the simplex start
@@ -260,9 +259,7 @@ def find_cycles(self):
             if len(cycles) == 1:
                 geom = Polygon(self.d.points[cycles[0]])
             else:
-                geom = MultiPolygon(
-                    [Polygon(self.d.points[c]) for c in cycles if len(c) >= 3]
-                )
+                geom = MultiPolygon([Polygon(self.d.points[c]) for c in cycles if len(c) >= 3])
         except Exception as e:
             print(e, cycles)
             return None
@@ -332,13 +329,7 @@ def generate_boundaries(df, x="x_location", y="y_location", cell_id="segger_cell
     res = []
     group_df = df.groupby(cell_id)
     for cell_id, t in tqdm(group_df, total=group_df.ngroups):
-        res.append(
-            {
-                "cell_id": cell_id,
-                "length": len(t),
-                "geom": generate_boundary(t, x=x, y=y),
-            }
-        )
+        res.append({"cell_id": cell_id, "length": len(t), "geom": generate_boundary(t, x=x, y=y)})
 
     return gpd.GeoDataFrame(
         data=[[b["cell_id"], b["length"]] for b in res],
@@ -350,7 +341,6 @@ def generate_boundaries(df, x="x_location", y="y_location", cell_id="segger_cell
 def generate_boundary(t, x="x_location", y="y_location"):
     if len(t) < 3:
         return None
-
     bi = BoundaryIdentification(t[[x, y]].values)
     bi.calculate_part_1(plot=False)
     bi.calculate_part_2(plot=False)
@@ -359,24 +349,9 @@ def generate_boundary(t, x="x_location", y="y_location"):
     return geom
 
 
-if __name__ == "__main__":
-    points = np.array(
-        [
-            [0, 0],  # Point 0
-            [3, 0],  # Point 1
-            [0, 4],  # Point 2
-            [5, 5],  # Point 3
-            [1, 6],  # Point 4
-        ]
-    )
 
-    simplices = triangles = np.array(
-        [
-            [0, 1, 2],  # Triangle formed by points 0, 1, 2
-            [1, 3, 4],  # Triangle formed by points 1, 3, 4
-        ]
-    )
 
-    angles = triangle_angles_from_points(points, triangles)
-    print("Angles of each triangle (in degrees):")
-    print(angles)
+def extract_largest_polygon(geom):
+    if isinstance(geom, MultiPolygon):
+        return max(geom.geoms, key=lambda p: p.area)  # Keep the largest polygon
+    return geom  # Keep the original polygon if it's already a Polygon