Sherman-Morrison/tests/test_internal_h5.cpp

134 lines
3.3 KiB
C++
Raw Normal View History

2021-02-16 10:49:15 +01:00
#include <iostream>
#include <string>
#include "hdf5/serial/hdf5.h"
#include "hdf5/serial/H5Cpp.h"
2021-02-16 10:49:15 +01:00
#include "SM_MaponiA3.hpp"
#include "SM_Standard.hpp"
#include "Helpers.hpp"
2021-02-16 10:49:15 +01:00
using namespace H5;
// #define DEBUG
2021-02-16 10:49:15 +01:00
2021-02-25 12:59:38 +01:00
const H5std_string FILE_NAME( "datasets.hdf5" );
2021-02-16 10:49:15 +01:00
double residual2(double * A, unsigned int Dim) {
double res = 0.0;
for (unsigned int i = 0; i < Dim; i++) {
for (unsigned int j = 0; j < Dim; j++) {
double delta = (A[i * Dim + j] - (i == j));
res += delta*delta;
}
}
return res;
}
2021-02-16 10:49:15 +01:00
void read_int(H5File file, std::string key, unsigned int * data) {
DataSet ds = file.openDataSet(key);
ds.read(data, PredType::STD_U32LE);
ds.close();
}
void read_double(H5File file, std::string key, double * data) {
DataSet ds = file.openDataSet(key);
ds.read(data, PredType::IEEE_F64LE);
ds.close();
}
int test_cycle(H5File file, int cycle) {
/* Read the data */
std::string group = "cycle_" + std::to_string(cycle);
unsigned int dim, nupdates, col, i, j;
2021-02-16 10:49:15 +01:00
read_int(file, group + "/slater_matrix_dim", &dim);
read_int(file, group + "/nupdates", &nupdates);
double * slater_matrix = new double[dim*dim];
read_double(file, group + "/slater_matrix", slater_matrix);
double * slater_inverse = new double[dim*dim];
read_double(file, group + "/slater_inverse", slater_inverse);
slater_inverse = transpose(slater_inverse, dim);
2021-02-16 10:49:15 +01:00
unsigned int * col_update_index = new unsigned int[nupdates];
read_int(file, group + "/col_update_index", col_update_index);
double * updates = new double[nupdates*dim];
read_double(file, group + "/updates", updates);
double * u = new double[nupdates*dim];
2021-02-16 10:49:15 +01:00
/* Test */
#ifdef DEBUG
showMatrix(slater_matrix, dim, "OLD Slater");
2021-02-16 10:49:15 +01:00
#endif
#ifdef DEBUG
showMatrix(slater_inverse, dim, "OLD Inverse");
#endif
for (j = 0; j < nupdates; j++) {
for (i = 0; i < dim; i++) {
col = col_update_index[j];
u[i + j*dim] = updates[i + j*dim] - slater_matrix[i*dim + (col - 1)];
slater_matrix[i*dim + (col - 1)] = updates[i + j*dim];
}
}
//MaponiA3(slater_inverse, dim, nupdates, updates, col_update_index);
SM(slater_inverse, dim, nupdates, u, col_update_index);
2021-02-16 10:49:15 +01:00
#ifdef DEBUG
showMatrix(slater_matrix, dim, "NEW Slater");
#endif
2021-02-16 10:49:15 +01:00
#ifdef DEBUG
showMatrix(slater_inverse, dim, "NEW Inverse");
2021-02-16 10:49:15 +01:00
#endif
2021-03-11 15:04:35 +01:00
double * res = new double[dim*dim] {0};
matMul(slater_matrix, slater_inverse, res, dim);
bool ok = is_identity(res, dim, 1e-4);
double res2 = residual2(res, dim);
std::cerr << "Residual = " << res2 << std::endl;
2021-02-16 10:49:15 +01:00
#ifdef DEBUG
showMatrix(res, dim, "Result");
#endif
delete [] res, updates, u, col_update_index,
slater_matrix, slater_inverse;
2021-02-16 10:49:15 +01:00
return ok;
}
int main(int argc, char **argv) {
if (argc != 3) {
std::cerr << "Execute from within 'datasets/'" << std::endl;
std::cerr << "usage: test_internal_h5 <start cycle> <stop cycle>" << std::endl;
2021-02-16 10:49:15 +01:00
return 1;
}
int start_cycle = std::stoi(argv[1]);
int stop_cycle = std::stoi(argv[2]);
2021-02-16 10:49:15 +01:00
H5File file(FILE_NAME, H5F_ACC_RDONLY);
bool ok;
for (int cycle = start_cycle; cycle < stop_cycle+1; cycle++) {
ok = test_cycle(file, cycle);
if (ok) {
std::cerr << "ok -- cycle " << std::to_string(cycle)
<< std::endl;
}
else {
std::cerr << "failed -- cycle " << std::to_string(cycle)
<< std::endl;
}
2021-02-16 10:49:15 +01:00
}
2021-02-16 10:49:15 +01:00
return ok;
}