66 lines
3.5 KiB
Python
66 lines
3.5 KiB
Python
from imports.geometry.citygml_tools import CityGmlTools
|
|
from city_model_structure.attributes.surface import Surface
|
|
from city_model_structure.attributes.polygon import Polygon
|
|
|
|
|
|
class CityGmlLod2(CityGmlTools):
|
|
|
|
@staticmethod
|
|
def _lod2_composite_surface(s):
|
|
solid_points = [
|
|
CityGmlTools._solid_points((CityGmlTools._remove_last_point(sm['Polygon']['exterior']['LinearRing']['posList'])))
|
|
for sm in s['CompositeSurface']['surfaceMember']]
|
|
return [Surface(Polygon(sp), Polygon(sp)) for sp in solid_points]
|
|
|
|
@staticmethod
|
|
def _lod2_multi_surface(s, surface_type):
|
|
# todo: this need to be changed into surface bounded?
|
|
try:
|
|
solid_points = [CityGmlTools._solid_points(CityGmlTools._remove_last_point(
|
|
s['Polygon']['exterior']['LinearRing']['posList']['#text']))]
|
|
except TypeError:
|
|
solid_points = [CityGmlTools._solid_points(CityGmlTools._remove_last_point(s['Polygon']['exterior']['LinearRing']
|
|
['posList']))]
|
|
return [Surface(Polygon(sp), Polygon(sp), surface_type=surface_type) for sp in solid_points]
|
|
|
|
@staticmethod
|
|
def lod2(bound):
|
|
surfaces = []
|
|
for surface_type in iter(bound):
|
|
for s in bound[surface_type]['lod2MultiSurface']['MultiSurface']['surfaceMember']:
|
|
if 'CompositeSurface' in s:
|
|
surfaces = surfaces + CityGmlLod2._lod2_composite_surface(s)
|
|
else:
|
|
surfaces = surfaces + CityGmlLod2._lod2_multi_surface(s, surface_type)
|
|
return surfaces
|
|
|
|
@staticmethod
|
|
def lod2_solid_multi_surface(o):
|
|
polygons = None
|
|
if 'boundedBy' in o['Building']['consistsOfBuildingPart']['BuildingPart']:
|
|
if 'RoofSurface' in o['Building']['consistsOfBuildingPart']['BuildingPart']['boundedBy']:
|
|
if o['Building']['consistsOfBuildingPart']['BuildingPart']['boundedBy']['RoofSurface']['lod2MultiSurface'] != 'None':
|
|
polygons = [Polygon(CityGmlTools._remove_last_point(s['Polygon']['exterior']['LinearRing']['posList']))
|
|
for s in o['Building']['consistsOfBuildingPart']['BuildingPart']['boundedBy']['RoofSurface']
|
|
['lod2MultiSurface']['MultiSurface']['surfaceMember']]
|
|
|
|
elif 'WallSurface' in o['Building']['consistsOfBuildingPart']['BuildingPart']['boundedBy']:
|
|
if o['Building']['consistsOfBuildingPart']['BuildingPart']['boundedBy']['WallSurface']['lod2MultiSurface'] != 'None':
|
|
polygons = [Polygon(CityGmlTools._remove_last_point(s['Polygon']['exterior']['LinearRing']['posList']))
|
|
for s in o['Building']['consistsOfBuildingPart']['BuildingPart']['boundedBy']['WallSurface']['lod2MultiSurface']['MultiSurface']['surfaceMember']]
|
|
else:
|
|
polygons = [Polygon(CityGmlTools._remove_last_point(s['Polygon']['exterior']['LinearRing']['posList']))
|
|
for s in o['Building']['lod2MultiSurface']['MultiSurface']['surfaceMember']]
|
|
return [Surface(p,p) for p in polygons]
|
|
|
|
|
|
@staticmethod
|
|
def lod2_solid(o):
|
|
try:
|
|
solid_points = [CityGmlTools._solid_points(CityGmlTools._remove_last_point(s['Polygon']['exterior']['LinearRing']['posList']['#text']))
|
|
for s in o['Building']['lod2Solid']['Solid']['exterior']['CompositeSurface']['surfaceMember']]
|
|
except TypeError:
|
|
solid_points = [CityGmlTools._solid_points(CityGmlTools._remove_last_point(s['Polygon']['exterior']['LinearRing']['posList']))
|
|
for s in o['Building']['lod2Solid']['Solid']['exterior']['CompositeSurface']['surfaceMember']]
|
|
return [Surface(Polygon(sp),Polygon(sp)) for sp in solid_points]
|