mirror of
https://github.com/QuantumPackage/qp2.git
synced 2024-06-24 06:02:26 +02:00
66 lines
1.5 KiB
Fortran
66 lines
1.5 KiB
Fortran
program print_energy
|
|
implicit none
|
|
BEGIN_DOC
|
|
! Prints the energy of the wave function stored in the |EZFIO| directory.
|
|
END_DOC
|
|
|
|
! this has to be done in order to be sure that N_det, psi_det and
|
|
! psi_coef_sorted are the wave function stored in the |EZFIO| directory.
|
|
read_wf = .True.
|
|
touch read_wf
|
|
PROVIDE N_states
|
|
if (is_complex) then
|
|
call run_complex
|
|
else
|
|
call run
|
|
endif
|
|
end
|
|
|
|
subroutine run
|
|
implicit none
|
|
integer :: i,j
|
|
double precision :: i_H_psi_array(N_states)
|
|
double precision :: E(N_states)
|
|
double precision :: norm(N_states)
|
|
|
|
E(1:N_states) = nuclear_repulsion
|
|
norm(1:N_states) = 0.d0
|
|
do i=1,N_det
|
|
call i_H_psi(psi_det(1,1,i), psi_det, psi_coef, N_int, N_det, &
|
|
size(psi_coef,1), N_states, i_H_psi_array)
|
|
do j=1,N_states
|
|
norm(j) += psi_coef(i,j)*psi_coef(i,j)
|
|
E(j) += i_H_psi_array(j) * psi_coef(i,j)
|
|
enddo
|
|
enddo
|
|
|
|
print *, 'Energy:'
|
|
do i=1,N_states
|
|
print *, E(i)/norm(i)
|
|
enddo
|
|
end
|
|
|
|
subroutine run_complex
|
|
implicit none
|
|
integer :: i,j
|
|
complex*16 :: i_h_psi_array(n_states)
|
|
double precision :: e(n_states)
|
|
double precision :: norm(n_states)
|
|
|
|
e(1:n_states) = nuclear_repulsion
|
|
norm(1:n_states) = 0.d0
|
|
do i=1,n_det
|
|
call i_H_psi_complex(psi_det(1,1,i), psi_det, psi_coef_complex, N_int, N_det, &
|
|
size(psi_coef_complex,1), N_states, i_H_psi_array)
|
|
do j=1,n_states
|
|
norm(j) += cdabs(psi_coef_complex(i,j))**2
|
|
E(j) += dble(i_h_psi_array(j) * dconjg(psi_coef_complex(i,j)))
|
|
enddo
|
|
enddo
|
|
|
|
print *, 'Energy:'
|
|
do i=1,N_states
|
|
print *, E(i)/norm(i)
|
|
enddo
|
|
end
|