batchtrackingstate.hpp

#pragma once
 
#include <vector>
#include <armadillo>
 
#include "composition.hpp"
 
class BatchTrackingState
{
public:
    friend class BatchTracking;
 
    struct Batch
    {
        Batch(const double position, const arma::vec& concentration):
            m_position(position),
            m_concentration(concentration)
        {}
 
        double m_position; 
        arma::vec m_concentration; 
 
    public:
        double position() const { return m_position; }
        const arma::vec& concentration() const { return m_concentration; }
    };
 
    BatchTrackingState(
            const arma::vec& gridPointsIncludingEndPoint,
            const arma::vec& concentration = Composition::defaultComposition,
            const arma::uword nBatches = 1);
 
    BatchTrackingState(
            const arma::vec& gridPointsIncludingEndPoint,
            const std::vector<Composition>& composition);
 
    std::vector<Composition> sample() const;
 
    std::vector<Composition> sample(const arma::vec& locations) const;
 
    std::vector<arma::vec> sampleToVec() const;
 
    std::vector<arma::vec> sampleToVec(const arma::vec& locations) const;
 
    std::vector<arma::vec> sampleInternal(const arma::vec& locations) const;
 
    const std::vector<Batch>& batches() const { return m_batches; }
 
protected:
    std::vector<Batch> m_batches; 
    arma::vec m_gridPoints; 
};