poissonnoiseextension.h

#ifndef CTL_POISSONNOISEEXTENSION_H
#define CTL_POISSONNOISEEXTENSION_H

#include "projectorextension.h"
#include "acquisition/acquisitionsetup.h"

#include <random>

namespace CTL {

class PoissonNoiseExtension : public ProjectorExtension
{
    CTL_TYPE_ID(103)

    // abstract interface
    public: void configure(const AcquisitionSetup& setup) override;

public:
    PoissonNoiseExtension() = default;
    using ProjectorExtension::ProjectorExtension;
    explicit PoissonNoiseExtension(uint fixedSeed, bool useParalellization = true);

    // ProjectorExtension interface
    bool isLinear() const override;

    // SerializationInterface interface
    QVariant toVariant() const override;
    QVariant parameter() const override;
    void setParameter(const QVariant& parameter) override;

    void setFixedSeed(uint seed);
    void setRandomSeedMode();
    void setParallelizationEnabled(bool enabled);

protected:
    ProjectionData extendedProject(const MetaProjector& nestedProjector) override;

private:
    static void processViewCompact(SingleViewData& view, const std::vector<float>& i_0, uint seed);

    std::mt19937 _rng;
    AcquisitionSetup _setup; 
    bool _useParallelization{ true };
    bool _useFixedSeed{ false };
    uint _seed{ 0u };
};

} // namespace CTL

#endif // CTL_POISSONNOISEEXTENSION_H