max/tug
1
0
Fork 0
mirror of https://git.gfz-potsdam.de/naaice/tug.git synced 2025-12-13 09:28:23 +01:00
Code Issues Packages Projects Releases Wiki Activity

udpate header + function

Browse Source
This commit is contained in:
Max Lübke 2021-10-26 11:54:08 +02:00
parent 0dea11744f
commit 04763406cb
2 changed files with 26 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
src/diffusion.cpp
View File

@ -2,6 +2,13 @@
#include <Eigen/SparseLU>
#include <Eigen/src/Core/Matrix.h>
#include <Eigen/src/Core/util/Constants.h>
#include <Eigen/src/OrderingMethods/Ordering.h>
#include <Eigen/src/SparseCore/SparseMap.h>
#include <Eigen/src/SparseCore/SparseMatrix.h>
#include <Eigen/src/SparseCore/SparseMatrixBase.h>
#include <Eigen/src/SparseLU/SparseLU.h>
#include <iostream>
void BTCS2D(int x, int y, std::vector<double> &c, std::vector<double> &alpha,
double timestep) {
@ -11,7 +18,8 @@ void BTCS2D(int x, int y, std::vector<double> &c, std::vector<double> &alpha,
int size = x * y;
Eigen::VectorXd d = Eigen::VectorXd::Constant(size, 0);
Eigen::VectorXd b = Eigen::VectorXd::Constant(size, 0);
Eigen::VectorXd x_out(size);
std::vector<T> tripletList;
tripletList.reserve(((x - 1) * (y - 1) * 5));
@ -25,6 +33,21 @@ void BTCS2D(int x, int y, std::vector<double> &c, std::vector<double> &alpha,
tripletList.push_back(T((i * x) + j, ((i * x) + j) - x, sy));
tripletList.push_back(T((i * x) + j, ((i * x) + j) + 1, sx));
tripletList.push_back(T((i * x) + j, ((i * x) + j) - 1, sx));
b[(i * x) + j] = -c[(i * x) + j];
}
}
Eigen::SparseMatrix<double> A(size, size);
A.setFromTriplets(tripletList.begin(), tripletList.end());
Eigen::SparseLU<Eigen::SparseMatrix<double>, Eigen::COLAMDOrdering<int>>
solver;
solver.analyzePattern(A);
solver.factorize(A);
std::cout << A << std::endl;
x_out = solver.solve(b);
}

4
src/diffusion.hpp
View File

@ -6,6 +6,6 @@
typedef Eigen::Triplet<double> T;
extern void BTCS2D(int x, int y, std::vector<double> &c, double alpha,
double timestep);
extern void BTCS2D(int x, int y, std::vector<double> &c,
std::vector<double> &alpha, double timestep);
#endif // DIFFUSION_H_