2016-11-27 14:03:39 +01:00
|
|
|
from collections import OrderedDict
|
2017-05-08 21:55:45 +02:00
|
|
|
|
2016-10-12 15:25:00 +02:00
|
|
|
from django.db import models
|
|
|
|
from django.utils.translation import ugettext_lazy as _
|
2017-05-07 12:06:13 +02:00
|
|
|
from shapely.geometry import CAP_STYLE, JOIN_STYLE, mapping
|
2016-10-12 15:25:00 +02:00
|
|
|
|
2017-05-08 16:05:44 +02:00
|
|
|
from c3nav.mapdata.fields import GeometryField
|
2017-05-08 16:40:22 +02:00
|
|
|
from c3nav.mapdata.models.geometry.base import GeometryMixin
|
2016-12-07 16:11:33 +01:00
|
|
|
from c3nav.mapdata.utils.json import format_geojson
|
2016-10-12 15:25:00 +02:00
|
|
|
|
2017-05-09 09:36:08 +02:00
|
|
|
SPACE_MODELS = OrderedDict()
|
2016-12-01 12:25:02 +01:00
|
|
|
|
|
|
|
|
2017-05-08 16:40:22 +02:00
|
|
|
class SpaceGeometryMixin(GeometryMixin):
|
2017-05-09 12:50:32 +02:00
|
|
|
space = models.ForeignKey('mapdata.Space', on_delete=models.CASCADE, verbose_name=_('space'))
|
|
|
|
|
|
|
|
class Meta:
|
|
|
|
abstract = True
|
|
|
|
|
2017-05-04 15:44:32 +02:00
|
|
|
def get_geojson_properties(self):
|
|
|
|
result = super().get_geojson_properties()
|
2017-05-06 17:40:06 +02:00
|
|
|
result['space'] = self.space.id
|
2017-05-04 15:44:32 +02:00
|
|
|
return result
|
|
|
|
|
|
|
|
|
2017-05-08 16:40:22 +02:00
|
|
|
class StuffedArea(SpaceGeometryMixin, models.Model):
|
2016-12-27 00:52:22 +01:00
|
|
|
"""
|
|
|
|
A slow area with many tables or similar. Avoid it from routing by slowing it a bit down
|
|
|
|
"""
|
2017-05-08 16:05:44 +02:00
|
|
|
geometry = GeometryField('polygon')
|
2016-12-27 00:52:22 +01:00
|
|
|
|
|
|
|
class Meta:
|
|
|
|
verbose_name = _('Stuffed Area')
|
|
|
|
verbose_name_plural = _('Stuffed Areas')
|
|
|
|
default_related_name = 'stuffedareas'
|
|
|
|
|
|
|
|
|
2017-05-08 16:40:22 +02:00
|
|
|
class Stair(SpaceGeometryMixin, models.Model):
|
2016-12-08 18:12:07 +01:00
|
|
|
"""
|
|
|
|
A stair
|
|
|
|
"""
|
2017-05-08 16:05:44 +02:00
|
|
|
geometry = GeometryField('polyline')
|
2017-05-04 12:28:17 +02:00
|
|
|
|
2016-12-08 18:12:07 +01:00
|
|
|
class Meta:
|
|
|
|
verbose_name = _('Stair')
|
|
|
|
verbose_name_plural = _('Stairs')
|
|
|
|
default_related_name = 'stairs'
|
|
|
|
|
2017-05-04 12:28:17 +02:00
|
|
|
def to_geojson(self):
|
|
|
|
result = super().to_geojson()
|
|
|
|
original_geometry = result['geometry']
|
|
|
|
draw = self.geometry.buffer(0.05, join_style=JOIN_STYLE.mitre, cap_style=CAP_STYLE.flat)
|
|
|
|
result['geometry'] = format_geojson(mapping(draw))
|
|
|
|
result['original_geometry'] = original_geometry
|
|
|
|
return result
|
|
|
|
|
|
|
|
def to_shadow_geojson(self):
|
|
|
|
shadow = self.geometry.parallel_offset(0.03, 'right', join_style=JOIN_STYLE.mitre)
|
|
|
|
shadow = shadow.buffer(0.019, join_style=JOIN_STYLE.mitre, cap_style=CAP_STYLE.flat)
|
|
|
|
return OrderedDict((
|
|
|
|
('type', 'Feature'),
|
|
|
|
('properties', OrderedDict((
|
|
|
|
('type', 'shadow'),
|
|
|
|
('original_type', self.__class__.__name__.lower()),
|
2017-05-05 16:21:48 +02:00
|
|
|
('original_id', self.id),
|
2017-05-06 17:40:06 +02:00
|
|
|
('space', self.space.id),
|
2017-05-04 12:28:17 +02:00
|
|
|
))),
|
|
|
|
('geometry', format_geojson(mapping(shadow), round=False)),
|
|
|
|
))
|
|
|
|
|
2016-12-08 18:12:07 +01:00
|
|
|
|
2017-05-08 16:40:22 +02:00
|
|
|
class Obstacle(SpaceGeometryMixin, models.Model):
|
2016-12-08 22:23:42 +01:00
|
|
|
"""
|
2016-12-09 14:49:20 +01:00
|
|
|
An obstacle
|
2016-12-08 22:23:42 +01:00
|
|
|
"""
|
2017-05-08 16:05:44 +02:00
|
|
|
geometry = GeometryField('polygon')
|
2016-12-08 22:23:42 +01:00
|
|
|
|
|
|
|
class Meta:
|
2016-12-09 14:49:20 +01:00
|
|
|
verbose_name = _('Obstacle')
|
|
|
|
verbose_name_plural = _('Obstacles')
|
|
|
|
default_related_name = 'obstacles'
|
2016-12-08 22:23:42 +01:00
|
|
|
|
|
|
|
|
2017-05-08 16:40:22 +02:00
|
|
|
class LineObstacle(SpaceGeometryMixin, models.Model):
|
2016-12-01 12:25:02 +01:00
|
|
|
"""
|
2016-12-09 14:49:20 +01:00
|
|
|
An obstacle that is a line with a specific width
|
2016-12-01 12:25:02 +01:00
|
|
|
"""
|
2017-05-08 16:05:44 +02:00
|
|
|
geometry = GeometryField('polyline')
|
2016-12-09 14:49:20 +01:00
|
|
|
width = models.DecimalField(_('obstacle width'), max_digits=4, decimal_places=2, default=0.15)
|
2016-12-04 20:01:37 +01:00
|
|
|
|
2016-12-01 12:25:02 +01:00
|
|
|
class Meta:
|
2016-12-09 14:49:20 +01:00
|
|
|
verbose_name = _('Line Obstacle')
|
|
|
|
verbose_name_plural = _('Line Obstacles')
|
|
|
|
default_related_name = 'lineobstacles'
|
|
|
|
|
2017-05-04 15:44:32 +02:00
|
|
|
@property
|
|
|
|
def buffered_geometry(self):
|
|
|
|
return self.geometry.buffer(self.width / 2, join_style=JOIN_STYLE.mitre, cap_style=CAP_STYLE.flat)
|
|
|
|
|
2016-12-09 14:49:20 +01:00
|
|
|
def to_geojson(self):
|
|
|
|
result = super().to_geojson()
|
|
|
|
original_geometry = result['geometry']
|
2017-05-04 15:44:32 +02:00
|
|
|
result['geometry'] = format_geojson(mapping(self.buffered_geometry))
|
2016-12-09 14:49:20 +01:00
|
|
|
result['original_geometry'] = original_geometry
|
|
|
|
return result
|
2016-12-01 12:25:02 +01:00
|
|
|
|
|
|
|
def get_geojson_properties(self):
|
|
|
|
result = super().get_geojson_properties()
|
2017-05-09 09:36:08 +02:00
|
|
|
# noinspection PyTypeChecker
|
2016-12-09 14:49:20 +01:00
|
|
|
result['width'] = float(self.width)
|
2016-12-01 12:25:02 +01:00
|
|
|
return result
|