wannier_opts.F90

Go to the documentation of this file.

!! Copyright (C) A Buccheri, J Reimann. 2026
!!
!! This Source Code Form is subject to the terms of the Mozilla Public
!! License, v. 2.0. If a copy of the MPL was not distributed with this
!! file, You can obtain one at https://mozilla.org/MPL/2.0/.
 
#include "global.h"
 
module wannier_opts_oct_m
  use, intrinsic :: iso_fortran_env
 
  use debug_oct_m
  use global_oct_m
  use io_oct_m
  use messages_oct_m
  use namespace_oct_m
  use parser_oct_m
  use restart_oct_m
 
  implicit none
  private
 
  public :: &
    wannier_opts_t, &
    wannier_opts_parse_oct, &
    wannier_opts_parse_win
 
  integer, parameter, public :: &
    TD_WANNIER_METHOD_NONE = 0, &
    td_wannier_method_wannier90 = 1
 
  type :: wannier_opts_t
    integer           :: num_bands = 0
    integer           :: num_wann = 0
    logical           :: use_scdm
    real(real64)      :: scdm_mu
    real(real64)      :: scdm_sigma
    real(real64)      :: threshold
    character(len=80) :: prefix
    integer           :: restart_from
    logical           :: plot
    logical           :: wannierize
    logical           :: calc_overlaps
    logical           :: dump_inputs
    logical           :: spinors = .false. 
    integer,      allocatable  :: spin_proj_component(:)
 
    ! TD-related options
    integer           :: td_output_interval
    integer           :: td_method
 
  contains
    procedure :: parse_oct => wannier_opts_parse_oct
    procedure :: parse_win => wannier_opts_parse_win
  end type wannier_opts_t
 
contains
 
  subroutine wannier_opts_parse_oct(this, namespace)
    class(wannier_opts_t), intent(inout) :: this
    type(namespace_t),  intent(in)    :: namespace
 
    logical :: read_from_td
 
    push_sub(wannier_opts_parse_oct)
 
    !%Variable Wannier90Prefix
    !%Type string
    !%Default w90
    !%Section Utilities::oct-wannier90
    !%Description
    !% Prefix for wannier90 files. Used by both oct-wannier90 and oct-wannier90lib
    !% as well as the time-dependent Wannier workflow.
    !%End
    call parse_variable(namespace, 'Wannier90Prefix', 'w90', this%prefix)
 
    !%Variable Wannier90UseTD
    !%Type logical
    !%Default no
    !%Section Utilities::oct-wannier90
    !%Description
    !% This variable controls which states are used as input into the standalone utilities
    !% oct-wannier90 and oct-wannier90lib.
    !% By default oct-wannier90 uses the ground-state states.
    !% By setting this variable to yes, oct-wannier90 will use the TD states instead.
    !% This parameter does not apply to the time-dependent Wannier workflow.
    !%End
    call parse_variable(global_namespace, 'Wannier90UseTD', .false., read_from_td)
    if (read_from_td) then
      this%restart_from = restart_td
    else
      this%restart_from = restart_gs
    end if
 
    !%Variable Wannier90DumpFiles
    !%Type logical
    !%Default no
    !%Section Utilities::oct-wannier90lib
    !%Description
    !% Whether or not to dump intermediate files to disk.
    !% This applies only to the oct-wannierlib utility.
    !% Files are: .amn, .mmn, .eig and .win with inputs.
    !%End
    call parse_variable(global_namespace, 'Wannier90DumpFiles', .false., this%dump_inputs)
 
    !%Variable Wannier90Wannierize
    !%Type logical
    !%Default yes
    !%Section Utilities::oct-wannier90lib
    !%Description
    !% Whether or not to perform the wannierization step (including disentanglement).
    !% This applies only to the standalone utility oct-wannier90lib.
    !%End
    call parse_variable(global_namespace, 'Wannier90Wannierize', .true., this%wannierize)
 
    !%Variable Wannier90Plot
    !%Type logical
    !%Default no
    !%Section Utilities::oct-wannier90lib
    !%Description
    !% If set to yes, oct-wannier90lib will generate the wannier functions on a real space
    !% grid and write them to file.
    !%End
    call parse_variable(global_namespace, 'Wannier90Plot', .false., this%plot)
 
    !%Variable Wannier90CalcOverlaps
    !%Type logical
    !%Default no
    !%Section Utilities::oct-wannier90lib
    !%Description
    !% Whether or not to calculate the amn and mmn matrices from octopus states.
    !% By setting this variable to no, oct-wannier90 will attempt to read them from files.
    !% This applies only to the standalone utility oct-wannier90lib.
    !%End
    call parse_variable(global_namespace, 'Wannier90CalcOverlaps', .false., this%calc_overlaps)
 
    !%Variable Wannier90UseSCDM
    !%Type logical
    !%Default no
    !%Section Utilities::oct-wannier90
    !%Description
    !% By default oct-wannier90 and oct-wannier90lib use the projection method to generate the .amn file.
    !% By setting this variable to yes, they will use the SCDM method instead.
    !%End
    call parse_variable(namespace, 'Wannier90UseSCDM', .false., this%use_scdm)
    if (this%use_scdm) then
      !%Variable SCDMsigma
      !%Type float
      !%Default 0.2
      !%Section Utilities::oct-wannier90
      !%Description
      !% Broadening of SCDM smearing function.
      !%End
      call parse_variable(namespace, 'SCDMsigma', 0.2_real64, this%scdm_sigma)
 
      !%Variable SCDMmu
      !%Type float
      !%Section Utilities::oct-wannier90
      !%Description
      !% Energy range up to which states are considered for SCDM.
      !%End
      call parse_variable(namespace, 'SCDMmu', m_huge, this%scdm_mu)
    end if
 
    ! We use the variable AOThreshold to deterine the threshold on the radii of the atomic orbitals
    ! TODO: Have the threshold for wannier functions different than for AO
    call parse_variable(namespace, 'AOThreshold', 0.001_real64, this%threshold)
 
    ! TODO: Parse the exclude_bands variable correctly from .win file here!
 
    !%Variable TDWannierMethod
    !%Type integer
    !%Default 0
    !%Section Time-Dependent
    !%Description
    !% Specifies which method to use for TD Wannier calculations.
    !%Option none 0
    !% TD Wannier functionality is disabled.
    !%Option wannier90 1
    !% Performs a full wannierization of the time-dependent Kohn-Sham states to update the U matrix.
    !%End
    call parse_variable(namespace, 'TDWannierMethod', 0, this%td_method)
    if (this%td_method /= td_wannier_method_none) then
      message(1) = 'Time-dependent Wannierization enabled, but not yet implemented.'
      call messages_warning(1)
    end if
 
    !%Variable TDWannierOutputInterval
    !%Type integer
    !%Default 1
    !%Section Time-Dependent::TD Output
    !%Description
    !% If TDWannierMethod is enabled, Wannier-related output is written
    !% every TDWannierOutputInterval iterations.
    !%End
    call parse_variable(namespace, 'TDWannierOutputInterval', 1, this%td_output_interval)
 
    pop_sub(wannier_opts_parse_oct)
 
  end subroutine wannier_opts_parse_oct
 
  subroutine wannier_opts_parse_win(this)
    class(wannier_opts_t), intent(inout) :: this
 
    character(len=80)       :: filename, line, dummy, dummy1, dummy2
    integer                 :: w90_win, io, tmp_int
    logical                 :: exist
 
    push_sub(wannier_opts_parse_win)
 
    filename = trim(adjustl(this%prefix)) //'.win'
    inquire(file=filename,exist=exist)
    if (.not. exist) then
      message(1) = 'Cannot find specified Wannier90 win file.'
      call messages_fatal(1)
    else
      message(1) = 'Reading file '//filename
      w90_win = io_open(trim(filename), action='read')
      do
        read(w90_win, '(A)', iostat=io) line
        if (is_iostat_end(io)) exit
 
        if (index(line, '=') > 0) then
          read(line, *, iostat=io) dummy, dummy2, dummy1
        else
          read(line, *, iostat=io) dummy, dummy1
        end if
        if (dummy == 'num_bands') then
          read(dummy1, *, iostat=io) tmp_int
          this%num_bands = tmp_int
        else if (dummy == 'num_wann') then
          read(dummy1, *, iostat=io) tmp_int
          this%num_wann = tmp_int
        end if
      end do
      call io_close(w90_win)
    end if
 
    pop_sub(wannier_opts_parse_win)
 
  end subroutine wannier_opts_parse_win
 
end module wannier_opts_oct_m