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
	EDProducer (const EDProducer &)=delete
bool	hasAbilityToProduceInBeginLumis () const final
bool	hasAbilityToProduceInBeginProcessBlocks () const final
bool	hasAbilityToProduceInBeginRuns () const final
bool	hasAbilityToProduceInEndLumis () const final
bool	hasAbilityToProduceInEndProcessBlocks () const final
bool	hasAbilityToProduceInEndRuns () const final
const EDProducer &	operator= (const EDProducer &)=delete

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<>
using	CacheTypes = CacheContexts< T... >
using	GlobalCache = typename CacheTypes::GlobalCache
using	HasAbility = AbilityChecker< T... >
using	InputProcessBlockCache = typename CacheTypes::InputProcessBlockCache
using	LuminosityBlockCache = typename CacheTypes::LuminosityBlockCache
using	LuminosityBlockContext = LuminosityBlockContextT< LuminosityBlockCache, RunCache, GlobalCache >
using	LuminosityBlockSummaryCache = typename CacheTypes::LuminosityBlockSummaryCache
using	RunCache = typename CacheTypes::RunCache
using	RunContext = RunContextT< RunCache, GlobalCache >
using	RunSummaryCache = typename CacheTypes::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.

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

        : 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>();
    }

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.

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().

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

produce()

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

override

Definition at line 142 of file CosmicTrackSplitter.cc.

References edm::AssociationMap< Tag >::begin(), DetId::det(), detsToIgnore_, dXYcut_, dZcut_, edm::AssociationMap< Tag >::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(), 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, DiMuonV_cfg::tracks, TrackingRecHit::type(), and findQualityFiles::v.

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

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().