mirror of
https://github.com/QuantumPackage/qp2.git
synced 2024-11-10 07:23:38 +01:00
127 lines
3.2 KiB
Fortran
127 lines
3.2 KiB
Fortran
! Predicted energy : e_model
|
|
|
|
! *Compute the energy predicted by the Taylor series*
|
|
|
|
! The energy is predicted using a Taylor expansion truncated at te 2nd
|
|
! order :
|
|
|
|
! \begin{align*}
|
|
! E_{k+1} = E_{k} + \textbf{g}_k^{T} \cdot \textbf{x}_{k+1} + \frac{1}{2} \cdot \textbf{x}_{k+1}^T \cdot \textbf{H}_{k} \cdot \textbf{x}_{k+1} + \mathcal{O}(\textbf{x}_{k+1}^2)
|
|
! \end{align*}
|
|
|
|
! Input:
|
|
! | n | integer | m*(m-1)/2 |
|
|
! | n2 | integer | m*(m-1)/2 or 1 if the hessian is diagonal |
|
|
! | v_grad(n) | double precision | gradient |
|
|
! | H(n,n) | double precision | hessian |
|
|
! | x(n) | double precision | Step in the trust region |
|
|
! | prev_energy | double precision | previous energy |
|
|
|
|
! Output:
|
|
! | e_model | double precision | predicted energy after the rotation of the MOs |
|
|
|
|
! Internal:
|
|
! | part_1 | double precision | v_grad^T.x |
|
|
! | part_2 | double precision | 1/2 . x^T.H.x |
|
|
! | part_2a | double precision | H.x |
|
|
! | i,j | integer | indexes |
|
|
|
|
! Function:
|
|
! | ddot | double precision | dot product (Lapack) |
|
|
|
|
|
|
subroutine trust_region_expected_e(n,n2,v_grad,H,x,prev_energy,e_model)
|
|
|
|
include 'pi.h'
|
|
|
|
!BEGIN_DOC
|
|
! Compute the expected criterion/energy after the application of the step x
|
|
!END_DOC
|
|
|
|
implicit none
|
|
|
|
! Variables
|
|
|
|
! in
|
|
integer, intent(in) :: n,n2
|
|
double precision, intent(in) :: v_grad(n),H(n,n2),x(n)
|
|
double precision, intent(in) :: prev_energy
|
|
|
|
! out
|
|
double precision, intent(out) :: e_model
|
|
|
|
! internal
|
|
double precision :: part_1, part_2, t1,t2,t3
|
|
double precision, allocatable :: part_2a(:)
|
|
|
|
integer :: i,j
|
|
|
|
!Function
|
|
double precision :: ddot
|
|
|
|
print*,''
|
|
print*,'---Trust_e_model---'
|
|
|
|
call wall_time(t1)
|
|
|
|
! Allocation
|
|
allocate(part_2a(n))
|
|
|
|
! Calculations
|
|
|
|
! part_1 corresponds to the product g.x
|
|
! part_2a corresponds to the product H.x
|
|
! part_2 corresponds to the product 0.5*(x^T.H.x)
|
|
|
|
! TODO: remove the dot products
|
|
|
|
|
|
! Product v_grad.x
|
|
part_1 = ddot(n,v_grad,1,x,1)
|
|
|
|
!if (debug) then
|
|
! print*,'g.x : ', part_1
|
|
!endif
|
|
|
|
! Product H.x
|
|
if (n == n2) then
|
|
call dgemv('N',n,n,1d0,H,size(H,1),x,1,0d0,part_2a,1)
|
|
else
|
|
! If the hessian is diagonal
|
|
do i = 1, n
|
|
part_2a(i) = H(i,1) * x(i)
|
|
enddo
|
|
endif
|
|
|
|
! Product 1/2 . x^T.H.x
|
|
part_2 = 0.5d0 * ddot(n,x,1,part_2a,1)
|
|
|
|
!if (debug) then
|
|
! print*,'1/2*x^T.H.x : ', part_2
|
|
!endif
|
|
|
|
|
|
! Sum
|
|
e_model = prev_energy + part_1 + part_2
|
|
|
|
! Writing the predicted energy
|
|
print*, 'prev_energy: ', prev_energy
|
|
print*, 'Predicted energy after the rotation:', e_model
|
|
print*, 'Previous energy - predicted energy: ', prev_energy - e_model
|
|
|
|
! Can be deleted, already in another subroutine
|
|
if (DABS(prev_energy - e_model) < 1d-12 ) then
|
|
print*,'WARNING: ABS(prev_energy - e_model) < 1d-12'
|
|
endif
|
|
|
|
! Deallocation
|
|
deallocate(part_2a)
|
|
|
|
call wall_time(t2)
|
|
t3 = t2 - t1
|
|
print*,'Time in trust e model:', t3
|
|
|
|
print*,'---End trust_e_model---'
|
|
|
|
end subroutine
|