R3BNeulandGeneratorFactory.h

Go to the documentation of this file.

#pragma once
 
#include "FairBoxGenerator.h"
#include "FairPrimaryGenerator.h"
#include "R3BNeulandCommon.h"
#include "R3BValueError.h"
#include <Math/Vector3D.h> // NOLINT
#include <Math/Vector3Dfwd.h>
#include <R3BMinMaxValue.h>
#include <TRandom3.h>
#include <cstdint>
#include <functional>
#include <memory>
#include <string>
 
namespace R3B::Neuland
{
    enum class GeneratorType : uint8_t
    {
        box,
        muon
    };
 
    class GeneratorFactory
    {
      public:
        struct Options
        {
            int multiplicity = 1;                              
            GeneratorType generator_type = GeneratorType::box; 
            std::string particle_type = "neutron";             
            ValueErrorD energy = DEFAULT_GENERATOR_ENERGY;     
            MinMaxValueD theta = DEFAULT_BOX_GENERATOR_THETA;  
            MinMaxValueD phi = DEFAULT_BOX_GENERATOR_PHI;      
            ROOT::Math::XYZVectorD position{ 0., 0., 0. };     
        };

        explicit GeneratorFactory(Options& options)
            : options_{ options }
        {
        }

        auto Create() -> std::unique_ptr<FairPrimaryGenerator>;
 
        // setters:
        void SetRandomGen(TRandom3* random_gen) { random_gen_ = random_gen; }
        void SetNumOfPlanes(int num_of_planes) { num_of_planes_ = num_of_planes; }
 
      private:
        int num_of_planes_ = MaxNumberOfPlanes; 
        std::reference_wrapper<Options> options_;
        TRandom3* random_gen_ = nullptr;
        auto create_muon_generator() -> std::unique_ptr<FairPrimaryGenerator>;
        auto create_box_generator() -> std::unique_ptr<FairPrimaryGenerator>;
    };
 
} // namespace R3B::Neuland