mirror of https://gitlab.com/QEF/q-e.git
268 lines
8.6 KiB
Fortran
268 lines
8.6 KiB
Fortran
!
|
|
! Copyright (C) 2001-2007 Quantum ESPRESSO PWSCF group
|
|
! This file is distributed under the terms of the
|
|
! GNU General Public License. See the file `License'
|
|
! in the root directory of the present distribution,
|
|
! or http://www.gnu.org/copyleft/gpl.txt .
|
|
!
|
|
!----------------------------------------------------------------------
|
|
subroutine add_for_charges (ik, uact)
|
|
!----------===============-----------------------------------------------
|
|
!! This subroutine calculates: \(\frac{dS}{du} P_c [x, H-eS] |\psi\rangle\)
|
|
!
|
|
|
|
USE kinds, only : DP
|
|
USE ions_base, ONLY : nat, ityp, ntyp => nsp
|
|
USE cell_base, ONLY : tpiba
|
|
USE gvect, ONLY : g
|
|
USE lsda_mod, ONLY: lsda, current_spin, isk
|
|
USE klist, ONLY : xk, ngk, igk_k
|
|
USE uspp, ONLY : nkb, qq_nt, qq_so, vkb
|
|
USE wvfct, ONLY : npwx, nbnd
|
|
USE becmod, ONLY: calbec, bec_type, allocate_bec_type, deallocate_bec_type
|
|
USE noncollin_module, ONLY : noncolin, npol, lspinorb
|
|
USE uspp_param, only: nh
|
|
USE eqv, ONLY : dvpsi, dpsi
|
|
USE control_lr, ONLY : lgamma
|
|
USE qpoint, ONLY : ikks
|
|
implicit none
|
|
!
|
|
! The dummy variables
|
|
!
|
|
|
|
integer :: ik
|
|
!! input: the k point
|
|
integer :: mode
|
|
! input: the actual perturbation
|
|
complex(DP) :: uact (3 * nat)
|
|
!! input: the pattern of displacements
|
|
!
|
|
! And the local variables
|
|
!
|
|
|
|
integer :: na, nb, mu, nu, ikk, ikq, ig, igg, nt, ibnd, ijkb0, &
|
|
ikb, jkb, ih, jh, ipol, is, js, ijs, npw
|
|
! counter on atoms
|
|
! counter on modes
|
|
! the point k
|
|
! the point k+q
|
|
! counter on G vectors
|
|
! auxiliary counter on G vectors
|
|
! counter on atomic types
|
|
! counter on bands
|
|
! auxiliary variable for counting
|
|
! counter on becp functions
|
|
! counter on becp functions
|
|
! counter on n index
|
|
! counter on m index
|
|
! counter on polarizations
|
|
|
|
real(DP), parameter :: eps = 1.d-12
|
|
|
|
complex(DP), allocatable :: ps1 (:,:), ps2 (:,:,:), aux (:)
|
|
complex(DP), allocatable :: ps1_nc (:,:,:), ps2_nc (:,:,:,:)
|
|
! the scalar product
|
|
! the scalar product
|
|
! a mesh space for psi
|
|
TYPE(bec_type) :: bedp, alphapp(3)
|
|
complex(DP), allocatable :: aux1(:,:)
|
|
|
|
logical :: ok
|
|
! used to save time
|
|
|
|
allocate (aux ( npwx))
|
|
allocate (aux1( npwx*npol, nbnd))
|
|
CALL allocate_bec_type(nkb,nbnd,bedp)
|
|
DO ipol=1,3
|
|
CALL allocate_bec_type(nkb,nbnd,alphapp(ipol))
|
|
ENDDO
|
|
IF (noncolin) THEN
|
|
allocate (ps1_nc ( nkb, npol, nbnd))
|
|
allocate (ps2_nc ( nkb, npol, nbnd , 3))
|
|
ELSE
|
|
allocate (ps1 ( nkb , nbnd))
|
|
allocate (ps2 ( nkb , nbnd , 3))
|
|
ENDIF
|
|
if (lgamma) then
|
|
ikk = ikks(ik)
|
|
ikq = ikk
|
|
npw =ngk(ikk)
|
|
else
|
|
call infomsg ('add_for_charges', 'called for lgamma .eq. false')
|
|
endif
|
|
if (lsda) current_spin = isk (ikk)
|
|
!
|
|
! we first compute the coefficients of the vectors
|
|
!
|
|
if (noncolin) then
|
|
ps1_nc = (0.d0, 0.d0)
|
|
ps2_nc = (0.d0, 0.d0)
|
|
bedp%nc = (0.d0,0.d0)
|
|
DO ipol=1,3
|
|
alphapp(ipol)%nc = (0.d0,0.d0)
|
|
END DO
|
|
else
|
|
ps1 = (0.d0, 0.d0)
|
|
ps2 = (0.d0, 0.d0)
|
|
bedp%k = (0.d0,0.d0)
|
|
DO ipol=1,3
|
|
alphapp(ipol)%k = (0.d0,0.d0)
|
|
END DO
|
|
endif
|
|
aux1 = (0.d0, 0.d0)
|
|
|
|
!
|
|
! first we calculate the products of the beta functions with dpsi
|
|
!
|
|
CALL calbec (npw, vkb, dpsi, bedp)
|
|
do ipol = 1, 3
|
|
aux1=(0.d0,0.d0)
|
|
do ibnd = 1, nbnd
|
|
do ig = 1, npw
|
|
aux1 (ig, ibnd) = dpsi(ig,ibnd) * &
|
|
tpiba * (0.d0,1.d0) * &
|
|
( xk(ipol,ikk) + g(ipol,igk_k(ig,ikk)) )
|
|
enddo
|
|
if (noncolin) then
|
|
do ig = 1, npw
|
|
aux1 (ig+npwx, ibnd) = dpsi(ig+npwx,ibnd) * &
|
|
tpiba * (0.d0,1.d0) * &
|
|
( xk(ipol,ikk) + g(ipol,igk_k(ig,ikk)) )
|
|
enddo
|
|
endif
|
|
enddo
|
|
CALL calbec ( npw, vkb, aux1, alphapp(ipol) )
|
|
enddo
|
|
|
|
|
|
ijkb0 = 0
|
|
do nt = 1, ntyp
|
|
do na = 1, nat
|
|
if (ityp (na) .eq.nt) then
|
|
mu = 3 * (na - 1)
|
|
if ( abs (uact (mu + 1) ) + &
|
|
abs (uact (mu + 2) ) + &
|
|
abs (uact (mu + 3) ) > eps) then
|
|
do ih = 1, nh (nt)
|
|
ikb = ijkb0 + ih
|
|
do jh = 1, nh (nt)
|
|
jkb = ijkb0 + jh
|
|
do ipol = 1, 3
|
|
do ibnd = 1, nbnd
|
|
if (noncolin) then
|
|
if (lspinorb) then
|
|
ijs=0
|
|
DO is=1,npol
|
|
DO js=1,npol
|
|
ijs=ijs+1
|
|
ps1_nc(ikb,is,ibnd)=ps1_nc(ikb,is,ibnd)+&
|
|
(qq_so (ih, jh, ijs, nt) * &
|
|
alphapp(ipol)%nc(jkb,js,ibnd))* &
|
|
uact (mu + ipol)
|
|
ps2_nc(ikb,is,ibnd,ipol)= &
|
|
ps2_nc(ikb,is,ibnd,ipol) + &
|
|
(qq_so (ih, jh, ijs, nt) * &
|
|
bedp%nc (jkb, js, ibnd))*(0.d0,-1.d0)* &
|
|
uact (mu + ipol) * tpiba
|
|
ENDDO
|
|
ENDDO
|
|
else
|
|
do is=1,npol
|
|
ps1_nc(ikb,is,ibnd)=ps1_nc(ikb,is,ibnd) + &
|
|
qq_nt (ih, jh, nt) * &
|
|
alphapp(ipol)%nc(jkb, is, ibnd) * &
|
|
uact (mu + ipol)
|
|
ps2_nc(ikb,is,ibnd,ipol)= &
|
|
ps2_nc(ikb,is, ibnd, ipol) + &
|
|
qq_nt (ih, jh, nt) * (0.d0, -1.d0) * &
|
|
bedp%nc (jkb, is, ibnd) * &
|
|
uact (mu + ipol) * tpiba
|
|
end do
|
|
endif
|
|
else
|
|
ps1 (ikb, ibnd) = ps1 (ikb, ibnd) + &
|
|
qq_nt (ih, jh, nt)*alphapp(ipol)%k(jkb, ibnd)* &
|
|
uact (mu + ipol)
|
|
ps2 (ikb, ibnd, ipol) = ps2 (ikb, ibnd, ipol) + &
|
|
qq_nt (ih, jh, nt) * (0.d0, -1.d0) * &
|
|
bedp%k(jkb, ibnd) *uact (mu + ipol) * tpiba
|
|
endif
|
|
enddo
|
|
enddo
|
|
enddo
|
|
enddo
|
|
endif
|
|
ijkb0 = ijkb0 + nh (nt)
|
|
endif
|
|
enddo
|
|
enddo
|
|
!
|
|
! This term is proportional to beta(k+q+G)
|
|
!
|
|
if (nkb.gt.0) then
|
|
if (noncolin) then
|
|
call zgemm ('N', 'N', npw, nbnd*npol, nkb, &
|
|
(1.d0, 0.d0), vkb, npwx, ps1_nc, nkb, (1.d0, 0.d0) , dvpsi, npwx)
|
|
else
|
|
call zgemm ('N', 'N', npw, nbnd*npol, nkb, &
|
|
(1.d0, 0.d0), vkb, npwx, ps1, nkb, (1.d0, 0.d0) , dvpsi, npwx)
|
|
! dvpsi = matmul(vkb,ps1) + dvpsi
|
|
endif
|
|
endif
|
|
!
|
|
! This term is proportional to (k+q+G)_\alpha*beta(k+q+G)
|
|
!
|
|
do ikb = 1, nkb
|
|
do ipol = 1, 3
|
|
ok = .false.
|
|
do ibnd = 1, nbnd
|
|
if (noncolin) then
|
|
ok = ok .or. (abs (ps2_nc (ikb, 1, ibnd, ipol) ) .gt.eps) &
|
|
.or. (abs (ps2_nc (ikb, 2, ibnd, ipol) ) .gt.eps)
|
|
else
|
|
ok = ok.or. (abs (ps2 (ikb, ibnd, ipol) ) .gt.eps)
|
|
endif
|
|
enddo
|
|
if (ok) then
|
|
do ig = 1, npw
|
|
igg = igk_k (ig,ikq)
|
|
aux (ig) = vkb(ig, ikb) * (xk(ipol, ikq) + g(ipol, igg) )
|
|
enddo
|
|
do ibnd = 1, nbnd
|
|
if (noncolin) then
|
|
do ig = 1, npw
|
|
dvpsi(ig,ibnd)=ps2_nc(ikb,1,ibnd,ipol)*aux(ig)+ &
|
|
dvpsi(ig,ibnd)
|
|
dvpsi(ig+npwx,ibnd)=ps2_nc(ikb,2,ibnd,ipol)*aux(ig)+ &
|
|
dvpsi(ig+npwx,ibnd)
|
|
enddo
|
|
else
|
|
do ig = 1, npw
|
|
dvpsi(ig,ibnd)=ps2(ikb,ibnd,ipol)*aux(ig)+dvpsi(ig,ibnd)
|
|
enddo
|
|
endif
|
|
enddo
|
|
endif
|
|
enddo
|
|
enddo
|
|
!
|
|
! Now dvpsi contains dS/du x |psi>
|
|
!
|
|
|
|
deallocate (aux)
|
|
deallocate (aux1)
|
|
IF (noncolin) THEN
|
|
deallocate (ps1_nc)
|
|
deallocate (ps2_nc)
|
|
ELSE
|
|
deallocate (ps1)
|
|
deallocate (ps2)
|
|
END IF
|
|
CALL deallocate_bec_type(bedp)
|
|
DO ipol=1,3
|
|
CALL deallocate_bec_type(alphapp(ipol))
|
|
END DO
|
|
|
|
return
|
|
end subroutine add_for_charges
|