class URBANopt::Scenario::ScenarioDefaultPostProcessor

# File lib/urbanopt/scenario/scenario_post_processor_default.rb, line 20
def initialize(scenario_base)
  super(scenario_base)

  @initialization_hash = { directory_name: scenario_base.run_dir, name: scenario_base.name, id: scenario_base.name, root_dir: scenario_base.root_dir }
  @scenario_result = URBANopt::Reporting::DefaultReports::ScenarioReport.new(@initialization_hash)
  @default_save_name = 'default_scenario_report'

  @@logger ||= URBANopt::Reporting::DefaultReports.logger
end

# File lib/urbanopt/scenario/scenario_post_processor_default.rb, line 46
def add_simulation_dir(simulation_dir)
  feature_reports = URBANopt::Reporting::DefaultReports::FeatureReport.from_simulation_dir(simulation_dir)

  feature_reports.each do |feature_report|
    if feature_report.to_hash[:simulation_status] == 'Complete'
      @scenario_result.add_feature_report(feature_report)
    else
      @@logger.error("Feature #{feature_report.id} failed to run!")
    end
  end

  return feature_reports
end

# File lib/urbanopt/scenario/scenario_post_processor_default.rb, line 65
def create_scenario_db_file(file_name = @default_save_name)
  new_db_file = File.join(@initialization_hash[:directory_name], "#{file_name}.db")
  scenario_db = SQLite3::Database.open new_db_file
  scenario_db.execute "CREATE TABLE IF NOT EXISTS ReportData(
    TimeIndex INTEGER,
    Year VARCHAR(255),
    Month VARCHAR(255),
    Day VARCHAR(255),
    Hour VARCHAR(255),
    Minute VARCHAR(255),
    Dst INTEGER,
    FuelType VARCHAR(255),
    Value INTEGER,
    FuelUnits VARCHAR(255)
    )"

  values_arr = []
  feature_list = Pathname.new(@initialization_hash[:directory_name]).children.select(&:directory?) # Folders in the run/scenario directory

  # get scenario CSV
  scenario_csv = File.join(@initialization_hash[:root_dir], "#{@initialization_hash[:name]}.csv")
  if File.exist?(scenario_csv)
    # csv found
    feature_ids = CSV.read(scenario_csv, headers: true)
    feature_list = []
    # loop through building feature ids from scenario csv
    feature_ids['Feature Id'].each do |feature|
      if Dir.exist?(File.join(@initialization_hash[:directory_name], feature))
        feature_list << File.join(@initialization_hash[:directory_name], feature)
      else
        puts "warning: did not find a directory for datapoint #{feature}...skipping"
      end
    end
  else
    raise "Couldn't find scenario CSV: #{scenario_csv}"
  end
  feature_list.each do |feature| # Loop through each feature in the scenario
    uo_output_sql_file = File.join(@initialization_hash[:directory_name], File.basename(feature), 'eplusout.sql')
    feature_db = SQLite3::Database.open uo_output_sql_file
    # Doing "db.results_as_hash = true" is prettier, but in this case significantly slower.

    elec_query = feature_db.query "SELECT ReportData.TimeIndex, Time.Year, Time.Month, Time.Day, Time.Hour,
      Time.Minute, Time.Dst, ReportData.Value
    FROM ReportData
    INNER JOIN Time ON Time.TimeIndex=ReportData.TimeIndex
    INNER JOIN ReportDataDictionary AS rddi ON rddi.ReportDataDictionaryIndex=ReportData.ReportDataDictionaryIndex
    WHERE rddi.IndexGroup == 'Facility:Electricity'
    AND rddi.ReportingFrequency == 'Zone Timestep'
    AND Time.Year > 1900
    ORDER BY ReportData.TimeIndex"

    elec_query.each do |row| # Add up all the values for electricity usage across all Features at this timestep
      # row[0] == TimeIndex, row[1] == Value

      arr_match = values_arr.find { |v| v[:time_index] == row[0] }
      if arr_match.nil?
        # add new row to value_arr
        values_arr << { time_index: row[0], year: row[1], month: row[2], day: row[3], hour: row[4], minute: row[5], dst: row[6], elec_val: Float(row[7]), gas_val: 0 }
      else
        # running sum
        arr_match[:elec_val] += Float(row[7])
      end
    end
    elec_query.close

    gas_query = feature_db.query "SELECT ReportData.TimeIndex, Time.Year, Time.Month, Time.Day, Time.Hour,
      Time.Minute, Time.Dst, ReportData.Value
    FROM ReportData
    INNER JOIN Time ON Time.TimeIndex=ReportData.TimeIndex
    INNER JOIN ReportDataDictionary AS rddi ON rddi.ReportDataDictionaryIndex=ReportData.ReportDataDictionaryIndex
    WHERE rddi.IndexGroup == 'Facility:Gas'
    AND rddi.ReportingFrequency == 'Zone Timestep'
    AND Time.Year > 1900
    ORDER BY ReportData.TimeIndex"

    gas_query.each do |row|
      # row[0] == TimeIndex, row[1] == Value
      arr_match = values_arr.find { |v| v[:time_index] == row[0] }
      if arr_match.nil?
        # add new row to value_arr
        values_arr << { time_index: row[0], year: row[1], month: row[2], day: row[3], hour: row[4], minute: row[5], dst: row[6], gas_val: Float(row[7]), elec_val: 0 }
      else
        # running sum
        arr_match[:gas_val] += Float(row[7])
      end
    end
    gas_query.close
    feature_db.close
  end

  elec_sql = []
  gas_sql = []
  values_arr.each do |i|
    elec_sql << "(#{i[:time_index]}, #{i[:year]}, #{i[:month]}, #{i[:day]}, #{i[:hour]}, #{i[:minute]}, #{i[:dst]}, 'Electricity', #{i[:elec_val]}, 'J')"
    gas_sql << "(#{i[:time_index]}, #{i[:year]}, #{i[:month]}, #{i[:day]}, #{i[:hour]}, #{i[:minute]}, #{i[:dst]}, 'Gas', #{i[:gas_val]}, 'J')"
  end

  # Put summed Values into the database
  scenario_db.execute("INSERT INTO ReportData VALUES #{elec_sql.join(', ')}")
  scenario_db.execute("INSERT INTO ReportData VALUES #{gas_sql.join(', ')}")
  scenario_db.close
end

# File lib/urbanopt/scenario/scenario_post_processor_default.rb, line 33
def run
  # this run method adds all the simulation_dirs, you can extend it to do more custom stuff
  @scenario_base.simulation_dirs.each do |simulation_dir|
    add_simulation_dir(simulation_dir)
  end
  return @scenario_result
end

# File lib/urbanopt/scenario/scenario_post_processor_default.rb, line 173
def save(file_name = @default_save_name)
  @scenario_result.save

  return @scenario_result
end