MuonMVAEstimator.h

Go to the documentation of this file.

//--------------------------------------------------------------------------------------------------
// $Id $
//
// MuonMVAEstimator
//
// Helper Class for applying MVA electron ID selection
//
// Authors: S.Xie
//--------------------------------------------------------------------------------------------------



#ifndef MuonMVAEstimator_H
#define MuonMVAEstimator_H

#include "MuonEffectiveArea.h"
#include <vector>
#include <TROOT.h>
#include "TMVA/Factory.h"
#include "TMVA/Tools.h"
#include "TMVA/Reader.h"
#include "DQM/PhysicsHWW/interface/HWW.h"

class MuonMVAEstimator{
  public:
    MuonMVAEstimator();
    ~MuonMVAEstimator(); 
  
    enum MVAType {
      kIDIsoRingsCombined = 0,  
      kIsoRings,
      kIsoDeltaR,
      kID
    };
  
    void     initialize( std::string methodName,
                         std::string weightsfile,
                         MuonMVAEstimator::MVAType type);
    void     initialize( std::string methodName,
                         MuonMVAEstimator::MVAType type,
                         Bool_t useBinnedVersion,
                         std::vector<std::string> weightsfiles );
    
    Bool_t   isInitialized() const { return fisInitialized; }
    UInt_t   GetMVABin( double eta, double pt, 
                        Bool_t isGlobal, Bool_t isTrackerMuon) const;
    
    void SetPrintMVADebug(bool b) { fPrintMVADebug = b; }
    
    void bindVariables();
    
    Double_t mvaValue_Iso(
            Double_t Pt,
            Double_t Eta,
            Bool_t isGlobalMuon,
            Bool_t isTrackerMuon,
            Double_t Rho,
            MuonEffectiveArea::MuonEffectiveAreaTarget EATarget,
            Double_t ChargedIso_DR0p0To0p1,
            Double_t ChargedIso_DR0p1To0p2,
            Double_t ChargedIso_DR0p2To0p3,
            Double_t ChargedIso_DR0p3To0p4,
            Double_t ChargedIso_DR0p4To0p5,
            Double_t GammaIso_DR0p0To0p1,
            Double_t GammaIso_DR0p1To0p2,
            Double_t GammaIso_DR0p2To0p3,
            Double_t GammaIso_DR0p3To0p4,
            Double_t GammaIso_DR0p4To0p5,
            Double_t NeutralHadronIso_DR0p0To0p1,
            Double_t NeutralHadronIso_DR0p1To0p2,
            Double_t NeutralHadronIso_DR0p2To0p3,
            Double_t NeutralHadronIso_DR0p3To0p4,
            Double_t NeutralHadronIso_DR0p4To0p5,
            Bool_t printDebug = kFALSE );

    Double_t mvaValueIso(HWW&, Int_t imu, Double_t rho, MuonEffectiveArea::MuonEffectiveAreaTarget EATarget,
                          std::vector<Int_t> IdentifiedEle, std::vector<Int_t> IdentifiedMu, Bool_t printDebug ); 

    
  private:
    std::vector<TMVA::Reader*> fTMVAReader;
    std::string                fMethodname;
    Bool_t                     fisInitialized;
    Bool_t                     fPrintMVADebug;
    MVAType                    fMVAType;
    Bool_t                     fUseBinnedVersion;
    UInt_t                     fNMVABins;
    
    // spectator categorization variables
    Float_t                    fMVAVar_MuEta;
    Float_t                    fMVAVar_MuPt;
    Float_t                    fMVAVar_MuTypeBits;

    //ID variables
    Float_t                    fMVAVar_MuTkNchi2;
    Float_t                    fMVAVar_MuGlobalNchi2;
    Float_t                    fMVAVar_MuNValidHits;
    Float_t                    fMVAVar_MuNTrackerHits;
    Float_t                    fMVAVar_MuNPixelHits;
    Float_t                    fMVAVar_MuNMatches;
    Float_t                    fMVAVar_MuD0;
    Float_t                    fMVAVar_MuIP3d;
    Float_t                    fMVAVar_MuIP3dSig;
    Float_t                    fMVAVar_MuTrkKink;
    Float_t                    fMVAVar_MuSegmentCompatibility;
    Float_t                    fMVAVar_MuCaloCompatibility;
    Float_t                    fMVAVar_MuHadEnergy;
    Float_t                    fMVAVar_MuEmEnergy;
    Float_t                    fMVAVar_MuHadS9Energy;
    Float_t                    fMVAVar_MuEmS9Energy;

    //isolation
    Float_t                   fMVAVar_ChargedIso_DR0p0To0p1;
    Float_t                   fMVAVar_ChargedIso_DR0p1To0p2;
    Float_t                   fMVAVar_ChargedIso_DR0p2To0p3;
    Float_t                   fMVAVar_ChargedIso_DR0p3To0p4;
    Float_t                   fMVAVar_ChargedIso_DR0p4To0p5;
    Float_t                   fMVAVar_GammaIso_DR0p0To0p1;
    Float_t                   fMVAVar_GammaIso_DR0p1To0p2;
    Float_t                   fMVAVar_GammaIso_DR0p2To0p3;
    Float_t                   fMVAVar_GammaIso_DR0p3To0p4;
    Float_t                   fMVAVar_GammaIso_DR0p4To0p5;
    Float_t                   fMVAVar_NeutralHadronIso_DR0p0To0p1;
    Float_t                   fMVAVar_NeutralHadronIso_DR0p1To0p2;
    Float_t                   fMVAVar_NeutralHadronIso_DR0p2To0p3;
    Float_t                   fMVAVar_NeutralHadronIso_DR0p3To0p4;
    Float_t                   fMVAVar_NeutralHadronIso_DR0p4To0p5;
    
    
    // isolation variables II
    Float_t                   fMVAVar_MuRelIsoPFCharged;
    Float_t                   fMVAVar_MuRelIsoPFNeutral;
    Float_t                   fMVAVar_MuRelIsoPFPhotons;
    Float_t                   fMVAVar_MuDeltaRMean;
    Float_t                   fMVAVar_MuDeltaRSum;
    Float_t                   fMVAVar_MuDensity;

};

#endif