44 lines
1.3 KiB
Python
44 lines
1.3 KiB
Python
"""
|
|
PhysicsFactory retrieve the specific physics module for the given region
|
|
SPDX - License - Identifier: LGPL - 3.0 - or -later
|
|
Copyright © 2020 Project Author Guille Gutierrez guillermo.gutierrezmorote@concordia.ca
|
|
"""
|
|
from factories.physics_feeders.us_new_york_city_physics_parameters import UsNewYorkCityPhysicsParameters
|
|
from factories.physics_feeders.us_physics_parameters import UsPhysicsParameters
|
|
from factories.physics_feeders.ca_physics_parameters import CaPhysicsParameters
|
|
from pathlib import Path
|
|
|
|
|
|
class PhysicsFactory:
|
|
"""
|
|
PhysicsFactor class
|
|
"""
|
|
def __init__(self, handler, city, base_path=Path(Path(__file__).parent.parent / 'data/physics')):
|
|
self._handler = '_' + handler.lower().replace(' ', '_')
|
|
self._city = city
|
|
self._base_path = base_path
|
|
self.factory()
|
|
|
|
# todo: adapt us_new_york and us to the new version with interface
|
|
def _us_new_york(self):
|
|
UsNewYorkCityPhysicsParameters(self._city, self._base_path)
|
|
|
|
def _us(self):
|
|
UsPhysicsParameters(self._city, self._base_path)
|
|
|
|
def _ca(self):
|
|
CaPhysicsParameters(self._city, self._base_path).enrich_buildings()
|
|
|
|
def _de(self):
|
|
raise NotImplementedError
|
|
|
|
def _es(self):
|
|
raise NotImplementedError
|
|
|
|
def factory(self):
|
|
"""
|
|
Enrich the city with the physics information
|
|
:return: None
|
|
"""
|
|
getattr(self, self._handler, lambda: None)()
|