completely rewrite map renderer

2016-11-26 13:48:44 +01:00 · 2016-11-26 13:48:44 +01:00 · b76cfa1985
commit b76cfa1985
parent 3c01b620fa
6 changed files with 138 additions and 38 deletions
--- a/src/c3nav/mapdata/render/renderer.py
+++ b/src/c3nav/mapdata/render/renderer.py
@ -0,0 +1,98 @@
+import os
+import xml.etree.ElementTree as ET
+
+from django.conf import settings
+from django.db.models import Max, Min
+from shapely.affinity import scale
+from shapely.geometry import box
+
+from c3nav.mapdata.models import Package
+
+
+class LevelRenderer():
+    def __init__(self, level):
+        self.level = level
+
+    @staticmethod
+    def get_dimensions():
+        aggregate = Package.objects.all().aggregate(Max('right'), Min('left'), Max('top'), Min('bottom'))
+        return (
+            (aggregate['right__max'] - aggregate['left__min']) * 10,
+            (aggregate['top__max'] - aggregate['bottom__min']) * 10
+        )
+
+    @staticmethod
+    def polygon_svg(geometry, fill_color=None, stroke_width=0.0, stroke_color=None, filter=None):
+        element = ET.fromstring(scale(geometry, xfact=10, yfact=10, origin=(0, 0)).svg(0, fill_color or '#FFFFFF'))
+        if element.tag != 'g':
+            new_element = ET.Element('g')
+            new_element.append(element)
+            element = new_element
+
+        for path in element.findall('path'):
+            path.attrib.pop('opacity')
+            path.set('stroke-width', str(stroke_width))
+
+            if fill_color is None and 'fill' in path.attrib:
+                path.attrib.pop('fill')
+                path.set('fill-opacity', '0')
+
+            if stroke_color is not None:
+                path.set('stroke', stroke_color)
+            elif 'stroke' in path.attrib:
+                path.attrib.pop('stroke')
+
+            if filter is not None:
+                path.set('filter', filter)
+
+        return element
+
+    def get_svg(self):
+        width, height = self.get_dimensions()
+
+        svg = ET.Element('svg', {
+            'width': str(width),
+            'height': str(height),
+            'xmlns:svg': 'http://www.w3.org/2000/svg',
+            'xmlns': 'http://www.w3.org/2000/svg',
+        })
+
+        svg.append(ET.fromstring("""
+        <filter id="area-filter" x="-50%" y="-50%" width="200%" height="200%">
+            <feComponentTransfer in="SourceAlpha">
+                <feFuncA type="table" tableValues="1 0" />
+            </feComponentTransfer>
+            <feGaussianBlur stdDeviation="15" result="offsetblur"/>
+            <feFlood flood-color="#B9B9B9" result="color"/>
+            <feComposite in2="offsetblur" operator="in"/>
+            <feComposite in2="SourceAlpha" operator="in" />
+            <feMerge>
+                <feMergeNode in="SourceGraphic" />
+                <feMergeNode />
+            </feMerge>
+        </filter>"""))
+
+        contents = ET.Element('g', {
+            'transform': 'scale(1 -1) translate(0 -%d)' % (height),
+        })
+        svg.append(contents)
+
+        contents.append(self.polygon_svg(box(0, 0, width, height), fill_color='#000000'))
+
+        contents.append(self.polygon_svg(self.level.geometries.buildings,
+                                         fill_color='#949494',
+                                         stroke_color='#757575',
+                                         stroke_width=1.5))
+
+        contents.append(self.polygon_svg(self.level.geometries.areas,
+                                         fill_color='#D5D5D5',
+                                         filter='url(#area-filter)'))
+        contents.append(self.polygon_svg(self.level.geometries.areas,
+                                         stroke_color='#757575',
+                                         stroke_width=1.5))
+        return ET.tostring(svg).decode()
+
+    def write_svg(self):
+        filename = os.path.join(settings.RENDER_ROOT, 'level-%s.svg' % self.level.name)
+        with open(filename, 'w') as f:
+            f.write(self.get_svg())