BeamHaloGeneratorAlg.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2024 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef BEAMHALOGENERATORALG_H
#define BEAMHALOGENERATORALG_H
 
#include "GaudiKernel/ServiceHandle.h"
#include "GeneratorModules/GenModule.h"
#include "GaudiKernel/ITHistSvc.h"
#include "TH1F.h"
#include "GeneratorModules/GenData.h"
 
#include <string>
#include <memory>
 
class BeamHaloGenerator;
 

class BeamHaloGeneratorAlg : public GenModule {
public:
  BeamHaloGeneratorAlg(const std::string& name, ISvcLocator *svcLocator);
  virtual ~BeamHaloGeneratorAlg() = default;
  virtual StatusCode genInitialize();
  virtual StatusCode genFinalize();

  virtual StatusCode callGenerator();

  virtual StatusCode fillEvt(HepMC::GenEvent* evt);
 
private:

  StringProperty m_inputTypeStr{this, "inputType", "MARS-NM"};
  
  StringProperty m_inputFile{this, "inputFile", "bgi-b2l1.1"};
  
  DoubleProperty m_interfacePlane{this, "interfacePlane", 20850.0}; // (mm)
  
  BooleanProperty m_enableFlip{this, "enableFlip", false};
  
  FloatProperty m_flipProbability{this, "flipProbability", 0.0};
  
  BooleanProperty m_enableSampling{this, "enableSampling", false};
  
  StringProperty m_bufferFileName{this, "bufferFileName", "BinaryBuffer.bin"};
  
  StringArrayProperty m_generatorSettings{this, "generatorSettings", {}, "A set of cuts to be applied to generated particles."};
  
  BooleanProperty m_doMonitoringPlots{this, "doMonitoringPlots", false};
  
  ServiceHandle<ITHistSvc> m_tHistSvc{this, "THistSvc", "THistSvc"};
  
  StringProperty m_randomStream{this, "randomStream", "BeamHalo"};
  
  BeamHaloGenerator *m_beamHaloGenerator{};

  HepMC::GenEvent m_evt{};
 
  enum validationPlotsEnum {
    PRI_R,
    PRI_Z,
    PRI_Z_TCT, 
    SP_R_ALL,
    SP_E_ALL,
    SP_PZ_ALL,
    SP_PT_ALL,
    SP_R_PROTONS,
    SP_E_PROTONS,
    SP_PZ_PROTONS,
    SP_PT_PROTONS,
    SP_R_MUONS,
    SP_E_MUONS,
    SP_PZ_MUONS,
    SP_PT_MUONS,
    NPLOTS};

  TH1F *m_validationPlots[NPLOTS];
  std::shared_ptr<GenData> m_gendata{nullptr};
 
};
 
#endif