APReweight3D.h

Go to the documentation of this file.

/*
  Copyright (C) 2002-2022 CERN for the benefit of the ATLAS collaboration
*/


 
#ifndef APReweight3D_h
#define APReweight3D_h
 
#include "APReweightBase.h"
#include <vector>
 
class TTree;
class TH3;
class TH3D;
class TAxis;
class APWeightEntry;
 
class APReweight3D : public APReweightBase {
 
public:
  APReweight3D();                                                                
  APReweight3D(TH3* denominator_in, TH3* numerator_in, bool isTrig = false);     
  virtual ~APReweight3D();                                                       
 
  void ReadEfficiency(TH3* efficiency_in, TH3* err_low_in, TH3* err_high_in = 0);
 
  APWeightEntry* GetBinWeight(unsigned int bin_x, unsigned int bin_y, unsigned int bin_z) const; 
  APWeightEntry* GetWeight(double value_x, double value_y, double value_z) const;
   
  double GetSampleScale() const;                                                 
  unsigned int NBins() const;                                                    
  unsigned int GetBinX(double value_x) const;                                    
  unsigned int GetBinY(double value_y) const;                                    
  unsigned int GetBinZ(double value_z) const;                                    
  
  const TH3D* GetDenominatorHist() const;                                              
  const TH3D* GetNumeratorHist() const;                                                
 
  void SetSystUncert(double rel_uncert);                                         
  void SetQuietMode(bool isQuiet = true);                                        
  
  ClassDef(APReweight3D,1)
 
private:
  TH3D* m_denominator_hist;                                                       
  TH3D* m_numerator_hist;                                                         
  std::vector< std::vector< std::vector< APWeightEntry* > > > m_weights;          
  unsigned int m_n_bins_x;                                                        
  unsigned int m_n_bins_y;                                                        
  unsigned int m_n_bins_z;                                                        
  TAxis *m_axis_x;                                                                
  TAxis *m_axis_y;                                                                
  TAxis *m_axis_z;                                                                
};
 
#endif