use HybridGeometry to precache triangulations

src/c3nav/mapdata/utils/geometry.py

@@ -5,7 +5,7 @@ from django.core import checks
 from matplotlib.patches import PathPatch
 from matplotlib.path import Path
 from shapely import speedups
-from shapely.geometry import LineString, Polygon
+from shapely.geometry import LinearRing, LineString, Polygon
 
 if speedups.available:
     speedups.enable()
@@ -91,3 +91,19 @@ def plot_geometry(geom, title=None, bounds=None):
     patch = PathPatch(path)
     axes.add_patch(patch)
     plt.show()
+
+
+def get_rings(geometry):
+    if isinstance(geometry, Polygon):
+        return chain((geometry.exterior, ), geometry.interiors)
+    try:
+        geoms = geometry.geoms
+    except AttributeError:
+        pass
+    else:
+        return chain(*(get_rings(geom) for geom in geoms))
+
+    if isinstance(geometry, LinearRing):
+        return (geometry, )
+
+    return ()

src/c3nav/mapdata/utils/mesh.py

@@ -14,6 +14,31 @@ def get_face_indizes(start, length):
     return np.vstack((indices, (indices[-1][-1], indices[0][0])))
 
 
+def triangulate_rings(rings, holes=None):
+    rings = tuple(tuple(tuple(vertex) for vertex in (np.array(ring.coords)*1000).astype(np.uint64)) for ring in rings)
+
+    if not rings:
+        return np.empty((0, 2), dtype=np.float32), np.empty((0, 3), dtype=np.int32)
+
+    vertices = tuple(set(chain(*rings)))
+    vertices_lookup = {vertex: i for i, vertex in enumerate(vertices)}
+
+    segments = set()
+    for ring in rings:
+        indices = tuple(vertices_lookup[vertex] for vertex in ring[:-1])
+        segments.update(zip(indices, indices[1:]+indices[:1]))
+
+    info = triangle.MeshInfo()
+    info.set_points(np.array(vertices).tolist())
+    info.set_facets(segments)
+
+    if holes is not None:
+        info.set_holes(holes)
+
+    mesh = triangle.build(info, quality_meshing=False)
+    return np.array(mesh.points)/1000, np.array(mesh.elements)
+
+
 def _triangulate_polygon(polygon: Polygon, keep_holes=False):
     vertices = deque()
     segments = deque()