AlignmentTrackSelector

ComparePt

Macros

Line Code
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107
#ifndef Alignment_CommonAlignmentAlgorithm_AlignmentTrackSelector_h
#define Alignment_CommonAlignmentAlgorithm_AlignmentTrackSelector_h

#include "DataFormats/Alignment/interface/AliClusterValueMapFwd.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrackerCommon/interface/TrackerTopology.h"
#include "DataFormats/TrackerRecHit2D/interface/SiStripMatchedRecHit2DCollection.h"
#include "DataFormats/TrackerRecHit2D/interface/SiStripRecHit2DCollection.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/Utilities/interface/EDGetToken.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "FWCore/ParameterSet/interface/ParameterSetDescription.h"
#include "Geometry/Records/interface/TrackerTopologyRcd.h"
#include <vector>

namespace edm {
  class Event;
  class EventSetup;
  class ParameterSet;
}  // namespace edm

class TrackingRecHit;
class SiStripRecHit1D;
class SiStripRecHit2D;

class AlignmentTrackSelector {
public:
  typedef std::vector<const reco::Track*> Tracks;

  /// constructor
  AlignmentTrackSelector(const edm::ParameterSet& cfg, edm::ConsumesCollector& iC);

  /// destructor
  ~AlignmentTrackSelector();

  /// select tracks
  Tracks select(const Tracks& tracks, const edm::Event& evt, const edm::EventSetup& eSetup) const;
  ///returns if any of the Filters is used.
  bool useThisFilter();

  static void fillPSetDescription(edm::ParameterSetDescription& desc);

private:
  /// apply basic cuts on pt,eta,phi,nhit
  Tracks basicCuts(const Tracks& tracks, const edm::Event& evt, const edm::EventSetup& eSetup) const;
  /// checking hit requirements beyond simple number of valid hits
  bool detailedHitsCheck(const reco::Track* track, const edm::Event& evt, const edm::EventSetup& eSetup) const;
  bool isHit2D(const TrackingRecHit& hit) const;
  /// if valid, check for minimum charge (currently only in strip), if invalid give true
  bool isOkCharge(const TrackingRecHit* therechit) const;
  bool isOkChargeStripHit(const SiStripRecHit1D& siStripRecHit1D) const;
  bool isOkChargeStripHit(const SiStripRecHit2D& siStripRecHit2D) const;
  bool isIsolated(const TrackingRecHit* therechit, const edm::Event& evt) const;
  bool isOkTrkQuality(const reco::Track* track) const;

  /// filter the n highest pt tracks
  Tracks theNHighestPtTracks(const Tracks& tracks) const;

  //filter tracks that do not have a min # of hits taken by the Skim&Prescale workflow
  Tracks checkPrescaledHits(const Tracks& tracks, const edm::Event& evt) const;

  /// compare two tracks in pt (used by theNHighestPtTracks)
  struct ComparePt {
    bool operator()(const reco::Track* t1, const reco::Track* t2) const { return t1->pt() > t2->pt(); }
  };
  ComparePt ptComparator;

  edm::ESGetToken<TrackerTopology, TrackerTopologyRcd> tTopoToken_;

  const bool applyBasicCuts_, applyNHighestPt_, applyMultiplicityFilter_;
  const int seedOnlyFromAbove_;
  const bool applyIsolation_, chargeCheck_;
  const int nHighestPt_, minMultiplicity_, maxMultiplicity_;
  const bool multiplicityOnInput_;  /// if true, cut min/maxMultiplicity on input instead of on final result
  const double ptMin_, ptMax_, pMin_, pMax_, etaMin_, etaMax_, phiMin_, phiMax_;
  const double nHitMin_, nHitMax_, chi2nMax_, d0Min_, d0Max_, dzMin_, dzMax_;
  const int theCharge_;
  const double minHitChargeStrip_, minHitIsolation_;
  edm::EDGetTokenT<SiStripRecHit2DCollection> rphirecHitsToken_;
  edm::EDGetTokenT<SiStripMatchedRecHit2DCollection> matchedrecHitsToken_;
  const bool countStereoHitAs2D_;  // count hits on stereo components of GluedDet for nHitMin2D_?
  const unsigned int nHitMin2D_;
  const int minHitsinTIB_, minHitsinTOB_, minHitsinTID_, minHitsinTEC_;
  const int minHitsinBPIX_, minHitsinFPIX_, minHitsinPIX_;
  const int minHitsinTIDplus_, minHitsinTIDminus_, minHitsinTECplus_, minHitsinTECminus_;
  const int minHitsinFPIXplus_, minHitsinFPIXminus_;
  const int minHitsinENDCAP_, minHitsinENDCAPplus_, minHitsinENDCAPminus_;
  const double maxHitDiffEndcaps_;
  const double nLostHitMax_;
  std::vector<double> RorZofFirstHitMin_;
  std::vector<double> RorZofFirstHitMax_;
  std::vector<double> RorZofLastHitMin_;
  std::vector<double> RorZofLastHitMax_;

  const edm::InputTag clusterValueMapTag_;                     // ValueMap containing association cluster - flag
  edm::EDGetTokenT<AliClusterValueMap> clusterValueMapToken_;  // ValueMap containing association cluster - flag
  const int minPrescaledHits_;
  const bool applyPrescaledHitsFilter_;

  std::vector<reco::TrackBase::TrackQuality> trkQualities_;

  std::vector<reco::TrackBase::TrackAlgorithm> trkSteps_;
  bool applyTrkQualityCheck_;
  bool applyIterStepCheck_;
};

#endif