sim_dplangevin.hpp

Go to the documentation of this file.

 
 
#ifndef SIMDP_HPP
#define SIMDP_HPP
 
#include "dplangevin.hpp"

class SimDP 
{
private:
    Coefficients coefficients;
    Parameters p;
    rng_t *rng; 
    DPLangevin *dpLangevin;
    void (DPLangevin::*integrator)(rng_t&);
    
    int n_epochs;
    int i_current_epoch;
    int i_next_epoch;
    double t_current_epoch;
    double t_next_epoch;
    dbl_vec_t t_epochs;
    dbl_vec_t mean_densities;
    py_array_t pyarray_t_epochs;
    py_array_t pyarray_mean_densities;
    py_array_t pyarray_density;
    bool did_integrate = false;
    bool is_initialized = false;

    int count_epochs() const;
    bool choose_integrator();
    bool integrate(const int n_next_epochs);

    bool pyprep_t_epochs();
    bool pyprep_mean_densities();
    bool pyprep_density_grid();
 
public:
    SimDP(
        const double linear, const double quadratic,
        const double diffusion, const double noise, 
        const double t_final, 
        const double dx, const double dt, 
        const int random_seed,
        const GridDimension grid_dimension,
        const int_vec_t grid_size,
        const gt_vec_t grid_topologies,
        const bc_vec_t boundary_conditions,
        const dbl_vec_t bc_values,
        const InitialCondition initial_condition,
        const dbl_vec_t ic_values,
        const IntegrationMethod integration_method
    );
    bool initialize();
    bool run(const int n_next_epochs);
    bool postprocess();
 
    // Utilities provided to Python via the wrapper

    int get_n_epochs() const;
    int get_i_current_epoch() const;
    int get_i_next_epoch() const;
    double get_t_current_epoch() const;
    double get_t_next_epoch() const;
    py_array_t get_t_epochs() const;
    py_array_t get_mean_densities() const;
    py_array_t get_density() const;
};
 
 
#endif