MomentumScaleCorrector.h

Go to the documentation of this file.

#ifndef MomentumScaleCorrector_h
#define MomentumScaleCorrector_h

#include <fstream>
#include <sstream>
#include "MuonAnalysis/MomentumScaleCalibration/interface/BaseFunction.h"
#include "MuonAnalysis/MomentumScaleCalibration/interface/Functions.h"
#include "FWCore/ParameterSet/interface/FileInPath.h"

class MomentumScaleCorrector : public BaseFunction
{
 public:
  MomentumScaleCorrector( TString identifier )
  {
    identifier.Prepend("MuonAnalysis/MomentumScaleCalibration/data/");
    identifier.Append(".txt");
    edm::FileInPath fileWithFullPath(identifier.Data());
    readParameters( fileWithFullPath.fullPath() );
  }
  MomentumScaleCorrector( const MuScleFitDBobject * dbObject ) : BaseFunction( dbObject )
  {
    std::vector<int>::const_iterator id = functionId_.begin();
    for( ; id != functionId_.end(); ++id ) {
      scaleFunctionVec_.push_back( scaleFunctionService( *id ) );
    }
    // Fill the arrays that will be used when calling the correction function.
    convertToArrays(scaleFunction_, scaleFunctionVec_);
  }

  ~MomentumScaleCorrector() {
    if( parArray_ != 0 ) {
      for( unsigned int i=0; i<functionId_.size(); ++i ) {
        delete[] parArray_[i];
        delete scaleFunction_[i];
      }
      delete[] parArray_;
      delete[] scaleFunction_;
    }
  }

  scaleFunctionBase<double * > * function(const int i) { return scaleFunction_[i]; }


  template <class U>
  double operator()( const U & track ) {

    // Loop on all the functions and apply them iteratively on the pt corrected by the previous function.
    double pt = track.pt();
    for( int i=0; i<=iterationNum_; ++i ) {
      // return ( scaleFunction_->scale( track.pt(), track.eta(), track.phi(), track.charge(), parScale_) );
      pt = ( scaleFunction_[i]->scale( pt, track.eta(), track.phi(), track.charge(), parArray_[i]) );
    }
    return pt;
  }

  template <class U>
  double correct( const U & lorentzVector ) {

    // Loop on all the functions and apply them iteratively on the pt corrected by the previous function.
    double pt = lorentzVector.Pt();
    for( int i=0; i<=iterationNum_; ++i ) {
      pt = ( scaleFunction_[i]->scale( pt, lorentzVector.Eta(), lorentzVector.Phi(), 1, parArray_[i]) );
    }
    return pt;
  }

 protected:
  void readParameters( TString fileName );

  scaleFunctionBase<double * > ** scaleFunction_;
  std::vector<scaleFunctionBase<double * > * > scaleFunctionVec_;
};

#endif // MomentumScaleCorrector_h