introducing multiplicity measurement with the ITS or TPC

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

diff --git a/PWG0/dNdEta/AliMultiplicityMCSelector.h b/PWG0/dNdEta/AliMultiplicityMCSelector.h
index 1e139150041957884af8b0f81133a7710dfdc904..675643b8cae5a64f0b33f0d263f94d0741503304 100644 (file)
--- a/PWG0/dNdEta/AliMultiplicityMCSelector.h
+++ b/PWG0/dNdEta/AliMultiplicityMCSelector.h
@@ -6,8 +6,9 @@
 #include "AliSelectorRL.h"
 
 class AliESDtrackCuts;
-class TH1F;
 class TH2F;
+class TH3F;
+class AliMultiplicityCorrection;
 
 class AliMultiplicityMCSelector : public AliSelectorRL {
   public:
@@ -16,6 +17,7 @@ class AliMultiplicityMCSelector : public AliSelectorRL {
 
     virtual void    Begin(TTree* tree);
     virtual void    SlaveBegin(TTree *tree);
+    virtual void    Init(TTree *tree);
     virtual Bool_t  Process(Long64_t entry);
     virtual void    SlaveTerminate();
     virtual void    Terminate();
@@ -23,12 +25,8 @@ class AliMultiplicityMCSelector : public AliSelectorRL {
  protected:
     void ReadUserObjects(TTree* tree);
 
-    TH1F* fMultiplicityESD; // multiplicity histogram
-    TH1F* fMultiplicityMC; // multiplicity histogram
-
-    TH2F* fCorrelation; // (gene multiplicity) vs (meas multiplicity)
-
-    AliESDtrackCuts*  fEsdTrackCuts;     // Object containing the parameters of the esd track cuts
+    AliMultiplicityCorrection* fMultiplicity; // object containing the extracted data
+    AliESDtrackCuts* fEsdTrackCuts;     // Object containing the parameters of the esd track cuts
 
  private:
     AliMultiplicityMCSelector(const AliMultiplicityMCSelector&);