mixing.h

Go to the documentation of this file.

#ifndef MIXING_H_
#define MIXING_H_
#include <functional>
 
#include "mixing_data.h"
 
namespace Base_Mixing
{
 
class Mixing
{
  public:
    Mixing(){};
    virtual ~Mixing(){};
 
    virtual void init_mixing_data(Mixing_Data& mdata, const int& length, const size_t& type_size) const
    {
        mdata.resize(data_ndim, length, type_size);
    }
 
    virtual void push_data(Mixing_Data& mdata,
                           const double* data_in,
                           const double* data_out,
                           std::function<void(double*)> screen,
                           std::function<void(double*, const double*, const double*)> mix,
                           const bool& need_calcoef)
        = 0;
    virtual void push_data(
        Mixing_Data& mdata,
        const std::complex<double>* data_in,
        const std::complex<double>* data_out,
        std::function<void(std::complex<double>*)> screen,
        std::function<void(std::complex<double>*, const std::complex<double>*, const std::complex<double>*)> mix,
        const bool& need_calcoef)
        = 0;
 
    virtual void push_data(Mixing_Data& mdata,
                           const double* data_in,
                           const double* data_out,
                           std::function<void(double*)> screen,
                           const bool& need_calcoef);
    virtual void push_data(Mixing_Data& mdata,
                           const std::complex<double>* data_in,
                           const std::complex<double>* data_out,
                           std::function<void(std::complex<double>*)> screen,
                           const bool& need_calcoef);
 
    virtual void cal_coef(const Mixing_Data& mdata, std::function<double(double*, double*)> inner_product) = 0;
    virtual void cal_coef(const Mixing_Data& mdata,
                          std::function<double(std::complex<double>*, std::complex<double>*)> inner_product)
        = 0;
 
    void mix_data(const Mixing_Data& mdata, double* data_mix);
    void mix_data(const Mixing_Data& mdata, std::complex<double>* data_mix);
 
    virtual void reset() = 0;
 
  public:
    // mixing_beta from INPUT
    double mixing_beta = 0.7;
    // coeficients for mixing
    std::vector<double> coef;
    // ndim for mixing
    int data_ndim = 1;
};
 
} // namespace Base_Mixing
 
#endif