include/WaveSimulation.hpp

@@ -0,0 +1,32 @@
+/** @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