Inheritance diagram for GeneralTracksImporterWithVeto:

Public Member Functions
	GeneralTracksImporterWithVeto (const edm::ParameterSet &conf, edm::ConsumesCollector &sumes)

void	importToBlock (const edm::Event &, ElementList &) const override

Public Member Functions inherited from BlockElementImporterBase
	BlockElementImporterBase (const BlockElementImporterBase &)=delete

	BlockElementImporterBase (const edm::ParameterSet &conf, edm::ConsumesCollector &sumes)

const std::string &	name () const

BlockElementImporterBase &	operator= (const BlockElementImporterBase &)=delete

virtual void	updateEventSetup (const edm::EventSetup &)

virtual	~BlockElementImporterBase ()=default

Private Member Functions
int	muAssocToTrack (const reco::TrackRef &trackref, const edm::Handle< reco::MuonCollection > &muonh) const

Private Attributes
const bool	cleanBadConvBrems_

const std::vector< double >	DPtovPtCut_

const double	muonMaxDPtOPt_

edm::EDGetTokenT< reco::MuonCollection >	muons_

const std::vector< unsigned >	NHitCut_

edm::EDGetTokenT< reco::PFRecTrackCollection >	src_

const reco::TrackBase::TrackQuality	trackQuality_

const bool	useIterTracking_

edm::EDGetTokenT< reco::PFRecTrackCollection >	veto_

Additional Inherited Members
Public Types inherited from BlockElementImporterBase
typedef std::vector< std::unique_ptr< reco::PFBlockElement > >	ElementList

Detailed Description

Definition at line 11 of file GeneralTracksImporterWithVeto.cc.

Constructor & Destructor Documentation

◆ GeneralTracksImporterWithVeto()

GeneralTracksImporterWithVeto::GeneralTracksImporterWithVeto	(	const edm::ParameterSet &	conf,
		edm::ConsumesCollector &	sumes
	)

inline

Definition at line 13 of file GeneralTracksImporterWithVeto.cc.

       : BlockElementImporterBase(conf, sumes),
         src_(sumes.consumes<reco::PFRecTrackCollection>(conf.getParameter<edm::InputTag>("source"))),
         veto_(sumes.consumes<reco::PFRecTrackCollection>(conf.getParameter<edm::InputTag>("veto"))),
         muons_(sumes.consumes<reco::MuonCollection>(conf.getParameter<edm::InputTag>("muonSrc"))),
         trackQuality_(reco::TrackBase::qualityByName(conf.getParameter<std::string>("trackQuality"))),
         DPtovPtCut_(conf.getParameter<std::vector<double> >("DPtOverPtCuts_byTrackAlgo")),
         NHitCut_(conf.getParameter<std::vector<unsigned> >("NHitCuts_byTrackAlgo")),
         useIterTracking_(conf.getParameter<bool>("useIterativeTracking")),
         cleanBadConvBrems_(conf.getParameter<bool>("cleanBadConvertedBrems")),
         muonMaxDPtOPt_(conf.getParameter<double>("muonMaxDPtOPt")) {}

Member Function Documentation

◆ importToBlock()

void GeneralTracksImporterWithVeto::importToBlock	(	const edm::Event &	e,
		BlockElementImporterBase::ElementList &	elems
	)		const

overridevirtual

Implements BlockElementImporterBase.

Definition at line 41 of file GeneralTracksImporterWithVeto.cc.

                                                                                                     {
   typedef BlockElementImporterBase::ElementList::value_type ElementType;
   auto tracks = e.getHandle(src_);
   auto vetosH = e.getHandle(veto_);
   const auto& vetos = *vetosH;
   std::unordered_set<unsigned> vetoed;
   for (unsigned i = 0; i < vetos.size(); ++i) {
     vetoed.insert(vetos[i].trackRef().key());
   }
   auto muons = e.getHandle(muons_);
   elems.reserve(elems.size() + tracks->size());
   std::vector<bool> mask(tracks->size(), true);
   reco::MuonRef muonref;
  
   // remove converted brems with bad pT resolution if requested
   // this reproduces the old behavior of PFBlockAlgo
   if (cleanBadConvBrems_) {
     auto itr = elems.begin();
     while (itr != elems.end()) {
       if ((*itr)->type() == reco::PFBlockElement::TRACK) {
         const reco::PFBlockElementTrack* trkel = static_cast<reco::PFBlockElementTrack*>(itr->get());
         const reco::ConversionRefVector& cRef = trkel->convRefs();
         const reco::PFDisplacedTrackerVertexRef& dvRef = trkel->displacedVertexRef(reco::PFBlockElement::T_FROM_DISP);
         const reco::VertexCompositeCandidateRef& v0Ref = trkel->V0Ref();
         // if there is no displaced vertex reference  and it is marked
         // as a conversion it's gotta be a converted brem
         if (trkel->trackType(reco::PFBlockElement::T_FROM_GAMMACONV) && cRef.empty() && dvRef.isNull() &&
             v0Ref.isNull()) {
           // if the Pt resolution is bad we kill this element
           if (!PFTrackAlgoTools::goodPtResolution(
                   trkel->trackRef(), DPtovPtCut_, NHitCut_, useIterTracking_, trackQuality_)) {
             itr = elems.erase(itr);
             continue;
           }
         }
       }
       ++itr;
     }  // loop on existing elements
   }
   // preprocess existing tracks in the element list and create a mask
   // so that we do not import tracks twice, tag muons we find
   // in this collection
   auto TKs_end = std::partition(
       elems.begin(), elems.end(), [](const ElementType& a) { return a->type() == reco::PFBlockElement::TRACK; });
   auto btk_elems = elems.begin();
   auto btrack = tracks->cbegin();
   auto etrack = tracks->cend();
   for (auto track = btrack; track != etrack; ++track) {
     auto tk_elem =
         std::find_if(btk_elems, TKs_end, [&](const ElementType& a) { return (a->trackRef() == track->trackRef()); });
     if (tk_elem != TKs_end) {
       mask[std::distance(tracks->cbegin(), track)] = false;
       // check and update if this track is a muon
       const int muId = muAssocToTrack((*tk_elem)->trackRef(), muons);
       if (muId != -1) {
         muonref = reco::MuonRef(muons, muId);
         if (PFMuonAlgo::isLooseMuon(muonref) || PFMuonAlgo::isMuon(muonref)) {
           static_cast<reco::PFBlockElementTrack*>(tk_elem->get())->setMuonRef(muonref);
         }
       }
     }
   }
   // now we actually insert tracks, again tagging muons along the way
   reco::PFRecTrackRef pftrackref;
   reco::PFBlockElementTrack* trkElem = nullptr;
   for (auto track = btrack; track != etrack; ++track) {
     const unsigned idx = std::distance(btrack, track);
     // since we already set muon refs in the previously imported tracks,
     // here we can skip everything that is already imported
     if (!mask[idx])
       continue;
     muonref = reco::MuonRef();
     pftrackref = reco::PFRecTrackRef(tracks, idx);
     // Get the eventual muon associated to this track
     const int muId = muAssocToTrack(pftrackref->trackRef(), muons);
     bool thisIsAPotentialMuon = false;
     if (muId != -1) {
       muonref = reco::MuonRef(muons, muId);
       thisIsAPotentialMuon =
           ((PFMuonAlgo::hasValidTrack(muonref, true, muonMaxDPtOPt_) && PFMuonAlgo::isLooseMuon(muonref)) ||
            (PFMuonAlgo::hasValidTrack(muonref, false, muonMaxDPtOPt_) && PFMuonAlgo::isMuon(muonref)));
     }
     if (thisIsAPotentialMuon || PFTrackAlgoTools::goodPtResolution(
                                     pftrackref->trackRef(), DPtovPtCut_, NHitCut_, useIterTracking_, trackQuality_)) {
       trkElem = new reco::PFBlockElementTrack(pftrackref);
       if (thisIsAPotentialMuon) {
         LogDebug("GeneralTracksImporterWithVeto")
             << "Potential Muon P " << pftrackref->trackRef()->p() << " pt " << pftrackref->trackRef()->p() << std::endl;
       }
       if (muId != -1)
         trkElem->setMuonRef(muonref);
       if (vetoed.count(pftrackref->trackRef().key()) == 0 || muonref.isNonnull()) {
         elems.emplace_back(trkElem);
       } else
         delete trkElem;
     }
   }
   elems.shrink_to_fit();
 }

◆ muAssocToTrack()

int GeneralTracksImporterWithVeto::muAssocToTrack	(	const reco::TrackRef &	trackref,
		const edm::Handle< reco::MuonCollection > &	muonh
	)		const

private

Definition at line 142 of file GeneralTracksImporterWithVeto.cc.

                                                                                                     {
   auto muon = std::find_if(muonh->cbegin(), muonh->cend(), [&](const reco::Muon& m) {
     return (m.track().isNonnull() && m.track() == trackref);
   });
   return (muon != muonh->cend() ? std::distance(muonh->cbegin(), muon) : -1);
 }