transitionschemeextension.h

#ifndef CTL_TRANSITIONSCHEMEEXTENSION_H
#define CTL_TRANSITIONSCHEMEEXTENSION_H

#include "abstractsubsetgenerator.h"
#include <memory>

namespace CTL {

class TransitionSchemeExtension : public AbstractSubsetGenerator
{
    CTL_TYPE_ID(1100)

    protected: std::vector<ProjectionDataView> generateSubsetsImpl(uint iteration) const override;

public:
    explicit TransitionSchemeExtension(AbstractFixedSizeSubsetGenerator* nestedGenerator);
    explicit TransitionSchemeExtension(std::unique_ptr<AbstractFixedSizeSubsetGenerator> nestedGenerator);

    virtual void setSetup(const AcquisitionSetup& setup) override;
    virtual void setProjections(ProjectionDataView projections) override;

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

    void setTransitionPeriod(uint transitionEveryNIterations);
    void setMaximumNbSubsets(uint maxNbSubsets);
    void setMinimumNbSubsets(uint minNbSubsets);
    void setSubsetBounds(uint initialNbSubsets, uint finalNbSubsets);

    void setSubsetGenerator(AbstractFixedSizeSubsetGenerator* nestedGenerator);
    void setSubsetGenerator(std::unique_ptr<AbstractFixedSizeSubsetGenerator> nestedGenerator);

protected:
    TransitionSchemeExtension() = default;

    std::unique_ptr<AbstractFixedSizeSubsetGenerator> _nestedGen;

    uint _minNbSubsets = 1u;
    uint _maxNbSubsets = 0u;
    uint _transitionPeriod = 1u;

    virtual uint numberOfSubsets(uint iteration) const;
    void shuffleSubsets(std::vector<ProjectionDataView> &subsets) const override;
};

} // namespace CTL

#endif // CTL_TRANSITIONSCHEMEEXTENSION_H