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