53 lines
1.4 KiB
Python
53 lines
1.4 KiB
Python
|
import glob
|
||
|
|
import os
|
||
|
|
import shutil
|
||
|
|
import subprocess
|
||
|
|
|
||
|
|
from hub.city_model_structure.city import City
|
||
|
|
from hub.exports.formats.obj import Obj
|
||
|
|
|
||
|
|
|
||
|
|
class GltExceptionError(Exception):
|
||
|
|
"""
|
||
|
|
Glt execution error
|
||
|
|
"""
|
||
|
|
|
||
|
|
|
||
|
|
class Glb:
|
||
|
|
def __init__(self, city, path, target_buildings=None):
|
||
|
|
self._city = city
|
||
|
|
self._path = path
|
||
|
|
if target_buildings is None:
|
||
|
|
target_buildings = [b.name for b in self._city.buildings]
|
||
|
|
self._target_buildings = target_buildings
|
||
|
|
self._export()
|
||
|
|
|
||
|
|
@property
|
||
|
|
def _obj2gtl(self):
|
||
|
|
"""
|
||
|
|
Get the SRA installation path
|
||
|
|
:return: str
|
||
|
|
"""
|
||
|
|
return shutil.which('obj2gltf')
|
||
|
|
|
||
|
|
def _export(self):
|
||
|
|
try:
|
||
|
|
for building in self._city.buildings:
|
||
|
|
city = City(building.lower_corner, building.upper_corner, self._city.srs_name)
|
||
|
|
city.name = building.name
|
||
|
|
city.add_city_object(building)
|
||
|
|
Obj( city, self._path)
|
||
|
|
glb = f'{self._path}/{building.name}.glb'
|
||
|
|
subprocess.run([
|
||
|
|
self._obj2gtl,
|
||
|
|
'-i', f'{self._path}/{building.name}.obj',
|
||
|
|
'-o', f'{glb}',
|
||
|
|
'-b',
|
||
|
|
'--normalTexture', f'{self._path}/{building.name}.mtl'
|
||
|
|
])
|
||
|
|
os.unlink(f'{self._path}/{building.name}.obj')
|
||
|
|
os.unlink(f'{self._path}/{building.name}.mtl')
|
||
|
|
except (subprocess.SubprocessError, subprocess.TimeoutExpired, subprocess.CalledProcessError) as err:
|
||
|
|
raise GltExceptionError from err
|
||
|
|
|