59 lines
1.2 KiB
C++
59 lines
1.2 KiB
C++
#include "diffusion/BTCSBoundaryCondition.hpp"
|
|
#include <diffusion/BTCSDiffusion.hpp>
|
|
#include <iomanip>
|
|
#include <iostream> // for std
|
|
#include <vector> // for vector
|
|
using namespace std;
|
|
|
|
using namespace Diffusion;
|
|
|
|
int main(int argc, char *argv[]) {
|
|
|
|
// dimension of grid
|
|
int dim = 2;
|
|
|
|
int n = 5;
|
|
int m = 5;
|
|
|
|
// create input + diffusion coefficients for each grid cell
|
|
std::vector<double> alpha(n * m, 1e-6);
|
|
std::vector<double> field(n * m, 0);
|
|
BTCSBoundaryCondition bc(n, m);
|
|
|
|
// create instance of diffusion module
|
|
BTCSDiffusion diffu(dim);
|
|
|
|
diffu.setXDimensions(n, n);
|
|
diffu.setYDimensions(m, m);
|
|
|
|
// set inlet to higher concentration without setting bc
|
|
field[12] = 1;
|
|
|
|
// set timestep for simulation to 1 second
|
|
diffu.setTimestep(1);
|
|
|
|
cout << setprecision(12);
|
|
|
|
for (int t = 0; t < 1000; t++) {
|
|
diffu.simulate(field.data(), alpha.data(), bc);
|
|
|
|
cout << "Iteration: " << t << "\n\n";
|
|
|
|
double sum = 0;
|
|
|
|
// iterate through rows
|
|
for (int i = 0; i < m; i++) {
|
|
// iterate through columns
|
|
for (int j = 0; j < n; j++) {
|
|
cout << left << std::setw(20) << field[i * n + j];
|
|
sum += field[i * n + j];
|
|
}
|
|
cout << "\n";
|
|
}
|
|
|
|
cout << "sum: " << sum << "\n" << endl;
|
|
}
|
|
|
|
return 0;
|
|
}
|