forked from s_ranjbar/city_retrofit
28 lines
937 B
Python
28 lines
937 B
Python
"""
|
|
export a city from trimesh into Triangular format (obj or stl)
|
|
SPDX - License - Identifier: LGPL - 3.0 - or -later
|
|
Copyright © 2020 Project Author Guille Gutierrez guillermo.gutierrezmorote@concordia.ca
|
|
"""
|
|
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._export()
|
|
|
|
def _export(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))
|