CMS 3D CMS Logo

Public Member Functions | Protected Attributes | Private Member Functions

edm::FlatRandomEGunSource Class Reference

#include <FlatRandomEGunSource.h>

Inheritance diagram for edm::FlatRandomEGunSource:
edm::BaseFlatGunSource edm::GeneratedInputSource edm::ConfigurableInputSource edm::InputSource edm::ProductRegistryHelper

List of all members.

Public Member Functions

 FlatRandomEGunSource (const ParameterSet &, const InputSourceDescription &)
virtual ~FlatRandomEGunSource ()

Protected Attributes

double fMaxE
double fMinE

Private Member Functions

virtual bool produce (Event &e)

Detailed Description

Definition at line 15 of file FlatRandomEGunSource.h.


Constructor & Destructor Documentation

FlatRandomEGunSource::FlatRandomEGunSource ( const ParameterSet pset,
const InputSourceDescription desc 
)

Definition at line 19 of file FlatRandomEGunSource.cc.

References gather_cfg::cout, fMaxE, fMinE, edm::ParameterSet::getUntrackedParameter(), and edm::InputSource::remainingEvents().

                                                                               :
   BaseFlatGunSource(pset, desc)
{

   ParameterSet defpset ;
   // ParameterSet pgun_params = pset.getParameter<ParameterSet>("PGunParameters") ;
   ParameterSet pgun_params = 
      pset.getUntrackedParameter<ParameterSet>("PGunParameters",defpset) ;
  
   // doesn't seem necessary to check if pset is empty - if this
   // is the case, default values will be taken for params
   fMinE = pgun_params.getUntrackedParameter<double>("MinE",0.99);
   fMaxE = pgun_params.getUntrackedParameter<double>("MaxE",1.01); 
  
   produces<HepMCProduct>();

   cout << "Internal FlatRandomEGun is initialzed" << endl ;
   cout << "It is going to generate " << remainingEvents() << "events" << endl ;
   
}
FlatRandomEGunSource::~FlatRandomEGunSource ( ) [virtual]

Definition at line 41 of file FlatRandomEGunSource.cc.

{
   // no need to cleanup fEvt since it's done in HepMCProduct
}

Member Function Documentation

bool FlatRandomEGunSource::produce ( Event e) [private, virtual]

Implements edm::ConfigurableInputSource.

Definition at line 46 of file FlatRandomEGunSource.cc.

References abs, funct::cos(), gather_cfg::cout, relval_parameters_module::energy, eta(), edm::ConfigurableInputSource::event(), funct::exp(), edm::BaseFlatGunSource::fAddAntiParticle, edm::BaseFlatGunSource::fEvt, fMaxE, edm::BaseFlatGunSource::fMaxEta, edm::BaseFlatGunSource::fMaxPhi, fMinE, edm::BaseFlatGunSource::fMinEta, edm::BaseFlatGunSource::fMinPhi, edm::BaseFlatGunSource::fPartIDs, edm::BaseFlatGunSource::fPDGTable, edm::BaseFlatGunSource::fRandomGenerator, edm::BaseFlatGunSource::fVerbosity, configurableAnalysis::GenParticle, scaleCards::mass, AlCaHLTBitMon_ParallelJobs::p, RecoTau_DiTaus_pt_20-420_cfg::ParticleID, phi, edm::Event::put(), funct::sin(), mathSSE::sqrt(), and theta().

{

   if ( fVerbosity > 0 )
   {
      cout << " FlatRandomEGunSource : Begin New Event Generation" << endl ; 
   }
   
   // event loop (well, another step in it...)
          
   // no need to clean up GenEvent memory - done in HepMCProduct

   // here re-create fEvt (memory)
   //
   fEvt = new HepMC::GenEvent() ;
   
   // now actualy, cook up the event from PDGTable and gun parameters
   //

   // 1st, primary vertex
   //
   HepMC::GenVertex* Vtx = new HepMC::GenVertex( HepMC::FourVector(0.,0.,0.));
   
   // loop over particles
   //
   int barcode = 1;
   for (unsigned int ip=0; ip<fPartIDs.size(); ip++)
   {
       double energy = fRandomGenerator->fire(fMinE, fMaxE) ;
       double eta    = fRandomGenerator->fire(fMinEta, fMaxEta) ;
       double phi    = fRandomGenerator->fire(fMinPhi, fMaxPhi) ;
       int PartID = fPartIDs[ip] ;
       const HepPDT::ParticleData* 
          PData = fPDGTable->particle(HepPDT::ParticleID(abs(PartID))) ;
       double mass   = PData->mass().value() ;
       double mom2   = energy*energy - mass*mass ;
       double mom    = 0. ;
       if (mom2 > 0.) 
       {
          mom = sqrt(mom2) ;
       }
       else
       {
          mom = 0. ;
       }
       double theta  = 2.*atan(exp(-eta)) ;
       double px     = mom*sin(theta)*cos(phi) ;
       double py     = mom*sin(theta)*sin(phi) ;
       double pz     = mom*cos(theta) ;

       HepMC::FourVector p(px,py,pz,energy) ;
       HepMC::GenParticle* Part = 
           new HepMC::GenParticle(p,PartID,1);
       Part->suggest_barcode( barcode ) ;
       barcode++ ;
       Vtx->add_particle_out(Part);
       
       if ( fAddAntiParticle )
       {
          HepMC::FourVector ap(-px,-py,-pz,energy) ;
          int APartID = -PartID ;
          if ( PartID == 22 || PartID == 23 )
          {
             APartID = PartID ;
          }
          HepMC::GenParticle* APart =
             new HepMC::GenParticle(ap,APartID,1);
          APart->suggest_barcode( barcode ) ;
          barcode++ ;
          Vtx->add_particle_out(APart) ;
       }
       
   }
   fEvt->add_vertex(Vtx) ;
   fEvt->set_event_number(event()) ;
   fEvt->set_signal_process_id(20) ;  
   
   
   if ( fVerbosity > 0 )
   {
      fEvt->print() ;  
   }  

   auto_ptr<HepMCProduct> BProduct(new HepMCProduct()) ;
   BProduct->addHepMCData( fEvt );
   e.put(BProduct);
    
   if ( fVerbosity > 0 )
   {
      // for testing purpose only
      //fEvt->print() ;  // for some strange reason, it prints NO event info
      // after it's been put into edm::Event...
      cout << " FlatRandomEGunSource : Event Generation Done " << endl;
   }
      
   return true;
}

Member Data Documentation

Definition at line 31 of file FlatRandomEGunSource.h.

Referenced by FlatRandomEGunSource(), and produce().

Definition at line 30 of file FlatRandomEGunSource.h.

Referenced by FlatRandomEGunSource(), and produce().