CosmicGenerator.h

Go to the documentation of this file.

// --------------------------------------------------
//
// File:  CosmicGenerator/CosmicGenerator.h
// Description:
// Ported from Atlas to BES
//                      by Beijiang Liu
 
//    The output will be stored in the transient event store so it can be
//    passed to the simulation.
//
// AuthorList:
//         W. Seligman: Initial Code 08-Nov-2002,
//         based on work by M. Shapiro and I. Hinchliffe
//         adapted for the CosmicGun by S Bentvelsen
 
#ifndef GENERATORMODULESCOSMICGEN_H
#define GENERATORMODULESCOSMICGEN_H
 
#include "BesRndmGenSvc/IBesRndmGenSvc.h"
 
#include "GaudiKernel/Algorithm.h"
#include "GaudiKernel/ISvcLocator.h"
#include "GaudiKernel/NTuple.h"
 
#include "CLHEP/Geometry/Normal3D.h"
#include "CLHEP/Geometry/Point3D.h"
#include "CLHEP/Vector/LorentzVector.h"
#include "HepMC/Polarization.h"
#include <fstream>
#include <string>
#include <vector>
 
using namespace std;
using namespace CLHEP;
#ifndef ENABLE_BACKWARDS_COMPATIBILITY
typedef HepGeom::Point3D<double> HepPoint3D;
typedef HepGeom::Normal3D<float> HepNormal3D;
#endif
 
class StoreGateSvc;
class ActiveStoreSvc;
 
class CosmicGun;
 
class CosmicGenerator : public Algorithm {
public:
  CosmicGenerator( const std::string& name, ISvcLocator* pSvcLocator );
  ~CosmicGenerator();
  StatusCode initialize();
  StatusCode execute();
  StatusCode finalize();
 
  HepLorentzVector generateVertex( void );
 
  static IBesRndmGenSvc* p_BesRndmGenSvc;
 
private:
  StoreGateSvc*        m_sgSvc;
  ActiveStoreSvc*      m_activeStore;
  NTuple::Tuple*       m_tuple;
  NTuple::Item<double> m_cosmicE;
  NTuple::Item<double> m_cosmicTheta;
  NTuple::Item<double> m_cosmicPhi;
  NTuple::Item<double> m_cosmicCharge;
  NTuple::Tuple*       m_tuple1;
  NTuple::Item<double> mc_phi;
  NTuple::Item<double> mc_theta;
  NTuple::Item<double> mc_px;
  NTuple::Item<double> mc_py;
  NTuple::Item<double> mc_pz;
 
  // event counter, used for event ID
  long int         m_events, m_rejected, m_accepted, m_tried;
  std::vector<int> m_pdgCode;
  float            m_emin, m_emax;
  float            m_ctcut;
  float            m_xlow, m_xhig, m_zlow, m_zhig, m_yval, m_IPx, m_IPy, m_IPz;
  float            m_radius, m_xpos, m_ypos, m_zpos;
  float m_xposMin_top, m_xposMax_top, m_ypos_top, m_zposMin_top, m_zposMax_top; // top plane
  float m_xposMin_bottom, m_xposMax_bottom, m_ypos_bottom, m_zposMin_bottom,
      m_zposMax_bottom; // bottom plane
 
  float m_tmin, m_tmax, m_tcor;
  bool  m_sphereOpt;
 
  bool     m_swapYZAxis;
  bool     m_cubeProj;
  bool     m_doublePlaneTrigger;
  long int m_printEvent, m_printMod;
  long int m_selection;
  int      m_dumpMC;
  float    m_thetamin, m_thetamax, m_phimin, m_phimax;
 
  float m_GeV;
  float m_mm;
 
  bool          m_readfile;
  std::string   m_infile;
  std::ifstream m_ffile;
 
  // Event scalars, three-vectors, and four-vectors:
  std::vector<HepLorentzVector>    m_fourPos;
  std::vector<HepLorentzVector>    m_fourMom;
  Hep3Vector                       m_center;
  std::vector<HepMC::Polarization> m_polarization;
 
  // Pointer to Athena MessageSvc.
  IMessageSvc* p_msgSvc;
 
  // Energy dependent position cut for muons to reach the detector.
  bool exzCut( const Hep3Vector& pos, const HepLorentzVector& p );
 
  // property for calling exzCut
  bool m_exzCut;
  // maximum r used in exzCut
  float m_rmax;
};
 
#endif