forked from s_ranjbar/city_retrofit
Merge remote-tracking branch 'origin/master'
This commit is contained in:
commit
f98a341ea2
|
@ -6,7 +6,7 @@
|
|||
<excludeFolder url="file://$MODULE_DIR$/venv2" />
|
||||
<excludeFolder url="file://$MODULE_DIR$/venv38" />
|
||||
</content>
|
||||
<orderEntry type="jdk" jdkName="Python 3.7 (venv)" jdkType="Python SDK" />
|
||||
<orderEntry type="jdk" jdkName="Python 3.8 (libs) (2)" jdkType="Python SDK" />
|
||||
<orderEntry type="sourceFolder" forTests="false" />
|
||||
</component>
|
||||
<component name="PackageRequirementsSettings">
|
||||
|
|
|
@ -25,7 +25,7 @@ class Polyhedron:
|
|||
self._volume = None
|
||||
self._faces = None
|
||||
self._vertices = None
|
||||
self._mesh = None
|
||||
self._trimesh = None
|
||||
self._centroid = None
|
||||
self._max_z = None
|
||||
self._max_y = None
|
||||
|
@ -348,10 +348,10 @@ class Polyhedron:
|
|||
return self._faces
|
||||
|
||||
@property
|
||||
def polyhedron_trimesh(self):
|
||||
if self._mesh is None:
|
||||
self._mesh = Trimesh(vertices=self.vertices, faces=self.faces)
|
||||
return self._mesh
|
||||
def trimesh(self) -> Trimesh:
|
||||
if self._trimesh is None:
|
||||
self._trimesh = Trimesh(vertices=self.vertices, faces=self.faces)
|
||||
return self._trimesh
|
||||
|
||||
@property
|
||||
def volume(self):
|
||||
|
@ -360,10 +360,10 @@ class Polyhedron:
|
|||
:return: float
|
||||
"""
|
||||
if self._volume is None:
|
||||
if not self.polyhedron_trimesh.is_volume:
|
||||
if not self.trimesh.is_volume:
|
||||
self._volume = np.inf
|
||||
else:
|
||||
self._volume = self.polyhedron_trimesh.volume
|
||||
self._volume = self.trimesh.volume
|
||||
return self._volume
|
||||
|
||||
@property
|
||||
|
@ -466,7 +466,7 @@ class Polyhedron:
|
|||
:param full_path: str
|
||||
:return: None
|
||||
"""
|
||||
self.polyhedron_trimesh.export(full_path, 'stl_ascii')
|
||||
self.trimesh.export(full_path, 'stl_ascii')
|
||||
|
||||
def obj_export(self, full_path):
|
||||
"""
|
||||
|
@ -474,7 +474,7 @@ class Polyhedron:
|
|||
:param full_path: str
|
||||
:return: None
|
||||
"""
|
||||
self.polyhedron_trimesh.export(full_path, 'obj')
|
||||
self.trimesh.export(full_path, 'obj')
|
||||
|
||||
def show(self):
|
||||
self.polyhedron_trimesh.show()
|
||||
self.trimesh.show()
|
||||
|
|
|
@ -311,18 +311,18 @@ class Building(CityObject):
|
|||
# height = self.average_storey_height
|
||||
number_of_storeys = 4
|
||||
height = 1.5
|
||||
mesh = self.simplified_polyhedron.polyhedron_trimesh
|
||||
trimesh = self.simplified_polyhedron.trimesh
|
||||
normal_plane = [0, 0, -1]
|
||||
rest_mesh = mesh
|
||||
rest_trimesh = trimesh
|
||||
for n in range(0, number_of_storeys - 1):
|
||||
# todo: I need the lower corner of the building!!
|
||||
# point_plane = [self._lower_corner[0], self._lower_corner[1], self._lower_corner[2] + height]
|
||||
point_plane = [self._lower_corner[0] + 0.5, self._lower_corner[1] + 0.5, self._lower_corner[2] + height * (n + 1)]
|
||||
meshes = gh.divide_mesh_by_plane(rest_mesh, normal_plane, point_plane)
|
||||
storey = meshes[0]
|
||||
rest_mesh = meshes[1]
|
||||
trimeshes = gh.divide_mesh_by_plane(rest_trimesh, normal_plane, point_plane)
|
||||
storey = trimeshes[0]
|
||||
rest_trimesh = trimeshes[1]
|
||||
storeys.append(storey)
|
||||
storeys.append(rest_mesh)
|
||||
storeys.append(rest_trimesh)
|
||||
return storeys
|
||||
|
||||
@property
|
||||
|
|
58
exports/exports_factory.py
Normal file
58
exports/exports_factory.py
Normal file
|
@ -0,0 +1,58 @@
|
|||
"""
|
||||
ExportsFactory export a city into several formats
|
||||
SPDX - License - Identifier: LGPL - 3.0 - or -later
|
||||
Copyright © 2020 Project Author Guille Gutierrez guillermo.gutierrezmorote@concordia.ca
|
||||
"""
|
||||
|
||||
from exports.formats.stl import Stl
|
||||
from exports.formats.obj import Obj
|
||||
|
||||
class ExportsFactory:
|
||||
"""
|
||||
Exports factory class
|
||||
"""
|
||||
def __init__(self, export_type, city, path):
|
||||
self._city = city
|
||||
self._export_type = '_' + export_type.lower()
|
||||
self._path = path
|
||||
|
||||
@property
|
||||
def _citygml(self):
|
||||
"""
|
||||
Export to citygml with application domain extensions
|
||||
:return: None
|
||||
"""
|
||||
raise NotImplementedError()
|
||||
|
||||
@property
|
||||
def _stl(self):
|
||||
"""
|
||||
Export the city geometry to stl
|
||||
:return: None
|
||||
"""
|
||||
return Stl(self._city, self._path)
|
||||
|
||||
@property
|
||||
def _obj(self):
|
||||
"""
|
||||
Export the city geometry to obj
|
||||
:return: None
|
||||
"""
|
||||
return Obj(self._city, self._path)
|
||||
|
||||
@property
|
||||
def _idf(self):
|
||||
"""
|
||||
Export the city to Energy+ idf format
|
||||
:return:
|
||||
"""
|
||||
raise NotImplementedError()
|
||||
|
||||
@property
|
||||
def export(self):
|
||||
"""
|
||||
Export the city model structure to the given export type
|
||||
:return: City
|
||||
"""
|
||||
return getattr(self, self._export_type, lambda: None)
|
||||
|
|
@ -10,7 +10,7 @@ from pathlib import Path
|
|||
import helpers.constants as cte
|
||||
|
||||
|
||||
class IdfHelper:
|
||||
class Idf:
|
||||
_THERMOSTAT = 'HVACTEMPLATE:THERMOSTAT'
|
||||
_IDEAL_LOAD_AIR_SYSTEM = 'HVACTEMPLATE:ZONE:IDEALLOADSAIRSYSTEM'
|
||||
_SURFACE = 'BUILDINGSURFACE:DETAILED'
|
5
exports/formats/obj.py
Normal file
5
exports/formats/obj.py
Normal file
|
@ -0,0 +1,5 @@
|
|||
from exports.formats.triangular import Triangular
|
||||
|
||||
class Obj(Triangular):
|
||||
def __init__(self, city, path):
|
||||
super().__init__(city, path, 'obj')
|
5
exports/formats/stl.py
Normal file
5
exports/formats/stl.py
Normal file
|
@ -0,0 +1,5 @@
|
|||
from exports.formats.triangular import Triangular
|
||||
|
||||
class Stl(Triangular):
|
||||
def __init__(self, city, path):
|
||||
super().__init__(city, path, 'stl')
|
21
exports/formats/triangular.py
Normal file
21
exports/formats/triangular.py
Normal file
|
@ -0,0 +1,21 @@
|
|||
from pathlib import Path
|
||||
from trimesh import Trimesh
|
||||
|
||||
class Triangular:
|
||||
def __init__(self, city, path, triangular_format):
|
||||
self._city = city
|
||||
self._path = path
|
||||
self._triangular_format = triangular_format
|
||||
self._files()
|
||||
|
||||
def _files(self):
|
||||
if self._city.name is None:
|
||||
self._city.name = 'unknown_city'
|
||||
file_name = self._city.name + '.' + self._triangular_format
|
||||
file_path = (Path(self._path).resolve() / file_name).resolve()
|
||||
print(file_path)
|
||||
trimesh = Trimesh()
|
||||
for building in self._city.buildings:
|
||||
trimesh = trimesh.union(building.simplified_polyhedron.trimesh)
|
||||
with open(file_path, 'w') as file:
|
||||
file.write(trimesh.export(file_type=self._triangular_format))
|
39
tests/test_exports.py
Normal file
39
tests/test_exports.py
Normal file
|
@ -0,0 +1,39 @@
|
|||
"""
|
||||
TestExports test and validate the city export formats
|
||||
SPDX - License - Identifier: LGPL - 3.0 - or -later
|
||||
Copyright © 2020 Project Author Guille Gutierrez guillermo.gutierrezmorote@concordia.ca
|
||||
"""
|
||||
|
||||
from pathlib import Path
|
||||
from unittest import TestCase
|
||||
from imports.geometry_factory import GeometryFactory
|
||||
from exports.exports_factory import ExportsFactory
|
||||
|
||||
class TestExports(TestCase):
|
||||
"""
|
||||
TestGeometryFactory TestCase 1
|
||||
"""
|
||||
def setUp(self) -> None:
|
||||
"""
|
||||
Test setup
|
||||
:return: None
|
||||
"""
|
||||
self._city_gml = None
|
||||
self._example_path = (Path(__file__).parent / 'tests_data').resolve()
|
||||
self._output_path = (Path(__file__).parent / 'tests_outputs').resolve()
|
||||
|
||||
def _get_city(self):
|
||||
if self._city_gml is None:
|
||||
file_path = (self._example_path / '20buildings.gml').resolve()
|
||||
self._city_gml = GeometryFactory('citygml', file_path).city
|
||||
return self._city_gml
|
||||
|
||||
def _export(self, export_type):
|
||||
self._city_gml = self._get_city()
|
||||
ExportsFactory(export_type, self._city_gml, self._output_path).export
|
||||
|
||||
def test_obj_export(self):
|
||||
self._export('obj')
|
||||
|
||||
def test_stl_export(self):
|
||||
self._export('stl')
|
|
@ -21,7 +21,7 @@ class TestGeometryFactory(TestCase):
|
|||
:return: None
|
||||
"""
|
||||
self._city_gml = None
|
||||
self._example_path = (Path(__file__).parent.parent / 'tests_data').resolve()
|
||||
self._example_path = (Path(__file__).parent / 'tests_data').resolve()
|
||||
|
||||
def _get_citygml(self, file):
|
||||
if self._city_gml is None:
|
||||
|
|
|
@ -9,7 +9,6 @@ from imports.geometry_factory import GeometryFactory
|
|||
from imports.physics_factory import PhysicsFactory
|
||||
from imports.usage_factory import UsageFactory
|
||||
from imports.schedules_factory import SchedulesFactory
|
||||
from exports.idf_helper import IdfHelper
|
||||
import os
|
||||
import glob
|
||||
|
||||
|
@ -25,8 +24,8 @@ class TestIdf(TestCase):
|
|||
:return: None
|
||||
"""
|
||||
self._city_gml = None
|
||||
self._example_path = (Path(__file__).parent.parent / 'tests_data').resolve()
|
||||
self._output_path = (Path(__file__).parent / 'ep_outputs').resolve()
|
||||
self._example_path = (Path(__file__).parent / 'tests_data').resolve()
|
||||
self._output_path = (Path(__file__).parent / 'tests_outputs').resolve()
|
||||
|
||||
def _get_city(self):
|
||||
if self._city_gml is None:
|
||||
|
|
|
@ -21,7 +21,7 @@ class TestPhysicsFactory(TestCase):
|
|||
"""
|
||||
self._city_gml = None
|
||||
self._nyc_with_physics = None
|
||||
self._example_path = (Path(__file__).parent.parent / 'tests_data').resolve()
|
||||
self._example_path = (Path(__file__).parent / 'tests_data').resolve()
|
||||
|
||||
def _get_citygml(self, file_path):
|
||||
if self._city_gml is None:
|
||||
|
|
|
@ -23,7 +23,7 @@ class TestSchedulesFactory(TestCase):
|
|||
:return: None
|
||||
"""
|
||||
self._city_gml_with_usage = None
|
||||
self._example_path = (Path(__file__).parent.parent / 'tests_data').resolve()
|
||||
self._example_path = (Path(__file__).parent / 'tests_data').resolve()
|
||||
|
||||
@property
|
||||
def _handler(self):
|
||||
|
|
|
@ -21,7 +21,7 @@ class TestUsageFactory(TestCase):
|
|||
"""
|
||||
self._city_gml = None
|
||||
self._nyc_with_usage = None
|
||||
self._example_path = (Path(__file__).parent.parent / 'tests_data').resolve()
|
||||
self._example_path = (Path(__file__).parent / 'tests_data').resolve()
|
||||
|
||||
def _get_citygml(self, file_path):
|
||||
if self._city_gml is None:
|
||||
|
|
|
@ -22,7 +22,7 @@ class TestWeatherFactory(TestCase):
|
|||
self._city_gml = None
|
||||
self._city_with_weather = None
|
||||
self._city_name = 'new_york_city'
|
||||
self._example_path = (Path(__file__).parent.parent / 'tests_data').resolve()
|
||||
self._example_path = (Path(__file__).parent / 'tests_data').resolve()
|
||||
|
||||
def _get_citygml(self, file_path):
|
||||
if self._city_gml is None:
|
||||
|
@ -32,7 +32,7 @@ class TestWeatherFactory(TestCase):
|
|||
|
||||
def _get_city_with_weather(self):
|
||||
if self._city_with_weather is None:
|
||||
file_path = (Path(__file__).parent.parent / 'tests_data' / '20buildings.gml').resolve()
|
||||
file_path = (Path(__file__).parent / 'tests_data' / '20buildings.gml').resolve()
|
||||
self._city_with_weather = self._get_citygml(file_path)
|
||||
WeatherFactory('dat', self._city_with_weather, city_name=self._city_name, base_path=self._example_path)
|
||||
return self._city_with_weather
|
||||
|
@ -54,7 +54,7 @@ class TestWeatherFactory(TestCase):
|
|||
self.assertFalse(values.empty, 'wrong value beam')
|
||||
|
||||
def test_weather_xls(self):
|
||||
file_path = (Path(__file__).parent.parent / 'tests_data' / 'iso_52016_1_2017_lod2.gml').resolve()
|
||||
file_path = (Path(__file__).parent / 'tests_data' / 'iso_52016_1_2017_lod2.gml').resolve()
|
||||
city_with_weather = self._get_citygml(file_path)
|
||||
WeatherFactory('xls', city_with_weather, city_name=self._city_name, base_path=self._example_path)
|
||||
for building in city_with_weather.buildings:
|
||||
|
|
4
tests/tests_outputs/.gitignore
vendored
Normal file
4
tests/tests_outputs/.gitignore
vendored
Normal file
|
@ -0,0 +1,4 @@
|
|||
# Ignore everything in this directory
|
||||
*
|
||||
# Except this file
|
||||
!.gitignore
|
Loading…
Reference in New Issue
Block a user