//===-- apps/ccam/ccam.cpp --------------------------------------*- C++ -*-===//
//
//                 The RoSA Framework -- Application CCAM
//
//===----------------------------------------------------------------------===//
///
/// \file apps/ccam/ccam.cpp
///
/// \author Maximilian Goetzinger (maximilian.goetzinger@tuwien.ac.at)
/// \author Benedikt Tutzer (benedikt.tutzer@tuwien.ac.at)
///
/// \date 2019
///
/// \brief The application CCAM implements the case study from the paper:
/// M. Goetzinger, N. TaheriNejad, H. A. Kholerdi, A. Jantsch, E. Willegger,
/// T. Glatzl, A.M. Rahmani, T.Sauter, P. Liljeberg: Model - Free Condition
/// Monitoring with Confidence
//===----------------------------------------------------------------------===//
#include "cxxopts/cxxopts.hpp"
#include "rosa/agent/Abstraction.hpp"
#include "rosa/agent/Confidence.hpp"
#include "rosa/agent/FunctionAbstractions.hpp"
#include <iostream>

#include "rosa/config/version.h"

#include "rosa/deluxe/DeluxeContext.hpp"

#include "rosa/support/csv/CSVReader.hpp"
#include "rosa/support/csv/CSVWriter.hpp"

#include <fstream>
#include <memory>
#include <streambuf>
/*
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wunused-parameter"
#pragma GCC diagnostic ignored "-Wmissing-field-initializers"
#pragma GCC diagnostic ignored "-Wreorder"
#pragma GCC diagnostic ignored "-Wunused-local-typedef"
#pragma GCC diagnostic ignored "-Wgnu-folding-constant"*/
#include "configuration.h"
//#pragma GCC diagnostic pop

using namespace rosa;
using namespace rosa::agent;
using namespace rosa::deluxe;
using namespace rosa::terminal;

const std::string AppName = "CCAM";

int main(int argc, char **argv) {

  LOG_INFO_STREAM << '\n'
                  << library_string() << " -- " << Color::Red << AppName
                  << "app" << Color::Default << '\n';

  cxxopts::Options Options(
      "CCAM", "Confidence-based Context-Aware condition Monitoring");

  // clang-format off
  Options.add_options()
    ("c,config", "Path to config file",
        cxxopts::value<std::string>()->default_value("config.json"))
    ("i,infile", "Path to input file",
        cxxopts::value<std::string>()->default_value("in.csv"))
    ("o,outfile", "Path to output file",
        cxxopts::value<std::string>()->default_value("out.csv"))
    ("b,broken", "Broken counter",
        cxxopts::value<std::uint32_t>()->default_value("10"))
    ("l,length", "Length of fuzzy function for match-check",
        cxxopts::value<std::uint32_t>()->default_value("10"))
    ("s,sig", "Signal",
        cxxopts::value<std::uint32_t>()->default_value("10"))
    ;
  // clang-format on

  auto ParsedOptions = Options.parse(argc, argv);
  std::string ConfigPath = ParsedOptions["config"].as<std::string>();

  if (!readConfigFile(ConfigPath)) {
    LOG_ERROR_STREAM << "Could not read config from \"" << ConfigPath << "\"\n";
    return 1;
  }

  /*
    std::string ConfigString((std::istreambuf_iterator<char>(ConfigFile)),
                             std::istreambuf_iterator<char>());

    AppConfiguration ConfigObj;
    json_object(s::InputFilePath, s::OutputFilePath, s::BrokenCounter,
    s::Length, s::SignalConfigurations = json_vector(s::Name, s::Output,
    s::InnerBound, s::OuterBound, s::InnerBoundDrift, s::OuterBoundDrift))
        .decode(ConfigString, ConfigObj);

    std::string test = iod::json_encode(ConfigObj);
    LOG_INFO_STREAM << test;
  */
  std::string InputFilePath, OutputFilePath;

  LOG_INFO("Creating Context\n");
  std::unique_ptr<DeluxeContext> C = DeluxeContext::create(AppName);

  //
  // Create deluxe sensors.
  //
  LOG_INFO("Creating sensors, SignalStateDetector functionalities and their "
           "Abstractions.");
  std::vector<AgentHandle> Sensors;
  std::vector<std::shared_ptr<PartialFunction<float, float>>>
      SampleMatchesFunctions;
  std::vector<std::shared_ptr<PartialFunction<float, float>>>
      SampleMismatchesFunctions;
  std::vector<std::shared_ptr<StepFunction<float, float>>>
      NumOfSamplesMatchFunctions;
  std::vector<std::shared_ptr<StepFunction<float, float>>>
      NumOfSamplesMismatchFunctions;
  for (auto SignalConfiguration : AppConfig.SignalConfigurations) {
    Sensors.emplace(Sensors.end(),
                    C->createSensor<float>(SignalConfiguration.Name));

    //
    // Create functionalities.
    //

    //
    // Create low-level deluxe agents with \c createLowLevelAgent.
    //

    /// AgentHandle HRAgent =
    ///    createLowLevelAgent(C, "HR Agent", HRConfidence, HRAbstraction);

    //
    // Connect sensors to low-level agents.
    //
    LOG_INFO("Connect sensors to their corresponding low-level agents.");

    /// C->connectSensor(HRAgent, 0, HRSensor, "HR Sensor Channel");
  }
  //
  // Create a high-level deluxe agent.
  //
  LOG_INFO("Create high-level agent.");

  // The new agent logs its input values and results in the the sum of them.
  /**  AgentHandle BodyAgent = C->createAgent(
        "Body Agent",
        DeluxeAgent::D<uint32_t, uint32_t, uint32_t, uint32_t, uint32_t,
                       uint32_t>(
            [](std::pair<uint32_t, bool> HR, std::pair<uint32_t, bool> BR,
               std::pair<uint32_t, bool> SpO2, std::pair<uint32_t, bool> BPSys,
               std::pair<uint32_t, bool> BodyTemp) -> Optional<uint32_t> {
              LOG_INFO_STREAM << "\n*******\nBody Agent trigged with values:\n"
                              << (HR.second ? "<New>" : "<Old>")
                              << " HR warning score: " << HR.first << "\n"
                              << (BR.second ? "<New>" : "<Old>")
                              << " BR warning score: " << BR.first << "\n"
                              << (SpO2.second ? "<New>" : "<Old>")
                              << " SpO2 warning score: " << SpO2.first << "\n"
                              << (BPSys.second ? "<New>" : "<Old>")
                              << " BPSys warning score: " << BPSys.first << "\n"
                              << (BodyTemp.second ? "<New>" : "<Old>")
                              << " BodyTemp warning score: " << BodyTemp.first
                              << "\n******\n";
              return {HR.first + BR.first + SpO2.first + BPSys.first +
                      BodyTemp.first};
            }));
  */
  //
  // Connect low-level agents to the high-level agent.
  //
  LOG_INFO("Connect low-level agents to the high-level agent.");

  /// C->connectAgents(BodyAgent, 0, HRAgent, "HR Agent Channel");

  //
  // For simulation output, create a logger agent writing the output of the
  // high-level agent into a CSV file.
  //
  LOG_INFO("Create a logger agent.");

  // Create CSV writer.
  /// std::ofstream ScoreCSV(ScoreCSVPath);
  /// csv::CSVWriter<uint32_t> ScoreWriter(ScoreCSV);

  // The agent writes each new input value into a CSV file and produces nothing.
  /** AgentHandle LoggerAgent = C->createAgent(
      "Logger Agent",
      DeluxeAgent::D<unit_t, uint32_t>(
          [&ScoreWriter](std::pair<uint32_t, bool> Score) -> Optional<unit_t> {
            if (Score.second) {
              // The state of \p ScoreWriter is not checked, expecting good.
              ScoreWriter << Score.first;
            }
            return {};
          }));
*/
  //
  // Connect the high-level agent to the logger agent.
  //
  LOG_INFO("Connect the high-level agent to the logger agent.");

  /// C->connectAgents(LoggerAgent, 0, BodyAgent, "Body Agent Channel");

  //
  // Do simulation.
  //
  LOG_INFO("Setting up and performing simulation.");

  //
  // Initialize deluxe context for simulation.
  //

  C->initializeSimulation();

  //
  // Open CSV files and register them for their corresponding sensors.
  //

  //
  // Simulate.
  //

  /// C->simulate(NumberOfSimulationCycles);

  return 0;
}