TrackRefitter.h

Go to the documentation of this file.

#ifndef TrackRefitter_h
#define TrackRefitter_h

#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "RecoTracker/TrackProducer/interface/KfTrackProducerBase.h"
#include "RecoTracker/TrackProducer/interface/TrackProducerAlgorithm.h"

class TrackRefitter : public KfTrackProducerBase, public edm::stream::EDProducer<> {
public:
  explicit TrackRefitter(const edm::ParameterSet& iConfig);

  void produce(edm::Event&, const edm::EventSetup&) override;

private:
  TrackProducerAlgorithm<reco::Track> theAlgo;
  enum Constraint { none, momentum, vertex, trackParameters };
  Constraint constraint_;
  edm::EDGetToken trkconstrcoll_;

  const edm::ESGetToken<TrackerTopology, TrackerTopologyRcd> ttopoToken_;
};

#endif