Inheritance diagram for GeneralTracksImporter:

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

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

Public Member Functions inherited from BlockElementImporterBase
	BlockElementImporterBase (const edm::ParameterSet &conf, edm::ConsumesCollector &sumes)

	BlockElementImporterBase (const BlockElementImporterBase &)=delete

const std::string &	name () const

BlockElementImporterBase &	operator= (const BlockElementImporterBase &)=delete

virtual void	updateEventSetup (const edm::EventSetup &)

Private Member Functions
bool	goodPtResolution (const reco::TrackRef &trackref) const

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

Private Attributes
const bool	_cleanBadConvBrems

const bool	_debug

const std::vector< double >	_DPtovPtCut

edm::EDGetTokenT < reco::MuonCollection >	_muons

const std::vector< unsigned >	_NHitCut

edm::EDGetTokenT < reco::PFRecTrackCollection >	_src

const bool	_useIterTracking

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

Detailed Description

Definition at line 10 of file GeneralTracksImporter.cc.

Constructor & Destructor Documentation

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

inline

Definition at line 12 of file GeneralTracksImporter.cc.

                                          :
     BlockElementImporterBase(conf,sumes),
     _src(sumes.consumes<reco::PFRecTrackCollection>(conf.getParameter<edm::InputTag>("source"))),
     _muons(sumes.consumes<reco::MuonCollection>(conf.getParameter<edm::InputTag>("muonSrc"))),
     _DPtovPtCut(conf.getParameter<std::vector<double> >("DPtOverPtCuts_byTrackAlgo")),
     _NHitCut(conf.getParameter<std::vector<unsigned> >("NHitCuts_byTrackAlgo")),
     _useIterTracking(conf.getParameter<bool>("useIterativeTracking")),
     _cleanBadConvBrems(conf.existsAs<bool>("cleanBadConvertedBrems") ? conf.getParameter<bool>("cleanBadConvertedBrems") : false),
     _debug(conf.getUntrackedParameter<bool>("debug",false)) {}

Member Function Documentation

bool GeneralTracksImporter::goodPtResolution ( const reco::TrackRef & trackref ) const

private

Definition at line 140 of file GeneralTracksImporter.cc.

References _debug, _DPtovPtCut, _NHitCut, _useIterTracking, gather_cfg::cout, reco::TrackBase::ctf, reco::TrackBase::detachedTripletStep, reco::TrackBase::hltIter0, reco::TrackBase::hltIter1, reco::TrackBase::hltIter2, reco::TrackBase::hltIter3, reco::TrackBase::hltIter4, reco::TrackBase::hltIterX, reco::TrackBase::initialStep, reco::TrackBase::jetCoreRegionalStep, reco::TrackBase::lowPtTripletStep, reco::TrackBase::mixedTripletStep, reco::TrackBase::muonSeededStepInOut, reco::TrackBase::muonSeededStepOutIn, P, reco::TrackBase::pixelLessStep, reco::TrackBase::pixelPairStep, reco::tau::disc::Pt(), mathSSE::sqrt(), reco::TrackBase::tobTecStep, and reco::HitPattern::TRACK_HITS.

Referenced by importToBlock().

                                                       {
 
   const double P = trackref->p();
   const double Pt = trackref->pt();
   const double DPt = trackref->ptError();
   const unsigned int NHit = 
     trackref->hitPattern().trackerLayersWithMeasurement();
   const unsigned int NLostHit = 
     trackref->hitPattern().trackerLayersWithoutMeasurement(reco::HitPattern::TRACK_HITS);
   const unsigned int LostHits = trackref->numberOfLostHits();
   const double sigmaHad = sqrt(1.20*1.20/P+0.06*0.06) / (1.+LostHits);
 
   // iteration 1,2,3,4,5 correspond to algo = 1/4,5,6,7,8,9
   unsigned int Algo = 0; 
   switch (trackref->algo()) {
   case reco::TrackBase::ctf:
   case reco::TrackBase::initialStep:
   case reco::TrackBase::lowPtTripletStep:
   case reco::TrackBase::pixelPairStep:
   case reco::TrackBase::jetCoreRegionalStep:
   case reco::TrackBase::muonSeededStepInOut:
   case reco::TrackBase::muonSeededStepOutIn:
     Algo = 0;
     break;
   case reco::TrackBase::detachedTripletStep:
     Algo = 1;
     break;
   case reco::TrackBase::mixedTripletStep:
     Algo = 2;
     break;
   case reco::TrackBase::pixelLessStep:
     Algo = 3;
     break;
   case reco::TrackBase::tobTecStep:
     Algo = 4;
     break;
   case reco::TrackBase::hltIter0:
   case reco::TrackBase::hltIter1:
   case reco::TrackBase::hltIter2:
     Algo = _useIterTracking ? 0 : 0;
     break;
   case reco::TrackBase::hltIter3:
     Algo = _useIterTracking ? 1 : 0;
     break;
   case reco::TrackBase::hltIter4:
     Algo = _useIterTracking ? 2 : 0;
     break;
   case reco::TrackBase::hltIterX:
     Algo = _useIterTracking ? 0 : 0;
     break;
   default:
     Algo = _useIterTracking ? 5 : 0;
     break;
   }
 
   // Protection against 0 momentum tracks
   if ( P < 0.05 ) return false;
 
   // Temporary : Reject all tracking iteration beyond 5th step. 
 //  if ( Algo > 4 ) return false;
  
   if (_debug) std::cout << " PFBlockAlgo: PFrecTrack->Track Pt= "
            << Pt << " DPt = " << DPt << std::endl;
   if ( ( _DPtovPtCut[Algo] > 0. && 
      DPt/Pt > _DPtovPtCut[Algo]*sigmaHad ) || 
        NHit < _NHitCut[Algo] ) { 
     // (Algo >= 3 && LostHits != 0) ) {
     if (_debug) std::cout << " PFBlockAlgo: skip badly measured track"
              << ", P = " << P 
              << ", Pt = " << Pt 
              << " DPt = " << DPt 
              << ", N(hits) = " << NHit << " (Lost : " << LostHits << "/" << NLostHit << ")"
              << ", Algo = " << Algo
              << std::endl;
     if (_debug) std::cout << " cut is DPt/Pt < " << _DPtovPtCut[Algo] * sigmaHad << std::endl;
     if (_debug) std::cout << " cut is NHit >= " << _NHitCut[Algo] << std::endl;
     /*
     std::cout << "Track REJECTED : ";
     std::cout << ", P = " << P 
           << ", Pt = " << Pt 
           << " DPt = " << DPt 
           << ", N(hits) = " << NHit << " (Lost : " << LostHits << "/" << NLostHit << ")"
           << ", Algo = " << Algo
           << std::std::endl;
     */
     return false;
   }
 
   /*
   std::cout << "Track Accepted : ";
   std::cout << ", P = " << P 
        << ", Pt = " << Pt 
        << " DPt = " << DPt 
        << ", N(hits) = " << NHit << " (Lost : " << LostHits << "/" << NLostHit << ")"
        << ", Algo = " << Algo
        << std::std::endl;
   */
   return true;
 }

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

overridevirtual

Implements BlockElementImporterBase.

Definition at line 43 of file GeneralTracksImporter.cc.

                                                               {
   typedef BlockElementImporterBase::ElementList::value_type ElementType;  
   edm::Handle<reco::PFRecTrackCollection> tracks;
   e.getByToken(_src,tracks);
    edm::Handle<reco::MuonCollection> muons;
   e.getByToken(_muons,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.size() == 0 && dvRef.isNull() && v0Ref.isNull() ) {
       // if the Pt resolution is bad we kill this element
       if( !goodPtResolution( trkel->trackRef() ) ) {
         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 = NULL;
   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::isLooseMuon(muonref) || 
                    PFMuonAlgo::isMuon(muonref)         );
     }
     if(thisIsAPotentialMuon || goodPtResolution( pftrackref->trackRef() ) ) {
       trkElem = new reco::PFBlockElementTrack( pftrackref );
       if (thisIsAPotentialMuon && _debug) {
     std::cout << "Potential Muon P " <<  pftrackref->trackRef()->p() 
           << " pt " << pftrackref->trackRef()->p() << std::endl; 
       }
       if( muId != -1 ) trkElem->setMuonRef(muonref);
       elems.emplace_back(trkElem);
     }
   }
   elems.shrink_to_fit();
 }

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

private

Definition at line 241 of file GeneralTracksImporter.cc.

References m, and trackingTruthProducerFastSim_cfi::muon.

Referenced by importToBlock().

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