ABCSMCController.h

#ifndef ABCSMCCONTROLLER_H
#define ABCSMCCONTROLLER_H

#include <string>
#include <vector>
#include <queue>
#include <memory>
#include <random>
#include <chrono>

#include "core/Command.h"

#include "AbstractController.h"

class LongOptions;
class Arguments;

class ABCSMCController : public AbstractController
{
    public:

        // Forward declaration of Input
        struct Input;

        ABCSMCController(const Input &input_obj);

        virtual ~ABCSMCController() override = default;

        virtual void iterate() override;

        virtual Command getSimulator() const override;

        static std::string help();

        static void addLongOptions(LongOptions& lopts);

        static ABCSMCController* makeController(const Arguments& args);

        struct Input
        {
            static Input makeInput(const Arguments& args);

            int population_size;

            std::vector<Epsilon> epsilons;

            Command simulator;

            std::vector<ParameterName> parameter_names;

            Command prior_sampler;

            Command perturber;

            Command prior_pdf;

            Command perturbation_pdf;

            unsigned long seed =
                std::chrono::system_clock::now().time_since_epoch().count();
        };

    private:

        // Sample parameter
        Parameter sampleParameter();

        // Epsilons
        std::vector<Epsilon> m_epsilons;

        // Iteration counter
        int m_t = 0;

        // Perturbation pdf for weights calculation
        Command m_perturbation_pdf;

        // Parameter names
        std::vector<ParameterName> m_parameter_names;

        // Population size
        int m_population_size;

        // New accepted parameters
        std::vector<Parameter> m_prmtr_accepted_new;

        // New weights
        std::vector<double> m_weights_new;

        // Number of parameters simulated
        int m_number_simulated = 0;

        // Simulator command
        Command m_simulator;

        // Prior pdf values for accepted parameters
        std::vector<double> m_prior_pdf_accepted;

        // Uniform distribution for sampling from population
        std::uniform_real_distribution<double> m_distribution;

        // Random number generator
        std::mt19937_64 m_generator;

        // Prior pdf values of pending parameters
        std::queue<double> m_prior_pdf_pending;

        // Parameters accepted in previous generation
        std::vector<Parameter> m_prmtr_accepted_old;

        // Weights of parameters accepted in previous generation
        std::vector<double> m_weights_old;

        // Cumulative sum of weights
        std::vector<double> m_weights_cumsum;

        // Prior sampler command
        Command m_prior_sampler;

        // Perturber command
        Command m_perturber;

        // Prior_pdf command
        Command m_prior_pdf;

        // First iteration
        bool m_first = true;

        // Entered iterate()
        bool m_entered = false;
};

#endif // ABCSMCCONTROLLER_H