city_retrofit/imports/geometry_factory.py
2021-08-11 10:29:54 -04:00

84 lines
2.0 KiB
Python

"""
GeometryFactory retrieve the specific geometric module to load the given format
SPDX - License - Identifier: LGPL - 3.0 - or -later
Copyright © 2020 Project Author Guille Gutierrez guillermo.gutierrezmorote@concordia.ca
"""
from trimesh.scene import Scene
from city_model_structure.city import City
from city_model_structure.city_object import CityObject
from imports.geometry.citygml import CityGml
from imports.geometry.osm_subway import OsmSubway
from imports.geometry.obj import Obj
class GeometryFactory:
"""
GeometryFactory class
"""
def __init__(self, file_type, path):
self._file_type = '_' + file_type.lower()
self._path = path
@property
def _citygml(self):
return CityGml(self._path).city
@property
def _stl(self):
raise Exception('Not implemented')
@property
def _obj(self):
return Obj(self._path).scene
@property
def _geojson(self):
raise Exception('Not implemented')
@property
def _bim(self):
raise Exception('Not implemented')
@property
def city(self) -> City:
"""
Load the city model structure from a geometry source
:return: City
"""
return getattr(self, self._file_type, lambda: None)
@property
def scene(self) -> Scene:
"""
Load the city model structure from a geometry source
:return: Trimesh scene
"""
return getattr(self, self._file_type, lambda: None)
@property
def _scene_debug(self):
"""
just for debug. More information is provided without the lambda parameter
"""
return getattr(self, self._file_type)
@property
def _city_debug(self):
"""
just for debug. More information is provided without the lambda parameter
"""
return CityGml(self._path).city
@property
def _osm_subway(self):
return OsmSubway(self._path).subway_entrances
@property
def features(self) -> [CityObject]:
"""
Load the city model structure from a geometry source
:return: [CityObject]
"""
return getattr(self, self._file_type, lambda: None)