forked from s_ranjbar/city_retrofit
52 lines
1.8 KiB
Python
52 lines
1.8 KiB
Python
"""
|
|
UsageFactory retrieve the specific usage module for the given region
|
|
This factory can only be called after calling the construction factory so the thermal zones are created.
|
|
SPDX - License - Identifier: LGPL - 3.0 - or -later
|
|
Copyright © 2022 Concordia CERC group
|
|
Project Coder Guille Gutierrez guillermo.gutierrezmorote@concordia.ca
|
|
Code contributors: Pilar Monsalvete Alvarez de Uribarri pilar.monsalvete@concordia.ca
|
|
"""
|
|
import logging
|
|
from hub.imports.usage.comnet_usage_parameters import ComnetUsageParameters
|
|
from hub.imports.usage.nrcan_usage_parameters import NrcanUsageParameters
|
|
from hub.helpers.utils import validate_import_export_type
|
|
|
|
|
|
class UsageFactory:
|
|
"""
|
|
UsageFactory class
|
|
"""
|
|
def __init__(self, handler, city):
|
|
self._handler = '_' + handler.lower().replace(' ', '_')
|
|
class_funcs = validate_import_export_type(UsageFactory)
|
|
if self._handler not in class_funcs:
|
|
error_message = f"Wrong import type [{self._handler}]. Valid functions include {class_funcs}"
|
|
logging.error(error_message)
|
|
raise Exception(error_message)
|
|
self._city = city
|
|
|
|
def _comnet(self):
|
|
"""
|
|
Enrich the city with COMNET usage library
|
|
"""
|
|
ComnetUsageParameters(self._city).enrich_buildings()
|
|
self._city.level_of_detail.usage = 2
|
|
for building in self._city.buildings:
|
|
building.level_of_detail.usage = 2
|
|
|
|
def _nrcan(self):
|
|
"""
|
|
Enrich the city with NRCAN usage library
|
|
"""
|
|
NrcanUsageParameters(self._city).enrich_buildings()
|
|
self._city.level_of_detail.usage = 2
|
|
for building in self._city.buildings:
|
|
building.level_of_detail.usage = 2
|
|
|
|
def enrich(self):
|
|
"""
|
|
Enrich the city given to the class using the usage factory given handler
|
|
:return: None
|
|
"""
|
|
getattr(self, self._handler, lambda: None)()
|