import hub.helpers.constants as cte

class HubFunctionToMatsimActivity:
  """
  Hub function to matsim activity construction function class
  """
  def __init__(self):
    self._dictionary = {
      cte.RESIDENTIAL: 'home',
      cte.SINGLE_FAMILY_HOUSE: 'home',
      cte.MULTI_FAMILY_HOUSE: 'home',
      cte.ROW_HOUSE: 'home',
      cte.MID_RISE_APARTMENT: 'home',
      cte.HIGH_RISE_APARTMENT: 'home',
      cte.OFFICE_AND_ADMINISTRATION: 'work',
      cte.SMALL_OFFICE: 'work',
      cte.MEDIUM_OFFICE: 'work',
      cte.LARGE_OFFICE: 'work',
      cte.COURTHOUSE: 'work',
      cte.FIRE_STATION: 'work',
      cte.PENITENTIARY: 'work',
      cte.POLICE_STATION: 'work',
      cte.POST_OFFICE: 'work',
      cte.LIBRARY: 'edu',
      cte.EDUCATION: 'edu',
      cte.PRIMARY_SCHOOL: 'edu',
      cte.PRIMARY_SCHOOL_WITH_SHOWER: 'edu',
      cte.SECONDARY_SCHOOL: 'edu',
      cte.UNIVERSITY: 'edu',
      cte.LABORATORY_AND_RESEARCH_CENTER: 'edu',
      cte.STAND_ALONE_RETAIL: 'work,secondary',
      cte.HOSPITAL: 'work',
      cte.OUT_PATIENT_HEALTH_CARE: 'work',
      cte.HEALTH_CARE: 'work',
      cte.RETIREMENT_HOME_OR_ORPHANAGE: 'home,secondary',
      cte.COMMERCIAL: 'work,secondary',
      cte.STRIP_MALL: 'work,secondary',
      cte.SUPERMARKET: 'work,secondary',
      cte.RETAIL_SHOP_WITHOUT_REFRIGERATED_FOOD: 'work,secondary',
      cte.RETAIL_SHOP_WITH_REFRIGERATED_FOOD: 'work,secondary',
      cte.RESTAURANT: 'work,secondary',
      cte.QUICK_SERVICE_RESTAURANT: 'work,secondary',
      cte.FULL_SERVICE_RESTAURANT: 'work,secondary',
      cte.HOTEL: 'work,secondary',
      cte.HOTEL_MEDIUM_CLASS: 'work,secondary',
      cte.SMALL_HOTEL: 'work,secondary',
      cte.LARGE_HOTEL: 'work,secondary',
      cte.DORMITORY: 'work',
      cte.EVENT_LOCATION: 'work,secondary',
      cte.CONVENTION_CENTER: 'work,secondary',
      cte.HALL: 'work',
      cte.GREEN_HOUSE: 'work',
      cte.INDUSTRY: 'work',
      cte.WORKSHOP: 'work',
      cte.WAREHOUSE: 'work',
      cte.WAREHOUSE_REFRIGERATED: 'work',
      cte.SPORTS_LOCATION: 'work,secondary',
      cte.SPORTS_ARENA: 'work,secondary',
      cte.GYMNASIUM: 'work,secondary',
      cte.MOTION_PICTURE_THEATRE: 'work,secondary',
      cte.MUSEUM: 'work,secondary',
      cte.PERFORMING_ARTS_THEATRE: 'work,secondary',
      cte.TRANSPORTATION: 'work',
      cte.AUTOMOTIVE_FACILITY: 'work',
      cte.PARKING_GARAGE: 'work',
      cte.RELIGIOUS: 'work,secondary',
      cte.NON_HEATED: 'work',
      cte.DATACENTER: 'work',
      cte.FARM: 'work'
    }

  @property
  def dictionary(self) -> dict:
    """
    Get the dictionary
    :return: {}
    """
    return self._dictionary