"""
export a city into Geojson format
SPDX - License - Identifier: LGPL - 3.0 - or -later
Copyright © 2022 Concordia CERC group
Project Coder Guille Gutierrez guillermo.gutierrezmorote@concordia.ca
"""
import json
from pathlib import Path

import numpy as np
import pyproj
from pyproj import Transformer

from hub.helpers.geometry_helper import GeometryHelper


class Geojson:
  """
  Export to geojson format
  """
  def __init__(self, city, path, target_buildings):
    self._city = city
    self._file_path = Path(path / f'{self._city.name}.geojson').resolve()
    try:
      srs_name = self._city.srs_name
      if self._city.srs_name in GeometryHelper.srs_transformations:
        srs_name = GeometryHelper.srs_transformations[self._city.srs_name]
      input_reference = pyproj.CRS(srs_name)  # Projected coordinate system from input data
    except pyproj.exceptions.CRSError as err:
      raise pyproj.exceptions.CRSError from err
    self._to_gps = Transformer.from_crs(input_reference, pyproj.CRS('EPSG:4326'))
    if target_buildings is None:
      target_buildings = [b.name for b in self._city.buildings]
    self._geojson_skeleton = {
      'type': 'FeatureCollection',
      'features': []
    }
    self._feature_skeleton = {
      'type': 'Feature',
      'geometry': {
        'type': 'Polygon',
        'coordinates': []
      },
      'properties': {}
    }
    self._export()

  def _export(self):
    for building in self._city.buildings:
      feature = self._feature_skeleton.copy()
      feature['id'] = building.name
      feature['properties']['height'] = f'{building.max_height}'
      feature['properties']['function'] = f'{building.function}'
      feature['properties']['year_of_construction'] = f'{building.year_of_construction}'
      feature['properties']['aliases'] = building.aliases
      feature['properties']['elevation'] = f'{building.lower_corner[2]}'
      for ground in building.grounds:

        ground_coordinates = []
        for coordinate in ground.solid_polygon.coordinates:
          gps_coordinate = self._to_gps.transform(coordinate[0], coordinate[1])
          ground_coordinates.insert(0, [gps_coordinate[1], gps_coordinate[0]])

        first_gps_coordinate = self._to_gps.transform(
          ground.solid_polygon.coordinates[0][0],
          ground.solid_polygon.coordinates[0][1]
        )
        ground_coordinates.insert(0, [first_gps_coordinate[1], first_gps_coordinate[0]])
        feature['geometry']['coordinates'].append(ground_coordinates)
      self._geojson_skeleton['features'].append(feature)
    with open(self._file_path, 'w', encoding='utf-8') as f:
      json.dump(self._geojson_skeleton, f, indent=2)