WaveSimulation.hpp

Go to the documentation of this file.

 
#ifndef __WAVE_SIMULATION__
#define __WAVE_SIMULATION__
 
#include "constants.hpp"
#include "literals.hpp"
 
#include <armadillo>
#include <cstdint>
 
class WaveSimulation {
private:
    int32_t M;     
    int32_t N;     
    arma::cx_mat V; 
    arma::cx_mat U; 
    arma::sp_cx_mat B; 
    arma::sp_cx_mat A; 
    double h; 
    double dt; 
    double T; 
 
    void initialize_U(double x_c, double y_c, double sigma_x, double sigma_y,
                      double p_x, double p_y);
 
    void initialize_V(double thickness, double pos_x,
                      double aperture_separation, double aperture,
                      uint32_t slits);
 
    void initialize_V();
 
    void initialize_A();
 
    void initialize_B();
 
    void write_U(std::ofstream &ofile);
public:
    WaveSimulation(double h, double dt, double T, double x_c, double y_c,
                   double sigma_x, double sigma_y, double p_x, double p_y,
                   double thickness, double pos_x, double ap_sep, double ap,
                   uint32_t slits);
 
    WaveSimulation(double h, double dt, double T, double x_c, double y_c,
                   double sigma_x, double sigma_y, double p_x, double p_y);
 
    void step();
 
    void simulate(std::string outfile, bool write_each_step = false);
    
    void simulate(std::string outfile, std::vector<double> &steps);
 
    void probability_deviation(std::string outfile,
                               bool write_each_step = false);
};
 
#endif