MonteCarloTransport.h

#ifndef MLMC_MONTECARLOTRANSPORT_HPP
#define MLMC_MONTECARLOTRANSPORT_HPP

#include "MonteCarlo.h"
#include "assemble/DGTransportAssemble.h"
#include "timestepping/TimeIntegrator.h"
//#include "timestepping/TimeIntegrator2.h"


class MonteCarloTransport : public MonteCarlo {
private:
    Plot plot;

    void solveTransport(int l,
                        bool fineLevel,
                        double &valueQ,
                        double &cost,
                        Vector &solution);

    TimeSeries getTimeSeries(bool fineLevel);

public:
    DGTransportAssemble *assemble;
    double startTime = 0.0;
    double endTime = 0.0;

    MatrixGraphs fluxMatrixGraphs;
    CellMatrixGraphs solMatrixGraphs;

    Transfer *transfer;

    MonteCarloTransport(int l, int dM, bool baseLevel, Meshes *meshes,
                        StochasticField *stochFields, DGTransportAssemble *assemble) :
        MonteCarlo(l, dM, baseLevel, meshes, stochFields),
        assemble(assemble),
        fluxMatrixGraphs(MatrixGraphs(*meshes, dof("cell", 3))),
        solMatrixGraphs(*meshes, *assemble->disc),
        plot(Plot((*meshes)[l], meshes->dim())) {

        config.get("startTime", startTime);
        config.get("endTime", endTime);

        transfer = new MatrixTransfer(*assemble);
        if (meshes->dim() != 1)
            transfer->Construct((solMatrixGraphs)[l - pLevel], (solMatrixGraphs)[l - pLevel - 1]);
    }

    ~MonteCarloTransport() {
        transfer->Destruct();
        delete transfer;
    }

    void UpscaleSolution(const Vector &coarseSol, Vector &coarseSolUp) {
        coarseSolUp = (*transfer) * coarseSol;
    }

    static void UpscaleFlux(const Vector &coarseFlux, Vector &coarseFluxUp) {
        coarseFluxUp = 0;
        for (cell c = coarseFlux.cells(); c != coarseFlux.cells_end(); c++) {
            row coarseRow = coarseFlux.find_row(c());
            for (int k = 0; k < c.Children(); ++k) {
                row fineRow = coarseFluxUp.find_row(c.Child(k));
                for (int i = 0; i < fineRow.n(); ++i)
                    coarseFluxUp(fineRow, i) += coarseFlux(coarseRow, i);
            }
        }
    }

    void Method() override;
};

#endif //MLMC_MONTECARLOTRANSPORT_HPP