CMS 3D CMS Logo

Public Member Functions | Private Member Functions | Private Attributes

TtSemiLepJetCombMVATrainer Class Reference

#include <TtSemiLepJetCombMVATrainer.h>

Inheritance diagram for TtSemiLepJetCombMVATrainer:
edm::EDAnalyzer edm::EDConsumerBase

List of all members.

Public Member Functions

 TtSemiLepJetCombMVATrainer (const edm::ParameterSet &)
 ~TtSemiLepJetCombMVATrainer ()

Private Member Functions

virtual void analyze (const edm::Event &evt, const edm::EventSetup &setup)
virtual void beginJob ()
virtual void endJob ()
WDecay::LeptonType readLeptonType (const std::string &str)

Private Attributes

edm::InputTag jets_
edm::InputTag leps_
WDecay::LeptonType leptonType_
edm::InputTag matching_
int maxNJets_
edm::InputTag mets_
PhysicsTools::MVAComputerCache mvaComputer
unsigned int nEvents [5]

Detailed Description

Definition at line 23 of file TtSemiLepJetCombMVATrainer.h.


Constructor & Destructor Documentation

TtSemiLepJetCombMVATrainer::TtSemiLepJetCombMVATrainer ( const edm::ParameterSet cfg) [explicit]

Definition at line 17 of file TtSemiLepJetCombMVATrainer.cc.

                                                                                :
  leps_      (cfg.getParameter<edm::InputTag>("leps"    )),
  jets_      (cfg.getParameter<edm::InputTag>("jets"    )),
  mets_      (cfg.getParameter<edm::InputTag>("mets"    )),
  matching_  (cfg.getParameter<edm::InputTag>("matching")),
  maxNJets_  (cfg.getParameter<int>          ("maxNJets")),
  leptonType_(readLeptonType(cfg.getParameter<std::string>("leptonType")))
{
}
TtSemiLepJetCombMVATrainer::~TtSemiLepJetCombMVATrainer ( )

Definition at line 27 of file TtSemiLepJetCombMVATrainer.cc.

{
}

Member Function Documentation

void TtSemiLepJetCombMVATrainer::analyze ( const edm::Event evt,
const edm::EventSetup setup 
) [private, virtual]

Implements edm::EDAnalyzer.

Definition at line 39 of file TtSemiLepJetCombMVATrainer.cc.

References PhysicsTools::Variable::ValueList::add(), evaluateTtSemiLepJetComb(), TtGenEvtProducer_cfi::genEvt, edm::Event::getByLabel(), i, fwrapper::jets, jets_, PhysicsTools::MVATrainer::kTargetId, leps_, EgammaValidation_Wenu_cff::leptons, leptonType_, TtSemiLepEvtPartons::LightQ, TtFullHadDaughter::LightQ, TtSemiLepEvtPartons::LightQBar, TtFullHadDaughter::LightQBar, matching_, maxNJets_, CaloMET_cfi::met, mets_, mvaComputer, nEvents, stdcomb::next_combination(), nPartons, HcalObjRepresent::setup(), PhysicsTools::MVAComputerCache::update(), and makeHLTPrescaleTable::values.

{
  mvaComputer.update<TtSemiLepJetCombMVARcd>("trainer", setup, "ttSemiLepJetCombMVA");

  // can occur in the last iteration when the 
  // MVATrainer is about to save the result
  if(!mvaComputer) return;

  nEvents[0]++;

  edm::Handle<TtGenEvent> genEvt;
  evt.getByLabel("genEvt", genEvt);

  edm::Handle< edm::View<reco::RecoCandidate> > leptons; 
  evt.getByLabel(leps_, leptons);

  // skip events with no appropriate lepton candidate
  if( leptons->empty() ) return;

  nEvents[1]++;

  const math::XYZTLorentzVector lepton = leptons->begin()->p4();

  edm::Handle< std::vector<pat::MET> > mets;
  evt.getByLabel(mets_, mets);

  // skip events with empty METs vector
  if( mets->empty() ) return;

  nEvents[2]++;

  const pat::MET *met = &(*mets)[0];

  edm::Handle< std::vector<pat::Jet> > jets;
  evt.getByLabel(jets_, jets);

  // skip events with less jets than partons
  unsigned int nPartons = 4;
  if( jets->size() < nPartons ) return;

  nEvents[3]++;

  edm::Handle< std::vector< std::vector<int> > > matchingHandle;
  std::vector<int> matching;
  // get jet-parton matching if signal channel
  if(genEvt->semiLeptonicChannel() == leptonType_) {
    evt.getByLabel(matching_, matchingHandle);
    if( matchingHandle->empty() )
      throw cms::Exception("EmptyProduct")
        << "Empty vector from jet-parton matching. This should not happen! \n";
    matching = *(matchingHandle->begin());
    if(matching.size() < nPartons) return;
    // skip events that were affected by the outlier 
    // rejection in the jet-parton matching
    for(unsigned int i = 0; i < matching.size(); ++i) {
      if(matching[i] == -3) continue; // -3: parton was chosen to be excluded from jet-parton matching
      if(matching[i] < 0 || matching[i] >= (int)jets->size())
        return;
    }
  }
  // use dummy for matching if not signal channel
  else
    for(unsigned int i = 0; i < nPartons; i++)
      matching.push_back( -1 );

  nEvents[4]++;

  // take into account indistinguishability of the two jets from the hadr. W decay,
  // reduces combinatorics by a factor of 2
  if(matching[TtSemiLepEvtPartons::LightQ] > matching[TtSemiLepEvtPartons::LightQBar]) {
    int iTemp = matching[TtSemiLepEvtPartons::LightQ];
    matching[TtSemiLepEvtPartons::LightQ] = matching[TtSemiLepEvtPartons::LightQBar];
    matching[TtSemiLepEvtPartons::LightQBar] = iTemp;
  }

  std::vector<int> jetIndices;
  for(unsigned int i=0; i<jets->size(); ++i) {
    if(maxNJets_ >= (int) nPartons && i == (unsigned int) maxNJets_) break;
    jetIndices.push_back(i);
  }

  std::vector<int> combi;
  for(unsigned int i = 0; i < nPartons; ++i) 
    combi.push_back(i);

  // use a ValueList to collect all variables for the MVAComputer
  PhysicsTools::Variable::ValueList values;

  // set target variable to false like for background
  // this is necessary when passing all combinations at once
  values.add( PhysicsTools::Variable::Value(PhysicsTools::MVATrainer::kTargetId, false) );

  do {
    // number of possible combinations from number of partons: e.g. 4! = 24
    for(unsigned int cnt = 0; cnt < TMath::Factorial( combi.size() ); ++cnt) {

      // take into account indistinguishability of the two jets from the hadr. W decay,
      // reduces combinatorics by a factor of 2
      if(combi[TtSemiLepEvtPartons::LightQ] < combi[TtSemiLepEvtPartons::LightQBar]) {
        
        TtSemiLepJetComb jetComb(*jets, combi, lepton, *met);

        bool trueCombi = false;
        // true combination only if signal channel
        // and in agreement with matching
        if(genEvt->semiLeptonicChannel()==leptonType_) {
          trueCombi = true;
          for(unsigned int i = 0; i < matching.size(); ++i) {
            if(combi[i]!=matching[i] && matching[i]!=-3) {
              trueCombi = false;
              break;
            }
          }
        }

        // feed MVA input variables for this jetComb into the ValueList
        values.add("target", trueCombi);
        evaluateTtSemiLepJetComb(values, jetComb);

      }

      next_permutation( combi.begin() , combi.end() );
    }
  }
  while(stdcomb::next_combination( jetIndices.begin(), jetIndices.end(), combi.begin(), combi.end() ));

  // pass MVA input variables for all jet combinations in this event
  // to the MVAComputer for training
  mvaComputer->eval( values );
}
void TtSemiLepJetCombMVATrainer::beginJob ( void  ) [private, virtual]

Reimplemented from edm::EDAnalyzer.

Definition at line 32 of file TtSemiLepJetCombMVATrainer.cc.

References i, and nEvents.

{
  for(unsigned int i = 0; i < 5; i++)
    nEvents[i] = 0;
}
void TtSemiLepJetCombMVATrainer::endJob ( void  ) [private, virtual]

Reimplemented from edm::EDAnalyzer.

Definition at line 171 of file TtSemiLepJetCombMVATrainer.cc.

References create_public_lumi_plots::log, and nEvents.

{
  edm::LogInfo log("TtSemiLepJetCombMVATrainer");
  log << "Number of events... \n"
      << "...passed to the trainer                   : " << std::setw(7) << nEvents[0] << "\n"
      << "...rejected since no lepton candidate      : " << std::setw(7) << nEvents[0]-nEvents[1] << "\n"
      << "...rejected since no MET object            : " << std::setw(7) << nEvents[1]-nEvents[2] << "\n"
      << "...rejected since not enough jets          : " << std::setw(7) << nEvents[2]-nEvents[3] << "\n"
      << "...rejected due to bad jet-parton matching : " << std::setw(7) << nEvents[3]-nEvents[4] << "\n"
      << "...accepted for training                   : " << std::setw(7) << nEvents[4] << "\n";
}
WDecay::LeptonType TtSemiLepJetCombMVATrainer::readLeptonType ( const std::string &  str) [private]

Definition at line 184 of file TtSemiLepJetCombMVATrainer.cc.

References Exception, WDecay::kElec, WDecay::kMuon, and WDecay::kTau.

{
  if     (str == "kElec") return WDecay::kElec;
  else if(str == "kMuon") return WDecay::kMuon;
  else if(str == "kTau" ) return WDecay::kTau;
  else throw cms::Exception("Configuration")
    << "Chosen leptonType is not supported: " << str << "\n";
}

Member Data Documentation

Definition at line 39 of file TtSemiLepJetCombMVATrainer.h.

Referenced by analyze().

Definition at line 38 of file TtSemiLepJetCombMVATrainer.h.

Referenced by analyze().

Definition at line 45 of file TtSemiLepJetCombMVATrainer.h.

Referenced by analyze().

Definition at line 41 of file TtSemiLepJetCombMVATrainer.h.

Referenced by analyze().

Definition at line 43 of file TtSemiLepJetCombMVATrainer.h.

Referenced by analyze().

Definition at line 40 of file TtSemiLepJetCombMVATrainer.h.

Referenced by analyze().

Definition at line 47 of file TtSemiLepJetCombMVATrainer.h.

Referenced by analyze().

unsigned int TtSemiLepJetCombMVATrainer::nEvents[5] [private]

Definition at line 49 of file TtSemiLepJetCombMVATrainer.h.

Referenced by analyze(), beginJob(), and endJob().