ObjectResolutionCalc.h

Go to the documentation of this file.

//
//

#ifndef PhysicsTools_PatUtils_ObjectResolutionCalc_h
#define PhysicsTools_PatUtils_ObjectResolutionCalc_h

#include "DataFormats/PatCandidates/interface/Electron.h"
#include "DataFormats/PatCandidates/interface/Muon.h"
#include "DataFormats/PatCandidates/interface/Tau.h"
#include "DataFormats/PatCandidates/interface/MET.h"
#include "DataFormats/PatCandidates/interface/Jet.h"

#include "TF1.h"
#include "TFile.h"
#include "TMultiLayerPerceptron.h"
#include "TString.h"

namespace pat {

  class ObjectResolutionCalc {
  public:
    ObjectResolutionCalc();
    ObjectResolutionCalc(const TString& resopath, bool useNN);
    ~ObjectResolutionCalc();

    float obsRes(int obs, int eta, float eT);
    int etaBin(float eta);

#ifdef OBSOLETE
    void operator()(Electron& obj);
    void operator()(Muon& obj);
    void operator()(Tau& obj);
    void operator()(Jet& obj);
    void operator()(MET& obj);
#else
    // WORKAROUND
    template <typename T>
    void operator()(T& obj) {}
#endif

  private:
    TFile* resoFile_;
    std::vector<float> etaBinVals_;
    TF1 fResVsEt_[10][10];
    TMultiLayerPerceptron* network_[10];
    bool useNN_;
  };

}  // namespace pat

#endif