TtSemiLepJetCombMVATrainer.h

Go to the documentation of this file.

#ifndef TtSemiLepJetCombMVATrainer_h
#define TtSemiLepJetCombMVATrainer_h

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/EDAnalyzer.h"
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "AnalysisDataFormats/TopObjects/interface/TopGenEvent.h"

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

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

class TtSemiLepJetCombMVATrainer : public edm::EDAnalyzer {
  
 public:
  
  explicit TtSemiLepJetCombMVATrainer(const edm::ParameterSet&);
  ~TtSemiLepJetCombMVATrainer();
  
 private:
  
  virtual void beginJob();
  virtual void analyze(const edm::Event& evt, const edm::EventSetup& setup);
  virtual void endJob();

  WDecay::LeptonType readLeptonType(const std::string& str);

  edm::InputTag leps_;
  edm::InputTag jets_;
  edm::InputTag mets_;
  edm::InputTag matching_;

  int maxNJets_;
  
  WDecay::LeptonType leptonType_;

  PhysicsTools::MVAComputerCache mvaComputer;

  unsigned int nEvents[5];
};

#endif