# frozen_string_literal: true

class Location
  def self.apply(model, runner, weather, epw_file, hpxml)
    apply_year(model, hpxml)
    apply_site(model, epw_file)
    apply_climate_zones(model, epw_file)
    apply_dst(model, hpxml)
    apply_ground_temps(model, weather)
    return weather
  end

  def self.apply_weather_file(model, runner, weather_file_path, weather_cache_path)
    if File.exist?(weather_file_path) && weather_file_path.downcase.end_with?('.epw')
      epw_file = OpenStudio::EpwFile.new(weather_file_path)
    else
      fail "'#{weather_file_path}' does not exist or is not an .epw file."
    end

    OpenStudio::Model::WeatherFile.setWeatherFile(model, epw_file).get

    # Obtain weather object
    # Load from cache .csv file if exists, as this is faster and doesn't require
    # parsing the weather file.
    if File.exist? weather_cache_path
      weather = WeatherProcess.new(nil, nil, weather_cache_path)
    else
      weather = WeatherProcess.new(model, runner)
    end

    return weather, epw_file
  end

  private

  def self.apply_site(model, epw_file)
    site = model.getSite
    site.setName("#{epw_file.city}_#{epw_file.stateProvinceRegion}_#{epw_file.country}")
    site.setLatitude(epw_file.latitude)
    site.setLongitude(epw_file.longitude)
    site.setTimeZone(epw_file.timeZone)
    site.setElevation(epw_file.elevation)
  end

  def self.apply_climate_zones(model, epw_file)
    ba_zone = get_climate_zone_ba(epw_file.wmoNumber)
    return if ba_zone.nil?

    climateZones = model.getClimateZones
    climateZones.setClimateZone(Constants.BuildingAmericaClimateZone, ba_zone)
  end

  def self.apply_year(model, hpxml)
    year_description = model.getYearDescription
    year_description.setCalendarYear(hpxml.header.sim_calendar_year)
  end

  def self.apply_dst(model, hpxml)
    return unless hpxml.header.dst_enabled

    month_names = ['Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun', 'Jul', 'Aug', 'Sep', 'Oct', 'Nov', 'Dec']
    dst_start_date = "#{month_names[hpxml.header.dst_begin_month - 1]} #{hpxml.header.dst_begin_day_of_month}"
    dst_end_date = "#{month_names[hpxml.header.dst_end_month - 1]} #{hpxml.header.dst_end_day_of_month}"

    run_period_control_daylight_saving_time = model.getRunPeriodControlDaylightSavingTime
    run_period_control_daylight_saving_time.setStartDate(dst_start_date)
    run_period_control_daylight_saving_time.setEndDate(dst_end_date)
  end

  def self.apply_ground_temps(model, weather)
    # Shallow ground temperatures only currently used for ducts located under slab
    sgts = model.getSiteGroundTemperatureShallow
    sgts.resetAllMonths
    sgts.setAllMonthlyTemperatures(weather.data.GroundMonthlyTemps.map { |t| UnitConversions.convert(t, 'F', 'C') })

    # Deep ground temperatures used by GSHP setpoint manager
    dgts = model.getSiteGroundTemperatureDeep
    dgts.resetAllMonths
    dgts.setAllMonthlyTemperatures([UnitConversions.convert(weather.data.AnnualAvgDrybulb, 'F', 'C')] * 12)
  end

  def self.get_climate_zone_ba(wmo)
    ba_zone = nil
    zones_csv = File.join(File.dirname(__FILE__), 'data_climate_zones.csv')
    if not File.exist?(zones_csv)
      fail 'Could not find data_climate_zones.csv'
    end

    require 'csv'
    CSV.foreach(zones_csv) do |row|
      if row[0].to_s == wmo.to_s
        ba_zone = row[5].to_s
        break
      end
    end

    return ba_zone
  end
end