reco::modules::CosmicTrackSplitter Class Reference

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	hasAbilityToProduceInBeginRuns () const final
bool	hasAbilityToProduceInEndLumis () 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::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 76 of file CosmicTrackSplitter.cc.

Constructor & Destructor Documentation

CosmicTrackSplitter()

reco::modules::CosmicTrackSplitter::CosmicTrackSplitter

(

const edm::ParameterSet &

iConfig

)

Definition at line 107 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"));
 
      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_, stripAllInvalidHits_, 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 380 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 (TrackCandidate::range hitR = cand.recHits(); hitR.first != hitR.second; ++hitR.first) {
        auto const &tmp = *hitR.first;
        LogTrace("CosmicTrackSplitter") << "     hit detid = " << hitR.first->geographicalId().rawId()
                                        << ", type  = " << typeid(tmp).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(), SurveyInfoScenario_cff::seed, reco::Track::seedDirection(), reco::Track::seedRef(), theGeometry, theMagField, and createJobs::tmp.

Referenced by produce().

produce()

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

override

Definition at line 137 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
      iSetup.get<TrackerDigiGeometryRecord>().get(theGeometry);
      iSetup.get<IdealMagneticFieldRecord>().get(theMagField);
 
      // 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 begin, edm::AssociationMap< Tag >::begin(), edm::AssociationMap< edm::OneToOne< std::vector< Trajectory >, reco::TrackCollection, unsigned short > >::const_iterator, DetId::det(), detsToIgnore_, dXYcut_, dZcut_, end, edm::AssociationMap< Tag >::end(), excludePixelHits_, TrackingRecHit::geographicalId(), reco::get(), edm::EventSetup::get(), 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(), tokenTracks, tokenTrajTrack, totalTracks_, DetId::Tracker, PDWG_EXOHSCP_cff::tracks, TrackingRecHit::type(), and findQualityFiles::v.

Member Data Documentation

detsToIgnore_

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

private

Definition at line 96 of file CosmicTrackSplitter.cc.

Referenced by CosmicTrackSplitter(), and produce().

dXYcut_

double reco::modules::CosmicTrackSplitter::dXYcut_

private

Definition at line 94 of file CosmicTrackSplitter.cc.

Referenced by produce().

dZcut_

double reco::modules::CosmicTrackSplitter::dZcut_

private

Definition at line 93 of file CosmicTrackSplitter.cc.

Referenced by produce().

excludePixelHits_

bool reco::modules::CosmicTrackSplitter::excludePixelHits_

private

Definition at line 91 of file CosmicTrackSplitter.cc.

Referenced by produce().

minimumHits_

size_t reco::modules::CosmicTrackSplitter::minimumHits_

private

Definition at line 85 of file CosmicTrackSplitter.cc.

Referenced by produce().

replaceWithInactiveHits_

bool reco::modules::CosmicTrackSplitter::replaceWithInactiveHits_

private

Definition at line 87 of file CosmicTrackSplitter.cc.

Referenced by CosmicTrackSplitter(), and produce().

stripAllInvalidHits_

bool reco::modules::CosmicTrackSplitter::stripAllInvalidHits_

private

Definition at line 90 of file CosmicTrackSplitter.cc.

Referenced by CosmicTrackSplitter(), and produce().

stripBackInvalidHits_

bool reco::modules::CosmicTrackSplitter::stripBackInvalidHits_

private

Definition at line 89 of file CosmicTrackSplitter.cc.

Referenced by produce().

stripFrontInvalidHits_

bool reco::modules::CosmicTrackSplitter::stripFrontInvalidHits_

private

Definition at line 88 of file CosmicTrackSplitter.cc.

Referenced by produce().

theGeometry

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

private

Definition at line 98 of file CosmicTrackSplitter.cc.

Referenced by makeCandidate(), and produce().

theMagField

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

private

Definition at line 99 of file CosmicTrackSplitter.cc.

Referenced by makeCandidate(), and produce().

tokenTracks

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

private

Definition at line 82 of file CosmicTrackSplitter.cc.

Referenced by CosmicTrackSplitter(), and produce().

tokenTrajTrack

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

private

Definition at line 83 of file CosmicTrackSplitter.cc.

Referenced by CosmicTrackSplitter(), and produce().

totalTracks_

int reco::modules::CosmicTrackSplitter::totalTracks_

private

Definition at line 84 of file CosmicTrackSplitter.cc.

Referenced by CosmicTrackSplitter(), and produce().