MuScleFitBase.h

Go to the documentation of this file.

#ifndef MUSCLEFITBASE_H
#define MUSCLEFITBASE_H

#include <map>
#include <string>
#include "TFile.h"
#include "Histograms.h"
#include "MuScleFitUtils.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "MuonAnalysis/MomentumScaleCalibration/interface/MuonPair.h"
#include "MuonAnalysis/MomentumScaleCalibration/interface/GenMuonPair.h"

class MuScleFitBase
{
public:
  MuScleFitBase(const edm::ParameterSet& iConfig) :
    probabilitiesFileInPath_( iConfig.getUntrackedParameter<std::string>( "ProbabilitiesFileInPath" , "MuonAnalysis/MomentumScaleCalibration/test/Probs_new_Horace_CTEQ_1000.root" ) ),
    probabilitiesFile_( iConfig.getUntrackedParameter<std::string>( "ProbabilitiesFile" , "" ) ),
    theMuonType_( iConfig.getParameter<int>( "MuonType" ) ),
    theMuonLabel_( iConfig.getParameter<edm::InputTag>( "MuonLabel" ) ),
    theRootFileName_( iConfig.getUntrackedParameter<std::string>("OutputFileName") ),
    theGenInfoRootFileName_( iConfig.getUntrackedParameter<std::string>("OutputGenInfoFileName", "genSimRecoPlots.root") ),
    debug_( iConfig.getUntrackedParameter<int>("debug",0) )
  {}
  virtual ~MuScleFitBase() {}
protected:
  void fillHistoMap(TFile* outputFile, unsigned int iLoop);
  void clearHistoMap();
  void writeHistoMap( const unsigned int iLoop );

  void readProbabilityDistributionsFromFile();

  std::string probabilitiesFileInPath_;
  std::string probabilitiesFile_;

  int theMuonType_;
  edm::InputTag theMuonLabel_;
  std::string theRootFileName_;
  std::string theGenInfoRootFileName_;

  int debug_;

  class ProbForIntegral
  {
  public:
    ProbForIntegral( const double & massResol, const int iRes, const int iY, const bool isZ ) :
      massResol_(massResol),
      iRes_(iRes), iY_(iY), isZ_(isZ)
    {}
    double operator()(const double * mass, const double *)
    {
      if( isZ_ ) {
        return( MuScleFitUtils::probability(*mass, massResol_, MuScleFitUtils::GLZValue, MuScleFitUtils::GLZNorm, iRes_, iY_) );
      }
      return( MuScleFitUtils::probability(*mass, massResol_, MuScleFitUtils::GLValue, MuScleFitUtils::GLNorm, iRes_, iY_) );
    }
  protected:
    double massResol_;
    int iRes_, iY_;
    bool isZ_;
  };

  std::vector<TFile*> theFiles_;

  std::map<std::string, Histograms*> mapHisto_;
  
  std::vector<MuonPair> muonPairs_;
  std::vector<GenMuonPair> genMuonPairs_;
};

#endif