max/TugJulia
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

use internal boundary condition mechanism

Browse Source
This commit is contained in:
Max Luebke 2021-12-13 13:23:42 +01:00
parent 0a5c271a4b
commit 496e4ee3bc
2 changed files with 28 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
src/BTCSDiffusion.cpp
View File

@ -97,7 +97,9 @@ void BTCSDiffusion::simulate1D(std::vector<double> &c, double bc_left,
// b[A_line] = this->bc[1]; // b[A_line] = this->bc[1];
/* /*
* Begin to solve the equation system * Begin to solve the equation system using LU solver of Eigen.
*
* But first fill the A matrix and b vector.
* *
* At this point there is some debugging output in the code. * At this point there is some debugging output in the code.
* TODO: remove output * TODO: remove output
@ -110,7 +112,7 @@ void BTCSDiffusion::simulate1D(std::vector<double> &c, double bc_left,
A_matrix.insert(size - 1, size - 1) = 1; A_matrix.insert(size - 1, size - 1) = 1;
b_vector[0] = bc_left; b_vector[0] = bc_left;
b_vector[size-1] = bc_right; b_vector[size - 1] = bc_right;
for (int i = 1; i < this->dim_x + 1; i++) { for (int i = 1; i < this->dim_x + 1; i++) {
double sx = (alpha[i - 1] * time_step) / (dx * dx); double sx = (alpha[i - 1] * time_step) / (dx * dx);
@ -130,7 +132,7 @@ void BTCSDiffusion::simulate1D(std::vector<double> &c, double bc_left,
// A_line++; // A_line++;
} }
std::cout << A_matrix << std::endl; // std::cout << A_matrix << std::endl;
// Eigen::SparseMatrix<double> A(size, size); // Eigen::SparseMatrix<double> A(size, size);
// A.setFromTriplets(tripletList.begin(), tripletList.end()); // A.setFromTriplets(tripletList.begin(), tripletList.end());
@ -159,22 +161,32 @@ void BTCSDiffusion::setTimestep(double time_step) {
void BTCSDiffusion::simulate(std::vector<double> &c, void BTCSDiffusion::simulate(std::vector<double> &c,
std::vector<double> &alpha) { std::vector<double> &alpha) {
if (this->grid_dim == 1) { if (this->grid_dim == 1) {
// double bc_left = getBCFromTuple(0); double bc_left = getBCFromTuple(0, c[0]);
// double bc_right = getBCFromTuple(1); double bc_right = getBCFromTuple(1, c[c.size() - 1]);
double bc_left = 5. * std::pow(10,-6); // double bc_left = 5. * std::pow(10,-6);
double bc_right = c[this->dim_x -1]; // double bc_right = c[this->dim_x -1];
simulate1D(c, bc_left, bc_right, alpha); simulate1D(c, bc_left, bc_right, alpha);
} }
} }
double BTCSDiffusion::getBCFromTuple(int index, std::vector<double> &c) { double BTCSDiffusion::getBCFromTuple(int index, double nearest_value) {
double val = std::get<0>(bc[index]); double val;
int type = std::get<0>(bc[index]);
if (type == BTCSDiffusion::BC_NEUMANN) {
// TODO implement gradient here
val = nearest_value;
} else if (type == BTCSDiffusion::BC_DIRICHLET) {
val = std::get<1>(bc[index]);
} else {
// some error handling here. Type was set to wrong value.
}
return val; return val;
} }
void BTCSDiffusion::setBoundaryCondition(int index, double val, int type) { void BTCSDiffusion::setBoundaryCondition(int index, double val, bctype type) {
std::get<0>(bc[index]) = val; std::get<0>(bc[index]) = type;
std::get<1>(bc[index]) = type; std::get<1>(bc[index]) = val;
} }

7
src/BTCSDiffusion.hpp
View File

@ -12,7 +12,8 @@
*/ */
typedef Eigen::Triplet<double> T; typedef Eigen::Triplet<double> T;
typedef std::vector<std::tuple<int, double>> boundary_condition; typedef int bctype;
typedef std::vector<std::tuple<bctype, double>> boundary_condition;
/*! /*!
* Class implementing a solution for a 1/2/3D diffusion equation using backward * Class implementing a solution for a 1/2/3D diffusion equation using backward
@ -60,7 +61,7 @@ public:
void setTimestep(double time_step); void setTimestep(double time_step);
void setBoundaryCondition(int index, double val, int type); void setBoundaryCondition(int index, double val, bctype type);
private: private:
void simulate1D(std::vector<double> &c, double bc_left, double bc_right, void simulate1D(std::vector<double> &c, double bc_left, double bc_right,
@ -68,7 +69,7 @@ private:
void simulate2D(std::vector<double> &c); void simulate2D(std::vector<double> &c);
void simulate3D(std::vector<double> &c); void simulate3D(std::vector<double> &c);
double getBCFromTuple(int index, std::vector<double> &c); double getBCFromTuple(int index, double nearest_value);
boundary_condition bc; boundary_condition bc;