ps_in_grid.F90

Go to the documentation of this file.

!! Copyright (C) 2002-2006 M. Marques, A. Castro, A. Rubio, G. Bertsch
!!
!! This program is free software; you can redistribute it and/or modify
!! it under the terms of the GNU General Public License as published by
!! the Free Software Foundation; either version 2, or (at your option)
!! any later version.
!!
!! This program is distributed in the hope that it will be useful,
!! but WITHOUT ANY WARRANTY; without even the implied warranty of
!! MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
!! GNU General Public License for more details.
!!
!! You should have received a copy of the GNU General Public License
!! along with this program; if not, write to the Free Software
!! Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
!! 02110-1301, USA.
!!
 
#include "global.h"
 
module ps_in_grid_oct_m
  use atomic_oct_m
  use debug_oct_m
  use global_oct_m
  use, intrinsic :: iso_fortran_env
  use logrid_oct_m
  use messages_oct_m
  use namespace_oct_m
  use profiling_oct_m
 
  implicit none
 
  private
  public ::                   &
    ps_in_grid_t,             &
    ps_in_grid_init,          &
    ps_in_grid_end,           &
    ps_in_grid_vlocal,        &
    ps_in_grid_kb_projectors, &
    ps_in_grid_kb_cosines,    &
    ps_in_grid_cutoff_radii,  &
    ps_in_grid_check_rphi,    &
    first_point_extrapolate
 
  type ps_in_grid_t
    ! Components are public by default
    type(logrid_t) :: g
 
    real(real64)       :: zval
 
    integer            :: no_l_channels
    real(real64), allocatable :: vps(:, :)
    real(real64), allocatable :: KB(:,:)
    real(real64), allocatable :: dkbcos(:)
    real(real64), allocatable, private :: dknorm(:)
    real(real64), allocatable :: kb_radius(:)
    integer,      allocatable :: kb_nrval(:)
 
    integer            :: so_no_l_channels
    real(real64), allocatable :: so_vps(:,:)
    real(real64), allocatable, private :: so_KB(:,:)
    real(real64), allocatable, private :: so_dkbcos(:)
    real(real64), allocatable, private :: so_dknorm(:)
    real(real64), allocatable, private :: so_kb_radius(:)
    integer,      allocatable :: so_kb_nrval(:)
 
    real(real64), allocatable :: vlocal(:)
    real(real64), allocatable :: rphi(:, :,:)
 
    logical            :: core_corrections
    real(real64), allocatable :: chcore(:)
 
  end type ps_in_grid_t
 
contains
 
  subroutine ps_in_grid_init(ps, flavor, a, b, nrval, no_l, so_no_l)
    type(ps_in_grid_t), intent(out) :: ps
    integer,            intent(in)  :: flavor, nrval
    real(real64),       intent(in)  :: a, b
    integer,            intent(in)  :: no_l, so_no_l
 
    push_sub(ps_in_grid_init)
 
    ! initialize logarithmic grid
    call logrid_init(ps%g, flavor, a, b, nrval)
 
    ! copy data
    ps%   no_l_channels =    no_l
    ps%so_no_l_channels = so_no_l
 
    ! Allocate some stuff
    safe_allocate(ps%vps    (1:ps%g%nrval, 1:no_l))
    safe_allocate(ps%chcore (1:ps%g%nrval))
    safe_allocate(ps%vlocal (1:ps%g%nrval))
 
    safe_allocate(ps%rphi  (1:ps%g%nrval, 1:no_l, 1:3))
    safe_allocate(ps%KB    (1:ps%g%nrval, 1:no_l))
    safe_allocate(ps%dkbcos(1:no_l))
    safe_allocate(ps%dknorm(1:no_l))
    safe_allocate(ps%kb_radius(1:no_l+1))
    safe_allocate(ps%kb_nrval(1:no_l+1))
 
    if (so_no_l > 0) then
      safe_allocate(ps%so_vps(1:ps%g%nrval, 1:so_no_l))
      safe_allocate(ps%so_KB (1:ps%g%nrval, 1:so_no_l))
      safe_allocate(ps%so_dkbcos(1:so_no_l))
      safe_allocate(ps%so_dknorm(1:so_no_l))
      safe_allocate(ps%so_kb_radius(1:so_no_l))
      safe_allocate(ps%so_kb_nrval(1:so_no_l))
    end if
 
    pop_sub(ps_in_grid_init)
  end subroutine ps_in_grid_init
 
 
  ! ---------------------------------------------------------
  subroutine ps_in_grid_end(ps)
    type(ps_in_grid_t), intent(inout) :: ps
 
    push_sub(ps_in_grid_end)
 
    safe_deallocate_a(ps%vps)
    safe_deallocate_a(ps%chcore)
    safe_deallocate_a(ps%vlocal)
 
    safe_deallocate_a(ps%rphi)
    safe_deallocate_a(ps%KB)
    safe_deallocate_a(ps%dkbcos)
    safe_deallocate_a(ps%dknorm)
    safe_deallocate_a(ps%kb_radius)
    safe_deallocate_a(ps%kb_nrval)
 
    if (ps%so_no_l_channels > 0) then
      safe_deallocate_a(ps%so_vps)
      safe_deallocate_a(ps%so_KB)
      safe_deallocate_a(ps%so_dkbcos)
      safe_deallocate_a(ps%so_dknorm)
      safe_deallocate_a(ps%so_kb_radius)
      safe_deallocate_a(ps%so_kb_nrval)
    end if
 
    call logrid_end(ps%g)
 
    pop_sub(ps_in_grid_end)
  end subroutine ps_in_grid_end
 
 
  ! ---------------------------------------------------------
  subroutine ps_in_grid_vlocal(ps, l_loc, rcore, namespace)
    type(ps_in_grid_t), intent(inout) :: ps
    integer,            intent(in)    :: l_loc
    real(real64),       intent(in)    :: rcore
    type(namespace_t),  intent(in)    :: namespace
 
    integer :: ir
    real(real64) :: a, b, qtot
    real(real64), allocatable :: rho(:)
 
    push_sub(ps_in_grid_vlocal)
 
    if (l_loc >= 0) then
      ps%vlocal(:) = ps%vps(:, l_loc+1)
 
    else if (l_loc == -1) then
      if (ps%g%flavor /= logrid_psf) then
        message(1) = "For the moment, Vanderbilt local potentials are only possible with tm grids."
        call messages_fatal(1, namespace=namespace)
      end if
 
      a = 1.82_real64 / rcore
      b = m_one
      safe_allocate(rho(1:ps%g%nrval))
 
      do ir = 1, ps%g%nrval
        rho(ir) = exp(-(sinh(a*b*ps%g%rofi(ir)) / sinh(b))**2)
        rho(ir) = m_four * m_pi * rho(ir) * ps%g%r2ofi(ir)
      end do
!      do ir =1,2
      qtot = sum(rho(:)*ps%g%drdi(:))
      rho(:) = - rho(:)*(ps%zval/qtot)
 
      call vhrtre(rho, ps%vlocal, ps%g%rofi, ps%g%drdi, ps%g%s, ps%g%nrval, ps%g%a)
      ps%vlocal(1) = ps%vlocal(2)
 
      safe_deallocate_a(rho)
    end if
 
    pop_sub(ps_in_grid_vlocal)
  end subroutine ps_in_grid_vlocal
 
 
  ! ---------------------------------------------------------
  subroutine ps_in_grid_kb_projectors(ps)
    type(ps_in_grid_t), intent(inout) :: ps
 
    integer :: l
 
    push_sub(ps_in_grid_kb_projectors)
 
    do l = 1, ps%no_l_channels
      ps%KB(2:, l) = (ps%vps(2:, l) - ps%vlocal(2:))*(ps%rphi(2:, l, 1)/ps%g%rofi(2:))*ps%dknorm(l)
 
      ps%KB(1, l) = first_point_extrapolate(ps%g%rofi, ps%KB(:, l))
    end do
 
    do l = 1, ps%so_no_l_channels
      ps%so_KB(2:, l) = ps%so_vps(2:, l)*(ps%rphi(2:, l, 1)/ps%g%rofi(2:))*ps%dknorm(l)
 
      ps%so_KB(1, l) = first_point_extrapolate(ps%g%rofi, ps%so_KB(:, l))
    end do
 
    pop_sub(ps_in_grid_kb_projectors)
  end subroutine ps_in_grid_kb_projectors
 
 
  ! ---------------------------------------------------------
  subroutine ps_in_grid_kb_cosines(ps, lloc)
    type(ps_in_grid_t), intent(inout) :: ps
    integer,            intent(in)    :: lloc
 
    integer :: ir, l
    real(real64) :: dnrm, avgv, vphi
    real(real64), parameter :: tol = 1.0e-20_real64
 
    push_sub(ps_in_grid_kb_cosines)
 
    do l = 1, ps%no_l_channels
      if (l-1 == lloc) then
        ps%dkbcos(l) = m_zero
        ps%dknorm(l) = m_zero
        cycle
      end if
 
      dnrm = m_zero
      avgv = m_zero
      do ir = 1, ps%g%nrval
        vphi = (ps%vps(ir, l) - ps%vlocal(ir))*ps%rphi(ir, l, 1)
        dnrm = dnrm + vphi*vphi*ps%g%drdi(ir)
        avgv = avgv + vphi*ps%rphi(ir, l, 1)*ps%g%drdi(ir)
      end do
      ps%dkbcos(l) = dnrm/safe_tol(avgv, tol)
      ps%dknorm(l) = m_one/safe_tol(sqrt(dnrm), tol)
    end do
 
    do l = 1, ps%so_no_l_channels
      dnrm = m_zero
      avgv = m_zero
      do ir = 1, ps%g%nrval
        vphi = ps%so_vps(ir, l)*ps%rphi(ir, l, 1)
        dnrm = dnrm + vphi*vphi*ps%g%drdi(ir)
        avgv = avgv + vphi*ps%rphi(ir, l, 1)*ps%g%drdi(ir)
      end do
      ps%so_dkbcos(l) = dnrm/safe_tol(avgv, tol)
      ps%so_dknorm(l) = m_one/safe_tol(sqrt(dnrm), tol)
    end do
 
    pop_sub(ps_in_grid_kb_cosines)
  end subroutine ps_in_grid_kb_cosines
 
 
  ! ---------------------------------------------------------
  subroutine ps_in_grid_cutoff_radii(ps, lloc)
    type(ps_in_grid_t), intent(inout) :: ps
    integer,            intent(in)    :: lloc
 
    integer          :: l, ir
    real(real64)     :: dincv, phi
    real(real64), parameter :: threshold = 1.0e-6_real64
 
    push_sub(ps_in_grid_cutoff_radii)
 
    ! local part ....
    do ir = ps%g%nrval-1, 2, -1
      dincv = abs(ps%vlocal(ir)*ps%g%rofi(ir) + m_two*ps%zval)
      if (dincv > threshold) exit
    end do
    ps%kb_radius(ps%no_l_channels+1) = ps%g%rofi(ir + 1)
    ps%kb_nrval(ps%no_l_channels+1) = ir + 1
 
    ! non-local part....
    do l = 1, ps%no_l_channels
      if (l-1 == lloc) then
        ps%kb_radius(l) = m_zero
        ps%kb_nrval(l) = 0
        cycle
      end if
 
      do ir = ps%g%nrval-1, 2, -1
        phi = (ps%rphi(ir, l, 1)/ps%g%rofi(ir))*ps%dknorm(l)
        dincv = abs((ps%vps(ir, l) - ps%vlocal(ir))*phi)
 
        if (dincv > threshold) exit
 
        phi = (ps%rphi(ir, l, 1)/ps%g%rofi(ir))*ps%dknorm(l)
      end do
 
      ps%kb_radius(l) = ps%g%rofi(ir + 1)
      ps%kb_nrval(l) = ir + 1
    end do
 
    ! now the SO part
    do l = 1, ps%so_no_l_channels
      do ir = ps%g%nrval-1, 2, -1
        phi = (ps%rphi(ir, l, 1)/ps%g%rofi(ir))*ps%so_dknorm(l)
        dincv = abs((ps%so_vps(ir, l))*phi)
 
        if (dincv > threshold) exit
      end do
 
      ps%so_kb_radius(l) = ps%g%rofi(ir + 1)
      ps%so_kb_nrval(l) = ir + 1
    end do
 
    pop_sub(ps_in_grid_cutoff_radii)
  end subroutine ps_in_grid_cutoff_radii
 
 
  ! ---------------------------------------------------------
  subroutine ps_in_grid_check_rphi(ps, namespace)
    type(ps_in_grid_t), intent(in) :: ps
    type(namespace_t),  intent(in) :: namespace
 
    integer :: l
    real(real64)   :: nrm
 
    push_sub(ps_in_grid_check_rphi)
 
    !  checking normalization of the wavefunctions
    do l = 1, ps%no_l_channels
      nrm = sqrt(sum(ps%g%drdi(:)*ps%rphi(:, l, 1)**2))
      nrm = abs(nrm - m_one)
      if (nrm > 1.0e-5_real64) then
        write(message(1), '(a,i2,a)') "Eigenstate for l = ", l-1, ' is not normalized.'
        write(message(2), '(a, f12.6,a)') '(abs(1 - norm) = ', nrm, ')'
        call messages_warning(2, namespace=namespace)
      end if
    end do
 
    pop_sub(ps_in_grid_check_rphi)
  end subroutine ps_in_grid_check_rphi
 
  ! ---------------------------------------------------------
 
  real(real64) function first_point_extrapolate(x, y, high_order) result(y0)
    real(real64),      intent(in)    :: x(:)
    real(real64),      intent(in)    :: y(:)
    logical, optional, intent(in)    :: high_order
 
    real(real64) :: x1, x2, x3
    real(real64) :: y1, y2, y3
 
    x1 = x(2) - x(1)
    x2 = x(3) - x(1)
    x3 = x(4) - x(1)
    y1 = y(2)
    y2 = y(3)
    y3 = y(4)
 
    if (optional_default(high_order, .false.)) then
 
      y0 = y1*x2*x3*(x2 - x3) + y2*x1*x3*(x3 - x1) + y3*x1*x2*(x1 - x2)
      y0 = y0/((x1 - x2)*(x1 - x3)*(x2 - x3))
 
    else
 
      y0 = y1 - (y2 - y1)*x1/(x2 - x1)
 
    end if
 
  end function first_point_extrapolate
 
end module ps_in_grid_oct_m
 
!! Local Variables:
!! mode: f90
!! coding: utf-8
!! End: