/**
 * @file pdevs/tests.cpp
 * @author The ARTIS Development Team
 * See the AUTHORS or Authors.txt file
 */

/*
 * ARTIS - the multimodeling and simulation environment
 * This file is a part of the ARTIS environment
 *
 * Copyright (C) 2013-2019 ULCO http://www.univ-littoral.fr
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

#include <tests/qss/graph_manager.hpp>

#include <artis-star/common/RootCoordinator.hpp>
#include <artis-star/common/observer/Iterator.hpp>
#include <artis-star/common/observer/Output.hpp>

#include <fstream>
#include <iostream>
#include <boost/archive/binary_oarchive.hpp>

using namespace artis::tests::qss;

void test_parabola()
{
  artis::qss::QSSParameters<ParabolaParameters> parameters = {{0.5},
                                                              {true, true, 0.001, 3},
                                                              {0.2}};

  class View : public artis::observer::View<artis::common::DoubleTime>
  {
  public:
    View()
    {
      selector("Value",
               {OnlyOneParabolaGraphManager::A, ParabolaGraphManager::submodel::S_Integrator,
                artis::qss::Integrator<artis::common::DoubleTime>::var::VALUE});
    }
  };

  artis::common::context::Context<artis::common::DoubleTime> context(0, 5);
  artis::common::RootCoordinator<
      artis::common::DoubleTime, artis::pdevs::Coordinator<
          artis::common::DoubleTime,
          OnlyOneParabolaGraphManager,
          artis::qss::QSSParameters<ParabolaParameters>>
  > rc(context, "root", parameters, artis::common::NoParameters());

  rc.attachView("Value", new View());

  rc.run(context);

  const View::Values &values = rc.observer().view("Value").get("Value");

  for (const auto &value: values) {
    double v;

    value.second(v);
    std::cout << value.first << ": " << v << std::endl;
  }
}

class PredatorPreyView : public artis::observer::View<artis::common::DoubleTime>
{
public:
  PredatorPreyView()
  {
    selector("PredatorView",
             {PreyPredatorGraphManager::PREDATOR, PredatorGraphManager::submodel::S_Integrator,
              artis::qss::Integrator<artis::common::DoubleTime>::var::VALUE});
    selector("PreyView",
             {PreyPredatorGraphManager::PREY, PreyGraphManager::submodel::S_Integrator,
              artis::qss::Integrator<artis::common::DoubleTime>::var::VALUE});
  }
};

void run_predator_prey(artis::common::context::Context<artis::common::DoubleTime> &context,
                       const PreyPredatorGraphManagerParameters &parameters)
{
  artis::common::RootCoordinator<
      artis::common::DoubleTime, artis::pdevs::Coordinator<
          artis::common::DoubleTime,
          PreyPredatorGraphManager,
          PreyPredatorGraphManagerParameters>
  > rc(context, "root", parameters, artis::common::NoParameters());

  rc.attachView("Value1", new PredatorPreyView());

  rc.run(context);

  artis::observer::Output<artis::common::DoubleTime,
                          artis::observer::TimedIterator<artis::common::DoubleTime>>
      output(rc.observer());

  output(context.begin(), context.end(), {context.begin(), 0.1});

  //  rc.save(context);
//
//  std::ofstream os("state");
//  boost::archive::binary_oarchive oa(os);
//
//  oa << context;
//
//  {
//    artis::observer::DiscreteTimeIterator<artis::common::DoubleTime> it(
//        rc.observer().view("Value").get("PredatorView"), context.begin(), 0.1);
//
//    while (it.has_next()) {
//      double v;
//
//      (*it).second(v);
//      std::cout << (*it).first << ";" << v << std::endl;
//      ++it;
//    }
//  }
//  {
//    artis::observer::DiscreteTimeIterator<artis::common::DoubleTime> it(
//        rc.observer().view("Value").get("PreyView"), context.begin(), 0.1);
//
//    while (it.has_next()) {
//      double v;
//
//      (*it).second(v);
//      std::cout << (*it).first << ";" << v << std::endl;
//      ++it;
//    }
//  }
}

void test_predator_prey()
{
  PreyPredatorGraphManagerParameters parameters = {{{45.},
                                                    {true, true, 0.1, 3},
                                                    {0.5, 0.01, 0.01, 0.2}},
                                                   {{5000.},
                                                    {true, true, 1, 3},
                                                    {0.5, 0.01, 0.01, 0.2}}};

  artis::common::context::Context<artis::common::DoubleTime> context(0, 100);

  run_predator_prey(context, parameters);

//  artis::common::context::Context<artis::common::DoubleTime> new_context(context);
//
//  new_context.end(200);
//  run_predator_prey(new_context, parameters);
}

class PredatorPreySmartGardenerView : public artis::observer::View<artis::common::DoubleTime>
{
public:
  PredatorPreySmartGardenerView()
  {
    selector("PredatorView",
             {PreyPredatorSmartGardenerGraphManager::PREY_PREDATOR,
              PreyPredatorGraphManager::PREDATOR, PredatorGraphManager::submodel::S_Integrator,
              artis::qss::Integrator<artis::common::DoubleTime>::var::VALUE});
    selector("PreyView",
             {PreyPredatorSmartGardenerGraphManager::PREY_PREDATOR,
              PreyPredatorGraphManager::PREY, PreyGraphManager::submodel::S_Integrator,
              artis::qss::Integrator<artis::common::DoubleTime>::var::VALUE});
  }
};

void test_predator_prey_smart_gardener()
{
  PreyPredatorSmartGardenerGraphManagerParameters parameters = {{{{45.},
                                                                  {true, true, 0.1, 3},
                                                                  {0.5, 0.01, 0.01, 0.2}},
                                                                 {{5000.},
                                                                  {true, true, 1, 3},
                                                                  {0.5, 0.01, 0.01, 0.2}}},
                                                                {2000, 0.25, 0.75, 5}};

  artis::common::context::Context<artis::common::DoubleTime> context(0, 100);
  artis::common::RootCoordinator<
      artis::common::DoubleTime, artis::pdevs::Coordinator<
          artis::common::DoubleTime,
          PreyPredatorSmartGardenerGraphManager,
          PreyPredatorSmartGardenerGraphManagerParameters>
  > rc(context, "root", parameters, artis::common::NoParameters());

  rc.attachView("Value2", new PredatorPreySmartGardenerView());

  rc.run(context);

  artis::observer::Output<artis::common::DoubleTime,
                          artis::observer::TimedIterator<artis::common::DoubleTime>>
      output(rc.observer());

  output(context.begin(), context.end(), {context.begin(), 0.1});
}

class MixedPredatorPreyView : public artis::observer::View<artis::common::DoubleTime>
{
public:
  MixedPredatorPreyView()
  {
    selector("PredatorView",
             {MixedPreyPredatorGraphManager::PREDATOR, DiscretePredatorGraphManager::PREDATOR,
              DiscretePredator::VALUE});
    selector("PreyView",
             {MixedPreyPredatorGraphManager::PREY, PreyGraphManager::submodel::S_Integrator,
              artis::qss::Integrator<artis::common::DoubleTime>::var::VALUE});
  }
};

void test_mixed_predator_prey()
{
  MixedPreyPredatorGraphManagerParameters parameters = {{0.0001, 45., 0.5, 0.01, 0.01, 0.2},
                                                        {{5000.},
                                                         {true, true, 1, 3},
                                                         {0.5, 0.01, 0.01, 0.2}}};
  artis::common::context::Context<artis::common::DoubleTime> context(0, 100);
  artis::common::RootCoordinator<
      artis::common::DoubleTime, artis::pdevs::Coordinator<
          artis::common::DoubleTime,
          MixedPreyPredatorGraphManager,
          MixedPreyPredatorGraphManagerParameters>
  > rc(context, "root", parameters, artis::common::NoParameters());

  rc.attachView("Value3", new MixedPredatorPreyView());

  rc.run(context);

  artis::observer::Output<artis::common::DoubleTime,
                          artis::observer::TimedIterator<artis::common::DoubleTime>>
      output(rc.observer());

  output(context.begin(), context.end(), {context.begin(), 0.1});
}

class MultiPredatorPreyView : public artis::observer::View<artis::common::DoubleTime>
{
public:
  MultiPredatorPreyView()
  {
    selector("PredatorView",
             {MultiPreyPredatorGraphManager::submodel::S_Integrator,
              artis::qss::Integrator<artis::common::DoubleTime>::var::VALUE});
    selector("PreyView",
             {MultiPreyPredatorGraphManager::submodel::S_Integrator + 1,
              artis::qss::Integrator<artis::common::DoubleTime>::var::VALUE});
    selector("PredatorDerivativeView",
             {MultiPreyPredatorGraphManager::submodel::S_Derivative,
              PreyPredator::var::VALUE});
    selector("PreyDerivativeView",
             {MultiPreyPredatorGraphManager::submodel::S_Derivative,
              PreyPredator::var::VALUE + 1});
  }
};

void test_multi_predator_prey()
{
  artis::qss::MultiQSSParameters<PreyPredatorParameters> parameters = {
      {{5000.}, {45.}},
      {{true, true, 1, 3}, {true, true, 0.1, 3}},
      {0.5, 0.01, 0.01, 0.2}
  };
  artis::common::context::Context<artis::common::DoubleTime> context(0, 100);
  artis::common::RootCoordinator<
      artis::common::DoubleTime, artis::pdevs::Coordinator<
          artis::common::DoubleTime,
          MultiPreyPredatorGraphManager,
          artis::qss::MultiQSSParameters<PreyPredatorParameters>>
  > rc(context, "root", parameters, artis::common::NoParameters());

  rc.attachView("Value4", new MultiPredatorPreyView());

  rc.run(context);

  artis::observer::Output<artis::common::DoubleTime,
                          artis::observer::TimedIterator<artis::common::DoubleTime>>
      output(rc.observer());

  output(context.begin(), context.end(), {context.begin(), 0.1});
}

int main()
{
  test_parabola();
  test_predator_prey();
  test_predator_prey_smart_gardener();
  test_mixed_predator_prey();
  test_multi_predator_prey();
}