CommonTools/RecoAlgos/interface/RecoTrackSelector.h - master - githubmirror/cmssw

CommonTools/RecoAlgos/interface/RecoTrackSelector.h

41 строка1 KB

Cms Build

Clang-Format

18 июн 2019, 07:17

18 июн 2019, 07:17bcd6d63

#ifndef RecoSelectors_RecoTrackSelector_h
#define RecoSelectors_RecoTrackSelector_h
/* \class RecoTrackSelector
 *
 * \author Giuseppe Cerati, INFN
 *
 */
#include "CommonTools/RecoAlgos/interface/RecoTrackSelectorBase.h"

class RecoTrackSelector : public RecoTrackSelectorBase {
public:
  typedef reco::TrackRef reference_type;
  typedef reco::TrackCollection collection;
  typedef std::vector<const reco::Track*> container;
  typedef container::const_iterator const_iterator;

  /// Constructors
  RecoTrackSelector() {}
  RecoTrackSelector(const edm::ParameterSet& cfg, edm::ConsumesCollector&& iC) : RecoTrackSelectorBase(cfg, iC) {}

  const_iterator begin() const { return selected_.begin(); }
  const_iterator end() const { return selected_.end(); }

  void select(const edm::Handle<collection>& c, const edm::Event& event, const edm::EventSetup& es) {
    init(event, es);
    selected_.clear();
    for (reco::TrackCollection::const_iterator trk = c->begin(); trk != c->end(); ++trk) {
      reference_type tkref(c, std::distance(c->begin(), trk));
      if (operator()(*tkref)) {
        selected_.push_back(&*trk);
      }
    }
  }

  size_t size() const { return selected_.size(); }

private:
  container selected_;
};

#endif