---

GenParticleProducer Class Reference

Inheritance diagram for GenParticleProducer:

List of all members.

Public Member Functions
	GenParticleProducer (const edm::ParameterSet &)
	constructor
	~GenParticleProducer ()
	destructor
Private Member Functions
void	beginJob (const edm::EventSetup &)
	module init at begin of job
int	chargeTimesThree (int) const
void	produce (edm::Event &e, const edm::EventSetup &)
	process one event
Private Attributes
bool	abortOnUnknownPDGCode_
	unknown code treatment flag
std::vector< int >	chargeM_
std::map< int, int >	chargeMap_
std::vector< int >	chargeP_
	charge indices
bool	saveBarCodes_
	save bar-codes
edm::InputTag	src_
	source collection name

---

Detailed Description

Definition at line 22 of file GenParticleProducer.cc.

---

Constructor & Destructor Documentation

GenParticleProducer::GenParticleProducer

(

const edm::ParameterSet &

cfg

)

constructor

Definition at line 65 of file GenParticleProducer.cc.

References edm::ParameterSet::getParameter(), and saveBarCodes_.

                                                                   :
  src_( cfg.getParameter<InputTag>( "src" ) ),
  abortOnUnknownPDGCode_( cfg.getUntrackedParameter<bool>( "abortOnUnknownPDGCode", true ) ),
  saveBarCodes_( cfg.getUntrackedParameter<bool>( "saveBarCodes", false ) ),
  chargeP_( PDGCacheMax, 0 ), chargeM_( PDGCacheMax, 0 ) {
  produces<GenParticleCollection>();
  if( saveBarCodes_ ) {
    std::string alias( cfg.getParameter<std::string>( "@module_label" ) );
    produces<vector<int> >().setBranchAlias( alias + "BarCodes" );
  }                               
}

GenParticleProducer::~GenParticleProducer

(

)

destructor

Definition at line 77 of file GenParticleProducer.cc.

                                          { 
}

---

Member Function Documentation

void GenParticleProducer::beginJob

(

const edm::EventSetup &

es

)

[private, virtual]

module init at begin of job

Reimplemented from edm::EDProducer.

Definition at line 94 of file GenParticleProducer.cc.

References funct::abs(), chargeM_, chargeMap_, chargeP_, edm::EventSetup::getData(), p, gen_jpsi2muons_cfg::ParticleID, and PDGCacheMax.

                                                          {
  ESHandle<HepPDT::ParticleDataTable> pdt;
  es.getData( pdt );
  for( HepPDT::ParticleDataTable::const_iterator p = pdt->begin(); p != pdt->end(); ++ p ) {
    const HepPDT::ParticleID & id = p->first;
    int pdgId = id.pid(), apdgId = abs( pdgId );
    int q3 = id.threeCharge();
    if ( apdgId < PDGCacheMax && pdgId > 0 ) {
      chargeP_[ apdgId ] = q3;
      chargeM_[ apdgId ] = -q3;
    } else if ( apdgId < PDGCacheMax ) {
      chargeP_[ apdgId ] = -q3;
      chargeM_[ apdgId ] = q3;
    } else {
      chargeMap_[ pdgId ] = q3; 
      chargeMap_[ -pdgId ] = -q3;
    } 
  }
}

int GenParticleProducer::chargeTimesThree

(

int

id

)

const [private]

Definition at line 80 of file GenParticleProducer.cc.

References abortOnUnknownPDGCode_, funct::abs(), chargeM_, chargeMap_, chargeP_, lat::endl(), f, edm::errors::LogicError, gen_jpsi2muons_cfg::ParticleID, and PDGCacheMax.

Referenced by produce().

                                                        {
  if( abs( id ) < PDGCacheMax ) 
    return id > 0 ? chargeP_[ id ] : chargeM_[ - id ];
  map<int, int>::const_iterator f = chargeMap_.find( id );
  if ( f == chargeMap_.end() ) 
    if ( abortOnUnknownPDGCode_ )
      throw edm::Exception( edm::errors::LogicError ) 
        << "invalid PDG id: " << id << endl;
    else {
      return HepPDT::ParticleID(id).threeCharge();
    }
  return f->second;
}

void GenParticleProducer::produce	(	edm::Event &	e,
		const edm::EventSetup &	es
	)			[private, virtual]

process one event

fill indices

Implements edm::EDProducer.

Definition at line 114 of file GenParticleProducer.cc.

References begin, chargeTimesThree(), d, end, lat::endl(), edm::Event::getByLabel(), edm::Event::getRefBeforePut(), i, edm::RefProd< C >::id(), edm::errors::InvalidReference, m, mmToCm, p, p4, edm::Event::put(), reco::CompositeRefCandidateT< D >::resetDaughters(), saveBarCodes_, edm::second(), reco::Particle::setP4(), reco::Particle::setPdgId(), reco::Particle::setStatus(), reco::Particle::setThreeCharge(), reco::Particle::setVertex(), size, src_, and v.

                                                                    {
  Handle<HepMCProduct> mcp;
  evt.getByLabel( src_, mcp );
  const GenEvent * mc = mcp->GetEvent();
  if( mc == 0 ) 
    throw edm::Exception( edm::errors::InvalidReference ) 
      << "HepMC has null pointer to GenEvent" << endl;
  const size_t size = mc->particles_size();
  
  vector<const HepMC::GenParticle *> particles( size );
  map<int, size_t> barcodes;

  auto_ptr<GenParticleCollection> candsPtr( new GenParticleCollection( size ) );
  auto_ptr<vector<int> > barCodeVector( new vector<int>( size ) );
  const GenParticleRefProd ref = evt.getRefBeforePut<GenParticleCollection>();
  GenParticleCollection & cands = * candsPtr;

  GenEvent::particle_const_iterator begin = mc->particles_begin(), end = mc->particles_end();
  size_t idx = 0;
  for( GenEvent::particle_const_iterator p = begin; p != end; ++ p ) {
    const HepMC::GenParticle * particle = * p;
    size_t barCode = particle->barcode();
    if( barcodes.find(barCode) != barcodes.end() )
      throw cms::Exception( "WrongReference" )
        << "barcodes are duplicated! " << endl;
    particles[idx] = particle;
    (*barCodeVector)[idx] = barCode;
    barcodes.insert( make_pair(barCode, idx ++) );
  }

  // fill output collection and save association
  for( size_t i = 0; i < particles.size(); ++ i ) {
    const HepMC::GenParticle * part = particles[ i ];
    Candidate::LorentzVector p4( part->momentum() );
    int pdgId = part->pdg_id();
    reco::GenParticle & cand = cands[ i ];
    cand.setThreeCharge( chargeTimesThree( pdgId ) );
    cand.setPdgId( pdgId );
    cand.setStatus( part->status() );
    cand.setP4( p4 );
    const GenVertex * v = part->production_vertex();
    if ( v != 0 ) {
      ThreeVector vtx = v->point3d();
      Candidate::Point vertex( vtx.x() * mmToCm, vtx.y() * mmToCm, vtx.z() * mmToCm );
      cand.setVertex( vertex );
    } else {
      cand.setVertex( Candidate::Point( 0, 0, 0 ) );
    }
    cand.resetDaughters( ref.id() );
  }

  // fill references to daughters
  for( size_t d = 0; d < cands.size(); ++ d ) {
    const HepMC::GenParticle * part = particles[ d ];
    const GenVertex * productionVertex = part->production_vertex();
    if ( productionVertex != 0 ) {
      size_t numberOfMothers = productionVertex->particles_in_size();
      if ( numberOfMothers > 0 ) {
        GenVertex::particles_in_const_iterator motherIt = productionVertex->particles_in_const_begin();
        for( ; motherIt != productionVertex->particles_in_const_end(); motherIt++) {
          const HepMC::GenParticle * mother = * motherIt;
          size_t m = barcodes.find( mother->barcode() )->second;
          cands[ m ].addDaughter( GenParticleRef( ref, d ) );  
          cands[ d ].addMother( GenParticleRef( ref, m ) );  
        }
      }
    }
  }

  evt.put( candsPtr );
  if(saveBarCodes_) evt.put( barCodeVector );
}

---

Member Data Documentation

bool GenParticleProducer::abortOnUnknownPDGCode_ [private]

unknown code treatment flag

Definition at line 37 of file GenParticleProducer.cc.

Referenced by chargeTimesThree().

std::vector<int> GenParticleProducer::chargeM_ [private]

Definition at line 41 of file GenParticleProducer.cc.

Referenced by beginJob(), and chargeTimesThree().

std::map<int, int> GenParticleProducer::chargeMap_ [private]

Definition at line 42 of file GenParticleProducer.cc.

Referenced by beginJob(), and chargeTimesThree().

std::vector<int> GenParticleProducer::chargeP_ [private]

charge indices

Definition at line 41 of file GenParticleProducer.cc.

Referenced by beginJob(), and chargeTimesThree().

bool GenParticleProducer::saveBarCodes_ [private]

save bar-codes

Definition at line 39 of file GenParticleProducer.cc.

Referenced by GenParticleProducer(), and produce().

edm::InputTag GenParticleProducer::src_ [private]

source collection name

Definition at line 35 of file GenParticleProducer.cc.

Referenced by produce().

---

The documentation for this class was generated from the following file:

• PhysicsTools/HepMCCandAlgos/plugins/GenParticleProducer.cc

---