Scheduler.hpp

#pragma once
#include <samchon/protocol/Entity.hpp>
#include <samchon/examples/tsp/Travel.hpp>

#include <iostream>
#include <samchon/library/GeneticAlgorithm.hpp>

namespace samchon
{
namespace examples
{
namespace tsp
{
    using namespace std;

    using namespace library;
    using namespace protocol;

    struct GAParameters
    {
        double mutationRate;
        size_t tournament;

        size_t population;
        size_t generation;
    };

    class Scheduler
        : public Entity<>
    {
    private:
        typedef Entity<std::string> super;

    protected:
        shared_ptr<Travel> travel;

        struct GAParameters gaParameters;

    public:
        /* -----------------------------------------------------------
            CONSTRUCTORS
        ----------------------------------------------------------- */
        Scheduler()
            : super()
        {
            this->travel = make_shared<Travel>();
            this->gaParameters = {.03, 50, 100, 300};
        };

        Scheduler(shared_ptr<Travel> travel, const struct GAParameters &gaParameteres)
            : super()
        {
            this->travel = travel;
            this->gaParameters = gaParameteres;
        };

        virtual ~Scheduler() = default;

        virtual void construct(shared_ptr<XML> xml) override
        {
            gaParameters.mutationRate = xml->getProperty<double>("mutationRate");
            gaParameters.tournament = xml->getProperty<size_t>("tournament");
            gaParameters.population = xml->getProperty<size_t>("population");
            gaParameters.generation = xml->getProperty<size_t>("generation");

            travel->construct(xml->get(travel->TAG())->at(0));
        };

        /* -----------------------------------------------------------
            CALCULATORS
        ----------------------------------------------------------- */
        auto optimize() -> shared_ptr<Travel>
        {
            GeneticAlgorithm<Travel> geneticAlgorithm
                (
                    true,
                    gaParameters.mutationRate,
                    gaParameters.tournament
                );

            travel = 
                geneticAlgorithm.evolveGeneArray
                (
                    travel, 
                    gaParameters.population, 
                    gaParameters.generation
                );

            return travel;
        };

        auto calcDistance() const -> double
        {
            return travel->calcDistance();
        };

        /* -----------------------------------------------------------
            EXPORTER
        ----------------------------------------------------------- */
        virtual auto TAG() const -> string override
        {
            return "scheduler";
        };

        virtual auto toXML() const -> shared_ptr<XML> override
        {
            shared_ptr<XML> &xml = super::toXML();
            xml->setProperty("mutationRate", gaParameters.mutationRate);
            xml->setProperty("tournament", gaParameters.tournament);
            xml->setProperty("population", gaParameters.population);
            xml->setProperty("generation", gaParameters.generation);

            xml->push_back(travel->toXML());
            return xml;
        };

        auto toString() const -> string
        {
            return travel->toString();
        };

        /* -----------------------------------------------------------
            MAIN
        ----------------------------------------------------------- */
        static void main()
        {
            shared_ptr<Travel> travel = make_shared<Travel>();
            for(int i = 0; i < 30; i++)
                travel->emplace_back(new GeometryPoint(i + 1));

            // OPTIMIZING
            struct GAParameters gaParameters = {.03, 30, 400, 400};

            Scheduler scheduler(travel, gaParameters);
            travel = scheduler.optimize();

            // PRINTING
            string &str = travel->toString();

            cout << str << endl;
        };
    };          
};
};
};