hub/factories/occupancy_feeders/demo_occupancy_parameters.py

"""
PhysicsFactory retrieve the specific physics module for the given region
SPDX - License - Identifier: LGPL - 3.0 - or -later
Copyright © 2020 Project Author Guille Gutierrez guillermo.gutierrezmorote@concordia.ca
contributors Pilar Monsalvete pilar_monsalvete@yahoo.es
"""
import pandas as pd
from factories.occupancy_feeders.helpers.occupancy_helper import OccupancyHelper


class DemoOccupancyParameters:

  def __init__(self, city, base_path):
    self._city = city
    self._demo_schedules_path = base_path / 'demo_schedules.xlsx'
    print('schedules: ', self._demo_schedules_path)
    xls = pd.ExcelFile(self._demo_schedules_path)
    # todo: review for more than one usage_zones per building
    for building in city.buildings:
      schedules = dict()
      occupancy = pd.read_excel(xls, sheet_name=OccupancyHelper.pluto_occupancy_function(building.function),
                                skiprows=[0, 1, 2, 3], nrows=39, usecols="A:AA")
      # todo: should we safe the data type? How?
      for index in range(0, 13):
        data = pd.DataFrame()
        columns_names = []
        name = ''
        data_type = ''
        for i in range(0, 3):
          row = occupancy.iloc[3*index + i]
          if i == 0:
            name = row[0]
            data_type = row[1]
          columns_names.append(row[2])
          data1 = row[3:]
          data = pd.concat([data, data1],  axis=1)
        data.columns = columns_names
        schedules[name] = data
      building.usage_zones[0].schedules = schedules