move everything from c3nav.routing.graph to c3nav.routing

src/c3nav/routing/level.py

@@ -0,0 +1,108 @@
+import os
+from itertools import permutations
+
+from django.conf import settings
+from PIL import Image, ImageDraw
+from shapely.geometry import JOIN_STYLE
+
+from c3nav.mapdata.utils import assert_multipolygon
+from c3nav.routing.point import GraphPoint
+from c3nav.routing.room import GraphRoom
+from c3nav.routing.utils.base import get_nearest_point
+from c3nav.routing.utils.draw import _ellipse_bbox, _line_coords
+
+
+class GraphLevel():
+    def __init__(self, graph, level):
+        self.graph = graph
+        self.level = level
+        self.points = []
+        self.rooms = []
+
+    def build(self):
+        self.collect_rooms()
+        self.create_points()
+
+    def collect_rooms(self):
+        accessibles = self.level.geometries.accessible
+        accessibles = assert_multipolygon(accessibles)
+        for geometry in accessibles:
+            room = GraphRoom(self, geometry)
+            if not room.empty:
+                self.rooms.append(room)
+
+    def create_points(self):
+        print('Level %s:' % self.level.name)
+        for room in self.rooms:
+            room.create_points()
+
+        doors = self.level.geometries.doors
+        doors = assert_multipolygon(doors)
+        for door in doors:
+            polygon = door.buffer(0.01, join_style=JOIN_STYLE.mitre)
+            center = door.centroid
+            points = []
+            for room in self.rooms:
+                if polygon.intersects(room.geometry):
+                    nearest_point = get_nearest_point(room.clear_geometry, center)
+                    point = GraphPoint(room, *nearest_point.coords[0])
+                    points.append(point)
+                    room.points.append(point)
+
+            if len(points) < 2:
+                print('door with <2 rooms (%d) detected at %s' % (len(points), center))
+
+            for from_point, to_point in permutations(points, 2):
+                from_point.connect_to(to_point)
+
+        for levelconnector in self.level.levelconnectors.all():
+            polygon = levelconnector.geometry
+            center = polygon.centroid
+
+            for room in self.rooms:
+                if not polygon.intersects(room.geometry):
+                    continue
+
+                point = center
+                if not point.within(room.clear_geometry):
+                    point = get_nearest_point(room.clear_geometry, point)
+
+                point = GraphPoint(room, *point.coords[0])
+                room.points.append(point)
+                self.graph.add_levelconnector_point(levelconnector, point)
+
+        for room in self.rooms:
+            room.connect_points()
+
+        self.points = sum((room.points for room in self.rooms), [])
+        print('%d points' % len(self.points))
+        print()
+
+    def draw_png(self, points=True, lines=True, transfer_points=False, transfer_lines=False):
+        filename = os.path.join(settings.RENDER_ROOT, 'level-%s.base.png' % self.level.name)
+        graph_filename = os.path.join(settings.RENDER_ROOT, 'level-%s.graph.png' % self.level.name)
+
+        im = Image.open(filename)
+        height = im.size[1]
+        draw = ImageDraw.Draw(im)
+        if lines:
+            for point in self.points:
+                for otherpoint, connection in point.connections.items():
+                    draw.line(_line_coords(point, otherpoint, height), fill=(255, 100, 100))
+
+        if points:
+            for point in self.points:
+                draw.ellipse(_ellipse_bbox(point.x, point.y, height), (200, 0, 0))
+
+        if transfer_lines:
+            for point in self.points:
+                if point.in_room_transfer_distances is not None:
+                    for otherpoint, distance in point.in_room_transfer_distances.items():
+                        draw.line(_line_coords(point, otherpoint, height), fill=(100, 100, 255))
+
+        if transfer_points:
+            for point in self.points:
+                if point.in_room_transfer_distances is not None:
+                    draw.ellipse(_ellipse_bbox(point.x, point.y, height), (0, 0, 200))
+
+        im.save(graph_filename)