mirror of
https://gitlab.com/scemama/qp_plugins_scemama.git
synced 2024-12-22 20:34:05 +01:00
204 lines
5.9 KiB
Fortran
204 lines
5.9 KiB
Fortran
program delta_2bTSVD_v0
|
|
|
|
BEGIN_DOC
|
|
! !!!
|
|
END_DOC
|
|
|
|
implicit none
|
|
|
|
read_wf = .True.
|
|
touch read_wf
|
|
|
|
call run()
|
|
|
|
end
|
|
|
|
|
|
|
|
subroutine run()
|
|
|
|
implicit none
|
|
|
|
integer :: ii, ii_i, ii_j, k, l
|
|
integer :: na, nb, nsvd
|
|
integer :: n_TSVD
|
|
integer :: n_tmp
|
|
double precision, allocatable :: U(:,:), V(:,:)
|
|
|
|
double precision, allocatable :: delta0_det(:), delta1_det(:)
|
|
double precision, allocatable :: delta_O(:,:), delta_D(:)
|
|
double precision, allocatable :: delta_tmp(:)
|
|
|
|
double precision :: t1, t2
|
|
|
|
na = n_det_alpha_unique
|
|
nb = n_det_beta_unique
|
|
|
|
call ezfio_get_spindeterminants_n_svd_coefs(nsvd)
|
|
allocate( u(na,nsvd) , v(nb,nsvd) )
|
|
|
|
print *, ' start read SVD vectors'
|
|
call wall_time(t1)
|
|
call ezfio_get_spindeterminants_psi_svd_alpha(U)
|
|
call ezfio_get_spindeterminants_psi_svd_beta (V)
|
|
call wall_time(t2)
|
|
print *, ' end read SVD vectors after (min)', (t2-t1)/60.d0
|
|
|
|
allocate( delta0_det(N_det) )
|
|
call ezfio_get_dmc_dress_dmc_delta_h(delta0_det)
|
|
|
|
|
|
! ---------------------------------------------
|
|
! ---------------------------------------------
|
|
! parameters
|
|
!
|
|
!n_TSVD = 100
|
|
read *, n_TSVD
|
|
|
|
print *, " --- delta_2bTSVD_v0 ---"
|
|
print *, " n_TSVD =", n_TSVD
|
|
print *, " tot =", n_TSVD * n_TSVD + &
|
|
nsvd - n_TSVD
|
|
!
|
|
! ---------------------------------------------
|
|
! ---------------------------------------------
|
|
! ---------------------------------------------
|
|
|
|
n_tmp = n_TSVD + 1
|
|
|
|
allocate( delta_O(1:n_TSVD,1:n_TSVD) )
|
|
allocate( delta_D(n_tmp:nsvd) )
|
|
call obtenir_deltaSVD_2b(nsvd, n_TSVD, U, V, delta0_det, delta_O, delta_D)
|
|
deallocate( delta0_det )
|
|
|
|
allocate( delta1_det(N_det) )
|
|
delta1_det(1:N_det) = 0.d0
|
|
|
|
!$OMP PARALLEL DEFAULT(NONE) &
|
|
!$OMP SHARED(nsvd, n_TSVD, n_tmp, N_det, U, V, delta1_det, delta_O, delta_D &
|
|
!$OMP , psi_bilinear_matrix_rows, psi_bilinear_matrix_columns) &
|
|
!$OMP PRIVATE(k, l, ii, ii_i, ii_j, delta_tmp)
|
|
allocate( delta_tmp(N_det) )
|
|
delta_tmp(1:N_det) = 0.d0
|
|
!$OMP DO
|
|
do ii = 1, N_det
|
|
ii_i = psi_bilinear_matrix_rows (ii)
|
|
ii_j = psi_bilinear_matrix_columns(ii)
|
|
do l = 1, n_TSVD
|
|
delta_tmp(ii) = delta_tmp(ii) + U(ii_i,l) * V(ii_j,l) * delta_D(l)
|
|
do k = 1, n_TSVD
|
|
delta_tmp(ii) = delta_tmp(ii) + U(ii_i,k) * V(ii_j,l) * delta_O(k,l)
|
|
enddo
|
|
enddo
|
|
enddo
|
|
!$OMP END DO
|
|
!$OMP CRITICAL
|
|
do ii = 1, N_det
|
|
delta1_det(ii) = delta1_det(ii) + delta_tmp(ii)
|
|
enddo
|
|
!$OMP END CRITICAL
|
|
deallocate( delta_tmp )
|
|
!$OMP END PARALLEL
|
|
|
|
deallocate( U , V )
|
|
deallocate( delta_O , delta_D )
|
|
|
|
call ezfio_set_dmc_dress_dmc_delta_h(delta1_det)
|
|
deallocate( delta1_det )
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
|
|
! _______________________________________________________________________________________________________
|
|
!
|
|
! [ delta_SVD ]_kl = < dup_SVD ddn_SVD | H_tilde - H | psi >
|
|
! = \sum_{i,j} U_{i,k} V_{j,l} < dup_i ddn_j | H_tilde - H | psi >
|
|
!
|
|
subroutine obtenir_deltaSVD_2b(nsvd, n_TSVD, U, V, delta0_det, delta_O, delta_D)
|
|
|
|
implicit none
|
|
|
|
integer, intent(in) :: nsvd, n_TSVD
|
|
double precision, intent(in) :: U(n_det_alpha_unique,nsvd), V(n_det_beta_unique,nsvd)
|
|
double precision, intent(in) :: delta0_det(N_det)
|
|
|
|
double precision, intent(out) :: delta_O(1:n_TSVD,1:n_TSVD)
|
|
double precision, intent(out) :: delta_D(n_TSVD+1:nsvd)
|
|
|
|
integer :: n_tmp
|
|
integer :: k, l, ii, ii_i, ii_j
|
|
double precision, allocatable :: delta_tmp(:,:), delta1d_tmp(:)
|
|
|
|
n_tmp = n_TSVD + 1
|
|
|
|
|
|
! --------------------------------------------------------------------------------
|
|
!
|
|
delta_O(1:n_TSVD,1:n_TSVD) = 0.d0
|
|
!$OMP PARALLEL DEFAULT(NONE) &
|
|
!$OMP SHARED(n_TSVD, N_det, U, V, delta0_det, delta_O &
|
|
!$OMP , psi_bilinear_matrix_rows, psi_bilinear_matrix_columns) &
|
|
!$OMP PRIVATE(k, l, ii, ii_i, ii_j, delta_tmp)
|
|
allocate( delta_tmp(1:n_TSVD,1:n_TSVD) )
|
|
delta_tmp(1:n_TSVD,1:n_TSVD) = 0.d0
|
|
!$OMP DO
|
|
do l = 1, n_TSVD
|
|
do k = 1, n_TSVD
|
|
do ii = 1, N_det
|
|
ii_i = psi_bilinear_matrix_rows (ii)
|
|
ii_j = psi_bilinear_matrix_columns(ii)
|
|
delta_tmp(k,l) = delta_tmp(k,l) + U(ii_i,k) * V(ii_j,l) * delta0_det(ii)
|
|
enddo
|
|
enddo
|
|
enddo
|
|
!$OMP END DO
|
|
!$OMP CRITICAL
|
|
do l = 1, n_TSVD
|
|
do k = 1, n_TSVD
|
|
delta_O(k,l) = delta_O(k,l) + delta_tmp(k,l)
|
|
enddo
|
|
enddo
|
|
!$OMP END CRITICAL
|
|
deallocate(delta_tmp)
|
|
!$OMP END PARALLEL
|
|
!
|
|
! --------------------------------------------------------------------------------
|
|
|
|
|
|
|
|
! --------------------------------------------------------------------------------
|
|
!
|
|
delta_D(n_tmp:nsvd) = 0.d0
|
|
!$OMP PARALLEL DEFAULT(NONE) &
|
|
!$OMP SHARED(nsvd, n_tmp, N_det, U, V, delta0_det, delta_D &
|
|
!$OMP , psi_bilinear_matrix_rows, psi_bilinear_matrix_columns) &
|
|
!$OMP PRIVATE(l, ii, ii_i, ii_j, delta1d_tmp)
|
|
allocate( delta1d_tmp(n_tmp:nsvd) )
|
|
delta1d_tmp(n_tmp:nsvd) = 0.d0
|
|
!$OMP DO
|
|
do l = n_tmp, nsvd
|
|
do ii = 1, N_det
|
|
ii_i = psi_bilinear_matrix_rows (ii)
|
|
ii_j = psi_bilinear_matrix_columns(ii)
|
|
delta1d_tmp(l) = delta1d_tmp(l) + U(ii_i,l) * V(ii_j,l) * delta0_det(ii)
|
|
enddo
|
|
enddo
|
|
!$OMP END DO
|
|
!$OMP CRITICAL
|
|
do l = n_tmp, nsvd
|
|
delta_D(l) = delta_D(l) + delta1d_tmp(l)
|
|
enddo
|
|
!$OMP END CRITICAL
|
|
deallocate(delta1d_tmp)
|
|
!$OMP END PARALLEL
|
|
!
|
|
! --------------------------------------------------------------------------------
|
|
|
|
return
|
|
end subroutine obtenir_deltaSVD_2b
|
|
! _______________________________________________________________________________________________________
|
|
! _______________________________________________________________________________________________________
|