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	hasAbilityToProduceInBeginRuns () const final
bool	hasAbilityToProduceInEndLumis () 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

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

Definition at line 48 of file PFSimParticleProducer.h.

Constructor & Destructor Documentation

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

explicit

Definition at line 65 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 98 of file PFSimParticleProducer.cc.

{ delete mySimEvent; }

Member Function Documentation

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

Definition at line 395 of file PFSimParticleProducer.cc.

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

                                                                                                   {
  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
}

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

override

Definition at line 100 of file PFSimParticleProducer.cc.

References 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(), runTheMatrix::err, edm::EventID::event(), Exception, newFWLiteAna::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(), RawParticle::px(), RawParticle::py(), RawParticle::pz(), hiRegitInitialStep_cff::recTracks, edm::EventID::run(), TrackCandidateProducer_cfi::simTracks, HGCalValidator_cfi::simVertices, 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));

  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 66 of file PFSimParticleProducer.h.

edm::InputTag PFSimParticleProducer::inputTagEcalRecHitsEE_

private

Definition at line 68 of file PFSimParticleProducer.h.

edm::InputTag PFSimParticleProducer::inputTagFastSimProducer_

private

Definition at line 60 of file PFSimParticleProducer.h.

edm::InputTag PFSimParticleProducer::inputTagRecTracks_

private

Definition at line 64 of file PFSimParticleProducer.h.

edm::InputTag PFSimParticleProducer::inputTagSim_

private

module label for retrieving input simtrack and simvertex

Definition at line 53 of file PFSimParticleProducer.h.

bool PFSimParticleProducer::mctruthMatchingInfo_

private

Definition at line 59 of file PFSimParticleProducer.h.

FSimEvent* PFSimParticleProducer::mySimEvent

private

Definition at line 74 of file PFSimParticleProducer.h.

edm::ParameterSet PFSimParticleProducer::particleFilter_

private

Definition at line 73 of file PFSimParticleProducer.h.

bool PFSimParticleProducer::processParticles_

private

process particles on/off

Definition at line 79 of file PFSimParticleProducer.h.

edm::EDGetTokenT<EcalRecHitCollection> PFSimParticleProducer::tokenEcalRecHitsEB_

private

Definition at line 67 of file PFSimParticleProducer.h.

edm::EDGetTokenT<EcalRecHitCollection> PFSimParticleProducer::tokenEcalRecHitsEE_

private

Definition at line 69 of file PFSimParticleProducer.h.

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

private

Definition at line 61 of file PFSimParticleProducer.h.

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

private

Definition at line 65 of file PFSimParticleProducer.h.

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

private

Definition at line 54 of file PFSimParticleProducer.h.

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

private

Definition at line 55 of file PFSimParticleProducer.h.

bool PFSimParticleProducer::verbose_

private

verbose ?

Definition at line 82 of file PFSimParticleProducer.h.