99 lines
3.6 KiB
Python
99 lines
3.6 KiB
Python
|
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())
|