Lumi3DReWeighting.h

Go to the documentation of this file.

#ifndef PhysicsTools_Utilities_interface_Lumi3DReWeighting_h
#define PhysicsTools_Utilities_interface_Lumi3DReWeighting_h
 
#include "TH1.h"
#include "TFile.h"
#include <cmath>
#include <string>
 
#include <vector>
 
namespace edm {
  class EventBase;
  class Lumi3DReWeighting {
  public:
    Lumi3DReWeighting(std::string generatedFile,
                      std::string dataFile,
                      std::string GenHistName,
                      std::string DataHistName,
                      std::string WeightOutputFile);
 
    Lumi3DReWeighting(const std::vector<float>& MC_distr,
                      const std::vector<float>& Lumi_distr,
                      std::string WeightOutputFile);
 
    Lumi3DReWeighting(){};
 
    double weight3D(const edm::EventBase& e);
 
    double weight3D(int, int, int);
 
    void weight3D_set(std::string generatedFile,
                      std::string dataFile,
                      std::string GenHistName,
                      std::string DataHistName,
                      std::string WeightOutputFile);
 
    void weight3D_init(float Scale);
 
    void weight3D_init(std::string WeightFileName);  // initialize from root file
 
    void weight3D_init(std::string MCFileName, std::string DataFileName);  // initialize from root files
 
  protected:
    std::string generatedFileName_;
    std::string dataFileName_;
    std::string GenHistName_;
    std::string DataHistName_;
    std::string weightFileName_;
    std::shared_ptr<TFile> generatedFile_;
    std::shared_ptr<TFile> dataFile_;
    std::shared_ptr<TH1> weights_;
 
    //keep copies of normalized distributions:
 
    std::shared_ptr<TH1> MC_distr_;
    std::shared_ptr<TH1> Data_distr_;
 
    double Weight3D_[50][50][50];
  };
}  // namespace edm
 
#endif