RecoAlgos/interface/MassKinFitterCandProducer.h

0001 #ifndef PhysicsTools_RecoAlgos_MassKinFitterCandProducer_h
0002 #define PhysicsTools_RecoAlgos_MassKinFitterCandProducer_h
0003 /* \class MassKinFitterCandProducer

0004  *

0005  * \author Luca Lista, INFN

0006  *

0007  */
0008 #include "FWCore/Utilities/interface/InputTag.h"
0009 #include "FWCore/Framework/interface/stream/EDProducer.h"
0010 #include "PhysicsTools/RecoUtils/interface/CandMassKinFitter.h"
0011
0012 #include "DataFormats/Candidate/interface/Candidate.h"
0013
0014 class MassKinFitterCandProducer : public edm::stream::EDProducer<> {
0015 public:
0016   explicit MassKinFitterCandProducer(const edm::ParameterSet &, CandMassKinFitter * = nullptr);
0017
0018 private:
0019   edm::EDGetTokenT<reco::CandidateCollection> srcToken_;
0020   std::unique_ptr<CandMassKinFitter> fitter_;
0021   void produce(edm::Event &, const edm::EventSetup &) override;
0022 };
0023
0024 #endif