mirror of https://gitlab.com/QEF/q-e.git
246 lines
7.5 KiB
Fortran
246 lines
7.5 KiB
Fortran
!
|
|
! Copyright (C) 2001-2008 Quantum ESPRESSO 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 solve_e2
|
|
!-----------------------------------------------------------------------
|
|
!
|
|
! Self consistent cycle to compute the second order derivatives
|
|
! of the wavefunctions with respect to electric fields
|
|
!
|
|
USe kinds, ONLY : DP
|
|
USE io_global, ONLY : stdout
|
|
USE cell_base, ONLY : tpiba2
|
|
USE klist, ONLY : lgauss, wk, xk
|
|
USE lsda_mod, ONLY : lsda, nspin
|
|
USE gvect, ONLY : nrxx, g
|
|
USE gsmooth, ONLY : nrxxs, doublegrid
|
|
USE wvfct, ONLY : npw, npwx, nbnd, igk, g2kin, et
|
|
USE io_files, ONLY: prefix, iunigk
|
|
USE ions_base, ONLY: nat
|
|
USE uspp, ONLY: okvan, nkb, vkb
|
|
USE uspp_param,ONLY : nhm
|
|
USE wavefunctions_module, ONLY: evc
|
|
USE eqv, ONLY : dpsi, dvpsi
|
|
USE qpoint, ONLY : npwq, igkq, nksq
|
|
USE control_ph, ONLY : convt, nmix_ph, alpha_mix, nbnd_occ, tr2_ph, &
|
|
niter_ph, lgamma, rec_code, flmixdpot, rec_code_read
|
|
USE units_ph, ONLY : lrwfc, iuwfc
|
|
USE ramanm, ONLY : lrba2, iuba2, lrd2w, iud2w
|
|
USE recover_mod, ONLY : read_rec, write_rec
|
|
|
|
USE check_stop, ONLY: check_stop_now
|
|
USE mp_global, ONLY : inter_pool_comm, intra_pool_comm
|
|
USE mp, ONLY : mp_sum
|
|
implicit none
|
|
|
|
real(DP) :: thresh, weight, avg_iter, dr2
|
|
! convergence threshold for the solution of the
|
|
! linear system
|
|
! used for summation over k points
|
|
! average number of iterations
|
|
! convergence limit
|
|
|
|
complex(DP) , pointer :: dvscfin (:,:,:), dvscfins (:,:,:)
|
|
! change of the scf potential (input)
|
|
! change of the scf potential (smooth)
|
|
|
|
complex(DP) , allocatable :: dvscfout (:,:,:), dbecsum (:,:), &
|
|
aux1 (:)
|
|
! change of the scf potential (output)
|
|
! auxiliary space
|
|
! auxiliary space
|
|
|
|
logical :: exst
|
|
! used to open the recover file
|
|
|
|
integer :: kter, iter0, ipol, ibnd, iter, ik, is, ig, iig, irr, ir, nrec, ios
|
|
! counter on iterations
|
|
! counter on perturbations
|
|
! counter on bands
|
|
! counter on iterations
|
|
! counter on k points
|
|
! counter on G vectors
|
|
! counter on g vectors
|
|
! counter on mesh points
|
|
! the record number
|
|
! integer variable for I/O control
|
|
|
|
external ch_psi_all, cg_psi
|
|
|
|
if (lsda) call errore ('solve_e2', ' LSDA not implemented', 1)
|
|
if (okvan) call errore ('solve_e2', ' Ultrasoft PP not implemented', 1)
|
|
|
|
call start_clock('solve_e2')
|
|
allocate (dvscfin( nrxx, nspin, 6))
|
|
if (doublegrid) then
|
|
allocate (dvscfins( nrxxs, nspin, 6))
|
|
else
|
|
dvscfins => dvscfin
|
|
endif
|
|
allocate (dvscfout( nrxx , nspin, 6))
|
|
allocate (dbecsum( nhm*(nhm+1)/2, nat))
|
|
allocate (aux1( nrxxs))
|
|
convt=.FALSE.
|
|
if (rec_code_read == -10) then
|
|
! restarting in Raman
|
|
CALL read_rec(dr2, iter0, 6, dvscfin, dvscfins)
|
|
else
|
|
iter0 = 0
|
|
end if
|
|
if (convt) goto 155
|
|
!
|
|
if (lgauss.or..not.lgamma) &
|
|
call errore ('solve_e2', 'called in the wrong case', 1)
|
|
!
|
|
! The outside loop is over the iterations
|
|
!
|
|
|
|
do kter = 1, niter_ph
|
|
|
|
iter = kter + iter0
|
|
avg_iter = 0.d0
|
|
|
|
dvscfout (:,:,:) = (0.d0, 0.d0)
|
|
dbecsum (:,:) = (0.d0, 0.d0)
|
|
if (nksq.gt.1) rewind (unit = iunigk)
|
|
|
|
do ik = 1, nksq
|
|
if (nksq.gt.1) then
|
|
read (iunigk, err = 100, iostat = ios) npw, igk
|
|
100 call errore ('solve_e', 'reading igk', abs (ios) )
|
|
endif
|
|
!
|
|
! reads unperturbed wavefuctions psi_k in G_space, for all bands
|
|
!
|
|
if (nksq.gt.1) call davcio (evc, lrwfc, iuwfc, ik, -1)
|
|
npwq = npw
|
|
call init_us_2 (npw, igk, xk (1, ik), vkb)
|
|
!
|
|
! compute the kinetic energy
|
|
!
|
|
do ig = 1, npwq
|
|
iig = igkq (ig)
|
|
g2kin (ig) = ( (xk (1, ik) + g (1, iig) ) **2 + &
|
|
(xk (2, ik) + g (2, iig) ) **2 + &
|
|
(xk (3, ik) + g (3, iig) ) **2 ) * tpiba2
|
|
enddo
|
|
!
|
|
! The counter on the polarizations runs only on the 6 inequivalent
|
|
! indexes --see the comment on raman.F--
|
|
!
|
|
do ipol = 1, 6
|
|
nrec = (ipol - 1) * nksq + ik
|
|
|
|
if (kter.eq.1) then
|
|
dpsi (:,:) = (0.d0, 0.d0)
|
|
else
|
|
call davcio (dpsi, lrd2w, iud2w, nrec, -1)
|
|
endif
|
|
|
|
if (iter.eq.1) then
|
|
dvscfin (:,:,:) = (0.d0, 0.d0)
|
|
call davcio (dvpsi, lrba2, iuba2, nrec, -1)
|
|
thresh = 1.0d-2
|
|
else
|
|
call davcio (dvpsi, lrba2, iuba2, nrec, -1)
|
|
do ibnd = 1, nbnd_occ (ik)
|
|
call cft_wave (evc (1, ibnd), aux1, +1)
|
|
do ir = 1, nrxxs
|
|
aux1 (ir) = aux1 (ir) * dvscfins (ir, 1, ipol)
|
|
enddo
|
|
call cft_wave (dvpsi (1, ibnd), aux1, -1)
|
|
enddo
|
|
thresh = min (0.1d0 * sqrt(dr2), 1.0d-2)
|
|
endif
|
|
|
|
call pcgreen (avg_iter, thresh, ik, et (1, ik) )
|
|
call davcio ( dpsi, lrd2w, iud2w, nrec, +1)
|
|
!
|
|
! calculates dvscf, sum over k => dvscf_q_ipert
|
|
!
|
|
weight = wk (ik)
|
|
call incdrhoscf (dvscfout (1,1,ipol), weight, ik, &
|
|
dbecsum (1, 1), dpsi)
|
|
enddo ! on perturbations
|
|
enddo ! on k points
|
|
#ifdef __PARA
|
|
call mp_sum ( dbecsum, intra_pool_comm )
|
|
#endif
|
|
if (doublegrid) then
|
|
do is = 1, nspin
|
|
do ipol = 1, 6
|
|
call cinterpolate (dvscfout (1, is, ipol), &
|
|
dvscfout (1, is, ipol), 1)
|
|
enddo
|
|
enddo
|
|
endif
|
|
|
|
! call addusddense (dvscfout, dbecsum)
|
|
|
|
!
|
|
! After the loop over the perturbations we have the change of the pote
|
|
! for all the modes, and we symmetrize this potential
|
|
!
|
|
#ifdef __PARA
|
|
call mp_sum ( dvscfout, inter_pool_comm )
|
|
#endif
|
|
do ipol = 1, 6
|
|
call dv_of_drho (0, dvscfout (1, 1, ipol), .false.)
|
|
enddo
|
|
|
|
#ifdef __PARA
|
|
call psyme2(dvscfout)
|
|
#else
|
|
call syme2(dvscfout)
|
|
#endif
|
|
!
|
|
! Mixing with the old potential
|
|
!
|
|
call mix_potential (2 * 6 * nrxx * nspin, dvscfout, dvscfin, &
|
|
alpha_mix (kter), dr2, 6 * tr2_ph, iter, &
|
|
nmix_ph, flmixdpot, convt)
|
|
|
|
if (doublegrid) then
|
|
do is = 1, nspin
|
|
do ipol = 1, 6
|
|
call cinterpolate (dvscfin (1, is, ipol), &
|
|
dvscfins (1, is, ipol), -1)
|
|
enddo
|
|
enddo
|
|
end if
|
|
|
|
write (6, "(//,5x,' iter # ',i3, &
|
|
& ' av.it.: ',f5.1)") iter, avg_iter / (6.d0 * nksq)
|
|
dr2 = dr2 / 6
|
|
write (6, "(5x,' thresh=',e10.3, ' alpha_mix = ',f6.3, &
|
|
& ' |ddv_scf|^2 = ',e10.3 )") thresh, alpha_mix (kter), dr2
|
|
!
|
|
CALL flush_unit( stdout )
|
|
!
|
|
! rec_code: state of the calculation
|
|
! rec_code=-10 to -19 Raman
|
|
rec_code=-10
|
|
CALL write_rec('solve_e2..', irr, dr2, iter, convt, 6, dvscfin)
|
|
|
|
if ( check_stop_now() ) call stop_smoothly_ph (.false.)
|
|
if ( convt ) goto 155
|
|
|
|
enddo
|
|
155 continue
|
|
deallocate (dvscfin )
|
|
if (doublegrid) deallocate (dvscfins )
|
|
deallocate (dvscfout )
|
|
deallocate (dbecsum )
|
|
deallocate (aux1 )
|
|
|
|
call stop_clock('solve_e2')
|
|
|
|
return
|
|
end subroutine solve_e2
|