mirror of
https://github.com/TREX-CoE/Sherman-Morrison.git
synced 2024-10-02 14:31:06 +02:00
8d63dd1701
S = [1,0,1,-1; 0,1,1,0; -1,0,-1,0; 1,1,1,1] S_inv = [1,-1,1,1; 1,0,2,1; -1,1,-2,-1; -1,0,-1,0] u1 = [0,-2,0,0] u2 = [0,-1,0,0] upd_idx = [2,4] To go from Maponi's examples where the number of updates is always equal to the the dimension of the matrix, and the decomposition is always diagonal, to cases with a non-diagonal decomposition and a number of updates unequal to its size, the following changed needed to be made: * in the calculation of the {y0k} an extra inner for-loop needs to be added to make it a full matrix-vector multiplication due to the fact that A0 is not a diagonal matrix * in some places the use of the update-order vector p needs the be replaced with that of upd_idx to make sure the correct component of the ylk is selected and the proper rank-1 matrices are constructed * when a matrix is passed from Fortran to C++ with 2D adressing, it is passed in colum-major order. The passed matrix needs to be transposed before passing to C++. Doing this inside the algorithm will break compatibility with called from C/C++.
79 lines
2.2 KiB
Fortran
79 lines
2.2 KiB
Fortran
program QMCChem_dataset_test
|
|
use Sherman_Morrison, only : MaponiA3
|
|
use Utils, only : Read_dataset
|
|
use, intrinsic :: iso_c_binding, only : c_int, c_double
|
|
implicit none
|
|
|
|
integer :: i, j, col !! Iterators
|
|
integer :: cycle_id, dim, n_updates
|
|
integer(c_int), dimension(:), allocatable :: Updates_index
|
|
real(c_double), dimension(:,:), allocatable :: S, S_inv, S_inv_trans, Updates
|
|
|
|
call Read_dataset("test.dataset.dat", &
|
|
cycle_id, &
|
|
dim, &
|
|
n_updates, &
|
|
S, &
|
|
S_inv, &
|
|
Updates_index, &
|
|
Updates)
|
|
|
|
allocate(S_inv_trans(dim,dim))
|
|
do i=1,dim
|
|
do j=1,dim
|
|
S_inv_trans(i,j) = S_inv(j,i)
|
|
end do
|
|
end do
|
|
|
|
!! Write current S and S_inv to file for check in Octave
|
|
open(unit = 2000, file = "Slater_old.dat")
|
|
open(unit = 3000, file = "Slater_old_inv.dat")
|
|
do i=1,dim
|
|
do j=1,dim
|
|
write(2000,"(E23.15, 1X)", advance="no") S(i,j)
|
|
write(3000,"(E23.15, 1X)", advance="no") S_inv_trans(i,j)
|
|
end do
|
|
write(2000,*)
|
|
write(3000,*)
|
|
end do
|
|
close(2000)
|
|
close(3000)
|
|
|
|
!! Write Updates to file to check
|
|
open(unit = 2000, file = "Updates.dat")
|
|
do i=1,dim
|
|
do j=1,n_updates
|
|
write(2000,"(E23.15, 1X)", advance="no") Updates(i,j)
|
|
end do
|
|
write(2000,*)
|
|
end do
|
|
close(2000)
|
|
|
|
!! Update S
|
|
do j=1,n_updates
|
|
do i=1,dim
|
|
col = Updates_index(j)
|
|
S(i,col) = S(i,col) + Updates(i,j)
|
|
end do
|
|
end do
|
|
|
|
!! Send S_inv and Updates to MaponiA3 algo
|
|
call MaponiA3(S_inv_trans, dim, n_updates, Updates, Updates_index)
|
|
|
|
!! Write new S and S_inv to file for check in Octave
|
|
open(unit = 4000, file = "Slater.dat")
|
|
open(unit = 5000, file = "Slater_inv.dat")
|
|
do i=1,dim
|
|
do j=1,dim
|
|
write(4000,"(E23.15, 1X)", advance="no") S(i,j)
|
|
write(5000,"(E23.15, 1X)", advance="no") S_inv(i,j)
|
|
end do
|
|
write(4000,*)
|
|
write(5000,*)
|
|
end do
|
|
close(4000)
|
|
close(5000)
|
|
|
|
deallocate(S, S_inv, Updates, Updates_index)
|
|
end program
|