PFSimParticleProducer Class Reference

Producer for PFRecTracks and PFSimParticles. More...

#include <PFSimParticleProducer.h>

Inheritance diagram for PFSimParticleProducer:

Public Types
typedef edm::Handle< reco::PFRecTrackCollection >	TrackHandle
Public Types inherited from edm::stream::EDProducer<>
typedef CacheContexts< T... >	CacheTypes
typedef CacheTypes::GlobalCache	GlobalCache
typedef AbilityChecker< T... >	HasAbility
typedef CacheTypes::LuminosityBlockCache	LuminosityBlockCache
typedef LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >	LuminosityBlockContext
typedef CacheTypes::LuminosityBlockSummaryCache	LuminosityBlockSummaryCache
typedef CacheTypes::RunCache	RunCache
typedef RunContextT< RunCache, GlobalCache >	RunContext
typedef CacheTypes::RunSummaryCache	RunSummaryCache

Public Member Functions
void	getSimIDs (const TrackHandle &trackh, std::vector< unsigned > &recTrackSimID)
	PFSimParticleProducer (const edm::ParameterSet &)
void	produce (edm::Event &, const edm::EventSetup &) override
	~PFSimParticleProducer () override
Public Member Functions inherited from edm::stream::EDProducer<>
	EDProducer ()=default
bool	hasAbilityToProduceInLumis () const final
bool	hasAbilityToProduceInRuns () const final

Private Attributes
edm::InputTag	inputTagEcalRecHitsEB_
edm::InputTag	inputTagEcalRecHitsEE_
edm::InputTag	inputTagFastSimProducer_
edm::InputTag	inputTagRecTracks_
edm::InputTag	inputTagSim_
	module label for retrieving input simtrack and simvertex More...
bool	mctruthMatchingInfo_
FSimEvent *	mySimEvent
edm::ParameterSet	particleFilter_
bool	processParticles_
	process particles on/off More...
edm::EDGetTokenT< EcalRecHitCollection >	tokenEcalRecHitsEB_
edm::EDGetTokenT< EcalRecHitCollection >	tokenEcalRecHitsEE_
edm::EDGetTokenT< edm::PCaloHitContainer >	tokenFastSimProducer_
edm::EDGetTokenT< reco::PFRecTrackCollection >	tokenRecTracks_
edm::EDGetTokenT< std::vector< SimTrack > >	tokenSim_
edm::EDGetTokenT< std::vector< SimVertex > >	tokenSimVertices_
bool	verbose_
	verbose ? More...

Detailed Description

Producer for PFRecTracks and PFSimParticles.

Author

Colin Bernet

Date

April 2007

Definition at line 44 of file PFSimParticleProducer.h.

Member Typedef Documentation

typedef edm::Handle<reco::PFRecTrackCollection> PFSimParticleProducer::TrackHandle

Definition at line 53 of file PFSimParticleProducer.h.

Constructor & Destructor Documentation

PFSimParticleProducer::PFSimParticleProducer ( const edm::ParameterSet &  iConfig )

explicit

Definition at line 66 of file PFSimParticleProducer.cc.

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

{


  processParticles_ = 
    iConfig.getUntrackedParameter<bool>("process_Particles",true);
    
  
  inputTagSim_ = iConfig.getParameter<InputTag>("sim"); 
  tokenSim_    = consumes<std::vector<SimTrack> >(inputTagSim_);
  tokenSimVertices_    = consumes<std::vector<SimVertex> >(inputTagSim_);

  //retrieving collections for MC Truth Matching

  //modif-beg
  inputTagFastSimProducer_  = iConfig.getUntrackedParameter<InputTag>("fastSimProducer");
  tokenFastSimProducer_  = consumes<edm::PCaloHitContainer>(inputTagFastSimProducer_);
  mctruthMatchingInfo_ = 
    iConfig.getUntrackedParameter<bool>("MCTruthMatchingInfo",false);
  //modif-end

  inputTagRecTracks_ = iConfig.getParameter<InputTag>("RecTracks");
  tokenRecTracks_    = consumes<reco::PFRecTrackCollection>(inputTagRecTracks_);

  inputTagEcalRecHitsEB_ = iConfig.getParameter<InputTag>("ecalRecHitsEB");
  tokenEcalRecHitsEB_    = consumes<EcalRecHitCollection>(inputTagEcalRecHitsEB_);
  inputTagEcalRecHitsEE_ = iConfig.getParameter<InputTag>("ecalRecHitsEE");
  tokenEcalRecHitsEE_ 
    = consumes<EcalRecHitCollection>(inputTagEcalRecHitsEE_);

  verbose_ = 
    iConfig.getUntrackedParameter<bool>("verbose",false);


  // register products
  produces<reco::PFSimParticleCollection>();
  
  particleFilter_ = iConfig.getParameter<ParameterSet>
    ( "ParticleFilter" );   
  
  mySimEvent =  new FSimEvent( particleFilter_ );
}

PFSimParticleProducer::~PFSimParticleProducer ( )

override

Definition at line 111 of file PFSimParticleProducer.cc.

{ 
  delete mySimEvent; 
}

Member Function Documentation

void PFSimParticleProducer::getSimIDs	(	const TrackHandle &	trackh,
		std::vector< unsigned > &	recTrackSimID
	)

Definition at line 485 of file PFSimParticleProducer.cc.

References mps_fire::i, edm::HandleBase::isValid(), FastTrackerRecHit::nSimTrackIds(), FastTrackerRecHit::simTrackId(), and reco::PFRecTrack::trackRef().

{

  if( trackh.isValid() ) {

    for(unsigned i=0;i<trackh->size(); i++) {
      
      reco::PFRecTrackRef ref( trackh,i );
      const reco::PFRecTrack& PFT   = *ref;
      const reco::TrackRef& trackref = PFT.trackRef();
      
      trackingRecHit_iterator rhitbeg 
    = trackref->recHitsBegin();
      trackingRecHit_iterator rhitend 
    = trackref->recHitsEnd();
      for (trackingRecHit_iterator it = rhitbeg;  
       it != rhitend; it++){

    if( (*it)->isValid() ){

      const FastTrackerRecHit * rechit 
        = (const FastTrackerRecHit*) (*it);

      for(unsigned int st_index=0;st_index<rechit->nSimTrackIds();++st_index){
          recTrackSimID.push_back(rechit->simTrackId(st_index));
      }
      break;
    }
      }//loop track rechit
    }//loop recTracks
  }//track handle valid
}

void PFSimParticleProducer::produce ( edm::Event &  iEvent, const edm::EventSetup &  iSetup  )

override

Definition at line 116 of file PFSimParticleProducer.cc.

References reco::PFTrajectoryPoint::BeamPipeOrEndVertex, edm::SortedCollection< T, SORT >::begin(), FSimTrack::charge(), reco::PFTrajectoryPoint::ClosestApproach, FSimTrack::daughters(), reco::PFTrajectoryPoint::ECALEntrance, FSimTrack::ecalEntrance(), edm::SortedCollection< T, SORT >::end(), FSimTrack::endVertex(), edm::EventID::event(), Exception, runEdmFileComparison::found, edm::Event::getByToken(), edm::EventSetup::getData(), ecalpyutils::hashedIndex(), reco::PFTrajectoryPoint::HCALEntrance, FSimTrack::hcalEntrance(), reco::PFTrajectoryPoint::HCALExit, FSimTrack::hcalExit(), FSimTrack::hoEntrance(), reco::PFTrajectoryPoint::HOLayer, mps_fire::i, edm::EventBase::id(), FSimTrack::id(), edm::HandleBase::isValid(), FSimTrack::layer1Entrance(), FSimTrack::layer2Entrance(), LogDebug, FSimTrack::momentum(), FSimTrack::mother(), objects.autophobj::motherId, eostools::move(), FSimTrack::noEndVertex(), FSimTrack::noMother(), FSimTrack::onEcal(), FSimTrack::onHcal(), FSimTrack::onLayer1(), FSimTrack::onLayer2(), FSimVertex::position(), edm::Handle< T >::product(), reco::PFTrajectoryPoint::PS1, reco::PFTrajectoryPoint::PS2, edm::Event::put(), hiRegitInitialStep_cff::recTracks, edm::EventID::run(), tkConvValidator_cfi::simTracks, CoreSimTrack::type(), FSimTrack::vertex(), RawParticle::x(), RawParticle::y(), and RawParticle::z().

{  
  // init Particle data table (from Pythia)
  edm::ESHandle < HepPDT::ParticleDataTable > pdt;
  //  edm::ESHandle < DefaultConfig::ParticleDataTable > pdt;
  iSetup.getData(pdt);
  mySimEvent->initializePdt(&(*pdt));

  ParticleTable::Sentry ptable(mySimEvent->theTable());
  LogDebug("PFSimParticleProducer")<<"START event: "<<iEvent.id().event()
                   <<" in run "<<iEvent.id().run()<<endl;
 
  //MC Truth Matching only with Famos and UnFoldedMode option to true!!

  //vector to store the trackIDs of rectracks corresponding
  //to the simulated particle.
  std::vector<unsigned> recTrackSimID;
  
  //In order to know which simparticule contribute to 
  //a given Ecal RecHit energy, we need to access 
  //the PCAloHit from FastSim. 
  
  typedef std::pair<double, unsigned> hitSimID;
  typedef std::list< std::pair<double, unsigned> >::iterator ITM;
  std::vector< std::list <hitSimID> > caloHitsEBID(62000);
  std::vector<double> caloHitsEBTotE(62000,0.0);
  
  if(mctruthMatchingInfo_){
     
    //getting the PCAloHit
    edm::Handle<edm::PCaloHitContainer> pcalohits;
    //   bool found_phit 
    //     = iEvent.getByLabel("fastSimProducer","EcalHitsEB",
    //          pcalohits);  
    //modif-beg
    bool found_phit 
      = iEvent.getByToken(tokenFastSimProducer_,pcalohits);
    //modif-end
    
    if(!found_phit) {
      ostringstream err;
      err<<"could not find pcaloHit "<<"fastSimProducer:EcalHitsEB";
      LogError("PFSimParticleProducer")<<err.str()<<endl;
      
      throw cms::Exception( "MissingProduct", err.str());
    }
    else {
      assert( pcalohits.isValid() );

      edm::PCaloHitContainer::const_iterator it    
    = pcalohits.product()->begin();
      edm::PCaloHitContainer::const_iterator itend 
    = pcalohits.product()->end();
      
      //loop on the PCaloHit from FastSim Calorimetry
      for(;it!=itend;++it)
    {
      EBDetId detid(it->id());
      
      if(it->energy() > 0.0) {
        std::pair<double, unsigned> phitsimid
          = make_pair(it->energy(),it->geantTrackId());
        caloHitsEBID[detid.hashedIndex()].push_back(phitsimid);
        caloHitsEBTotE[detid.hashedIndex()] 
          += it->energy(); //summing pcalhit energy   
      }//energy > 0
      
    }//loop PcaloHits    
    }//pcalohit handle access
    
    //Retrieving the PFRecTrack collection for
    //Monte Carlo Truth Matching tool
    Handle< reco::PFRecTrackCollection > recTracks;
    try{      
      LogDebug("PFSimParticleProducer")<<"getting PFRecTracks"<<endl;
      iEvent.getByToken(tokenRecTracks_, recTracks);
      
    } catch (cms::Exception& err) { 
      LogError("PFSimParticleProducer")<<err
                       <<" cannot get collection "
                       <<"particleFlowBlock"<<":"
                       <<""
                       <<endl;
    }//pfrectrack handle access
    
    //getting the simID corresponding to 
    //each of the PFRecTracks
    getSimIDs( recTracks, recTrackSimID );
    
  }//mctruthMatchingInfo_ //modif
  
  // deal with true particles 
  if( processParticles_) {

    auto pOutputPFSimParticleCollection = std::make_unique<reco::PFSimParticleCollection>(); 

    Handle<vector<SimTrack> > simTracks;
    bool found = iEvent.getByToken(tokenSim_,simTracks);
    if(!found) {

      ostringstream err;
      err<<"cannot find sim tracks: "<<inputTagSim_;
      LogError("PFSimParticleProducer")<<err.str()<<endl;
      
      throw cms::Exception( "MissingProduct", err.str());
    }
      
    
    
    Handle<vector<SimVertex> > simVertices;
    found = iEvent.getByToken(tokenSimVertices_,simVertices);
    if(!found) {
      LogError("PFSimParticleProducer")
    <<"cannot find sim vertices: "<<inputTagSim_<<endl;
      return;
    }

    mySimEvent->fill( *simTracks, *simVertices );
      
    if(verbose_) 
      mySimEvent->print();

    for(unsigned i=0; i<mySimEvent->nTracks(); i++) {
    
      const FSimTrack& fst = mySimEvent->track(i);

      int motherId = -1;
      if( ! fst.noMother() ) // a mother exist
    motherId = fst.mother().id();

      //This is finding out the simID corresponding 
      //to the recTrack

      //GETTING THE TRACK ID
      unsigned         recTrackID = 99999;
      vector<unsigned> recHitContrib; //modif
      vector<double>   recHitContribFrac; //modif

      if(mctruthMatchingInfo_){ //modif

    for(unsigned lo=0; lo<recTrackSimID.size(); 
        lo++) {
      if( i == recTrackSimID[lo] ) {

        recTrackID = lo;
      }//match track
    }//loop rectrack
    
    // get the ecalBarrel rechits for MC truth matching tool
    edm::Handle<EcalRecHitCollection> rhcHandle;
    bool found = iEvent.getByToken(tokenEcalRecHitsEB_, 
                       rhcHandle);
    if(!found) {
      ostringstream err;
      err<<"could not find rechits "<< inputTagEcalRecHitsEB_;
      LogError("PFSimParticleProducer")<<err.str()<<endl;
      
      throw cms::Exception( "MissingProduct", err.str());
    }
    else {
      assert( rhcHandle.isValid() );
      
      EBRecHitCollection::const_iterator it_rh    
        = rhcHandle.product()->begin();
      EBRecHitCollection::const_iterator itend_rh 
        = rhcHandle.product()->end();
      
      for(;it_rh!=itend_rh;++it_rh)
        {
          unsigned rhit_hi 
        = EBDetId(it_rh->id()).hashedIndex();
          EBDetId detid(it_rh->id());
        
          ITM it_phit    = caloHitsEBID[rhit_hi].begin();
          ITM itend_phit = caloHitsEBID[rhit_hi].end();    
          for(;it_phit!=itend_phit;++it_phit)
        {
          if(i == it_phit->second)
            {
              //Alex (08/10/08) TO BE REMOVED, eliminating
              //duplicated rechits
              bool alreadyin = false;
              for( unsigned ihit = 0; ihit < recHitContrib.size(); 
               ++ihit )
            if(detid.rawId() == recHitContrib[ihit]) 
              alreadyin = true;
              
              if(!alreadyin){       
            double pcalofraction = 0.0;
            if(caloHitsEBTotE[rhit_hi] != 0.0)
              pcalofraction 
                = (it_phit->first/caloHitsEBTotE[rhit_hi])*100.0;
            
            //store info
            recHitContrib.push_back(it_rh->id());
            recHitContribFrac.push_back(pcalofraction);
              }//selected rechits       
            }//matching
        }//loop pcalohit
          
        }//loop rechits
      
    }//getting the rechits

      }//mctruthMatchingInfo_ //modif

      reco::PFSimParticle particle(  fst.charge(), 
                     fst.type(), 
                     fst.id(), 
                     motherId,
                     fst.daughters(),
                     recTrackID,
                     recHitContrib,
                     recHitContribFrac);


      const FSimVertex& originVtx = fst.vertex();

      math::XYZPoint          posOrig( originVtx.position().x(), 
                       originVtx.position().y(), 
                       originVtx.position().z() );

      math::XYZTLorentzVector momOrig( fst.momentum().px(), 
                       fst.momentum().py(), 
                       fst.momentum().pz(), 
                       fst.momentum().e() );
      reco::PFTrajectoryPoint 
    pointOrig(-1, 
          reco::PFTrajectoryPoint::ClosestApproach,
          posOrig, momOrig);

      // point 0 is origin vertex
      particle.addPoint(pointOrig);
    

      if( ! fst.noEndVertex() ) {
    const FSimVertex& endVtx = fst.endVertex();
    
    math::XYZPoint          posEnd( endVtx.position().x(), 
                    endVtx.position().y(), 
                    endVtx.position().z() );
    
    math::XYZTLorentzVector momEnd;
    
    reco::PFTrajectoryPoint 
      pointEnd( -1, 
            reco::PFTrajectoryPoint::BeamPipeOrEndVertex,
            posEnd, momEnd);
    
    particle.addPoint(pointEnd);
      }
      else { // add a dummy point
    reco::PFTrajectoryPoint dummy;
    particle.addPoint(dummy); 
      }


      if( fst.onLayer1() ) { // PS layer1
    const RawParticle& rp = fst.layer1Entrance();
      
    math::XYZPoint posLayer1( rp.x(), rp.y(), rp.z() );
    math::XYZTLorentzVector momLayer1( rp.px(), rp.py(), rp.pz(), rp.e() );
    reco::PFTrajectoryPoint layer1Pt(-1, reco::PFTrajectoryPoint::PS1, 
                     posLayer1, momLayer1);
    
    particle.addPoint( layer1Pt ); 

    // extrapolate to cluster depth
      }
      else { // add a dummy point
    reco::PFTrajectoryPoint dummy;
    particle.addPoint(dummy); 
      }

      if( fst.onLayer2() ) { // PS layer2
    const RawParticle& rp = fst.layer2Entrance();
      
    math::XYZPoint posLayer2( rp.x(), rp.y(), rp.z() );
    math::XYZTLorentzVector momLayer2( rp.px(), rp.py(), rp.pz(), rp.e() );
    reco::PFTrajectoryPoint layer2Pt(-1, reco::PFTrajectoryPoint::PS2, 
                     posLayer2, momLayer2);
    
    particle.addPoint( layer2Pt ); 

    // extrapolate to cluster depth
      }
      else { // add a dummy point
    reco::PFTrajectoryPoint dummy;
    particle.addPoint(dummy); 
      }

      if( fst.onEcal() ) {
    const RawParticle& rp = fst.ecalEntrance();
    
    math::XYZPoint posECAL( rp.x(), rp.y(), rp.z() );
    math::XYZTLorentzVector momECAL( rp.px(), rp.py(), rp.pz(), rp.e() );
    reco::PFTrajectoryPoint ecalPt(-1, 
                       reco::PFTrajectoryPoint::ECALEntrance, 
                       posECAL, momECAL);
    
    particle.addPoint( ecalPt ); 

    // extrapolate to cluster depth
      }
      else { // add a dummy point
    reco::PFTrajectoryPoint dummy;
    particle.addPoint(dummy); 
      }
    
      // add a dummy point for ECAL Shower max
      reco::PFTrajectoryPoint dummy;
      particle.addPoint(dummy); 

      if( fst.onHcal() ) {

    const RawParticle& rpin = fst.hcalEntrance();
    
    math::XYZPoint posHCALin( rpin.x(), rpin.y(), rpin.z() );
    math::XYZTLorentzVector momHCALin( rpin.px(), rpin.py(), rpin.pz(), 
                       rpin.e() );
    reco::PFTrajectoryPoint hcalPtin(-1, 
                     reco::PFTrajectoryPoint::HCALEntrance,
                     posHCALin, momHCALin);
    
    particle.addPoint( hcalPtin ); 

    const RawParticle& rpout = fst.hcalExit();
    
    math::XYZPoint posHCALout( rpout.x(), rpout.y(), rpout.z() );
    math::XYZTLorentzVector momHCALout( rpout.px(), rpout.py(), rpout.pz(),
                        rpout.e() );
    reco::PFTrajectoryPoint 
      hcalPtout(-1, reco::PFTrajectoryPoint::HCALExit, 
             posHCALout, momHCALout);
    
    particle.addPoint( hcalPtout );     

    const RawParticle& rpho = fst.hoEntrance();
    
    math::XYZPoint posHOEntrance( rpho.x(), rpho.y(), rpho.z() );
    math::XYZTLorentzVector momHOEntrance( rpho.px(), rpho.py(), rpho.pz(),
                        rpho.e() );
    reco::PFTrajectoryPoint 
      hoPtin(-1, reco::PFTrajectoryPoint::HOLayer, 
         posHOEntrance, momHOEntrance);
    
    particle.addPoint( hoPtin );    




      }
      else { // add a dummy point
    reco::PFTrajectoryPoint dummy;
    particle.addPoint(dummy); 
      }
          
      pOutputPFSimParticleCollection->push_back( particle );
    }
    
    iEvent.put(std::move(pOutputPFSimParticleCollection));
  }

  
  LogDebug("PFSimParticleProducer")<<"STOP event: "<<iEvent.id().event()
                   <<" in run "<<iEvent.id().run()<<endl;
}

Member Data Documentation

edm::InputTag PFSimParticleProducer::inputTagEcalRecHitsEB_

private

Definition at line 74 of file PFSimParticleProducer.h.

edm::InputTag PFSimParticleProducer::inputTagEcalRecHitsEE_

private

Definition at line 76 of file PFSimParticleProducer.h.

edm::InputTag PFSimParticleProducer::inputTagFastSimProducer_

private

Definition at line 68 of file PFSimParticleProducer.h.

edm::InputTag PFSimParticleProducer::inputTagRecTracks_

private

Definition at line 72 of file PFSimParticleProducer.h.

edm::InputTag PFSimParticleProducer::inputTagSim_

private

module label for retrieving input simtrack and simvertex

Definition at line 61 of file PFSimParticleProducer.h.

bool PFSimParticleProducer::mctruthMatchingInfo_

private

Definition at line 67 of file PFSimParticleProducer.h.

FSimEvent* PFSimParticleProducer::mySimEvent

private

Definition at line 82 of file PFSimParticleProducer.h.

edm::ParameterSet PFSimParticleProducer::particleFilter_

private

Definition at line 81 of file PFSimParticleProducer.h.

bool PFSimParticleProducer::processParticles_

private

process particles on/off

Definition at line 87 of file PFSimParticleProducer.h.

edm::EDGetTokenT<EcalRecHitCollection> PFSimParticleProducer::tokenEcalRecHitsEB_

private

Definition at line 75 of file PFSimParticleProducer.h.

edm::EDGetTokenT<EcalRecHitCollection> PFSimParticleProducer::tokenEcalRecHitsEE_

private

Definition at line 77 of file PFSimParticleProducer.h.

edm::EDGetTokenT<edm::PCaloHitContainer> PFSimParticleProducer::tokenFastSimProducer_

private

Definition at line 69 of file PFSimParticleProducer.h.

edm::EDGetTokenT<reco::PFRecTrackCollection> PFSimParticleProducer::tokenRecTracks_

private

Definition at line 73 of file PFSimParticleProducer.h.

edm::EDGetTokenT<std::vector<SimTrack> > PFSimParticleProducer::tokenSim_

private

Definition at line 62 of file PFSimParticleProducer.h.

edm::EDGetTokenT<std::vector<SimVertex> > PFSimParticleProducer::tokenSimVertices_

private

Definition at line 63 of file PFSimParticleProducer.h.

bool PFSimParticleProducer::verbose_

private

verbose ?

Definition at line 90 of file PFSimParticleProducer.h.