city_retrofit/hub/exports/formats/glb.py

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',
          '-b',
          '-o', f'{glb}',
          '--triangleWindingOrderSanitization',
          '--inputUpAxis', 'Y',
          '--outputUpAxis', 'Y'
        ])

        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