esolver_ks.h

Go to the documentation of this file.

#ifndef ESOLVER_KS_H
#define ESOLVER_KS_H
 
#include <cstring>
//#include <fstream>
 
// for first-principles esolver
#include "esolver_fp.h"
// for plane wave basis set 
#include "source_basis/module_pw/pw_basis_k.h"
// for k-points in Brillouin zone
#include "source_cell/klist.h"
// for charge mixing
#include "source_estate/module_charge/charge_mixing.h"
// for electronic wave functions
#include "source_psi/psi.h"
// for Hamiltonian
#include "source_hamilt/hamilt.h"
 
namespace ModuleESolver
{
 
template <typename T, typename Device = base_device::DEVICE_CPU>
class ESolver_KS : public ESolver_FP
{
  public:
    ESolver_KS();
 
    virtual ~ESolver_KS();
 
    virtual void before_all_runners(UnitCell& ucell, const Input_para& inp) override;
 
    virtual void runner(UnitCell& ucell, const int istep) override;
 
    virtual void after_all_runners(UnitCell& ucell) override;
 
  protected:
    virtual void before_scf(UnitCell& ucell, const int istep) override;
 
    virtual void iter_init(UnitCell& ucell, const int istep, const int iter);
 
    virtual void iter_finish(UnitCell& ucell, const int istep, int& iter, bool& conv_esolver) override;
 
    // calculate electron density from a specific Hamiltonian with ethr
    virtual void hamilt2rho_single(UnitCell& ucell, const int istep, const int iter, const double ethr);
 
    // calculate electron density from a specific Hamiltonian
    void hamilt2rho(UnitCell& ucell, const int istep, const int iter, const double ethr);
 
    virtual void after_scf(UnitCell& ucell, const int istep, const bool conv_esolver) override;
 
    virtual void update_pot(UnitCell& ucell, const int istep, const int iter, const bool conv_esolver){};
 
    hamilt::Hamilt<T, Device>* p_hamilt = nullptr;
 
    ModulePW::PW_Basis_K* pw_wfc = nullptr;
 
    Charge_Mixing* p_chgmix = nullptr;
 
    pseudopot_cell_vnl ppcell;
 
    psi::Psi<T>* psi = nullptr;
 
    std::string basisname;      
    double esolver_KS_ne = 0.0; 
    double diag_ethr;           
    double scf_thr;             
    double scf_ene_thr;         
    double drho;                
    double hsolver_error;       
    int maxniter;               
    int niter;                  
    bool oscillate_esolver = false; // whether esolver is oscillated
};
} // namespace ModuleESolver
#endif