ResolutionFunction.h

Go to the documentation of this file.

#ifndef ResolutionFunction_h
#define ResolutionFunction_h

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

class ResolutionFunction : public BaseFunction
{
public:
  ResolutionFunction( TString identifier )
  {
    identifier.Prepend("MuonAnalysis/MomentumScaleCalibration/data/");
    identifier.Append(".txt");
    edm::FileInPath fileWithFullPath(identifier.Data());
    readParameters( fileWithFullPath.fullPath() );

    std::vector<int>::const_iterator idIt = functionId_.begin();
    for( ; idIt != functionId_.end(); ++idIt ) std::cout << "idIt = " << *idIt << std::endl;
  }
  ResolutionFunction( const MuScleFitDBobject * dbObject ) : BaseFunction( dbObject )
  {
    std::vector<int>::const_iterator id = functionId_.begin();
    for( ; id != functionId_.end(); ++id ) {
      resolutionFunctionVec_.push_back( resolutionFunctionService( *id ) );
    }
    // Fill the arrays that will be used when calling the correction function.
    convertToArrays(resolutionFunction_, resolutionFunctionVec_);
  }

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

  // EM+SC: 2013.01.11
  // lorentzVector have both capital and lower case methods for pt(), eta() and phi
  // if a lorentzVector is passed use, parArray form iteration i=0
 
  template <class U>
  double sigmaPt(const U & track, const int i=0) const {
    if( i > iterationNum_ || i < 0 ) {
      std::cout << "Error: wrong iteration number, there are " << iterationNum_ << "iterations, ther first one is 0" << std::endl;
      exit(1);
    }
    return resolutionFunction_[i]->sigmaPt(track.pt(), track.eta(), parArray_[i]);
  }
  template <class U>
  double sigmaCotgTh(const U & track, const int i=0) const {
    if( i > iterationNum_ || i < 0 ) {
      std::cout << "Error: wrong iteration number, there are " << iterationNum_ << "iterations, ther first one is 0" << std::endl;
      exit(1);
    }
    return resolutionFunction_[i]->sigmaCotgTh(track.pt(), track.eta(), parArray_[i]);
  }
  template <class U>
  double sigmaPhi(const U & track, const int i=0) const {
    if( i > iterationNum_ || i < 0 ) {
      std::cout << "Error: wrong iteration number, there are " << iterationNum_ << "iterations, ther first one is 0" << std::endl;
      exit(1);
    }
    return resolutionFunction_[i]->sigmaPhi(track.pt(), track.eta(), parArray_[i]);
  }
  resolutionFunctionBase<double * > * function( const unsigned int i )
  {
    if( resolutionFunctionVec_.size() > i ) return resolutionFunction_[i];
    else return nullptr;
  }

protected:
  void readParameters( TString fileName );

  resolutionFunctionBase<double * > ** resolutionFunction_;
  std::vector<resolutionFunctionBase<double * > * > resolutionFunctionVec_;
};

#endif // ResolutionFunction_h