SamplesAccumulator.h

// Samples Accumulator class interface
// Written by Sergey G. Kosov in 2015, 2017 for Project X
#pragma once

#include "types.h"

namespace DirectGraphicalModels
{
    // ================================ Samples Accumulator Class ==============================
    class CSamplesAccumulator
    {
    public:
        CSamplesAccumulator(byte nStates, size_t maxSamples)
            : m_vSamplesAcc(vec_mat_t(nStates))
            , m_vNumInputSamples(vec_int_t(nStates, 0))
            , m_maxSamples(maxSamples ? static_cast<int>(maxSamples) : std::numeric_limits<int>::max())
        { }
        CSamplesAccumulator(const CSamplesAccumulator&) = delete;
        ~CSamplesAccumulator(void) {}

        CSamplesAccumulator& operator = (const CSamplesAccumulator&) = delete;

        void    reset(void);
        void    addSample(const Mat &featureVector, byte state);
        Mat     getSamplesContainer(byte state) const { return m_vSamplesAcc[state]; }
        int     getNumSamples(byte state) const;
        int     getNumInputSamples(byte state) const;
        void    release(byte state);


    private:
        vec_mat_t   m_vSamplesAcc;                      // = vec_mat_t(nStates);    // Samples container for all states
        vec_int_t   m_vNumInputSamples;                 // = vec_int_t(nStates, 0); // Amount of input samples for all states       
        int         m_maxSamples;                       // = INFINITY;              // for optimisation purposes
    };
}