SingleParticleGun.h

Go to the documentation of this file.

// --------------------------------------------------
// 
// File:  SingleParticleGun/SingleParticleGun.h
// Description:
//    This code is used to generate a single particle or a poissonly
//    distributed set of particles with pt, theta and phi selected as
//    specified by the JobOptionsService.  Three modes of generation are
//    possible:
//           FixedMode:  Generation with a fixed value
//           GaussianMode: Generation with a gaussian of specified mean and
//                         sigma
//           FlatMode: Generation of a flat distribution between a specified
//                     minimum and maximum value
//    The mode can be separately selected for pt, theta and phi.
// or for  E, theta and phi
// in PtMode user specifies Pt theta and phi
// in EMode user specifies E, theta and phi 
//    The particle species can also be specified
//
//    The output will be stored in the transient event store so it can be
//    passed to the simulation.
 
 
#ifndef GENERATORMODULESSINGLEPARTICLEGUN_H
#define GENERATORMODULESSINGLEPARTICLEGUN_H
 
#include "GeneratorModule/GenModule.h"
 
#include "CLHEP/Random/RandFlat.h"
#include "CLHEP/Random/RandGauss.h"
#include "CLHEP/Vector/LorentzVector.h"
 
class IBesRndmGenSvc;
 
struct SingleParticleGunGenMode {
  enum {FixedMode = 1, GaussMode = 2,FlatMode  = 3};
};
struct SingleEnergyMode {
  enum {EMode = 1, PtMode = 2};
};
 
class SingleParticleGun:public GenModule {
public:
  SingleParticleGun(const std::string& name, ISvcLocator* pSvcLocator);
  virtual ~SingleParticleGun();
  virtual StatusCode genInitialize();
  virtual StatusCode callGenerator();
  virtual StatusCode genFinalize();
  virtual StatusCode fillEvt(GenEvent* evt);
 
private:
  // Setable Properties:-
  double m_requestedPt;
  double m_requestedPhi;
  double m_requestedE;
  double m_requestedTheta;
  double m_requestedX;
  double m_requestedY;
  double m_requestedZ;
  double m_requestedT;
  double m_minPt;
  double m_minE;
  double m_minTheta;
  double m_minPhi;
  double m_maxE;
  double m_maxPt;
  double m_maxTheta;
  double m_maxPhi;
  double m_sigmaPt;
  double m_sigmaE;
  double m_sigmaTheta;
  double m_sigmaPhi;
  int m_PtGenMode;
  int m_EGenMode;
  int m_ThetaGenMode;
  int m_PhiGenMode;
  int m_pdgCode;
  int m_Emode;
  // event counter
  int m_events ;
 
  // Local Member Data:-
  IBesRndmGenSvc* p_BesRndmGenSvc;
  std::vector<long int> m_seeds;
 
  double m_mass;
  CLHEP::HepLorentzVector m_fourMom;  
  CLHEP::HepLorentzVector m_fourPos;  
      
 
  // Private Methods:=
  double generateValue(int mode,double val,double sigma,double min,
               double max);
 
};
 
#endif