abstractsubsetgenerator.h

#ifndef CTL_ABSTRACTSUBSETGENERATOR_H
#define CTL_ABSTRACTSUBSETGENERATOR_H

#include "img/projectiondataview.h"
#include "io/serializationinterface.h"

#include <random>

namespace CTL {

class AcquisitionSetup;

class AbstractSubsetGenerator : public SerializationInterface
{
    CTL_TYPE_ID(1000)

    protected:virtual std::vector<ProjectionDataView> generateSubsetsImpl(uint iteration) const = 0;

public:
    virtual ~AbstractSubsetGenerator() = default;

    // SerializationInterface interface
    void fromVariant(const QVariant& variant) override;
    QVariant toVariant() const override;

    virtual void setSetup(const AcquisitionSetup&) {}
    virtual void setProjections(ProjectionDataView projections);

    // convenience combined setter
    void setData(ProjectionDataView projections, const AcquisitionSetup& setup);

    std::vector<ProjectionDataView> generateSubsets(uint iteration) const;
    virtual std::vector<std::vector<ProjectionDataView>> generateAllSubsets(uint nbIterations);

    bool isSubsetPermutationEnabled() const;
    const ProjectionDataView& projections() const;
    uint randomGeneratorSeed() const;
    void setRandomGeneratorSeed(uint seed = std::random_device{}());
    void setSubsetPermutationEnabled(bool enabled);

protected:
    AbstractSubsetGenerator();
    AbstractSubsetGenerator(const AbstractSubsetGenerator&) = default;
    AbstractSubsetGenerator(AbstractSubsetGenerator&&) = default;
    AbstractSubsetGenerator& operator=(const AbstractSubsetGenerator&) = default;
    AbstractSubsetGenerator& operator=(AbstractSubsetGenerator&&) = default;

    virtual void shuffleSubsets(std::vector<ProjectionDataView>& subsets) const;

    ProjectionDataView _fullProjections = ProjectionDataView::invalidView(); 
    mutable std::mt19937 _rng;      
    uint _rngSeed = 42;             
    bool _permuteSubsets = true;    
};

class AbstractFixedSizeSubsetGenerator : public AbstractSubsetGenerator
{
    CTL_TYPE_ID(1001)

public:
    uint nbSubsets() const;
    void setNbSubsets(uint nbSubsets);

    // SerializationInterface interface
    void fromVariant(const QVariant& variant) override;
    QVariant toVariant() const override;

protected:
    AbstractFixedSizeSubsetGenerator(uint nbSubsets = 1u);

    uint _nbSubsets = 1u;
};


namespace assist {

enum class SubsetArrangement { Evenly, LastIsLargest, LastIsSmallest };

std::vector<uint> subsetSizes(uint nbViews, uint nbSubsets, SubsetArrangement arrangement = SubsetArrangement::Evenly);

} // namespace assist


} // namespace CTL

#endif // CTL_ABSTRACTSUBSETGENERATOR_H