#include <FlatEGunASCIIWriter.h>

Inheritance diagram for edm::FlatEGunASCIIWriter:

Public Member Functions
virtual void	analyze (const edm::Event &, const edm::EventSetup &) override

virtual void	beginJob ()

virtual void	beginRun (const edm::Run &, const EventSetup &) override

	FlatEGunASCIIWriter (const edm::ParameterSet &)

virtual	~FlatEGunASCIIWriter ()

Public Member Functions inherited from edm::EDAnalyzer
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)

	EDAnalyzer ()

ModuleDescription const &	moduleDescription () const

std::string	workerType () const

virtual	~EDAnalyzer ()

Public Member Functions inherited from edm::EDConsumerBase
std::vector< ConsumesInfo >	consumesInfo () const

	EDConsumerBase ()

ProductHolderIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const

void	itemsMayGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const

void	itemsToGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const

std::vector < ProductHolderIndexAndSkipBit > const &	itemsToGetFromEvent () const

void	labelsForToken (EDGetToken iToken, Labels &oLabels) const

void	modulesDependentUpon (std::string const &iProcessName, std::string const &iModuleLabel, bool iPrint, std::vector< char const * > &oModuleLabels) const

void	modulesWhoseProductsAreConsumed (std::vector< ModuleDescription const * > &modules, ProductRegistry const &preg, std::map< std::string, ModuleDescription const * > const &labelsToDesc, std::string const &processName) const

bool	registeredToConsume (ProductHolderIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) const

void	updateLookup (BranchType iBranchType, ProductHolderIndexHelper const &)

virtual	~EDConsumerBase ()

Private Attributes
int	fCurrentEvent

HepMC::GenEvent *	fEvt

double	fMaxE

double	fMaxEta

double	fMaxPhi

double	fMinE

double	fMinEta

double	fMinPhi

std::string	fOutFileName

HepMC::IO_GenEvent *	fOutStream

std::vector< int >	fPartIDs

ESHandle < HepPDT::ParticleDataTable >	fPDGTable

Additional Inherited Members
Public Types inherited from edm::EDAnalyzer
typedef EDAnalyzer	ModuleType

Public Types inherited from edm::EDConsumerBase
typedef ProductLabels	Labels

Static Public Member Functions inherited from edm::EDAnalyzer
static const std::string &	baseType ()

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &)

Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	consumes (edm::InputTag const &tag)

EDGetToken	consumes (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	consumes (TypeToGet const &id, edm::InputTag const &tag)

ConsumesCollector	consumesCollector ()
	Use a ConsumesCollector to gather consumes information from helper functions. More...

template<typename ProductType , BranchType B = InEvent>
void	consumesMany ()

void	consumesMany (const TypeToGet &id)

template<BranchType B>
void	consumesMany (const TypeToGet &id)

template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	mayConsume (edm::InputTag const &tag)

EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

Detailed Description

Definition at line 34 of file FlatEGunASCIIWriter.h.

Constructor & Destructor Documentation

FlatEGunASCIIWriter::FlatEGunASCIIWriter ( const edm::ParameterSet & pset )

explicit

Definition at line 37 of file FlatEGunASCIIWriter.cc.

References fMaxE, fMaxEta, fMaxPhi, fMinE, fMinEta, fMinPhi, fPartIDs, and edm::ParameterSet::getParameter().

    : fEvt(0), 
      fOutFileName( pset.getUntrackedParameter<string>("OutFileName","FlatEGunHepMC.dat") ),
      fCurrentEvent(0)
 {
       
   ParameterSet pgun_params = pset.getParameter<ParameterSet>("PGunParameters") ;
   fPartIDs    = pgun_params.getParameter< vector<int> >("PartID");
   fMinEta     = pgun_params.getParameter<double>("MinEta");
   fMaxEta     = pgun_params.getParameter<double>("MaxEta");
   fMinPhi     = pgun_params.getParameter<double>("MinPhi");
   fMaxPhi     = pgun_params.getParameter<double>("MaxPhi");
   fMinE       = pgun_params.getParameter<double>("MinE");
   fMaxE       = pgun_params.getParameter<double>("MaxE");
             
 }

FlatEGunASCIIWriter::~FlatEGunASCIIWriter ( )

virtual

Definition at line 54 of file FlatEGunASCIIWriter.cc.

References fEvt, fOutStream, and NULL.

 {
    
   if ( fEvt != NULL ) delete fEvt ;
   if( fOutStream) delete fOutStream;
 
 }

Member Function Documentation

void FlatEGunASCIIWriter::analyze	(	const edm::Event &	,
		const edm::EventSetup &
	)

overridevirtual

Implements edm::EDAnalyzer.

Definition at line 81 of file FlatEGunASCIIWriter.cc.

References funct::abs(), funct::cos(), relval_parameters_module::energy, eta, create_public_lumi_plots::exp, fCurrentEvent, fEvt, fMaxE, fMaxEta, fMaxPhi, fMinE, fMinEta, fMinPhi, fOutStream, fPartIDs, fPDGTable, GenParticle::GenParticle, NULL, AlCaHLTBitMon_ParallelJobs::p, phi(), reco::return(), funct::sin(), mathSSE::sqrt(), and theta().

 {
          
    // clean up GenEvent memory : also deletes all vtx/part in it
    // 
    if ( fEvt != NULL ) delete fEvt ;
    
    // here re-create fEvt (memory)
    fEvt = new HepMC::GenEvent() ;
 
    // now actualy, cook up the event from PDGTable and gun parameters
    //
    //HepMC::GenVertex* Vtx = new HepMC::GenVertex( CLHEP::HepLorentzVector(0.,0.,0.) );
    HepMC::GenVertex* Vtx = new HepMC::GenVertex( HepMC::FourVector(0.,0.,0.) );
 
    // loop over particles
    //
    for ( unsigned int ip=0; ip<fPartIDs.size(); ip++ )
    {
        double energy = CLHEP::RandFlat::shoot( fMinE, fMaxE ) ;
        double eta    = CLHEP::RandFlat::shoot( fMinEta, fMaxEta ) ;
        double phi    = CLHEP::RandFlat::shoot( fMinPhi, fMaxPhi ) ;
        const HepPDT::ParticleData* 
           PData = fPDGTable->particle(HepPDT::ParticleID(abs(fPartIDs[ip]))) ;
        double mass   = PData->mass().value() ;
        double mom2   = energy*energy - mass*mass ;
        double mom    = 0. ;
        if ( mom2 > 0. ) mom = sqrt(mom2) ;
        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) ;
        //CLHEP::Hep3Vector p(px,py,pz) ;
        //HepMC::GenParticle* Part = 
        //    new HepMC::GenParticle(CLHEP::HepLorentzVector(p,energy),fPartIDs[ip],1);
        HepMC::FourVector p(px,py,pz,energy);
        HepMC::GenParticle* Part = 
            new HepMC::GenParticle(p,fPartIDs[ip],1);
        Vtx->add_particle_out(Part);
    }
    fEvt->add_vertex( Vtx ) ;
    fEvt->set_event_number( fCurrentEvent+1 ) ;
    fEvt->set_signal_process_id(20) ; 
    
    // for testing purpose only
    // fEvt->print() ;     
 
    fOutStream->write_event( fEvt );
 
    fCurrentEvent++ ;
 
    return ;
    
 }

void FlatEGunASCIIWriter::beginJob ( void )

virtual

Reimplemented from edm::EDAnalyzer.

Definition at line 69 of file FlatEGunASCIIWriter.cc.

References Exception, fOutFileName, fOutStream, and reco::return().

 { 
   fOutStream = new HepMC::IO_GenEvent( fOutFileName.c_str() ); 
   if ( fOutStream->rdstate() == std::ios::failbit ) {
     throw cms::Exception("FileNotOpen", "FlatEGunASCIIWriter::beginJob()")
       << "File " << fOutFileName << " was not open.\n";
   }
   
   return ;
 
 }

void FlatEGunASCIIWriter::beginRun	(	const edm::Run &	r,
		const EventSetup &	es
	)

overridevirtual

Reimplemented from edm::EDAnalyzer.

Definition at line 62 of file FlatEGunASCIIWriter.cc.

References fPDGTable, and edm::EventSetup::getData().

 {
 
   es.getData( fPDGTable );
 
 }

Member Data Documentation

int edm::FlatEGunASCIIWriter::fCurrentEvent

private

Definition at line 72 of file FlatEGunASCIIWriter.h.

Referenced by analyze().

HepMC::GenEvent* edm::FlatEGunASCIIWriter::fEvt

private

Definition at line 63 of file FlatEGunASCIIWriter.h.

Referenced by analyze(), and ~FlatEGunASCIIWriter().

double edm::FlatEGunASCIIWriter::fMaxE

private

Definition at line 60 of file FlatEGunASCIIWriter.h.

Referenced by analyze(), and FlatEGunASCIIWriter().

double edm::FlatEGunASCIIWriter::fMaxEta

private

Definition at line 56 of file FlatEGunASCIIWriter.h.

Referenced by analyze(), and FlatEGunASCIIWriter().

double edm::FlatEGunASCIIWriter::fMaxPhi

private

Definition at line 58 of file FlatEGunASCIIWriter.h.

Referenced by analyze(), and FlatEGunASCIIWriter().

double edm::FlatEGunASCIIWriter::fMinE

private

Definition at line 59 of file FlatEGunASCIIWriter.h.

Referenced by analyze(), and FlatEGunASCIIWriter().

double edm::FlatEGunASCIIWriter::fMinEta

private

Definition at line 55 of file FlatEGunASCIIWriter.h.

Referenced by analyze(), and FlatEGunASCIIWriter().

double edm::FlatEGunASCIIWriter::fMinPhi

private

Definition at line 57 of file FlatEGunASCIIWriter.h.

Referenced by analyze(), and FlatEGunASCIIWriter().

std::string edm::FlatEGunASCIIWriter::fOutFileName

private

Definition at line 69 of file FlatEGunASCIIWriter.h.

Referenced by beginJob().

HepMC::IO_GenEvent* edm::FlatEGunASCIIWriter::fOutStream

private

Definition at line 70 of file FlatEGunASCIIWriter.h.

Referenced by analyze(), beginJob(), and ~FlatEGunASCIIWriter().

std::vector<int> edm::FlatEGunASCIIWriter::fPartIDs

private

Definition at line 54 of file FlatEGunASCIIWriter.h.

Referenced by analyze(), and FlatEGunASCIIWriter().

ESHandle<HepPDT::ParticleDataTable> edm::FlatEGunASCIIWriter::fPDGTable

private

Definition at line 67 of file FlatEGunASCIIWriter.h.

Referenced by analyze(), and beginRun().

Public Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation