forked from s_ranjbar/city_retrofit
119 lines
3.4 KiB
Python
119 lines
3.4 KiB
Python
"""
|
|
GeometryFactory retrieve the specific geometric module to load the given format
|
|
SPDX - License - Identifier: LGPL - 3.0 - or -later
|
|
Copyright © 2022 Concordia CERC group
|
|
Project Coder Guille Gutierrez guillermo.gutierrezmorote@concordia.ca
|
|
"""
|
|
|
|
import geopandas
|
|
from hub.city_model_structure.city import City
|
|
from hub.imports.geometry.citygml import CityGml
|
|
from hub.imports.geometry.obj import Obj
|
|
from hub.imports.geometry.osm_subway import OsmSubway
|
|
from hub.imports.geometry.rhino import Rhino
|
|
from hub.imports.geometry.gpandas import GPandas
|
|
from hub.imports.geometry.geojson import Geojson
|
|
from hub.helpers.utils import validate_import_export_type
|
|
from hub.hub_logger import logger
|
|
|
|
|
|
class GeometryFactory:
|
|
"""
|
|
GeometryFactory class
|
|
"""
|
|
def __init__(self, file_type,
|
|
path=None,
|
|
data_frame=None,
|
|
height_field=None,
|
|
year_of_construction_field=None,
|
|
function_field=None,
|
|
function_to_hub=None):
|
|
self._file_type = '_' + file_type.lower()
|
|
class_funcs = validate_import_export_type(GeometryFactory)
|
|
if self._file_type not in class_funcs:
|
|
err_msg = f"Wrong import type. Valid functions include {class_funcs}"
|
|
logger.error(err_msg)
|
|
raise Exception(err_msg)
|
|
self._path = path
|
|
self._data_frame = data_frame
|
|
self._height_field = height_field
|
|
self._year_of_construction_field = year_of_construction_field
|
|
self._function_field = function_field
|
|
self._function_to_hub = function_to_hub
|
|
|
|
@property
|
|
def _citygml(self) -> City:
|
|
"""
|
|
Enrich the city by using CityGML information as data source
|
|
:return: City
|
|
"""
|
|
return CityGml(self._path,
|
|
self._height_field,
|
|
self._year_of_construction_field,
|
|
self._function_field,
|
|
self._function_to_hub).city
|
|
|
|
@property
|
|
def _obj(self) -> City:
|
|
"""
|
|
Enrich the city by using OBJ information as data source
|
|
:return: City
|
|
"""
|
|
return Obj(self._path).city
|
|
|
|
@property
|
|
def _gpandas(self) -> City:
|
|
"""
|
|
Enrich the city by using GeoPandas information as data source
|
|
:return: City
|
|
"""
|
|
if self._data_frame is None:
|
|
self._data_frame = geopandas.read_file(self._path)
|
|
return GPandas(self._data_frame).city
|
|
|
|
@property
|
|
def _geojson(self) -> City:
|
|
"""
|
|
Enrich the city by using Geojson information as data source
|
|
:return: City
|
|
"""
|
|
return Geojson(self._path,
|
|
self._height_field,
|
|
self._year_of_construction_field,
|
|
self._function_field,
|
|
self._function_to_hub).city
|
|
|
|
@property
|
|
def _osm_subway(self) -> City:
|
|
"""
|
|
Enrich the city by using OpenStreetMap information as data source
|
|
:return: City
|
|
"""
|
|
return OsmSubway(self._path).city
|
|
|
|
@property
|
|
def _rhino(self) -> City:
|
|
"""
|
|
Enrich the city by using Rhino information as data source
|
|
:return: City
|
|
"""
|
|
return Rhino(self._path).city
|
|
|
|
@property
|
|
def city(self) -> City:
|
|
"""
|
|
Enrich the city given to the class using the class given handler
|
|
:return: City
|
|
"""
|
|
return getattr(self, self._file_type, lambda: None)
|
|
|
|
@property
|
|
def city_debug(self) -> City:
|
|
"""
|
|
Enrich the city given to the class using the class given handler
|
|
:return: City
|
|
"""
|
|
if self._data_frame is None:
|
|
self._data_frame = geopandas.read_file(self._path)
|
|
return GPandas(self._data_frame).city
|