mixing_data.h

Go to the documentation of this file.

#ifndef MIXING_DATA_H_
#define MIXING_DATA_H_
#include <vector>
 
#include "source_base/module_container/ATen/tensor.h"
namespace Base_Mixing
{
 
class Mixing_Data
{
  public:
    Mixing_Data() = default;
    Mixing_Data(const int& ndim, const std::size_t& length, const size_t& type_size);
 
    ~Mixing_Data();
 
    void resize(const int& ndim, const std::size_t& length, const size_t& type_size);
 
    template <typename FPTYPE>
    void push(const FPTYPE* data_in)
    {
        this->start = (this->start + 1) % this->ndim_tot;
        this->ndim_use = std::min(this->ndim_use + 1, this->ndim_tot);
        ++this->ndim_history;
        FPTYPE* FP_startdata = static_cast<FPTYPE*>(this->data) + this->start * this->length;
#ifdef _OPENMP
#pragma omp parallel for schedule(static, 4096/sizeof(FPTYPE))
#endif
        for (std::size_t i = 0; i < length; ++i)
        {
            FP_startdata[i] = data_in[i];
        }
    }
 
    void reset()
    {
        this->ndim_use = 0;
        this->ndim_history = 0;
        this->start = -1;
    }
 
    int index_move(const int& n) const
    {
        return (n + this->start + ndim_tot) % ndim_tot;
    }
 
  public:
    // Tensor pointer to store the data
    void* data = nullptr;
    // the number of vectors for mixing
    int ndim_tot = 0;
    // the length of each vector
    std::size_t length = 0;
    // the start index for vector: start = this->index_move(0)
    int start = -1;
    // the number of used vectors for mixing
    int ndim_use = 0;
    // the number of history vectors
    int ndim_history = 0;
};
 
} // namespace Base_Mixing
#endif