d9/d5d/APReweight3D_8h_source.html

/*

  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