target_tdlocal.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_tdlocal_oct_m
  use batch_ops_oct_m
  use debug_oct_m
  use electron_space_oct_m
  use global_oct_m
  use grid_oct_m
  use io_oct_m
  use ions_oct_m
  use ion_dynamics_oct_m
  use io_function_oct_m
  use, intrinsic :: iso_fortran_env
  use mesh_oct_m
  use mesh_function_oct_m
  use messages_oct_m
  use namespace_oct_m
  use output_low_oct_m
  use parser_oct_m
  use profiling_oct_m
  use space_oct_m
  use states_elec_oct_m
  use string_oct_m
  use target_low_oct_m
  use td_oct_m
  use unit_oct_m
  use unit_system_oct_m
 
 
  implicit none
 
  private
  public ::                &
    target_init_tdlocal,   &
    target_end_tdlocal,    &
    target_j1_tdlocal,     &
    target_output_tdlocal, &
    target_tdcalc_tdlocal, &
    target_build_tdlocal,  &
    target_chi_tdlocal
 
 
contains
 
  ! ----------------------------------------------------------------------
  subroutine target_init_tdlocal(gr, namespace, tg, td)
    type(grid_t),      intent(in)    :: gr
    type(namespace_t), intent(in)    :: namespace
    type(target_t),    intent(inout) :: tg
    type(td_t),        intent(in)    :: td
 
    type(block_t)       :: blk
    push_sub(target_init_tdlocal)
 
    tg%move_ions = ion_dynamics_ions_move(td%ions_dyn)
    tg%dt = td%dt
 
    !%Variable OCTTdTarget
    !%Type block
    !%Section Calculation Modes::Optimal Control
    !%Description
    !% (Experimental) If <tt>OCTTargetOperator = oct_tg_td_local</tt>, then you must supply
    !% a OCTTdTarget block. The block should only contain one element, a string cotaining the
    !% definition of the time-dependent local target, <i>i.e.</i> a function of x,y,z and t that
    !% is to be maximized along the evolution.
    !%End
    if (parse_block(namespace, 'OCTTdTarget', blk) == 0) then
      call parse_block_string(blk, 0, 0, tg%td_local_target)
      call conv_to_c_string(tg%td_local_target)
      safe_allocate(tg%rho(1:gr%np))
      call parse_block_end(blk)
    else
      message(1) = 'If OCTTargetOperator = oct_tg_td_local, you must supply a OCTTdTarget block.'
      call messages_fatal(1, namespace=namespace)
    end if
    safe_allocate(tg%td_fitness(0:td%max_iter))
    tg%td_fitness = m_zero
    call target_build_tdlocal(tg, gr, m_zero)
 
    pop_sub(target_init_tdlocal)
  end subroutine target_init_tdlocal
 
 
  ! ----------------------------------------------------------------------
  subroutine target_end_tdlocal(tg)
    type(target_t),   intent(inout) :: tg
 
    push_sub(target_end_tdlocal)
 
    safe_deallocate_a(tg%rho)
    safe_deallocate_a(tg%td_fitness)
 
    pop_sub(target_end_tdlocal)
  end subroutine target_end_tdlocal
 
 
  ! ----------------------------------------------------------------------
  subroutine target_output_tdlocal(tg, namespace, space, gr, dir, ions, outp)
    type(target_t),    intent(inout) :: 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(output_t),    intent(in)    :: outp
 
    integer :: ierr
    push_sub(target_output_tdlocal)
 
    call io_mkdir(trim(dir), namespace)
    call target_build_tdlocal(tg, gr, m_zero)
    call dio_function_output(outp%how(0), trim(dir), 'td_local_target', namespace, space, gr, &
      tg%rho, units_out%length**(-space%dim), ierr, pos=ions%pos, atoms=ions%atom)
 
    pop_sub(target_output_tdlocal)
  end subroutine target_output_tdlocal
  ! ----------------------------------------------------------------------
 
 
  ! ----------------------------------------------------------------------
  real(real64) function target_j1_tdlocal(tg) result(j1)
    type(target_t), intent(in) :: tg
 
    integer :: maxiter
    push_sub(target_j1_tdlocal)
 
    maxiter = size(tg%td_fitness) - 1
    j1 = m_half * tg%dt * tg%td_fitness(0) + &
      m_half * tg%dt * tg%td_fitness(maxiter) + &
      tg%dt * sum(tg%td_fitness(1:maxiter-1))
 
 
    pop_sub(target_j1_tdlocal)
  end function target_j1_tdlocal
 
 
  ! ----------------------------------------------------------------------
  subroutine target_chi_tdlocal(chi_out)
    type(states_elec_t), intent(inout) :: chi_out
 
    integer :: ik, ib
    push_sub(target_chi_tdlocal)
 
    !We assume that there is no time-independent operator.
 
    do ik = chi_out%d%kpt%start, chi_out%d%kpt%end
      do ib = chi_out%group%block_start, chi_out%group%block_end
        call batch_set_zero(chi_out%group%psib(ib, ik))
      end do
    end do
 
    pop_sub(target_chi_tdlocal)
  end subroutine target_chi_tdlocal
 
 
  ! ---------------------------------------------------------
  subroutine target_tdcalc_tdlocal(tg, gr, psi, time)
    type(target_t),      intent(inout) :: tg
    type(grid_t),        intent(in)    :: gr
    type(states_elec_t), intent(in)    :: psi
    integer,             intent(in)    :: time
 
    complex(real64), allocatable :: opsi(:, :), zpsi(:, :)
    integer :: ist, ip
    push_sub(target_tdcalc_tdlocal)
 
    tg%td_fitness(time) = m_zero
 
    safe_allocate(zpsi(1:gr%np, 1:psi%d%dim))
 
    !!!! WARNING Here one should build the time-dependent target.
    select case (psi%d%ispin)
    case (unpolarized)
      assert(psi%nik == 1)
      safe_allocate(opsi(1:gr%np_part, 1))
      opsi = m_z0
      do ist  = psi%st_start, psi%st_end
 
        call states_elec_get_state(psi, gr, ist, 1, zpsi)
 
        do ip = 1, gr%np
          opsi(ip, 1) = tg%rho(ip)*zpsi(ip, 1)
        end do
 
        tg%td_fitness(time) = tg%td_fitness(time) + psi%occ(ist, 1)*real(zmf_dotp(gr, psi%d%dim, zpsi, opsi), real64)
 
      end do
      safe_deallocate_a(opsi)
    case (spin_polarized)
      message(1) = 'Error in target.target_tdcalc: spin_polarized.'
      call messages_fatal(1)
    case (spinors)
      message(1) = 'Error in target.target_tdcalc: spinors.'
      call messages_fatal(1)
    end select
 
    safe_deallocate_a(zpsi)
 
    pop_sub(target_tdcalc_tdlocal)
  end subroutine target_tdcalc_tdlocal
  ! ----------------------------------------------------------------------
 
 
  !----------------------------------------------------------
  subroutine target_build_tdlocal(tg, gr, time)
    type(target_t), intent(inout) :: tg
    type(grid_t),   intent(in)    :: gr
    real(real64),   intent(in)    :: time
 
    integer :: ip
    real(real64) :: xx(gr%box%dim), rr, re, im
 
    push_sub(target_build_tdlocal)
 
    do ip = 1, gr%np
      call mesh_r(gr, ip, rr, coords = xx)
      call parse_expression(re, im, gr%box%dim, xx, rr, time, tg%td_local_target)
      tg%rho(ip) = re
    end do
 
    pop_sub(target_build_tdlocal)
  end subroutine target_build_tdlocal
  !----------------------------------------------------------
 
end module target_tdlocal_oct_m
 
!! Local Variables:
!! mode: f90
!! coding: utf-8
!! End: