simulator.hpp

#pragma once
 
#include <string>
#include <vector>
#include <memory>
#include <optional>
#include <armadillo>
 
// can replace some of these with forward declarations, but keep them for now
// so the program is easier to use (no need to include so much stuff)
// TODO: should probably make a "transflow.hpp" file with the most useful stuff
// included
#include "pipeline.hpp"
#include "solver/solver.hpp"
#include "physics.hpp"
#include "timeseries.hpp"
#include "config.hpp"
#include "sampler.hpp"
 
class Simulator
{
public:
    Simulator(
            const Pipeline& pipeline = Pipeline(),
            const Config& config = Config());
 
    Simulator(
            const Pipeline& pipeline,
            std::unique_ptr<Physics>& physics,
            std::unique_ptr<Solver>& solver);
 
    arma::vec simulate(const TimeSeries& timeSeries);
 
    void enableBatchTracking();
 
    arma::uword size() const { return m_state->size(); }
 
    const Pipeline& pipeline() const { return *m_state; }
 
    const Pipeline::State& state() const { return m_state->state(); }
 
    const Solver& solver() const { return *m_solver; }
 
    const Physics& physics() const { return *m_physics; }
 
    Sampler& sampler() { return m_sampler.value(); }
 
private:
    std::unique_ptr<Pipeline> m_state; 
    std::unique_ptr<Physics> m_physics; 
    std::unique_ptr<Solver> m_solver; 
    std::optional<Sampler> m_sampler; 
 
    std::optional<Sampler> makeSampler(const Config& config);
};