city_retrofit/city_model_structure/transport/join.py

"""
Join module
SPDX - License - Identifier: LGPL - 3.0 - or -later
Copyright © 2020 Project Author Pilar Monsalvete Alvarez de Uribarri pilar.monsalvete@concordia.ca
Contributor Milad milad.aghamohamadnia@concordia.ca
Contributor Guille guille.gutierrezmorote@concordia.ca
"""

from typing import List

from city_model_structure.attributes.node import Node
from city_model_structure.transport.traffic_node import TrafficNode


class Join(TrafficNode):
  """
  Join class
  """

  def __init__(self, name, coordinates, nodes):
    self._nodes = nodes
    edges = []
    prohibitions = []
    connections = []
    for node in self._nodes:
      edges = edges + node.edges
      prohibitions = prohibitions + node.prohibitions
      connections = connections + node.connections
    super().__init__(name, coordinates, edges=edges, prohibitions=prohibitions, connections=connections,
                     node_type='Join')