HitExtractorSTRP.h

Go to the documentation of this file.

#ifndef RecoTracker_TkSeedingLayers_HitExtractorSTRP_H
#define RecoTracker_TkSeedingLayers_HitExtractorSTRP_H

#include "RecoTracker/TkSeedingLayers/interface/SeedingLayer.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "HitExtractor.h"

#include "DataFormats/TrackerRecHit2D/interface/SiStripRecHit2D.h"
#include "DataFormats/TrackerRecHit2D/interface/SiStripMatchedRecHit2DCollection.h"
#include "DataFormats/TrackerRecHit2D/interface/SiStripRecHit2DCollection.h"

#include <vector>
#include <tuple>
class DetLayer;

namespace edm {
  template< typename T> class ContainerMask;
}

namespace ctfseeding {

class HitExtractorSTRP : public HitExtractor {

public:
  typedef SiStripRecHit2D::ClusterRef SiStripClusterRef;

  HitExtractorSTRP(GeomDetEnumerators::SubDetector subdet, SeedingLayer::Side & side, int idLayer);
  virtual ~HitExtractorSTRP(){}

  virtual HitExtractor::Hits hits( const TransientTrackingRecHitBuilder &ttrhBuilder, const edm::Event& , const edm::EventSetup& ) const;
  virtual HitExtractorSTRP * clone() const { return new HitExtractorSTRP(*this); }

  void useMatchedHits( const edm::InputTag & m, edm::ConsumesCollector& iC) { hasMatchedHits = true; theMatchedHits = iC.consumes<SiStripMatchedRecHit2DCollection>(m); }
  void useRPhiHits(    const edm::InputTag & m, edm::ConsumesCollector& iC) { hasRPhiHits    = true; theRPhiHits = iC.consumes<SiStripRecHit2DCollection>(m); }
  void useStereoHits(  const edm::InputTag & m, edm::ConsumesCollector& iC) { hasStereoHits = true; theStereoHits = iC.consumes<SiStripRecHit2DCollection>(m); }
  void useRingSelector(int minRing, int maxRing);
  void useSimpleRphiHitsCleaner(bool use) {hasSimpleRphiHitsCleaner = use;}

  void cleanedOfClusters( const TransientTrackingRecHitBuilder& ttrhBuilder, const edm::Event& ev, HitExtractor::Hits & hits, bool matched, unsigned int cleanFrom=0) const;

  bool skipThis(const TransientTrackingRecHitBuilder& ttrhBuilder, TransientTrackingRecHit::ConstRecHitPointer & ptr,edm::Handle<edm::ContainerMask<edmNew::DetSetVector<SiStripCluster> > > & stripClusterMask,
        TransientTrackingRecHit::ConstRecHitPointer & replaceMe) const;

  bool skipThis(OmniClusterRef const& clus, edm::Handle<edm::ContainerMask<edmNew::DetSetVector<SiStripCluster> > > & stripClusterMask) const;

  void project(const TransientTrackingRecHitBuilder& ttrhBuilder,
               TransientTrackingRecHit::ConstRecHitPointer & ptr,
           const SiStripRecHit2D * hit,
           TransientTrackingRecHit::ConstRecHitPointer & replaceMe) const;
  void setNoProjection() const {failProjection=true;};
  void setMinAbsZ(double minZToSet) {minAbsZ=minZToSet;}

  bool useRingSelector() const { return hasRingSelector; }
  std::tuple<int, int> getMinMaxRing() const { return std::make_tuple(theMinRing, theMaxRing); }
private:
  bool ringRange(int ring) const;

  typedef edm::ContainerMask<edmNew::DetSetVector<SiStripCluster> > SkipClustersCollection;
  void useSkipClusters_(const edm::InputTag & m, edm::ConsumesCollector& iC) override;
private:
  const GeomDetEnumerators::SubDetector theLayerSubDet;
  SeedingLayer::Side theSide;
  mutable const SeedingLayer * theSLayer;
  int theIdLayer;
  double minAbsZ;
  int theMinRing, theMaxRing;
  edm::EDGetTokenT<SkipClustersCollection> theSkipClusters;
  edm::EDGetTokenT<SiStripMatchedRecHit2DCollection> theMatchedHits;
  edm::EDGetTokenT<SiStripRecHit2DCollection> theRPhiHits;
  edm::EDGetTokenT<SiStripRecHit2DCollection> theStereoHits;
  bool hasMatchedHits;
  bool hasRPhiHits;
  bool hasStereoHits;
  bool hasRingSelector;
  bool hasSimpleRphiHitsCleaner;
  mutable bool failProjection;
};

}
#endif