ZSPJPTJetCorrector.cc

Go to the documentation of this file.

//
// Original Author:  Olga Kodolova, September 2007
//
// ZSPJPT Jet Corrector
//
#include "ZSPJPTJetCorrector.h"
#include "SimpleZSPJPTJetCorrector.h"
//#include "CondFormats/JetMETObjects/interface/SimpleL1OffsetCorrector.h"

#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ParameterSet/interface/FileInPath.h"
#include "DataFormats/JetReco/interface/Jet.h"
#include "DataFormats/Provenance/interface/LuminosityBlockID.h"

using namespace std;


ZSPJPTJetCorrector::ZSPJPTJetCorrector (const edm::ParameterSet& fConfig) {

 iPU = fConfig.getParameter<int>("PU");
 fixedPU = fConfig.getParameter<int>("FixedPU"); 

 if( iPU >= 0 || fixedPU > 0 ) {
 theFilesL1Offset = fConfig.getParameter <vector<string> > ("tagNameOffset");
 for(vector<string>::iterator it=theFilesL1Offset.begin(); it != theFilesL1Offset.end(); it++) {
   std::string file="CondFormats/JetMETObjects/data/"+(*it)+".txt";
   edm::FileInPath f2(file);
   mSimpleCorrectorOffset.push_back(new SimpleZSPJPTJetCorrector (f2.fullPath()));
 }
 }

 theFilesZSP = fConfig.getParameter <vector<string> > ("tagName");
 for(vector<string>::iterator it=theFilesZSP.begin(); it != theFilesZSP.end(); it++) {
   std::string file="CondFormats/JetMETObjects/data/"+(*it)+".txt";
   edm::FileInPath f1(file);
   mSimpleCorrector.push_back(new SimpleZSPJPTJetCorrector (f1.fullPath()));
 }
//   cout<<" Size of correctors "<<mSimpleCorrector.size()<<" "<<mSimpleCorrectorOffset.size()<<endl;
}

ZSPJPTJetCorrector::~ZSPJPTJetCorrector () {
} 

double ZSPJPTJetCorrector::correction( const reco::Jet& fJet, const edm::Event& iEvent, const edm::EventSetup& iSetup) const
{
   double b=1.;
   int nPU = 0;
   if(iPU > 0) {
// Look for the Lumi section
//     LuminosityBlock lbID = iEvent.getLuminosityBlock();
//       cout<<" Not implemented yet "<<iEvent.run()<<endl;
       nPU=setPU();
   } else { if(iPU==0) nPU=setPU(); } 

  double a = mSimpleCorrector[nPU]->correctionPtEtaPhiE (fJet.p4().Pt(), fJet.p4().Eta(), fJet.p4().Phi(),fJet.p4().E());

  if(iPU >= 0) {
    if(!mSimpleCorrectorOffset.empty()) {
     b = mSimpleCorrectorOffset[nPU]->correctionPUEtEtaPhiP (fJet.p4().Pt(), fJet.p4().Eta(), fJet.p4().Phi(),fJet.p4().E());
   } 
  }
   double c = a * b; 
   return c;
}