#ifndef PhysicsTools_RecoAlgos_MassKinFitterCandProducer_h
#define PhysicsTools_RecoAlgos_MassKinFitterCandProducer_h
/* \class MassKinFitterCandProducer
 *
 * \author Luca Lista, INFN
 *
 */
#include "FWCore/Utilities/interface/InputTag.h"
#include "FWCore/Framework/interface/stream/EDProducer.h"
#include "PhysicsTools/RecoUtils/interface/CandMassKinFitter.h"

#include "DataFormats/Candidate/interface/Candidate.h"

class MassKinFitterCandProducer : public edm::stream::EDProducer<> {
public:
  explicit MassKinFitterCandProducer(const edm::ParameterSet &, CandMassKinFitter * = nullptr);

private:
  edm::EDGetTokenT<reco::CandidateCollection> srcToken_;
  std::unique_ptr<CandMassKinFitter> fitter_;
  void produce(edm::Event &, const edm::EventSetup &) override;
};

#endif
