o) making selector proof ready

[u/mrichter/AliRoot.git] / PWG0 / dNdEta / dNdEtaCorrection.h

diff --git a/PWG0/dNdEta/dNdEtaCorrection.h b/PWG0/dNdEta/dNdEtaCorrection.h
index 2091b85bdfedbe7e0d875b76b720e6975bb4cfc3..ccf7e7f1b2c4ce861e7badd14f3d5e27c23d3c0c 100644 (file)
--- a/PWG0/dNdEta/dNdEtaCorrection.h
+++ b/PWG0/dNdEta/dNdEtaCorrection.h
@@ -40,6 +40,12 @@ protected:
 public:
   dNdEtaCorrection(Char_t* name="dndeta_correction");
 
+  TH2F* GetGeneratedHistogram() { return hEtaVsVtx_gene; }
+  TH2F* GetMeasuredHistogram() { return hEtaVsVtx_meas; }
+
+  void SetGeneratedHistogram(TH2F* aGeneratedHistogram) { hEtaVsVtx_gene = aGeneratedHistogram; }
+  void SetMeasuredHistogram(TH2F* aMeasuredHistogram) { hEtaVsVtx_meas = aMeasuredHistogram; }
+
   void FillMeas(Float_t vtx, Float_t eta) {hEtaVsVtx_meas->Fill(vtx, eta);}
   void FillGene(Float_t vtx, Float_t eta) {hEtaVsVtx_gene->Fill(vtx, eta);}