HitExtractor.h

Go to the documentation of this file.

#ifndef RecoTracker_TkSeedingLayers_HitExtractor_H
#define RecoTracker_TkSeedingLayers_HitExtractor_H

#include <vector>
#include <iterator>
#include "RecoTracker/TransientTrackingRecHit/interface/TkTransientTrackingRecHitBuilder.h"
#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/Utilities/interface/EDGetToken.h"
#include "DataFormats/Common/interface/ContainerMask.h"

#include "DataFormats/TrackerRecHit2D/interface/BaseTrackerRecHit.h"
#include "DataFormats/TrackingRecHit/interface/mayown_ptr.h"
#include <limits>

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

namespace ctfseeding {

  class HitExtractor {
  public:
    using TkHit = BaseTrackerRecHit;
    using TkHitRef = BaseTrackerRecHit const&;
    using HitPointer = mayown_ptr<BaseTrackerRecHit>;
    using Hits = std::vector<HitPointer>;

    virtual ~HitExtractor() {}
    HitExtractor() {}

    virtual Hits hits(const TkTransientTrackingRecHitBuilder& ttrhBuilder,
                      const edm::Event&,
                      const edm::EventSetup&) const = 0;
    virtual HitExtractor* clone() const = 0;

    //skip clusters
    void useSkipClusters(const edm::InputTag& m, edm::ConsumesCollector& iC) {
      skipClusters = true;
      maskCluster = true;
      useSkipClusters_(m, iC);
    }
    bool skipClusters = false;
    bool filterCluster = false;
    bool maskCluster = false;
    float minGoodCharge = 0;

  protected:
    virtual void useSkipClusters_(const edm::InputTag& m, edm::ConsumesCollector& iC) = 0;
  };

  template <typename DSTV, typename A, typename B>
  inline void range2SeedingHits(DSTV const& dstv, HitExtractor::Hits& v, std::pair<A, B> const& sel) {
    typename DSTV::Range range = dstv.equal_range(sel.first, sel.second);
    size_t ts = v.size();
    for (typename DSTV::const_iterator id = range.first; id != range.second; id++)
      ts += std::distance((*id).begin(), (*id).end());
    v.reserve(ts);
    for (typename DSTV::const_iterator id = range.first; id != range.second; id++) {
      for (auto const& h : (*id))
        v.emplace_back(h);
    }
  }
}  // namespace ctfseeding
#endif