mirror of
https://github.com/TREX-CoE/Sherman-Morrison.git
synced 2024-12-25 05:43:54 +01:00
63 lines
1.6 KiB
C++
63 lines
1.6 KiB
C++
// main.cpp
|
|
#include "Helpers.hpp"
|
|
#include "SM_Maponi.hpp"
|
|
|
|
int main() {
|
|
|
|
unsigned int M = 3; // Dimension of the Slater-matrix
|
|
unsigned int i, j; // Indices for iterators
|
|
|
|
// Declare, allocate all vectors and matrices and fill them with zeros
|
|
unsigned int *Ar_index = new unsigned int[M];
|
|
double *A = new double[M * M]; // The matrix to be inverted
|
|
double *A0 =
|
|
new double[M * M]; // A diagonal matrix with the digonal elements of A
|
|
double *Ar = new double[M * M]; // The update matrix
|
|
double *A0_inv = new double[M * M]; // The inverse
|
|
double breakdown = 1e-3;
|
|
|
|
// Fill with zeros
|
|
for (i = 0; i < M; i++) {
|
|
for (j = 0; j < M; j++) {
|
|
A0[i * M + j] = 0;
|
|
Ar[i * M + j] = 0;
|
|
A0_inv[i * M + j] = 0;
|
|
}
|
|
}
|
|
|
|
// Initialize A with M=3 and fill acc. to Eq. (17) from paper
|
|
A[0] = 1;
|
|
A[3] = 1;
|
|
A[6] = -1;
|
|
A[1] = 1;
|
|
A[4] = 1;
|
|
A[7] = 0;
|
|
A[2] = -1;
|
|
A[5] = 0;
|
|
A[8] = -1;
|
|
|
|
showMatrix(A, M, "A");
|
|
|
|
// Initialize the diagonal matrix A0,
|
|
// the inverse of A0_inv of diagonal matrix A0_inv
|
|
// and the update matrix Ar
|
|
for (i = 0; i < M; i++) {
|
|
A0[i * M + i] = A[i * M + i];
|
|
A0_inv[i * M + i] = 1.0 / A[i * M + i];
|
|
Ar_index[i] = i + 1; // ! First column needs to start with 1 !
|
|
for (j = 0; j < M; j++) {
|
|
Ar[i * M + j] = A[i * M + j] - A0[i * M + j];
|
|
}
|
|
}
|
|
|
|
// Define pointers dim and n_updates to use in Sherman-Morrison(...) function
|
|
// call
|
|
MaponiA3(A0_inv, M, M, Ar, Ar_index, breakdown);
|
|
showMatrix(A0_inv, M, "A0_inv");
|
|
|
|
// Deallocate all vectors and matrices
|
|
delete[] A, A0, A0_inv, Ar, Ar_index;
|
|
|
|
return 0;
|
|
}
|