added BTCSv2.cpp

This commit is contained in:
philippun 2023-08-10 14:21:00 +02:00
parent cc46a893a4
commit 2810db0d52
3 changed files with 52 additions and 1 deletions

View File

@ -7,7 +7,6 @@
#ifndef BOUNDARY_H_
#define BOUNDARY_H_
#include <Eigen/Core>
#include <cstddef>
#include "Grid.hpp"

View File

@ -6,6 +6,7 @@
*/
#include <Eigen/Core>
#include <Eigen/Sparse>
using namespace Eigen;

51
src/BTCSv2.cpp Normal file
View File

@ -0,0 +1,51 @@
#include "TugUtils.hpp"
#include <tug/Boundary.hpp>
using namespace Eigen;
// calculates arithmetic or harmonic mean of alpha between two cells
static double calcAlphaIntercell(double &alpha1, double &alpha2, bool useHarmonic = true) {
if (useHarmonic) {
return double(2) / ((double(1)/alpha1) + (double(1)/alpha2));
} else {
return 0.5 * (alpha1 + alpha2);
}
}
static MatrixXd createCoeffMatrix(Grid &grid, int rowIndex, double sx) {
// square matrix of column^2 dimension for the coefficients
int dim = grid.getCol();
SparseMatrix<double, RowMajor> cm(dim, dim);
// top left
cm.coeffRef(0,0) = 1 + sx * (calcAlphaIntercell(grid.getAlphaX()(rowIndex,0), grid.getAlphaX()(rowIndex,1)));
}
// BTCS solution for 1D grid
static void BTCS_1D(Grid &grid, Boundary &bc, double &timestep) {
}
// BTCS solution for 2D grid
static void BTCS_2D(Grid &grid, Boundary &bc, double &timestep) {
}
// entry point; differentiate between 1D and 2D grid
static void BTCS(Grid &grid, Boundary &bc, double &timestep) {
if (grid.getDim() == 1) {
BTCS_1D(grid, bc, timestep);
} else if (grid.getDim() == 2) {
BTCS_2D(grid, bc, timestep);
} else {
throw_invalid_argument("Error: Only 1- and 2-dimensional grids are defined!");
}
}