SimpleJetCorrector Class Reference

#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 (const std::string &fDataFile, const std::string &fOption="")
	SimpleJetCorrector (const JetCorrectorParameters &fParameters)

Private Member Functions
float	correctionBin (unsigned fBin, const std::vector< float > &fY) const
unsigned	findInvertVar ()
float	invert (const double *args, const double *params) const
SimpleJetCorrector &	operator= (const SimpleJetCorrector &)
void	setFuncParameters ()
	SimpleJetCorrector (const SimpleJetCorrector &)

Private Attributes
bool	mDoInterpolation
reco::FormulaEvaluator	mFunc
unsigned	mInvertVar
JetCorrectorParameters	mParameters

Detailed Description

Definition at line 15 of file SimpleJetCorrector.h.

Constructor & Destructor Documentation

SimpleJetCorrector() [1/3]

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

Definition at line 13 of file SimpleJetCorrector.cc.

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

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

SimpleJetCorrector() [2/3]

SimpleJetCorrector::SimpleJetCorrector

(

const JetCorrectorParameters &

fParameters

)

Definition at line 23 of file SimpleJetCorrector.cc.

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

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

SimpleJetCorrector() [3/3]

SimpleJetCorrector::SimpleJetCorrector ( const SimpleJetCorrector &  )

private

Member Function Documentation

correction()

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

Definition at line 33 of file SimpleJetCorrector.cc.

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

                                                                                                 {
  float result = 1.;
  float tmp = 0.0;
  float cor = 0.0;
  int bin = -1;
  bin = (fX.size() <= 3 && !fX.empty()) ? mParameters.binIndexN(fX) : 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();
  }
  if (result <= 0) {
    edm::LogWarning("SimpleJetCorrector")
        << "Null or negative jet energy correction factor evaluated: " << result << ". Truncating to 10e-10.";
    result = 10e-10;
  }
  return result;
}

correctionBin()

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

private

Definition at line 75 of file SimpleJetCorrector.cc.

References JetCorrectorParameters::definitions(), reco::FormulaEvaluator::evaluate(), mps_fire::i, invert(), JetCorrectorParameters::Definitions::isResponse(), mFunc, mParameters, N, JetCorrectorParameters::Record::parameters(), submitPVValidationJobs::params, JetCorrectorParameters::record(), JetCorrectorParameters::size(), 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 (!fY.empty() && N <= 4) {
    const std::vector<float>& par = mParameters.record(fBin).parameters();
    auto nParams = static_cast<int>(par.size() - 2 * N);
    if (nParams > 0) {
      double params[nParams];
      for (unsigned int i = 2 * N; i < par.size(); i++) {
        params[i - 2 * N] = par[i];
      }
      double 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.evaluate(reco::formula::ArrayAdaptor(x, N), reco::formula::ArrayAdaptor(params, par.size() - 2 * N));
    } else {
      return 1.0;
    }
  } else {
    edm::LogWarning("SimpleJetCorrector") << "Expect number of variables to be 0<N<=4 but is " << N << "in bin " << fBin
                                          << ". Returning zero" << std::endl;
    return 0.0;
  }
}

findInvertVar()

unsigned SimpleJetCorrector::findInvertVar ( )

private

Definition at line 110 of file SimpleJetCorrector.cc.

References JetCorrectorParameters::definitions(), mps_fire::i, mParameters, JetCorrectorParameters::Definitions::parVar(), and mps_fire::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;
}

invert()

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

private

Definition at line 125 of file SimpleJetCorrector.cc.

References writedatasetfile::args, MillePedeFileConverter_cfg::e, reco::FormulaEvaluator::evaluate(), mFunc, mInvertVar, metDiagnosticParameterSet_cfi::nMax, reco::FormulaEvaluator::numberOfParameters(), submitPVValidationJobs::params, boostedTaus_cff::precision, createJobs::tmp, and x.

Referenced by correctionBin().

                                                                               {
  unsigned nMax = 50;
  float precision = 0.0001;
  float rsp = 1.0;
  float e = 1.0;
  double x[4];
  unsigned nLoop = 0;

  // 4 dimensions (x, y, z, t) supported in TFormula
  memcpy(&x, args, sizeof(double) * 4);

  while (e > precision && nLoop < nMax) {
    rsp = mFunc.evaluate(reco::formula::ArrayAdaptor(x, 4),
                         reco::formula::ArrayAdaptor(params, mFunc.numberOfParameters()));
    float tmp = x[mInvertVar] * rsp;
    e = fabs(tmp - args[mInvertVar]) / args[mInvertVar];
    x[mInvertVar] = args[mInvertVar] / rsp;
    nLoop++;
  }
  return 1. / rsp;
}

operator=()

SimpleJetCorrector& SimpleJetCorrector::operator= ( const SimpleJetCorrector &  )

private

parameters()

const JetCorrectorParameters& SimpleJetCorrector::parameters ( void  ) const

inline

Definition at line 23 of file SimpleJetCorrector.h.

References mParameters.

{ return mParameters; }

setFuncParameters()

void SimpleJetCorrector::setFuncParameters ( )

private

setInterpolation()

void SimpleJetCorrector::setInterpolation ( bool  fInterpolation )

inline

Definition at line 21 of file SimpleJetCorrector.h.

References mDoInterpolation.

{ mDoInterpolation = fInterpolation; }

Member Data Documentation

mDoInterpolation

bool SimpleJetCorrector::mDoInterpolation

private

Definition at line 37 of file SimpleJetCorrector.h.

Referenced by correction(), setInterpolation(), and SimpleJetCorrector().

mFunc

reco::FormulaEvaluator SimpleJetCorrector::mFunc

private

Definition at line 35 of file SimpleJetCorrector.h.

Referenced by correctionBin(), and invert().

mInvertVar

unsigned SimpleJetCorrector::mInvertVar

private

Definition at line 36 of file SimpleJetCorrector.h.

Referenced by invert(), and SimpleJetCorrector().

mParameters

JetCorrectorParameters SimpleJetCorrector::mParameters

private

Definition at line 34 of file SimpleJetCorrector.h.

Referenced by correction(), correctionBin(), findInvertVar(), parameters(), and SimpleJetCorrector().