PileupJetIdAlgo.h

Go to the documentation of this file.

//--------------------------------------------------------------------------------------------------
//
// PileupJetIdAlgo
//
// Author: P. Musella, P. Harris
//--------------------------------------------------------------------------------------------------

#ifndef RecoJets_JetProducers_plugins_PileupJetIdAlgo_h
#define RecoJets_JetProducers_plugins_PileupJetIdAlgo_h

#include "DataFormats/JetReco/interface/Jet.h"
#include "DataFormats/JetReco/interface/PFJetCollection.h"
#include "DataFormats/VertexReco/interface/VertexFwd.h"
#include "DataFormats/PatCandidates/interface/Jet.h"
#include "DataFormats/PatCandidates/interface/PackedCandidate.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "TMVA/Tools.h"
#include "TMVA/Reader.h"
#include "TMVA/Tools.h"
#include "TMVA/Reader.h"

#include "DataFormats/JetReco/interface/PileupJetIdentifier.h"
#include "CondFormats/EgammaObjects/interface/GBRForest.h"

// ----------------------------------------------------------------------------------------------------
class PileupJetIdAlgo {
public:
    enum version_t { USER=-1, PHILv0=0 };
    
    PileupJetIdAlgo(int version=PHILv0, const std::string & tmvaWeight="", const std::string & tmvaMethod="", 
            Float_t impactParTkThreshod_=1., const std::vector<std::string> & tmvaVariables= std::vector<std::string>(), bool runMvas=true);
    PileupJetIdAlgo(const edm::ParameterSet & ps, bool runMvas);
    ~PileupJetIdAlgo(); 
    
    PileupJetIdentifier computeIdVariables(const reco::Jet * jet, 
                           float jec, const reco::Vertex *, const reco::VertexCollection &, double rho);

    void set(const PileupJetIdentifier &);
        std::unique_ptr<const GBRForest> getMVA(const std::vector<std::string> &, const std::string &);
        float getMVAval(const std::vector<std::string> &, const std::unique_ptr<const GBRForest> &);
    PileupJetIdentifier computeMva();
    const std::string method() const { return tmvaMethod_; }
    
    std::string dumpVariables() const;

    typedef std::map<std::string,std::pair<float *,float> > variables_list_t;

    std::pair<int,int> getJetIdKey(float jetPt, float jetEta);
    int computeCutIDflag(float betaStarClassic,float dR2Mean,float nvtx, float jetPt, float jetEta);
    int computeIDflag   (float mva, float jetPt, float jetEta);
    int computeIDflag   (float mva,int ptId,int etaId);

    const variables_list_t & getVariables() const { return variables_; };

protected:

    void setup(); 
    void runMva(); 
    void bookReader();  
    void resetVariables();
    void initVariables();

    
    PileupJetIdentifier internalId_;
    variables_list_t variables_;

    std::unique_ptr<const GBRForest> reader_;
        std::vector<std::unique_ptr<const GBRForest>> etaReader_;
    std::string tmvaWeights_, tmvaMethod_;
        std::vector<std::string> tmvaEtaWeights_;
    std::vector<std::string> tmvaVariables_;
        std::vector<std::vector<std::string>> tmvaEtaVariables_;
    std::vector<std::string> tmvaSpectators_;
    std::map<std::string,std::string> tmvaNames_;
    
    int   version_;
    float impactParTkThreshod_;
    bool    cutBased_; 
    bool    etaBinnedWeights_;
        int   nEtaBins_;
        std::vector<double> jEtaMin_;
        std::vector<double> jEtaMax_;
    bool runMvas_;
    float mvacut_     [3][4][4]; //Keep the array fixed
    float rmsCut_     [3][4][4]; //Keep the array fixed
    float betaStarCut_[3][4][4]; //Keep the array fixed
};

#endif