dm_ls_scf_types.F

Go to the documentation of this file.

!--------------------------------------------------------------------------------------------------!
!   CP2K: A general program to perform molecular dynamics simulations                              !
!   Copyright 2000-2024 CP2K developers group <https://cp2k.org>                                   !
!                                                                                                  !
!   SPDX-License-Identifier: GPL-2.0-or-later                                                      !
!--------------------------------------------------------------------------------------------------!
 
! **************************************************************************************************
!> \brief Types needed for a linear scaling quickstep SCF run based on the density
!>        matrix
!> \par History
!>       2010.10 created [Joost VandeVondele]
!> \author Joost VandeVondele
! **************************************************************************************************
MODULE dm_ls_scf_types
   USE dbcsr_api,                       ONLY: dbcsr_release,&
                                              dbcsr_type
   USE input_constants,                 ONLY: ls_cluster_atomic,&
                                              ls_cluster_molecular
   USE input_section_types,             ONLY: section_vals_release,&
                                              section_vals_type
   USE kinds,                           ONLY: dp
   USE message_passing,                 ONLY: mp_para_env_release,&
                                              mp_para_env_type
   USE pao_types,                       ONLY: pao_env_type,&
                                              pao_finalize
   USE pexsi_types,                     ONLY: lib_pexsi_env,&
                                              lib_pexsi_finalize
   USE qs_density_mixing_types,         ONLY: mixing_storage_release,&
                                              mixing_storage_type
#include "./base/base_uses.f90"
 
   IMPLICIT NONE
 
   PRIVATE
 
   CHARACTER(len=*), PARAMETER, PRIVATE :: moduleN = 'dm_ls_scf_types'
 
   PUBLIC :: ls_scf_env_type, ls_mstruct_type, ls_cluster_atomic, ls_cluster_molecular, &
             ls_scf_curvy_type
 
   TYPE ls_mstruct_type
      INTEGER :: cluster_type = -1
      LOGICAL :: do_pao = .false.
      INTEGER, DIMENSION(:), ALLOCATABLE :: atom_to_molecule
      TYPE(dbcsr_type)                :: matrix_a
      TYPE(dbcsr_type)                :: matrix_b
   END TYPE
 
   TYPE ls_mat_history_type
      INTEGER :: istore = 0, nstore = 0
      TYPE(dbcsr_type), DIMENSION(:, :), ALLOCATABLE :: matrix
   END TYPE
 
   TYPE ls_scf_curvy_type
      TYPE(dbcsr_type), DIMENSION(:), ALLOCATABLE :: matrix_dp
      TYPE(dbcsr_type), DIMENSION(:), ALLOCATABLE :: matrix_p
      TYPE(dbcsr_type), DIMENSION(:, :), ALLOCATABLE :: matrix_psave
      TYPE(dbcsr_type), DIMENSION(:, :), ALLOCATABLE :: matrix_bch
      REAL(kind=dp), DIMENSION(2)                      :: step_size = 0.0_dp
      REAL(kind=dp), DIMENSION(2)                      :: shift = 0.0_dp
      REAL(kind=dp), DIMENSION(2)                      :: cg_denom = 0.0_dp
      REAL(kind=dp), DIMENSION(2)                      :: cg_numer = 0.0_dp
      REAL(kind=dp), DIMENSION(6)                      :: energies = 0.0_dp
      INTEGER                                          :: line_search_step = 0
      INTEGER, DIMENSION(2)                            :: bch_saved = 0
      LOGICAL                                          :: double_step_size = .false.
      LOGICAL, DIMENSION(2)                            :: fix_shift = .false.
 
      INTEGER                                          :: line_search_type = 0
      INTEGER                                          :: n_bch_hist = 0
      REAL(kind=dp)                                    :: scale_filter = 0.0_dp
      REAL(kind=dp)                                    :: filter_factor = 0.0_dp
      REAL(kind=dp)                                    :: min_shift = 0.0_dp
      REAL(kind=dp)                                    :: min_filter = 0.0_dp
   END TYPE
 
   TYPE chebyshev_type
      LOGICAL :: compute_chebyshev = .false.
      INTEGER :: n_chebyshev = 0
      INTEGER :: n_gridpoint_dos = 0
      REAL(kind=dp), DIMENSION(:), POINTER :: min_energy => null()
      REAL(kind=dp), DIMENSION(:), POINTER :: max_energy => null()
      TYPE(section_vals_type), POINTER :: print_key_dos => null()
      TYPE(section_vals_type), POINTER :: print_key_cube => null()
   END TYPE
 
   TYPE ls_scf_env_type
      INTEGER               :: nspins = 0, natoms = 0
      INTEGER               :: nelectron_total = 0
      INTEGER, DIMENSION(2) :: nelectron_spin = 0
      REAL(kind=dp), DIMENSION(2) ::  mu_spin = 0.0_dp
      REAL(kind=dp), DIMENSION(2) ::  homo_spin = 0.0_dp
      REAL(kind=dp), DIMENSION(2) ::  lumo_spin = 0.0_dp
 
      ! This prevents a bug in GCC up to version 9.x (9.5 seems to work though)
      TYPE(ls_mat_history_type) :: scf_history = ls_mat_history_type(matrix=null())
      INTEGER :: extrapolation_order = -1
 
      LOGICAL :: has_unit_metric = .false.
 
      LOGICAL :: curvy_steps = .false.
      INTEGER :: s_preconditioner_type = 0
      INTEGER :: s_inversion_type = 0
      INTEGER :: purification_method = 0
      INTEGER :: sign_method = 0
      INTEGER :: sign_order = 0
      LOGICAL :: sign_symmetric = .false.
      INTEGER :: submatrix_sign_method = -1
      INTEGER :: s_sqrt_method = 0
      INTEGER :: s_sqrt_order = 0
 
      LOGICAL               :: needs_s_inv = .false., has_s_preconditioner = .false., fixed_mu = .false., &
                               dynamic_threshold = .false., check_s_inv = .false.
      LOGICAL               :: restart_read = .false., restart_write = .false., non_monotonic = .false.
      REAL(kind=dp)         :: eps_filter = 0.0_dp, eps_scf = 0.0_dp
 
      REAL(kind=dp)         :: eps_lanczos = 0.0_dp
      INTEGER               :: max_iter_lanczos = 0
 
      REAL(kind=dp)         :: mixing_fraction = 0.0_dp
      INTEGER               :: max_scf = 0
      LOGICAL               :: ls_diis = .false.
      INTEGER               :: iter_ini_diis = 0
      INTEGER               :: nmixing = 0, max_diis = 0
      REAL(kind=dp)         :: eps_diis = 0.0_dp
      REAL(kind=dp)         :: energy_init = 0.0_dp
 
      TYPE(dbcsr_type)   :: matrix_s_inv
      TYPE(dbcsr_type)   :: matrix_s
      TYPE(dbcsr_type)   :: matrix_bs_sqrt, matrix_bs_sqrt_inv
      TYPE(dbcsr_type)   :: matrix_s_sqrt, matrix_s_sqrt_inv
      TYPE(dbcsr_type), DIMENSION(:), ALLOCATABLE :: matrix_ks
      TYPE(dbcsr_type), DIMENSION(:), ALLOCATABLE :: matrix_p
 
      LOGICAL  :: report_all_sparsities = .false., perform_mu_scan = .false., use_s_sqrt = .false.
 
      TYPE(ls_mstruct_type) :: ls_mstruct
      TYPE(ls_scf_curvy_type) :: curvy_data = ls_scf_curvy_type(matrix_dp=null(), matrix_p=null(), &
                                                                matrix_psave=null(), matrix_bch=null())
 
      TYPE(chebyshev_type) :: chebyshev = chebyshev_type()
 
      LOGICAL :: do_rho_mixing = .false.
      INTEGER :: density_mixing_method = 0
      TYPE(mixing_storage_type), POINTER :: mixing_store => null()
 
      LOGICAL :: do_transport = .false.
      LOGICAL :: do_pexsi = .false.
 
      LOGICAL :: calculate_forces = .false.
 
      TYPE(lib_pexsi_env) :: pexsi
 
      TYPE(mp_para_env_type), POINTER :: para_env => null()
      LOGICAL                 :: do_pao = .false.
      TYPE(pao_env_type)      :: pao_env
   END TYPE ls_scf_env_type
 
   PUBLIC :: ls_scf_release
 
CONTAINS
 
! **************************************************************************************************
!> \brief release the LS type.
!> \param ls_scf_env ...
!> \par History
!>       2012.11 created [Joost VandeVondele]
!> \author Joost VandeVondele
! **************************************************************************************************
   SUBROUTINE ls_scf_release(ls_scf_env)
      TYPE(ls_scf_env_type), POINTER                     :: ls_scf_env
 
      CHARACTER(len=*), PARAMETER                        :: routinen = 'ls_scf_release'
 
      INTEGER                                            :: handle, ispin, istore
 
      CALL timeset(routinen, handle)
 
      CALL mp_para_env_release(ls_scf_env%para_env)
 
      DEALLOCATE (ls_scf_env%ls_mstruct%atom_to_molecule)
 
      ! set up the buffer for the history of matrices
      DO istore = 1, min(ls_scf_env%scf_history%istore, ls_scf_env%scf_history%nstore)
         DO ispin = 1, SIZE(ls_scf_env%scf_history%matrix, 1)
            CALL dbcsr_release(ls_scf_env%scf_history%matrix(ispin, istore))
         END DO
      END DO
      DEALLOCATE (ls_scf_env%scf_history%matrix)
 
      IF (ALLOCATED(ls_scf_env%matrix_p)) THEN
         DO ispin = 1, SIZE(ls_scf_env%matrix_p)
            CALL dbcsr_release(ls_scf_env%matrix_p(ispin))
         END DO
         DEALLOCATE (ls_scf_env%matrix_p)
      END IF
 
      IF (ASSOCIATED(ls_scf_env%chebyshev%print_key_dos)) &
         CALL section_vals_release(ls_scf_env%chebyshev%print_key_dos)
      IF (ASSOCIATED(ls_scf_env%chebyshev%print_key_cube)) &
         CALL section_vals_release(ls_scf_env%chebyshev%print_key_cube)
      IF (ASSOCIATED(ls_scf_env%chebyshev%min_energy)) THEN
         DEALLOCATE (ls_scf_env%chebyshev%min_energy)
      END IF
      IF (ASSOCIATED(ls_scf_env%chebyshev%max_energy)) THEN
         DEALLOCATE (ls_scf_env%chebyshev%max_energy)
      END IF
 
      IF (ASSOCIATED(ls_scf_env%mixing_store)) THEN
         CALL mixing_storage_release(ls_scf_env%mixing_store)
         DEALLOCATE (ls_scf_env%mixing_store)
      END IF
 
      IF (ls_scf_env%do_pexsi) THEN
         CALL lib_pexsi_finalize(ls_scf_env%pexsi)
      END IF
 
      IF (ls_scf_env%do_pao) &
         CALL pao_finalize(ls_scf_env%pao_env)
 
      DEALLOCATE (ls_scf_env)
 
      CALL timestop(handle)
 
   END SUBROUTINE ls_scf_release
 
END MODULE dm_ls_scf_types