MLMCMain.h

#ifndef MLMC_MLMCMAIN_H
#define MLMC_MLMCMAIN_H

#include "assemble/AssembleHeader.h"
#include "problem/StochasticEllipticProblem.h"
#include "problem/StochasticTransportProblem.h"
#include "MultilevelMonteCarlo.h"
#include "discgalerkin/DGDiscretization.h"

class MLMCMain {
public:
    int enablePython = 1;
    double epsilon = 0.1;
    int size_init = 3, plevel = 2, Lmax = 8, size_lst = 3;
    vector<int> l_init = {3, 4, 5}, M_init = {12, 6, 3};
    vector<double> epsilon_lst = {0.1, 0.05, 0.01};
    string problemName = "", modelName = "";
    string fieldName = "", experimentName = "";

    Meshes *meshes = nullptr;
    Discretization *disc = nullptr;
    StochasticFields *stochFields = nullptr;
    StochasticProblem *problem = nullptr;
    Assemble *assemble = nullptr;
    MatrixGraphs *graphs = nullptr;

    MultilevelMonteCarlo *mlmc = nullptr;

    MLMCMain() {
        ReadConfig(Settings, "enablePython", enablePython);
        ReadConfig(Settings, "epsilon", epsilon);
        ReadConfig(Settings, "plevel", plevel);
        ReadConfig(Settings, "Lmax", Lmax);
        ReadConfig(Settings, "size_init", size_init);
        ReadConfig(Settings, "l_init", l_init, size_init);
        ReadConfig(Settings, "M_init", M_init, size_init);
        ReadConfig(Settings, "size_lst", size_lst);
        ReadConfig(Settings, "epsilon_lst", epsilon_lst, size_lst);
        ReadConfig(Settings, "Problem", problemName);
        ReadConfig(Settings, "Model", modelName);
        ReadConfig(Settings, "StochasticField", fieldName);
        ReadConfig(Settings, "Experiment", experimentName);

        initialize();
    }

    ~MLMCMain() {
        delete meshes, delete disc, delete stochFields;
        delete problem, delete assemble, delete graphs;
        delete mlmc;
    }

    void initialize();

    Meshes *getMeshes();

    Discretization *getDiscretization();

    StochasticFields *getStochasticFields();

    StochasticProblem *getStochasticProblem();

    Assemble *getAssemble();

    MatrixGraphs *getMatrixGraphs();

    void run();

    static void headConvergenceTest() {
        mout << endl << "**********************************************************"
             << endl << "*** Convergence tests, kurtosis, telescoping sum check ***"
             << endl << "**********************************************************"
             << endl;
    }

    static void headMLMCExperiment() {
        mout << endl << "**********************************************************"
             << endl << "***           MultilevelMonteCarlo test run            ***"
             << endl << "**********************************************************"
             << endl;
    }

    static void headMLMCOverEpsilon() {
        mout << endl << "**********************************************************"
             << endl << "***         MultilevelMonteCarlo over epsilon          ***"
             << endl << "**********************************************************"
             << endl;
    }
};

#endif //MLMC_MLMCMAIN_H