GsfTrackRefitter.h

Go to the documentation of this file.

#ifndef GsfTrackRefitter_h
#define GsfTrackRefitter_h

#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "RecoTracker/TrackProducer/interface/GsfTrackProducerBase.h"
#include "RecoTracker/TrackProducer/interface/TrackProducerAlgorithm.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "TrackingTools/GsfTracking/interface/GsfTrackConstraintAssociation.h"

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

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

private:
  TrackProducerAlgorithm<reco::GsfTrack> theAlgo;
  enum Constraint {
    none,
    //          momentum,
    vertex
  };
  Constraint constraint_;
  edm::EDGetTokenT<GsfTrackVtxConstraintAssociationCollection> gsfTrackVtxConstraintTag_;
};

#endif