#include <SimpleJetCorrector.h>

Public Member Functions
float	correction (const std::vector< float > &fX, const std::vector< float > &fY) const

const JetCorrectorParameters &	parameters () const

void	setInterpolation (bool fInterpolation)

	SimpleJetCorrector ()

	SimpleJetCorrector (const std::string &fDataFile, const std::string &fOption="")

	SimpleJetCorrector (const JetCorrectorParameters &fParameters)

	~SimpleJetCorrector ()

Private Member Functions
float	correctionBin (unsigned fBin, const std::vector< float > &fY) const

unsigned	findInvertVar ()

float	invert (const Double_t args, const Double_t params) const

SimpleJetCorrector &	operator= (const SimpleJetCorrector &)

void	setFuncParameters ()

	SimpleJetCorrector (const SimpleJetCorrector &)

Private Attributes
bool	mDoInterpolation

TFormula	mFunc

unsigned	mInvertVar

JetCorrectorParameters	mParameters

Detailed Description

Definition at line 15 of file SimpleJetCorrector.h.

Constructor & Destructor Documentation

SimpleJetCorrector::SimpleJetCorrector ( )

Definition at line 11 of file SimpleJetCorrector.cc.

References mDoInterpolation, and mInvertVar.

 {
   mDoInterpolation = false;
   mInvertVar       = 9999;
 }

SimpleJetCorrector::SimpleJetCorrector	(	const std::string &	fDataFile,
		const std::string &	fOption = `""`
	)

Definition at line 20 of file SimpleJetCorrector.cc.

References JetCorrectorParameters::definitions(), findInvertVar(), JetCorrectorParameters::Definitions::isResponse(), mDoInterpolation, mInvertVar, and mParameters.

                                                                                           :
   mParameters(fDataFile,fOption),
   mFunc("function",((mParameters.definitions()).formula()).c_str())
 {
   mDoInterpolation = false;
   if (mParameters.definitions().isResponse())
     mInvertVar = findInvertVar();
 }

SimpleJetCorrector::SimpleJetCorrector ( const JetCorrectorParameters & fParameters )

Definition at line 32 of file SimpleJetCorrector.cc.

References JetCorrectorParameters::definitions(), findInvertVar(), JetCorrectorParameters::Definitions::isResponse(), mDoInterpolation, mInvertVar, and mParameters.

                                                                                :
   mParameters(fParameters),
   mFunc("function",((mParameters.definitions()).formula()).c_str())
 {
   mDoInterpolation = false;
   if (mParameters.definitions().isResponse())
     mInvertVar = findInvertVar();
 }

SimpleJetCorrector::~SimpleJetCorrector ( )

Definition at line 43 of file SimpleJetCorrector.cc.

44 {

45 }

SimpleJetCorrector::SimpleJetCorrector ( const SimpleJetCorrector & )

private

Member Function Documentation

float SimpleJetCorrector::correction	(	const std::vector< float > &	fX,
		const std::vector< float > &	fY
	)		const

Definition at line 50 of file SimpleJetCorrector.cc.

References newFWLiteAna::bin, JetCorrectorParameters::binIndex(), correctionBin(), JetCorrectorParameters::definitions(), i, mDoInterpolation, mParameters, JetCorrectorParameters::Definitions::nBinVar(), JetCorrectorParameters::neighbourBin(), JetCorrectorParameters::record(), query::result, tmp, and JetCorrectorParameters::Record::xMiddle().

 {
   float result = 1.;
   float tmp    = 0.0;
   float cor    = 0.0;
   int bin = mParameters.binIndex(fX);
   if (bin<0)
     return result;
   if (!mDoInterpolation)
     result = correctionBin(bin,fY);
   else
     {
       for(unsigned i=0;i<mParameters.definitions().nBinVar();i++)
         {
           float xMiddle[3];
           float xValue[3];
           int prevBin = mParameters.neighbourBin((unsigned)bin,i,false);
           int nextBin = mParameters.neighbourBin((unsigned)bin,i,true);
           if (prevBin>=0 && nextBin>=0)
             {
               xMiddle[0] = mParameters.record(prevBin).xMiddle(i);
               xMiddle[1] = mParameters.record(bin).xMiddle(i);
               xMiddle[2] = mParameters.record(nextBin).xMiddle(i);
               xValue[0]  = correctionBin(prevBin,fY);
               xValue[1]  = correctionBin(bin,fY);
               xValue[2]  = correctionBin(nextBin,fY);
               cor = quadraticInterpolation(fX[i],xMiddle,xValue);
               tmp+=cor;
             }
           else
             {
               cor = correctionBin(bin,fY);
               tmp+=cor;
             }
         }
       result = tmp/mParameters.definitions().nBinVar();
     }
   return result;
 }

float SimpleJetCorrector::correctionBin	(	unsigned	fBin,
		const std::vector< float > &	fY
	)		const

private

Definition at line 92 of file SimpleJetCorrector.cc.

References JetCorrectorParameters::definitions(), i, invert(), JetCorrectorParameters::Definitions::isResponse(), mFunc, mParameters, N, JetCorrectorParameters::Record::parameters(), JetCorrectorParameters::record(), query::result, JetCorrectorParameters::size(), tmp, and x.

Referenced by correction().

 {
   if (fBin >= mParameters.size())
     {
       std::stringstream sserr;
       sserr<<"wrong bin: "<<fBin<<": only "<<mParameters.size()<<" available!";
       handleError("SimpleJetCorrector",sserr.str());
     }
   unsigned N = fY.size();
   if (N > 4)
     {
       std::stringstream sserr;
       sserr<<"two many variables: "<<N<<" maximum is 4";
       handleError("SimpleJetCorrector",sserr.str());
     }
   const std::vector<float>& par = mParameters.record(fBin).parameters();
 
 #if ROOT_VERSION_CODE >= ROOT_VERSION(6,03,00)
   Double_t params[par.size() - 2 * N];
   for(unsigned int i=2*N;i<par.size();i++)
     {
       params[i-2*N] = par[i];
     }
   Double_t x[4] = {};
   for(unsigned i=0;i<N;i++)
     {
       x[i] = (fY[i] < par[2*i]) ? par[2*i] : (fY[i] > par[2*i+1]) ? par[2*i+1] : fY[i];
     }
   if (mParameters.definitions().isResponse()) {
     return invert(x, params);
   }
   return mFunc.EvalPar(x, params);
 #else
   float result = -1;
   //Have to do calculation using a temporary TFormula to avoid
   // thread safety issues
   TFormula tFunc(mFunc);
 
   for(unsigned int i=2*N;i<par.size();i++)
     tFunc.SetParameter(i-2*N,par[i]);
   float x[4] = {};
   std::vector<float> tmp;
   for(unsigned i=0;i<N;i++)
     {
       x[i] = (fY[i] < par[2*i]) ? par[2*i] : (fY[i] > par[2*i+1]) ? par[2*i+1] : fY[i];
       tmp.push_back(x[i]);
     }
   if (mParameters.definitions().isResponse())
     result = invert(tmp,tFunc);
   else
     result = tFunc.Eval(x[0],x[1],x[2],x[3]);
   return result;
 #endif
 }

unsigned SimpleJetCorrector::findInvertVar ( )

private

Definition at line 149 of file SimpleJetCorrector.cc.

References JetCorrectorParameters::definitions(), i, mParameters, JetCorrectorParameters::Definitions::parVar(), and query::result.

Referenced by SimpleJetCorrector().

 {
   unsigned result = 9999;
   std::vector<std::string> vv = mParameters.definitions().parVar();
   for(unsigned i=0;i<vv.size();i++)
     if (vv[i]=="JetPt")
       {
         result = i;
         break;
       }
   if (result >= vv.size())
     handleError("SimpleJetCorrector","Response inversion is required but JetPt is not specified as parameter");
   return result;
 }

float SimpleJetCorrector::invert	(	const Double_t *	args,
		const Double_t *	params
	)		const

private

Definition at line 167 of file SimpleJetCorrector.cc.

References alignCSCRings::e, mFunc, mInvertVar, tmp, and x.

Referenced by correctionBin().

 {
   unsigned nMax = 50;
   float precision = 0.0001;
   float rsp = 1.0;
   float e = 1.0;
   Double_t x[4];
   unsigned nLoop=0;
 
   // 4 dimensions (x, y, z, t) supported in TFormula
   memcpy(&x, args, sizeof(Double_t) * 4);
 
   while(e > precision && nLoop < nMax)
     {
       rsp = mFunc.EvalPar(x, params);
       float tmp = x[mInvertVar] * rsp;
       e = fabs(tmp - args[mInvertVar])/args[mInvertVar];
       x[mInvertVar] = args[mInvertVar]/rsp;
       nLoop++;
     }
   return 1./rsp;
 }