show geometries in editor… introducing the editor API

src/c3nav/editor/api.py

@@ -0,0 +1,92 @@
+from itertools import chain
+
+from django.utils import timezone
+from rest_framework.decorators import detail_route, list_route
+from rest_framework.exceptions import ValidationError
+from rest_framework.generics import get_object_or_404
+from rest_framework.response import Response
+from rest_framework.viewsets import ViewSet
+from shapely.ops import cascaded_union
+
+from c3nav.mapdata.models import Area, Section, Space
+
+
+class EditorViewSet(ViewSet):
+    """
+    Editor API
+    /geometries/ returns a list of geojson features, you have to specify ?section=<id> or ?space=<id>
+    /geometrystyles/ returns styling information for all geometry types
+    """
+    @list_route(methods=['get'])
+    def geometries(self, request, *args, **kwargs):
+        section = request.GET.get('section')
+        space = request.GET.get('space')
+        if section is not None:
+            if space is not None:
+                raise ValidationError('Only section or space can be specified.')
+            section = get_object_or_404(Section, pk=section)
+            holes = section.holes.all()
+            holes_geom = cascaded_union([hole.geometry for hole in holes])
+            buildings = section.buildings.all()
+            spaces = section.spaces.all()
+            spaces_geom = cascaded_union([space.geometry for space in spaces if space.level == ''])
+            holes_geom = holes_geom.intersection(spaces_geom)
+            doors = section.doors.all()
+            for obj in chain(buildings, (s for s in spaces if s.level == '')):
+                obj.geometry = obj.geometry.difference(holes_geom)
+
+            results = []
+
+            def add_spaces(level):
+                results.extend(space for space in spaces if space.level == level)
+                results.extend((area for area in Area.objects.filter(space__section=section, space__level=level)
+                                if area.get_color()))
+
+            add_spaces('lower')
+
+            results.extend(buildings)
+            for door in section.doors.all():
+                results.append(door)
+
+            add_spaces('')
+            add_spaces('upper')
+            return Response([obj.to_geojson() for obj in results])
+        elif space is not None:
+            space = get_object_or_404(Space, pk=space)
+            section = space.section
+
+            doors = [door for door in section.doors.all() if door.geometry.intersects(space.geometry)]
+            doors_geom = cascaded_union([door.geometry for door in doors])
+
+            spaces = [space for space in section.spaces.all() if space.geometry.intersects(doors_geom)]
+
+            results = []
+            results.extend(section.buildings.all())
+            results.extend(doors)
+            results.extend(spaces)
+
+            results.extend(chain(
+                space.areas.all(),
+                space.stairs.all(),
+                space.obstacles.all(),
+                space.lineobstacles.all(),
+                space.points.all(),
+            ))
+
+            return Response([obj.to_geojson() for obj in results])
+        else:
+            raise ValidationError('No section or space specified.')
+
+    @list_route(methods=['get'])
+    def geometrystyles(self, request, *args, **kwargs):
+        return Response({
+            'building': '#929292',
+            'space': '#d1d1d1',
+            'hole': 'rgba(255, 0, 0, 0.3)',
+            'door': '#ffffff',
+            'area': '#55aaff',
+            'step': '#ff0099',
+            'obstacle': '#999999',
+            'lineobstacle': '#999999',
+            'point': '#4488cc',
+        })