Inheritance diagram for reco::modules::CosmicTrackSplitter:

Public Member Functions
	CosmicTrackSplitter (const edm::ParameterSet &iConfig)

void	produce (edm::Event &iEvent, const edm::EventSetup &iSetup) 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 Member Functions
TrackCandidate	makeCandidate (const reco::Track &tk, std::vector< TrackingRecHit * >::iterator hitsBegin, std::vector< TrackingRecHit * >::iterator hitsEnd)

Private Attributes
std::vector< uint32_t >	detsToIgnore_

double	dXYcut_

double	dZcut_

bool	excludePixelHits_

size_t	minimumHits_

bool	replaceWithInactiveHits_

bool	stripAllInvalidHits_

bool	stripBackInvalidHits_

bool	stripFrontInvalidHits_

edm::ESHandle< TrackerGeometry >	theGeometry

edm::ESHandle< MagneticField >	theMagField

edm::ESGetToken< TrackerGeometry, TrackerDigiGeometryRecord >	tokenGeometry

edm::ESGetToken< MagneticField, IdealMagneticFieldRecord >	tokenMagField

edm::EDGetTokenT< reco::TrackCollection >	tokenTracks

edm::EDGetTokenT< TrajTrackAssociationCollection >	tokenTrajTrack

int	totalTracks_

Additional Inherited Members
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

Detailed Description

Definition at line 77 of file CosmicTrackSplitter.cc.

Constructor & Destructor Documentation

◆ CosmicTrackSplitter()

reco::modules::CosmicTrackSplitter::CosmicTrackSplitter ( const edm::ParameterSet & iConfig )

Definition at line 110 of file CosmicTrackSplitter.cc.

         : minimumHits_(iConfig.getParameter<uint32_t>("minimumHits")),
           replaceWithInactiveHits_(iConfig.getParameter<bool>("replaceWithInactiveHits")),
           stripFrontInvalidHits_(iConfig.getParameter<bool>("stripFrontInvalidHits")),
           stripBackInvalidHits_(iConfig.getParameter<bool>("stripBackInvalidHits")),
           stripAllInvalidHits_(iConfig.getParameter<bool>("stripAllInvalidHits")),
           excludePixelHits_(iConfig.getParameter<bool>("excludePixelHits")),
           dZcut_(iConfig.getParameter<double>("dzCut")),
           dXYcut_(iConfig.getParameter<double>("dxyCut")),
           detsToIgnore_(iConfig.getParameter<std::vector<uint32_t> >("detsToIgnore")) {
       // sanity check
       if (stripAllInvalidHits_ && replaceWithInactiveHits_) {
         throw cms::Exception("Configuration") << "Inconsistent Configuration: you can't set both 'stripAllInvalidHits' "
                                                  "and 'replaceWithInactiveHits' to true\n";
       }
       tokenTracks = consumes<reco::TrackCollection>(iConfig.getParameter<edm::InputTag>("tracks"));
       tokenTrajTrack =
           consumes<TrajTrackAssociationCollection>(iConfig.getParameter<edm::InputTag>("tjTkAssociationMapTag"));
       tokenGeometry = esConsumes<TrackerGeometry, TrackerDigiGeometryRecord>();
       tokenMagField = esConsumes<MagneticField, IdealMagneticFieldRecord>();
  
       LogDebug("CosmicTrackSplitter") << "sanity check";
  
       // sort detids to ignore
       std::sort(detsToIgnore_.begin(), detsToIgnore_.end());
  
       totalTracks_ = 0;
  
       // issue the produce<>
       produces<TrackCandidateCollection>();
     }

References detsToIgnore_, Exception, edm::ParameterSet::getParameter(), LogDebug, replaceWithInactiveHits_, jetUpdater_cfi::sort, stripAllInvalidHits_, tokenGeometry, tokenMagField, tokenTracks, tokenTrajTrack, and totalTracks_.

Member Function Documentation

◆ makeCandidate()

TrackCandidate reco::modules::CosmicTrackSplitter::makeCandidate	(	const reco::Track &	tk,
		std::vector< TrackingRecHit * >::iterator	hitsBegin,
		std::vector< TrackingRecHit * >::iterator	hitsEnd
	)

private

Definition at line 385 of file CosmicTrackSplitter.cc.

                                                                                                    {
       LogDebug("CosmicTrackSplitter") << "Making a candidate!";
  
       PropagationDirection pdir = tk.seedDirection();
       PTrajectoryStateOnDet state;
       if (pdir == anyDirection)
         throw cms::Exception("UnimplementedFeature") << "Cannot work with tracks that have 'anyDirecton' \n";
       //if ( (pdir == alongMomentum) == ( tk.p() >= tk.outerP() ) ) {
       if ((pdir == alongMomentum) == ((tk.outerPosition() - tk.innerPosition()).Dot(tk.momentum()) >= 0)) {
         // use inner state
         TrajectoryStateOnSurface originalTsosIn(
             trajectoryStateTransform::innerStateOnSurface(tk, *theGeometry, &*theMagField));
         state = trajectoryStateTransform::persistentState(originalTsosIn, DetId(tk.innerDetId()));
       } else {
         // use outer state
         TrajectoryStateOnSurface originalTsosOut(
             trajectoryStateTransform::outerStateOnSurface(tk, *theGeometry, &*theMagField));
         state = trajectoryStateTransform::persistentState(originalTsosOut, DetId(tk.outerDetId()));
       }
  
       TrajectorySeed seed(state, TrackCandidate::RecHitContainer(), pdir);
  
       TrackCandidate::RecHitContainer ownHits;
       ownHits.reserve(hitsEnd - hitsBegin);
       for (; hitsBegin != hitsEnd; ++hitsBegin) {
         ownHits.push_back(*hitsBegin);
       }
  
       TrackCandidate cand(ownHits, seed, state, tk.seedRef());
  
 #ifdef EDM_ML_DEBUG
       LogDebug("CosmicTrackSplitter") << "   dumping the hits now: ";
       for (auto const &hit : cand.recHits()) {
         LogTrace("CosmicTrackSplitter") << "     hit detid = " << hit.geographicalId().rawId()
                                         << ", type  = " << typeid(hit).name();
       }
 #endif
  
       return cand;
     }

References alongMomentum, anyDirection, Exception, reco::Track::innerDetId(), reco::Track::innerPosition(), trajectoryStateTransform::innerStateOnSurface(), LogDebug, LogTrace, reco::TrackBase::momentum(), Skims_PA_cff::name, reco::Track::outerDetId(), reco::Track::outerPosition(), trajectoryStateTransform::outerStateOnSurface(), trajectoryStateTransform::persistentState(), edm::OwnVector< T, P >::push_back(), edm::OwnVector< T, P >::reserve(), fileCollector::seed, reco::Track::seedDirection(), reco::Track::seedRef(), theGeometry, and theMagField.

Referenced by produce().

◆ produce()

void reco::modules::CosmicTrackSplitter::produce	(	edm::Event &	iEvent,
		const edm::EventSetup &	iSetup
	)

override

Definition at line 142 of file CosmicTrackSplitter.cc.

                                                                                    {
       LogDebug("CosmicTrackSplitter") << "IN THE SPLITTER!!!!!";
  
       // read with View, so we can read also a TrackRefVector
       edm::Handle<std::vector<reco::Track> > tracks;
       iEvent.getByToken(tokenTracks, tracks);
  
       // also need trajectories ...
       // Retrieve trajectories and tracks from the event
       // -> merely skip if collection is empty
       edm::Handle<TrajTrackAssociationCollection> m_TrajTracksMap;
       iEvent.getByToken(tokenTrajTrack, m_TrajTracksMap);
  
       // read from EventSetup
       theGeometry = iSetup.getHandle(tokenGeometry);
       theMagField = iSetup.getHandle(tokenMagField);
  
       // prepare output collection
       auto output = std::make_unique<TrackCandidateCollection>();
       output->reserve(tracks->size());
  
       // working area and tools
       std::vector<TrackingRecHit *> hits;
  
       // Form pairs of trajectories and tracks
       //ConstTrajTrackPairCollection trajTracks;
       LogDebug("CosmicTrackSplitter") << "size of map: " << m_TrajTracksMap->size();
       int HITTOSPLITFROM = 0;
       for (TrajTrackAssociationCollection::const_iterator iPair = m_TrajTracksMap->begin();
            iPair != m_TrajTracksMap->end();
            iPair++) {
         const Trajectory *trajFromMap = &(*(*iPair).key);
         const reco::Track *trackFromMap = &(*(*iPair).val);
  
         // loop to find the hit to split from (by taking dot product of pT and transverse position
         std::vector<TrajectoryMeasurement> measurements = trajFromMap->measurements();
         int totalNumberOfHits = measurements.size();
         int numberOfHits = 0;
         double previousDotProduct = 0;
         for (trackingRecHit_iterator ith = trackFromMap->recHitsBegin(), edh = trackFromMap->recHitsEnd(); ith != edh;
              ++ith) {
           GlobalVector stateMomentum = measurements[numberOfHits].forwardPredictedState().globalMomentum();
           GlobalPoint statePosition = measurements[numberOfHits].forwardPredictedState().globalPosition();
           double dotProduct = stateMomentum.x() * statePosition.x() + stateMomentum.y() * statePosition.y();
           if (dotProduct * previousDotProduct < 0) {
             //found hit to split from...
             HITTOSPLITFROM = numberOfHits;
           }
  
           previousDotProduct = dotProduct;
           numberOfHits++;
         }
         LogDebug("CosmicTrackSplitter") << "number of rechits: " << numberOfHits;
  
         // check if the trajectories and rechits are in reverse order...
         trackingRecHit_iterator bIt = trackFromMap->recHitsBegin();
         trackingRecHit_iterator fIt = trackFromMap->recHitsEnd() - 1;
         const TrackingRecHit *bHit = (*bIt);
         const TrackingRecHit *fHit = (*fIt);
         // hit type valid = 0, missing = 1, inactive = 2, bad = 3
         if (bHit->type() != 0 || bHit->isValid() != 1) {
           //loop over hits forwards until first Valid hit is found
           trackingRecHit_iterator ihit;
           for (ihit = trackFromMap->recHitsBegin(); ihit != trackFromMap->recHitsEnd(); ++ihit) {
             const TrackingRecHit *iHit = (*ihit);
             if (iHit->type() == 0 && iHit->isValid() == 1) {
               bHit = iHit;
               break;
             }
           }
         }
         DetId bdetid = bHit->geographicalId();
         GlobalPoint bPosHit = theGeometry->idToDetUnit(bdetid)->surface().toGlobal(bHit->localPosition());
         if (fHit->type() != 0 || fHit->isValid() != 1) {
           //loop over hits backwards until first Valid hit is found
           trackingRecHit_iterator ihitf;
           for (ihitf = trackFromMap->recHitsEnd() - 1; ihitf != trackFromMap->recHitsBegin(); --ihitf) {
             const TrackingRecHit *iHit = (*ihitf);
             if (iHit->type() == 0 && iHit->isValid() == 1) {
               fHit = iHit;
               break;
             }
           }
         }
         DetId fdetid = fHit->geographicalId();
         GlobalPoint fPosHit = theGeometry->idToDetUnit(fdetid)->surface().toGlobal(fHit->localPosition());
         GlobalPoint bPosState = measurements[0].updatedState().globalPosition();
         GlobalPoint fPosState = measurements[measurements.size() - 1].updatedState().globalPosition();
         bool trajReversedFlag = false;
         /*
             DetId bdetid = bHit->geographicalId();
             DetId fdetid = fHit->geographicalId();
             GlobalPoint bPosHit =  theGeometry->idToDetUnit( bdetid )->surface().toGlobal(bHit->localPosition());
             GlobalPoint fPosHit =  theGeometry->idToDetUnit( fdetid )->surface().toGlobal(fHit->localPosition());
             GlobalPoint bPosState = measurements[0].updatedState().globalPosition();
             GlobalPoint fPosState = measurements[measurements.size() - 1].updatedState().globalPosition();
             bool trajReversedFlag = false;
             */
         if (((bPosHit - bPosState).mag() > (bPosHit - fPosState).mag()) &&
             ((fPosHit - fPosState).mag() > (fPosHit - bPosState).mag())) {
           trajReversedFlag = true;
         }
         if (trajReversedFlag) {
           int temp = HITTOSPLITFROM;
           HITTOSPLITFROM = totalNumberOfHits - temp;
         }
       }
  
       totalTracks_ = totalTracks_ + tracks->size();
       // loop on tracks
       for (std::vector<reco::Track>::const_iterator itt = tracks->begin(), edt = tracks->end(); itt != edt; ++itt) {
         hits.clear();  // extra safety
  
         LogDebug("CosmicTrackSplitter") << "ntracks: " << tracks->size();
  
         // try to find distance of closest approach
         GlobalPoint v(itt->vx(), itt->vy(), itt->vz());
  
         //checks on impact parameter
         bool continueWithTrack = true;
         if (fabs(v.z()) > dZcut_)
           continueWithTrack = false;
         if (v.perp() > dXYcut_)
           continueWithTrack = false;
         if (continueWithTrack == false)
           return;
  
         // LOOP TWICE, ONCE FOR TOP AND ONCE FOR BOTTOM
         for (int i = 0; i < 2; ++i) {
           hits.clear();  // extra safety
           LogDebug("CosmicTrackSplitter") << "   loop on hits of track #" << (itt - tracks->begin());
           int usedHitCtr = 0;
           int hitCtr = 0;
           for (trackingRecHit_iterator ith = itt->recHitsBegin(), edh = itt->recHitsEnd(); ith != edh; ++ith) {
             //hitCtr++;
             const TrackingRecHit *hit = (*ith);  // ith is an iterator on edm::Ref to rechit
             LogDebug("CosmicTrackSplitter") << "         hit number " << (ith - itt->recHitsBegin());
             // let's look at valid hits
             if (hit->isValid()) {
               LogDebug("CosmicTrackSplitter") << "            valid, detid = " << hit->geographicalId().rawId();
               DetId detid = hit->geographicalId();
  
               if (detid.det() == DetId::Tracker) {  // check for tracker hits
                 LogDebug("CosmicTrackSplitter") << "            valid, tracker ";
                 bool verdict = false;
  
                 //trying to get the global position of the hit
                 //const GeomDetUnit* geomDetUnit =  theGeometry->idToDetUnit( detid ).;
  
                 const GlobalPoint pos = theGeometry->idToDetUnit(detid)->surface().toGlobal(hit->localPosition());
                 LogDebug("CosmicTrackSplitter") << "hit pos: " << pos << ", dca pos: " << v;
  
                 // top half
                 if ((i == 0) && (hitCtr < HITTOSPLITFROM)) {
                   verdict = true;
                   LogDebug("CosmicTrackSplitter") << "tophalf";
                 }
                 // bottom half
                 if ((i == 1) && (hitCtr >= HITTOSPLITFROM)) {
                   verdict = true;
                   LogDebug("CosmicTrackSplitter") << "bottomhalf";
                 }
  
                 // if the hit is good, check again at module level
                 if (verdict && std::binary_search(detsToIgnore_.begin(), detsToIgnore_.end(), detid.rawId())) {
                   verdict = false;
                 }
  
                 // if hit is good check to make sure that we are keeping pixel hits
                 if (excludePixelHits_) {
                   // check for pixel hits
                   if ((detid.det() == DetId::Tracker) && ((detid.subdetId() == 1) || (detid.subdetId() == 2))) {
                     verdict = false;
                   }
                 }
  
                 LogDebug("CosmicTrackSplitter")
                     << "                   verdict after module list: " << (verdict ? "ok" : "no");
                 if (verdict == true) {
                   // just copy the hit
                   hits.push_back(hit->clone());
                   usedHitCtr++;
                 } else {
                   // still, if replaceWithInactiveHits is true we have to put a new hit
                   if (replaceWithInactiveHits_) {
                     hits.push_back(new InvalidTrackingRecHit(*hit->det(), TrackingRecHit::inactive));
                   }
                 }
               } else {  // just copy non tracker hits
                 hits.push_back(hit->clone());
               }
             } else {
               if (!stripAllInvalidHits_) {
                 hits.push_back(hit->clone());
               }
             }  // is valid hit
             LogDebug("CosmicTrackSplitter") << "         end of hit " << (ith - itt->recHitsBegin());
             hitCtr++;
           }  // loop on hits
           LogDebug("CosmicTrackSplitter") << "   end of loop on hits of track #" << (itt - tracks->begin());
  
           std::vector<TrackingRecHit *>::iterator begin = hits.begin(), end = hits.end();
  
           LogDebug("CosmicTrackSplitter") << "   selected " << hits.size() << " hits ";
  
           // strip invalid hits at the beginning
           if (stripFrontInvalidHits_) {
             while ((begin != end) && ((*begin)->isValid() == false))
               ++begin;
           }
  
           LogDebug("CosmicTrackSplitter") << "   after front stripping we have " << (end - begin) << " hits ";
  
           // strip invalid hits at the end
           if (stripBackInvalidHits_ && (begin != end)) {
             --end;
             while ((begin != end) && ((*end)->isValid() == false))
               --end;
             ++end;
           }
  
           LogDebug("CosmicTrackSplitter") << "   after back stripping we have " << (end - begin) << " hits ";
  
           // if we still have some hits
           //if ((end - begin) >= int(minimumHits_)) {
           if (usedHitCtr >= int(minimumHits_)) {
             output->push_back(makeCandidate(*itt, begin, end));
             LogDebug("CosmicTrackSplitter") << "we made a candidate of " << hits.size() << " hits!";
           }
           // now delete the hits not used by the candidate
           for (begin = hits.begin(), end = hits.end(); begin != end; ++begin) {
             if (*begin)
               delete *begin;
           }
           LogDebug("CosmicTrackSplitter")
               << "loop: " << i << " has " << usedHitCtr << " active hits and " << hits.size() << " total hits...";
           hits.clear();
         }  // loop twice for top and bottom
       }    // loop on tracks
       LogDebug("CosmicTrackSplitter") << "totalTracks_ = " << totalTracks_;
       iEvent.put(std::move(output));
     }

References edm::AssociationMap< Tag >::begin(), edm::AssociationMap< edm::OneToOne< std::vector< Trajectory >, reco::TrackCollection, unsigned short > >::const_iterator, DetId::det(), detsToIgnore_, dXYcut_, dZcut_, edm::AssociationMap< Tag >::end(), mps_fire::end, excludePixelHits_, TrackingRecHit::geographicalId(), edm::EventSetup::getHandle(), hfClusterShapes_cfi::hits, mps_fire::i, TrackerGeometry::idToDetUnit(), iEvent, TrackingRecHit::inactive, TrackingRecHit::isValid(), TrackingRecHit::localPosition(), LogDebug, mag(), makeCandidate(), Trajectory::measurements(), minimumHits_, eostools::move(), convertSQLitetoXML_cfg::output, DetId::rawId(), reco::Track::recHitsBegin(), reco::Track::recHitsEnd(), replaceWithInactiveHits_, edm::AssociationMap< Tag >::size(), stripAllInvalidHits_, stripBackInvalidHits_, stripFrontInvalidHits_, DetId::subdetId(), GeomDet::surface(), groupFilesInBlocks::temp, theGeometry, theMagField, Surface::toGlobal(), tokenGeometry, tokenMagField, tokenTracks, tokenTrajTrack, totalTracks_, DetId::Tracker, tracks, TrackingRecHit::type(), and findQualityFiles::v.

Member Data Documentation

◆ detsToIgnore_

std::vector<uint32_t> reco::modules::CosmicTrackSplitter::detsToIgnore_

private

Definition at line 97 of file CosmicTrackSplitter.cc.

Referenced by CosmicTrackSplitter(), and produce().

◆ dXYcut_

double reco::modules::CosmicTrackSplitter::dXYcut_

private

Definition at line 95 of file CosmicTrackSplitter.cc.

Referenced by produce().

◆ dZcut_

double reco::modules::CosmicTrackSplitter::dZcut_

private

Definition at line 94 of file CosmicTrackSplitter.cc.

Referenced by produce().

◆ excludePixelHits_

bool reco::modules::CosmicTrackSplitter::excludePixelHits_

private

Definition at line 92 of file CosmicTrackSplitter.cc.

Referenced by produce().

◆ minimumHits_

size_t reco::modules::CosmicTrackSplitter::minimumHits_

private

Definition at line 86 of file CosmicTrackSplitter.cc.

Referenced by produce().

◆ replaceWithInactiveHits_

bool reco::modules::CosmicTrackSplitter::replaceWithInactiveHits_

private

Definition at line 88 of file CosmicTrackSplitter.cc.

Referenced by CosmicTrackSplitter(), and produce().

◆ stripAllInvalidHits_

bool reco::modules::CosmicTrackSplitter::stripAllInvalidHits_

private

Definition at line 91 of file CosmicTrackSplitter.cc.

Referenced by CosmicTrackSplitter(), and produce().

◆ stripBackInvalidHits_

bool reco::modules::CosmicTrackSplitter::stripBackInvalidHits_

private

Definition at line 90 of file CosmicTrackSplitter.cc.

Referenced by produce().

◆ stripFrontInvalidHits_

bool reco::modules::CosmicTrackSplitter::stripFrontInvalidHits_

private

Definition at line 89 of file CosmicTrackSplitter.cc.

Referenced by produce().

◆ theGeometry

edm::ESHandle<TrackerGeometry> reco::modules::CosmicTrackSplitter::theGeometry

private

Definition at line 99 of file CosmicTrackSplitter.cc.

Referenced by makeCandidate(), and produce().

◆ theMagField

edm::ESHandle<MagneticField> reco::modules::CosmicTrackSplitter::theMagField

private

Definition at line 100 of file CosmicTrackSplitter.cc.

Referenced by makeCandidate(), and produce().

◆ tokenGeometry

edm::ESGetToken<TrackerGeometry, TrackerDigiGeometryRecord> reco::modules::CosmicTrackSplitter::tokenGeometry

private

Definition at line 101 of file CosmicTrackSplitter.cc.

Referenced by CosmicTrackSplitter(), and produce().

◆ tokenMagField

edm::ESGetToken<MagneticField, IdealMagneticFieldRecord> reco::modules::CosmicTrackSplitter::tokenMagField

private

Definition at line 102 of file CosmicTrackSplitter.cc.

Referenced by CosmicTrackSplitter(), and produce().

◆ tokenTracks

edm::EDGetTokenT<reco::TrackCollection> reco::modules::CosmicTrackSplitter::tokenTracks

private

Definition at line 83 of file CosmicTrackSplitter.cc.

Referenced by CosmicTrackSplitter(), and produce().

◆ tokenTrajTrack

edm::EDGetTokenT<TrajTrackAssociationCollection> reco::modules::CosmicTrackSplitter::tokenTrajTrack

private

Definition at line 84 of file CosmicTrackSplitter.cc.

Referenced by CosmicTrackSplitter(), and produce().

◆ totalTracks_

int reco::modules::CosmicTrackSplitter::totalTracks_

private

Definition at line 85 of file CosmicTrackSplitter.cc.

Referenced by CosmicTrackSplitter(), and produce().

Public Member Functions

Private Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

◆ CosmicTrackSplitter()

Member Function Documentation

◆ makeCandidate()

◆ produce()

Member Data Documentation

◆ detsToIgnore_

◆ dXYcut_

◆ dZcut_

◆ excludePixelHits_

◆ minimumHits_

◆ replaceWithInactiveHits_

◆ stripAllInvalidHits_

◆ stripBackInvalidHits_

◆ stripFrontInvalidHits_

◆ theGeometry

◆ theMagField

◆ tokenGeometry

◆ tokenMagField

◆ tokenTracks

◆ tokenTrajTrack

◆ totalTracks_