AlignmentTrackSelector.h

Go to the documentation of this file.

#ifndef Alignment_CommonAlignmentAlgorithm_AlignmentTrackSelector_h
#define Alignment_CommonAlignmentAlgorithm_AlignmentTrackSelector_h

#include "DataFormats/Alignment/interface/AliClusterValueMapFwd.h"
#include "DataFormats/TrackerRecHit2D/interface/SiStripMatchedRecHit2DCollection.h"
#include "DataFormats/TrackerRecHit2D/interface/SiStripRecHit2DCollection.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/Utilities/interface/EDGetToken.h"
#include <vector>

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

class TrackingRecHit;
class SiStripRecHit1D;
class SiStripRecHit2D;

class AlignmentTrackSelector
{

 public:

  typedef std::vector<const reco::Track*> Tracks; 

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

  ~AlignmentTrackSelector();

  Tracks select(const Tracks& tracks, const edm::Event& evt, const edm::EventSetup& eSetup) const;
  bool useThisFilter();


 private:

  Tracks basicCuts(const Tracks& tracks, const edm::Event& evt, const edm::EventSetup& eSetup) const;
  bool detailedHitsCheck(const reco::Track* track, const edm::Event& evt, const edm::EventSetup& eSetup) const;
  bool isHit2D(const TrackingRecHit &hit) const;
  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;

  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;

  struct ComparePt {
    bool operator()( const reco::Track* t1, const reco::Track* t2 ) const {
      return t1->pt()> t2->pt();
    }
  };
  ComparePt ptComparator;

  const bool applyBasicCuts_, applyNHighestPt_, applyMultiplicityFilter_;
  const bool seedOnlyFromAbove_, applyIsolation_, chargeCheck_ ;
  const int nHighestPt_, minMultiplicity_, maxMultiplicity_;
  const bool multiplicityOnInput_; 
  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