TtSemiLepJetCombMVAComputer.h

Go to the documentation of this file.

#ifndef TtSemiLepJetCombMVAComputer_h
#define TtSemiLepJetCombMVAComputer_h

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EDProducer.h"

#include "PhysicsTools/MVAComputer/interface/HelperMacros.h"
#include "PhysicsTools/MVAComputer/interface/MVAComputerCache.h"

#include "AnalysisDataFormats/TopObjects/interface/TtSemiLepEvtPartons.h"
#include "DataFormats/PatCandidates/interface/Jet.h"
#include "DataFormats/RecoCandidate/interface/RecoCandidate.h"

#include "TopQuarkAnalysis/TopJetCombination/interface/TtSemiLepJetCombEval.h"

#ifndef TtSemiLepJetCombMVARcd_defined  // to avoid conflicts with the TtSemiLepJetCombMVATrainer
#define TtSemiLepJetCombMVARcd_defined
MVA_COMPUTER_CONTAINER_DEFINE(TtSemiLepJetCombMVA);  // defines TtSemiLepJetCombMVARcd
#endif

class TtSemiLepJetCombMVAComputer : public edm::EDProducer {

 public:

  explicit TtSemiLepJetCombMVAComputer(const edm::ParameterSet&);
  ~TtSemiLepJetCombMVAComputer() override;

 private:

  void beginJob() override;
  void produce(edm::Event& evt, const edm::EventSetup& setup) override;
  void endJob() override;

  edm::EDGetTokenT< edm::View<reco::RecoCandidate>> lepsToken_;
  edm::EDGetTokenT< std::vector<pat::Jet> > jetsToken_;
  edm::EDGetTokenT< std::vector<pat::MET> > metsToken_;

  int maxNJets_;
  int maxNComb_;

  PhysicsTools::MVAComputerCache mvaComputer;

};

#endif