remove old mapdata.render package
This commit is contained in:
Laura Klünder
parent
46dc1627e1
commit
f42d4dbeaf
5 changed files with 1 additions and 284 deletions
|
|
@ -1,23 +0,0 @@
|
|||
def render_all_levels(show_accessibles=False):
|
||||
from c3nav.mapdata.models.section import Section
|
||||
from c3nav.mapdata.render.renderer import LevelRenderer # noqa
|
||||
|
||||
renderers = []
|
||||
for level in Section.objects.all():
|
||||
renderers.append(LevelRenderer(level, only_public=False))
|
||||
renderers.append(LevelRenderer(level, only_public=True))
|
||||
|
||||
for renderer in renderers:
|
||||
renderer.render_base(show_accessibles=show_accessibles)
|
||||
|
||||
for renderer in renderers:
|
||||
if not renderer.level.intermediate:
|
||||
renderer.render_simple()
|
||||
|
||||
for renderer in renderers:
|
||||
if not renderer.level.intermediate:
|
||||
renderer.render_full()
|
||||
|
||||
for renderer in renderers:
|
||||
if not renderer.level.intermediate:
|
||||
renderer.render_segments()
|
||||
|
|
@ -1,260 +0,0 @@
|
|||
import os
|
||||
import subprocess
|
||||
import xml.etree.ElementTree as ET
|
||||
|
||||
from shapely.geometry import JOIN_STYLE, box
|
||||
|
||||
from c3nav.mapdata.inclusion import get_maybe_invisible_areas
|
||||
from c3nav.mapdata.utils.misc import get_dimensions, get_public_private_area, get_render_dimensions, get_render_path
|
||||
|
||||
|
||||
class LevelRenderer():
|
||||
def __init__(self, level, only_public):
|
||||
self.level = level
|
||||
self.only_public = only_public
|
||||
|
||||
self.geometries = self.get_geometries(level)
|
||||
|
||||
def get_geometries(self, level):
|
||||
return level.public_geometries if self.only_public else level.geometries
|
||||
|
||||
def get_filename(self, mode, filetype, level=None):
|
||||
return get_render_path(filetype, self.level.name if level is None else level, mode, self.only_public)
|
||||
|
||||
def create_svg(self):
|
||||
width, height = get_render_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',
|
||||
'xmlns:xlink': 'http://www.w3.org/1999/xlink',
|
||||
})
|
||||
return svg
|
||||
|
||||
def add_svg_content(self, svg):
|
||||
width, height = get_render_dimensions()
|
||||
contents = ET.Element('g', {
|
||||
'transform': 'scale(1 -1) translate(0 -%d)' % (height),
|
||||
})
|
||||
svg.append(contents)
|
||||
return contents
|
||||
|
||||
def add_svg_image(self, svg, image):
|
||||
width, height = get_render_dimensions()
|
||||
contents = ET.Element('image', {
|
||||
'x': '0',
|
||||
'y': '0',
|
||||
'width': str(width),
|
||||
'height': str(height),
|
||||
'xlink:href': 'file://'+os.path.abspath(image)
|
||||
})
|
||||
svg.append(contents)
|
||||
return contents
|
||||
|
||||
def render_base(self, png=True, show_accessibles=False):
|
||||
svg = self.create_svg()
|
||||
contents = self.add_svg_content(svg)
|
||||
|
||||
MITRE = JOIN_STYLE.mitre
|
||||
if not self.level.intermediate:
|
||||
width, height = get_dimensions()
|
||||
holes = self.geometries.holes.buffer(0.1, join_style=MITRE)
|
||||
contents.append(self.polygon_svg(box(0, 0, width, height).difference(holes),
|
||||
fill_color='#000000'))
|
||||
|
||||
contents.append(self.polygon_svg(self.geometries.buildings_with_holes,
|
||||
fill_color='#D5D5D5'))
|
||||
|
||||
contents.append(self.polygon_svg(self.geometries.outsides_with_holes,
|
||||
fill_color='#DCE6DC'))
|
||||
|
||||
for location in self.level.arealocations.all():
|
||||
color = None
|
||||
if location.color:
|
||||
color = location.color
|
||||
else:
|
||||
colorgroup = location.groups.filter(color__isnull=False).first()
|
||||
if colorgroup:
|
||||
color = colorgroup.color
|
||||
|
||||
if color:
|
||||
contents.append(self.polygon_svg(location.geometry.intersection(self.geometries.accessible),
|
||||
fill_color=color))
|
||||
|
||||
contents.append(self.polygon_svg(self.geometries.stuffedareas,
|
||||
fill_color='#000000',
|
||||
fill_opacity=0.03))
|
||||
|
||||
contents.append(self.polygon_svg(self.geometries.stair_areas,
|
||||
fill_color='#000000',
|
||||
fill_opacity=0.03))
|
||||
|
||||
contents.append(self.polygon_svg(self.geometries.stairs,
|
||||
stroke_color='#000000',
|
||||
stroke_width=0.06,
|
||||
stroke_opacity=0.2))
|
||||
|
||||
contents.append(self.polygon_svg(self.geometries.escalators,
|
||||
fill_color='#B3B3B3'))
|
||||
|
||||
contents.append(self.polygon_svg(self.geometries.walls_shadow,
|
||||
fill_color='#000000',
|
||||
fill_opacity=0.06))
|
||||
|
||||
if show_accessibles:
|
||||
narrowed_geometry = self.geometries.accessible.buffer(-0.6, join_style=MITRE)
|
||||
clear_geometry = self.geometries.accessible.buffer(-0.3, join_style=JOIN_STYLE.mitre)
|
||||
wide_geometry = narrowed_geometry.buffer(0.31, join_style=MITRE).intersection(clear_geometry)
|
||||
missing_geometry = clear_geometry.difference(wide_geometry.buffer(0.01, join_style=MITRE))
|
||||
|
||||
contents.append(self.polygon_svg(clear_geometry,
|
||||
fill_color='#FFFF00',
|
||||
fill_opacity=0.5))
|
||||
|
||||
contents.append(self.polygon_svg(narrowed_geometry,
|
||||
fill_color='#009900',
|
||||
fill_opacity=0.5))
|
||||
|
||||
contents.append(self.polygon_svg(missing_geometry,
|
||||
fill_color='#FF9900',
|
||||
fill_opacity=0.5))
|
||||
|
||||
contents.append(self.polygon_svg(self.geometries.elevatorlevels,
|
||||
fill_color='#9EF8FB'))
|
||||
|
||||
contents.append(self.polygon_svg(self.geometries.doors,
|
||||
fill_color='#FFFFFF',
|
||||
stroke_color='#3c3c3c',
|
||||
stroke_width=0.05))
|
||||
|
||||
contents.append(self.polygon_svg(self.geometries.uncropped_obstacles,
|
||||
fill_color='#A6A6A6',
|
||||
stroke_color='#919191',
|
||||
stroke_width=0.05))
|
||||
|
||||
contents.append(self.polygon_svg(self.geometries.cropped_obstacles.buffer(-0.06, join_style=MITRE),
|
||||
fill_color='#A6A6A6',
|
||||
stroke_color='#919191',
|
||||
stroke_width=0.05))
|
||||
|
||||
wider_escalators = self.geometries.escalators.buffer(0.3, join_style=MITRE)
|
||||
contents.append(self.polygon_svg(wider_escalators.intersection(self.geometries.uncropped_obstacles),
|
||||
fill_color='#666666',
|
||||
stroke_color='#666666',
|
||||
stroke_width=0.05))
|
||||
|
||||
contents.append(self.polygon_svg(self.geometries.walls,
|
||||
fill_color='#949494',
|
||||
stroke_color='#3c3c3c',
|
||||
stroke_width=0.05))
|
||||
|
||||
filename = self.get_filename('base', 'svg')
|
||||
with open(filename, 'w') as f:
|
||||
f.write(ET.tostring(svg).decode())
|
||||
|
||||
if png:
|
||||
png_filename = self.get_filename('base', 'png')
|
||||
subprocess.call(['rsvg-convert', filename, '-o', png_filename])
|
||||
|
||||
def render_simple(self, png=True):
|
||||
svg = self.create_svg()
|
||||
|
||||
dark_lower = []
|
||||
lower = []
|
||||
for level in self.level.lower():
|
||||
lower.append(level)
|
||||
if not level.intermediate:
|
||||
dark_lower.extend(lower)
|
||||
lower = []
|
||||
lower.append(self.level)
|
||||
|
||||
width, height = get_dimensions()
|
||||
contents = self.add_svg_content(svg)
|
||||
contents.append(self.polygon_svg(box(0, 0, width, height),
|
||||
fill_color='#000000'))
|
||||
|
||||
for level in dark_lower:
|
||||
self.add_svg_image(svg, self.get_filename('base', 'png', level=level))
|
||||
|
||||
contents = self.add_svg_content(svg)
|
||||
contents.append(self.polygon_svg(box(0, 0, width, height),
|
||||
fill_color='#000000',
|
||||
fill_opacity=0.05))
|
||||
|
||||
for level in lower:
|
||||
self.add_svg_image(svg, self.get_filename('base', 'png', level=level))
|
||||
|
||||
filename = self.get_filename('simple', 'svg')
|
||||
with open(filename, 'w') as f:
|
||||
f.write(ET.tostring(svg).decode())
|
||||
|
||||
if png:
|
||||
png_filename = self.get_filename('simple', 'png')
|
||||
subprocess.call(['rsvg-convert', filename, '-o', png_filename])
|
||||
|
||||
def render_full(self, png=True):
|
||||
svg = self.create_svg()
|
||||
|
||||
self.add_svg_image(svg, self.get_filename('simple', 'png'))
|
||||
|
||||
higher = []
|
||||
for level in self.level.higher():
|
||||
if not level.intermediate:
|
||||
break
|
||||
higher.append(level)
|
||||
|
||||
contents = self.add_svg_content(svg)
|
||||
for level in higher:
|
||||
contents.append(self.polygon_svg(self.get_geometries(level).intermediate_shadows,
|
||||
fill_color='#000000',
|
||||
fill_opacity=0.07))
|
||||
|
||||
for level in higher:
|
||||
self.add_svg_image(svg, self.get_filename('base', 'png', level=level))
|
||||
|
||||
filename = self.get_filename('full', 'svg')
|
||||
with open(filename, 'w') as f:
|
||||
f.write(ET.tostring(svg).decode())
|
||||
|
||||
if png:
|
||||
png_filename = self.get_filename('full', 'png')
|
||||
subprocess.call(['rsvg-convert', filename, '-o', png_filename])
|
||||
|
||||
def add_clippath(self, svg):
|
||||
defs = ET.Element('defs')
|
||||
svg.append(defs)
|
||||
clippath = ET.Element('clipPath', {
|
||||
'id': 'clip-path',
|
||||
})
|
||||
defs.append(clippath)
|
||||
return self.add_svg_content(clippath)
|
||||
|
||||
def render_segments(self, png=True):
|
||||
if self.only_public:
|
||||
return
|
||||
|
||||
for area in get_maybe_invisible_areas():
|
||||
svg = self.create_svg()
|
||||
|
||||
public_area, private_area = get_public_private_area(self.level)
|
||||
|
||||
if area.level == self.level:
|
||||
clippath = self.add_clippath(svg)
|
||||
|
||||
geometry = area.geometry.intersection(self.level.geometries.areas_and_doors)
|
||||
geometry = geometry.buffer(-0.1, join_style=JOIN_STYLE.mitre)
|
||||
geometry = geometry.intersection(private_area).buffer(1.5, join_style=JOIN_STYLE.mitre)
|
||||
polygon = self.polygon_svg(geometry)
|
||||
clippath.append(polygon)
|
||||
|
||||
image = self.add_svg_image(svg, self.get_filename('simple', 'png'))
|
||||
image.attrib['clip-path'] = 'url(#clip-path)'
|
||||
|
||||
filename = self.get_filename('full', area.location_id+'.svg')
|
||||
with open(filename, 'w') as f:
|
||||
f.write(ET.tostring(svg).decode())
|
||||
|
||||
if png:
|
||||
png_filename = self.get_filename('full', area.location_id+'.png')
|
||||
subprocess.call(['rsvg-convert', filename, '-o', png_filename])
|
||||
|
|
@ -1,114 +0,0 @@
|
|||
import re
|
||||
import xml.etree.ElementTree as ET
|
||||
from itertools import chain
|
||||
|
||||
from shapely.affinity import scale
|
||||
|
||||
|
||||
class SVGImage:
|
||||
def __init__(self, width: int, height: int, scale: float=1):
|
||||
self.width = width
|
||||
self.height = height
|
||||
self.scale = scale
|
||||
self.g = ET.Element('g', {})
|
||||
self.defs = ET.Element('defs')
|
||||
self.def_i = 0
|
||||
|
||||
blur_filter = ET.Element('filter', {'id': 'wallblur'})
|
||||
blur_filter.append(ET.Element('feGaussianBlur',
|
||||
{'in': 'SourceGraphic',
|
||||
'stdDeviation': str(int(0.7 * self.scale))}))
|
||||
self.defs.append(blur_filter)
|
||||
|
||||
def get_element(self):
|
||||
root = ET.Element('svg', {
|
||||
'width': str(self.width*self.scale),
|
||||
'height': str(self.height*self.scale),
|
||||
'xmlns:svg': 'http://www.w3.org/2000/svg',
|
||||
'xmlns': 'http://www.w3.org/2000/svg',
|
||||
'xmlns:xlink': 'http://www.w3.org/1999/xlink',
|
||||
})
|
||||
root.append(self.defs)
|
||||
root.append(self.g)
|
||||
return root
|
||||
|
||||
def get_xml(self):
|
||||
return ET.tostring(self.get_element()).decode()
|
||||
|
||||
def new_defid(self):
|
||||
defid = 's'+str(self.def_i)
|
||||
self.def_i += 1
|
||||
return defid
|
||||
|
||||
def _trim_decimals(self, data):
|
||||
return re.sub(r'([0-9]+)\.0', r'\1', re.sub(r'([0-9]+\.[0-9])[0-9]+', r'\1', data))
|
||||
|
||||
def _create_geometry(self, geometry):
|
||||
geometry = scale(geometry, xfact=1, yfact=-1, origin=(self.width / 2, self.height / 2))
|
||||
geometry = scale(geometry, xfact=self.scale, yfact=self.scale, origin=(0, 0))
|
||||
element = ET.fromstring(self._trim_decimals(geometry.svg(0, '#FFFFFF')))
|
||||
if element.tag != 'g':
|
||||
new_element = ET.Element('g')
|
||||
new_element.append(element)
|
||||
element = new_element
|
||||
|
||||
for elem in chain(element.findall('polyline'), element.findall('path')):
|
||||
elem.attrib.pop('opacity', None)
|
||||
elem.attrib.pop('fill', None)
|
||||
elem.attrib.pop('fill-rule', None)
|
||||
elem.attrib.pop('stroke', None)
|
||||
elem.attrib.pop('stroke-width', None)
|
||||
return element
|
||||
|
||||
def register_geometry(self, geometry, defid=None, as_clip_path=False, comment=None):
|
||||
if defid is None:
|
||||
defid = self.new_defid()
|
||||
|
||||
element = self._create_geometry(geometry)
|
||||
|
||||
if as_clip_path:
|
||||
element.tag = 'clipPath'
|
||||
element.set('id', defid)
|
||||
self.defs.append(element)
|
||||
return defid
|
||||
|
||||
def add_clip_path(self, *geometries, inverted=False, subtract=False, defid=None):
|
||||
if defid is None:
|
||||
defid = self.new_defid()
|
||||
|
||||
clippath = ET.Element('clipPath', {'id': defid})
|
||||
clippath.append(ET.Element('use', {'xlink:href': '#' + geometries[0]}))
|
||||
self.defs.append(clippath)
|
||||
return defid
|
||||
|
||||
def add_geometry(self, geometry=None, fill_color=None, fill_opacity=None, opacity=None, filter=None,
|
||||
stroke_width=0.0, stroke_color=None, stroke_opacity=None, stroke_linejoin=None, clip_path=None):
|
||||
if geometry is not None:
|
||||
if not geometry:
|
||||
return
|
||||
if isinstance(geometry, str):
|
||||
element = ET.Element('use', {'xlink:href': '#'+geometry})
|
||||
else:
|
||||
element = self._create_geometry(geometry)
|
||||
else:
|
||||
element = ET.Element('rect', {'width': '100%', 'height': '100%'})
|
||||
element.set('fill', fill_color or 'none')
|
||||
if fill_opacity:
|
||||
element.set('fill-opacity', str(fill_opacity)[:4])
|
||||
if stroke_width:
|
||||
element.set('stroke-width', self._trim_decimals(str(stroke_width * self.scale)))
|
||||
if stroke_color:
|
||||
element.set('stroke', stroke_color)
|
||||
if stroke_opacity:
|
||||
element.set('stroke-opacity', str(stroke_opacity)[:4])
|
||||
if stroke_linejoin:
|
||||
element.set('stroke-linejoin', stroke_linejoin)
|
||||
if opacity:
|
||||
element.set('opacity', str(opacity)[:4])
|
||||
if filter:
|
||||
element.set('filter', 'url(#'+filter+')')
|
||||
if clip_path:
|
||||
element.set('clip-path', 'url(#'+clip_path+')')
|
||||
|
||||
self.g.append(element)
|
||||
return element
|
||||
Loading…
Add table
Add a link
Reference in a new issue