/** @file WaveSimulation.hpp
 *
 *  @author Cory Alexander Balaton (coryab)
 *  @author Janita Ovidie Sandtrøen Willumsen (janitaws)
 *
 *  @version 0.1
 *
 *  @brief The definition of the WaveSimulation class
 *
 *  @bug No known bugs
 * */
#ifndef __WAVE_SIMULATION__
#define __WAVE_SIMULATION__

#include <armadillo>

class WaveSimulation {
protected:
    int M;
    arma::cx_mat U;
    arma::cx_mat V;
    arma::cx_mat A;
    arma::cx_mat B;
    double h;
    double dt;
public:
    virtual void solve() = 0;
    void build_A(arma::cx_vec A_vec);
    void build_B(arma::cx_vec B_vec);
};

#endif