FYS3150/Project-1
3
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Make some changes

Browse source
This commit is contained in:
Cory Balaton 2023-09-10 22:44:28 +02:00
parent 85e469f101
commit 54f5d4ab5d
No known key found for this signature in database
GPG key ID: 3E5FCEBFD80F432B
7 changed files with 260 additions and 177 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

52
src/specialAlgorithm.cpp Normal file
View file

@ -0,0 +1,52 @@
#include "specialAlgorithm.hpp"
#include <iomanip>
arma::vec* special_algorithm(
double sub_sig,
double main_sig,
double sup_sig,
arma::vec* g_vec
)
{
int n = g_vec->n_elem;
arma::vec diag = arma::vec(n);
for (int i = 1; i < n; i++) {
// Calculate values for main diagonal based on indices
diag(i-1) = (double)(i+1) / i;
(*g_vec)(i) += (*g_vec)(i-1) / diag(i-1);
}
// The last element in main diagonal has value (i+1)/i = (n+1)/n
(*g_vec)(n-1) /= (double)(n+1) / (n);
for (int i = n-2; i >= 0; i--) {
(*g_vec)(i) = ((*g_vec)(i) + (*g_vec)(i+1))/ diag(i);
}
return g_vec;
}
void special_algorithm_main()
{
arma::vec g_vec, *v_vec;
std::ofstream ofile;
int steps;
double sub_sig, main_sig, sup_sig, step_size;
for (int i = 0; i < 6; i++) {
steps = std::pow(10, i+1);
step_size = 1./(double) steps;
g_vec.resize(steps - 1);
v_vec = special_algorithm(sub_sig, main_sig, sup_sig, &g_vec);
ofile.open("special_algorithm_" + std::to_string(steps) + ".txt");
for (int j=0; j < v_vec->n_elem; j++) {
ofile << std::setprecision(4) << std::scientific << step_size*(i+1) << ","
<< std::setprecision(4) << std::scientific << (*v_vec)(i) << std::endl;
}
ofile.close();
}
}