/**
 * @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::S_Integrator,
                               artis::qss::Integrator<artis::common::DoubleTime>::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::S_Integrator,
                 artis::qss::Integrator<artis::common::DoubleTime>::VALUE});
        selector("PreyView",
                {PreyPredatorGraphManager::PREY, PreyGraphManager::S_Integrator,
                 artis::qss::Integrator<artis::common::DoubleTime>::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);

    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::S_Integrator,
                 artis::qss::Integrator<artis::common::DoubleTime>::VALUE});
        selector("PreyView",
                {PreyPredatorSmartGardenerGraphManager::PREY_PREDATOR,
                 PreyPredatorGraphManager::PREY, PreyGraphManager::S_Integrator,
                 artis::qss::Integrator<artis::common::DoubleTime>::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> output(rc.observer());

    output(0, 100, 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::S_Integrator,
                 artis::qss::Integrator<artis::common::DoubleTime>::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> output(rc.observer());

    output(0, 100, 0.1);
}

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