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	hasAbilityToProduceInBeginLumis () const final
bool	hasAbilityToProduceInBeginProcessBlocks () const final
bool	hasAbilityToProduceInBeginRuns () const final
bool	hasAbilityToProduceInEndLumis () const final
bool	hasAbilityToProduceInEndProcessBlocks () const final
bool	hasAbilityToProduceInEndRuns () 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 40 of file PFSimParticleProducer.h.

Member Typedef Documentation

TrackHandle

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

Definition at line 48 of file PFSimParticleProducer.h.

Constructor & Destructor Documentation

PFSimParticleProducer()

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

explicit

Definition at line 65 of file PFSimParticleProducer.cc.

                                                                           {
  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_);
}

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

~PFSimParticleProducer()

PFSimParticleProducer::~PFSimParticleProducer ( )

override

Definition at line 98 of file PFSimParticleProducer.cc.

{ delete mySimEvent; }

Member Function Documentation

getSimIDs()

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

Definition at line 395 of file PFSimParticleProducer.cc.

                                                                                                   {
  if (trackh.isValid()) {
    for (unsigned i = 0; i < trackh->size(); i++) {
      const reco::PFRecTrackRef ref(trackh, i);
 
      for (auto const& hit : ref->trackRef()->recHits()) {
        if (hit->isValid()) {
          auto rechit = dynamic_cast<const FastTrackerRecHit*>(hit);
 
          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
}

References mps_fire::i, and edm::HandleBase::isValid().

produce()

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

override

Definition at line 100 of file PFSimParticleProducer.cc.

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

References cms::cuda::assert(), reco::PFTrajectoryPoint::BeamPipeOrEndVertex, edm::SortedCollection< T, SORT >::begin(), FSimTrack::charge(), reco::PFTrajectoryPoint::ClosestApproach, FSimTrack::daughters(), RawParticle::e(), reco::PFTrajectoryPoint::ECALEntrance, FSimTrack::ecalEntrance(), edm::SortedCollection< T, SORT >::end(), FSimTrack::endVertex(), submitPVResolutionJobs::err, Exception, newFWLiteAna::found, edm::EventSetup::getData(), ecalpyutils::hashedIndex(), reco::PFTrajectoryPoint::HCALEntrance, FSimTrack::hcalEntrance(), reco::PFTrajectoryPoint::HCALExit, FSimTrack::hcalExit(), FSimTrack::hoEntrance(), reco::PFTrajectoryPoint::HOLayer, mps_fire::i, FSimTrack::id(), iEvent, 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, RawParticle::px(), RawParticle::py(), RawParticle::pz(), hiRegitInitialStep_cff::recTracks, TrackCandidateProducer_cfi::simTracks, HGCalValidator_cfi::simVertices, CoreSimTrack::type(), FSimTrack::vertex(), RawParticle::x(), RawParticle::y(), and RawParticle::z().

Member Data Documentation

inputTagEcalRecHitsEB_

edm::InputTag PFSimParticleProducer::inputTagEcalRecHitsEB_

private

Definition at line 66 of file PFSimParticleProducer.h.

inputTagEcalRecHitsEE_

edm::InputTag PFSimParticleProducer::inputTagEcalRecHitsEE_

private

Definition at line 68 of file PFSimParticleProducer.h.

inputTagFastSimProducer_

edm::InputTag PFSimParticleProducer::inputTagFastSimProducer_

private

Definition at line 60 of file PFSimParticleProducer.h.

inputTagRecTracks_

edm::InputTag PFSimParticleProducer::inputTagRecTracks_

private

Definition at line 64 of file PFSimParticleProducer.h.

inputTagSim_

edm::InputTag PFSimParticleProducer::inputTagSim_

private

module label for retrieving input simtrack and simvertex

Definition at line 53 of file PFSimParticleProducer.h.

mctruthMatchingInfo_

bool PFSimParticleProducer::mctruthMatchingInfo_

private

Definition at line 59 of file PFSimParticleProducer.h.

mySimEvent

FSimEvent* PFSimParticleProducer::mySimEvent

private

Definition at line 74 of file PFSimParticleProducer.h.

particleFilter_

edm::ParameterSet PFSimParticleProducer::particleFilter_

private

Definition at line 73 of file PFSimParticleProducer.h.

processParticles_

bool PFSimParticleProducer::processParticles_

private

process particles on/off

Definition at line 79 of file PFSimParticleProducer.h.

tokenEcalRecHitsEB_

edm::EDGetTokenT<EcalRecHitCollection> PFSimParticleProducer::tokenEcalRecHitsEB_

private

Definition at line 67 of file PFSimParticleProducer.h.

tokenEcalRecHitsEE_

edm::EDGetTokenT<EcalRecHitCollection> PFSimParticleProducer::tokenEcalRecHitsEE_

private

Definition at line 69 of file PFSimParticleProducer.h.

tokenFastSimProducer_

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

private

Definition at line 61 of file PFSimParticleProducer.h.

tokenRecTracks_

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

private

Definition at line 65 of file PFSimParticleProducer.h.

tokenSim_

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

private

Definition at line 54 of file PFSimParticleProducer.h.

tokenSimVertices_

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

private

Definition at line 55 of file PFSimParticleProducer.h.

verbose_

bool PFSimParticleProducer::verbose_

private

verbose ?

Definition at line 82 of file PFSimParticleProducer.h.