BeamHaloGenerator.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/
 
#ifndef BEAMHALOGENERATOR_H
#define BEAMHALOGENERATOR_H
 
#include "BeamHaloGenerator/BeamHaloParticle.h"
#include <string>
#include <vector>
#include "AtlasHepMC/GenEvent.h"
 
namespace HepPDT {
  class ParticleDataTable;
}
namespace CLHEP {
  class HepRandomEngine;
}
class BeamHaloGeneratorAlg;
class AsciiInput;
class BeamHaloGeneratorSettings;
class BeamHaloParticleBuffer;

class BeamHaloGenerator {
 public:
 
  BeamHaloGenerator(const HepPDT::ParticleDataTable* particleTable,
            const std::string& inputFile,
            const std::vector<std::string>& generatorSettings);
 
  virtual ~BeamHaloGenerator();

  virtual int genInitialize();

  virtual int genFinalize();

  virtual int fillEvt(HepMC::GenEvent* evt,
            CLHEP::HepRandomEngine* engine) = 0;
  
 protected:

  virtual int readEvent(std::vector<BeamHaloParticle> *beamHaloEvent,
                        CLHEP::HepRandomEngine* engine) = 0;

  virtual int readParticle(BeamHaloParticle *beamHaloParticle) = 0;
 
 public:
 

  void setInterfacePlane(double interfacePlane) { m_interfacePlane = interfacePlane; }

  void setEnableFlip(bool enableFlip) { m_enableFlip = enableFlip; }

  void setFlipProbability(float flipProbability) { m_flipProbability = flipProbability; }

  void setEnableSampling(bool enableSampling) { m_enableSampling = enableSampling; }

  void setBufferFileName(const std::string& bufferFileName) { m_bufferFileName = bufferFileName; }

  void setDebugEnable(bool debug) { m_debug = debug; }
 
 protected:

  enum enumGeneratorProcessIds {
    MARS_READ,
    MARS_SINGLE,
    MARS_SHOWER,
    FLUKA_READ,
    FLUKA_SINGLE,
    FLUKA_SHOWER };
 

  bool flipEvent(CLHEP::HepRandomEngine* engine);

  int convertEvent(std::vector<BeamHaloParticle>* beamHaloEvent,
                   HepMC::GenEvent* evt,
                   CLHEP::HepRandomEngine* engine);

  const HepPDT::ParticleDataTable* m_particleTable;

  std::string m_inputFile;

  double m_interfacePlane;
  
  bool m_enableFlip;

  float m_flipProbability;
  
  bool m_enableSampling;

  std::string m_bufferFileName;

  BeamHaloParticleBuffer *m_beamHaloParticleBuffer;

  AsciiInput *m_asciiInput;

  BeamHaloGeneratorSettings *m_beamHaloGeneratorSettings;

  long m_eventNumber;

  enum enumCounters {
    TOT_READ, 
    TOT_AFTER,
    TOT_GEN,
    NUM_COUNTERS};

  long m_counters[NUM_COUNTERS];

  double m_wsums[NUM_COUNTERS];

  bool m_debug;
 
 private:

  std::vector<std::string> m_generatorSettings;
 
};
 
#endif