FdSimulation/PrimaryGeneratorKG/PrimaryGenerator.h

Go to the documentation of this file.

#ifndef _PrimaryGeneratorKG_PrimaryGenerator_h_
#define _PrimaryGeneratorKG_PrimaryGenerator_h_

#include <fwk/VModule.h>
#include <vector>


namespace utl {
  class RandomEngine;
  class Branch;
  class RandomSamplerFromPDF;
}

namespace evt {
  class Event;
}


namespace PrimaryGeneratorKG {

  class PrimaryGenerator : public fwk::VModule {

  public:
    PrimaryGenerator() { }
    virtual ~PrimaryGenerator();

    fwk::VModule::ResultFlag Init();
    fwk::VModule::ResultFlag Run(evt::Event& theEvent);
    fwk::VModule::ResultFlag Finish();

  private:
    double DiceEnergy();

    double fMinLogE = 0;
    double fMaxLogE = 0;
    double fIndex = 0;
    unsigned int fVerbosity = 1;
    std::vector<int> fPrimZ;
    std::vector<int> fPrimA;

    utl::RandomEngine* fRandomEngine = nullptr;
    utl::RandomSamplerFromPDF* fRandomPDF = nullptr;

    REGISTER_MODULE("PrimaryGeneratorKG", PrimaryGenerator);

  };

}


#endif