---

TtEvent.h

Go to the documentation of this file.

#ifndef TopObjects_TtEvent_h
#define TopObjects_TtEvent_h

#include <vector>
#include <string>

#include "DataFormats/Common/interface/Handle.h"
#include "DataFormats/Common/interface/RefProd.h"
#include "DataFormats/Candidate/interface/CompositeCandidate.h"
#include "AnalysisDataFormats/TopObjects/interface/TtGenEvent.h"

// ----------------------------------------------------------------------
// common base class for: 
//
//  * TtSemiLeptonicEvent
//  * TtFullLeptonicEvent
//  * TtFullHadronicEvent (still to be implemented)
//
//  the structure holds information on the leptonic decay channels, 
//  all event hypotheses of different classes (user defined during
//  production) and a reference to the TtGenEvent (if available) 
//  and provides access and administration
// ----------------------------------------------------------------------

class TtEvent {

 public:

  enum HypoClassKey {kGeom, kWMassMaxSumPt, kMaxSumPtWMass, kGenMatch, kMVADisc, kKinFit};
  typedef std::pair<reco::CompositeCandidate, std::vector<int> > HypoCombPair;
  
 public:

  TtEvent(){};
  virtual ~TtEvent(){};

  std::pair<WDecay::LeptonType, WDecay::LeptonType> lepDecays() const { return lepDecays_; }
  const reco::CompositeCandidate& eventHypo(const HypoClassKey& key, const unsigned& cmb=0) const { return (evtHyp_.find(key)->second)[cmb].first; };
  const edm::RefProd<TtGenEvent>& genEvent() const { return genEvt_; };

  bool isHypoClassAvailable(const HypoClassKey& key) const { return (evtHyp_.find(key)!=evtHyp_.end()); };
  bool isHypoAvailable(const HypoClassKey& key, const unsigned& cmb=0) const { return isHypoClassAvailable(key) ? (cmb<evtHyp_.find(key)->second.size()) : false; };
  bool isHypoValid(const HypoClassKey& key, const unsigned& cmb=0) const { return isHypoAvailable(key,cmb) ? !eventHypo(key,cmb).roles().empty() : false; };
  unsigned int numberOfAvailableHypoClasses() const { return evtHyp_.size();};
  unsigned int numberOfAvailableHypos(const HypoClassKey& key) const { return isHypoAvailable(key) ? evtHyp_.find(key)->second.size() : 0;};
  std::vector<int> jetLepComb(const HypoClassKey& key, const unsigned& cmb=0) const { return (evtHyp_.find(key)->second)[cmb].second; };
  double genMatchSumPt(const unsigned& cmb=0) const { return (cmb<genMatchSumPt_.size() ? genMatchSumPt_[cmb] : -1.); };
  double genMatchSumDR(const unsigned& cmb=0) const { return (cmb<genMatchSumDR_.size() ? genMatchSumDR_[cmb] : -1.); };
  std::string mvaMethod() const { return mvaMethod_; }
  double mvaDisc(const unsigned& cmb=0) const { return (cmb<mvaDisc_.size() ? mvaDisc_[cmb] : -1.); }
  double fitChi2(const unsigned& cmb=0) const { return (cmb<fitChi2_.size() ? fitChi2_[cmb] : -1.); }
  double fitProb(const unsigned& cmb=0) const { return (cmb<fitProb_.size() ? fitProb_[cmb] : -1.); }
  int correspondingHypo(const HypoClassKey& key1, const unsigned& hyp1, const HypoClassKey& key2) const;

  void setLepDecays(const WDecay::LeptonType& lepDecTop1, const WDecay::LeptonType& lepDecTop2) { lepDecays_=std::make_pair(lepDecTop1, lepDecTop2); };
  void setGenEvent(const edm::Handle<TtGenEvent>& evt) { genEvt_=edm::RefProd<TtGenEvent>(evt); };
  void addEventHypo(const HypoClassKey& key, HypoCombPair hyp) { evtHyp_[key].push_back(hyp); };
  void setGenMatchSumPt(const std::vector<double>& val) {genMatchSumPt_=val;};
  void setGenMatchSumDR(const std::vector<double>& val) {genMatchSumDR_=val;};
  void setMvaMethod(const std::string& name) { mvaMethod_=name; };
  void setMvaDiscriminators(const std::vector<double>& val) { mvaDisc_=val; };
  void setFitChi2(const std::vector<double>& val) { fitChi2_=val; };
  void setFitProb(const std::vector<double>& val) { fitProb_=val; };

 protected:

   std::pair<WDecay::LeptonType, WDecay::LeptonType> lepDecays_;
  edm::RefProd<TtGenEvent> genEvt_;
  std::map<HypoClassKey, std::vector<HypoCombPair> > evtHyp_;
  
  std::vector<double> fitChi2_;        
  std::vector<double> fitProb_;        
  std::vector<double> genMatchSumPt_;  
  std::vector<double> genMatchSumDR_;   
  std::string mvaMethod_;               
  std::vector<double> mvaDisc_;         
};

#endif

---