target_gstransformation.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 target_gstransformation_oct_m
  use debug_oct_m
  use density_oct_m
  use global_oct_m
  use grid_oct_m
  use hamiltonian_elec_oct_m
  use io_oct_m
  use ions_oct_m
  use ion_dynamics_oct_m
  use, intrinsic :: iso_fortran_env
  use kpoints_oct_m
  use mesh_function_oct_m
  use messages_oct_m
  use namespace_oct_m
  use output_oct_m
  use output_low_oct_m
  use parser_oct_m
  use profiling_oct_m
  use restart_oct_m
  use space_oct_m
  use states_elec_oct_m
  use states_elec_restart_oct_m
  use target_low_oct_m
  use td_oct_m
 
  implicit none
 
  private
  public ::                         &
    target_init_gstransformation,   &
    target_j1_gstransformation,     &
    target_output_gstransformation, &
    target_chi_gstransformation
 
 
contains
 
  ! ----------------------------------------------------------------------
  subroutine target_init_gstransformation(gr, namespace, space, tg, td, restart, kpoints)
    type(grid_t),      intent(in)    :: gr
    type(namespace_t), intent(in)    :: namespace
    class(space_t),    intent(in)    :: space
    type(target_t),    intent(inout) :: tg
    type(td_t),        intent(in)    :: td
    type(restart_t),   intent(inout) :: restart
    type(kpoints_t),   intent(in)    :: kpoints
 
    push_sub(target_init_gstransformation)
 
    message(1) =  'Info: Using Superposition of States for TargetOperator'
    call messages_info(1, namespace=namespace)
 
    tg%move_ions = ion_dynamics_ions_move(td%ions_dyn)
    tg%dt = td%dt
 
    !%Variable OCTTargetTransformStates
    !%Type block
    !%Default no
    !%Section Calculation Modes::Optimal Control
    !%Description
    !% If <tt>OCTTargetOperator = oct_tg_gstransformation</tt>, you must specify a
    !% <tt>OCTTargetTransformStates</tt> block, in order to specify which linear
    !% combination of the states present in <tt>restart/gs</tt> is used to
    !% create the target state.
    !%
    !% The syntax is the same as the <tt>TransformStates</tt> block.
    !%End
    call states_elec_transform(tg%st, namespace, space, restart, gr, kpoints, prefix = "OCTTarget")
 
    if (.not. parse_is_defined(namespace, 'OCTTargetTransformStates')) then
      message(1) = 'If "OCTTargetOperator = oct_tg_superposition", then you must'
      message(2) = 'supply an "OCTTargetTransformStates" block to create the superposition.'
      call messages_fatal(2, namespace=namespace)
    end if
    call density_calc(tg%st, gr, tg%st%rho)
 
    pop_sub(target_init_gstransformation)
  end subroutine target_init_gstransformation
 
 
  ! ----------------------------------------------------------------------
  subroutine target_output_gstransformation(tg, namespace, space, gr, dir, ions, hm, outp)
    type(target_t),      intent(in) :: tg
    type(namespace_t),   intent(in) :: namespace
    class(space_t),      intent(in) :: space
    type(grid_t),        intent(in) :: gr
    character(len=*),    intent(in) :: dir
    type(ions_t),        intent(in) :: ions
    type(hamiltonian_elec_t), intent(in) :: hm
    type(output_t),      intent(in) :: outp
    push_sub(target_output_gstransformation)
 
    call io_mkdir(trim(dir), namespace)
    call output_states(outp, namespace, space, trim(dir), tg%st, gr, ions, hm, -1)
 
    pop_sub(target_output_gstransformation)
  end subroutine target_output_gstransformation
  ! ----------------------------------------------------------------------
 
 
 
  ! ----------------------------------------------------------------------
  real(real64) function target_j1_gstransformation(tg, gr, psi) result(j1)
    type(target_t),      intent(in) :: tg
    type(grid_t),        intent(in) :: gr
    type(states_elec_t), intent(in) :: psi
 
    integer :: ik, ist
 
    complex(real64), allocatable :: zpsi(:, :), zst(:, :)
 
    push_sub(target_j1_gstransformation)
 
    safe_allocate(zpsi(1:gr%np, 1:tg%st%d%dim))
    safe_allocate(zst(1:gr%np, 1:tg%st%d%dim))
 
    j1 = m_zero
    do ik = 1, psi%nik
      do ist = psi%st_start, psi%st_end
 
        call states_elec_get_state(psi, gr, ist, ik, zpsi)
        call states_elec_get_state(tg%st, gr, ist, ik, zst)
 
        j1 = j1 + abs(zmf_dotp(gr, psi%d%dim, zpsi, zst))**2
 
      end do
    end do
 
    safe_deallocate_a(zpsi)
    safe_deallocate_a(zst)
 
    pop_sub(target_j1_gstransformation)
  end function target_j1_gstransformation
 
 
  ! ----------------------------------------------------------------------
  subroutine target_chi_gstransformation(tg, gr, psi_in, chi_out)
    type(target_t),      intent(in)    :: tg
    type(grid_t),        intent(in)    :: gr
    type(states_elec_t), intent(in)    :: psi_in
    type(states_elec_t), intent(inout) :: chi_out
 
    integer :: ik, ist
    complex(real64) :: olap
    complex(real64), allocatable :: zpsi(:, :), zst(:, :), zchi(:, :)
 
    push_sub(target_chi_gstransformation)
 
    safe_allocate(zpsi(1:gr%np, 1:tg%st%d%dim))
    safe_allocate(zst(1:gr%np, 1:tg%st%d%dim))
    safe_allocate(zchi(1:gr%np, 1:tg%st%d%dim))
 
    do ik = 1, psi_in%nik
      do ist = psi_in%st_start, psi_in%st_end
 
        call states_elec_get_state(psi_in, gr, ist, ik, zpsi)
        call states_elec_get_state(tg%st, gr, ist, ik, zst)
 
        olap = zmf_dotp(gr, zst(:, 1), zpsi(:, 1))
        zchi(1:gr%np, 1:tg%st%d%dim) = olap*zst(1:gr%np, 1:tg%st%d%dim)
 
        call states_elec_set_state(chi_out, gr, ist, ik, zchi)
 
      end do
    end do
 
    safe_deallocate_a(zpsi)
    safe_deallocate_a(zst)
    safe_deallocate_a(zchi)
 
    pop_sub(target_chi_gstransformation)
  end subroutine target_chi_gstransformation
 
end module target_gstransformation_oct_m
 
!! Local Variables:
!! mode: f90
!! coding: utf-8
!! End: