Mix_DMk_2D.h

Go to the documentation of this file.

//=======================
// AUTHOR : Peize Lin
// DATE :   2023-05-09
//=======================
 
#ifndef MIX_DMK_2D_H
#define MIX_DMK_2D_H
 
#include "Mix_Matrix.h"
#include "source_base/module_mixing/mixing.h"
#include "source_base/matrix.h"
#include "source_base/complexmatrix.h"
 
#include <vector>
 
class Mix_DMk_2D
{
public:
    Mix_DMk_2D &set_nks(const int nks, const bool gamma_only_in);
 
    Mix_DMk_2D &set_mixing(Base_Mixing::Mixing* mixing_in);
 
    Mix_DMk_2D &set_mixing_beta(const double mixing_beta);
 
    void mix(const std::vector<std::vector<double>>& dm, const bool flag_restart);
 
    void mix(const std::vector<std::vector<std::complex<double>>>& dm, const bool flag_restart);
 
    std::vector<const std::vector<double>*> get_DMk_gamma_out() const;
    std::vector<const std::vector<std::complex<double>>*> get_DMk_k_out() const;
 
private:
    std::vector<Mix_Matrix<std::vector<double>>> mix_DMk_gamma;
    std::vector<Mix_Matrix<std::vector<std::complex<double>>>> mix_DMk_k;
    bool gamma_only;
};
 
#endif