hamilt_sdft_pw.h

Go to the documentation of this file.

#ifndef HAMILTSDFTPW_H
#define HAMILTSDFTPW_H
 
#include "source_pw/module_pwdft/hamilt_pw.h"
 
namespace hamilt
{
 
template <typename T, typename Device = base_device::DEVICE_CPU>
class HamiltSdftPW : public HamiltPW<T, Device>
{
  public:
    using Real = typename GetTypeReal<T>::type;
    HamiltSdftPW(elecstate::Potential* pot_in,
                 ModulePW::PW_Basis_K* wfc_basis,
                 K_Vectors* p_kv,
                 pseudopot_cell_vnl* nlpp,
                 const UnitCell* ucell,
                 const int& npol,
                 Real* emin_in,
                 Real* emax_in);
    ~HamiltSdftPW(){};
 
    void hPsi(const T* psi_in, T* hpsi, const int& nbands = 1);
 
    void hPsi_norm(const T* psi_in, T* hpsi, const int& nbands = 1);
 
    Real* emin = nullptr; 
    Real* emax = nullptr; 
 
  private:
    int npwk_max = 0;      
    int npol = 0;          
    std::vector<int>& ngk; 
};
 
} // namespace hamilt
 
#endif