First update for AOD compatibility (distributions and efficiency), still missing...

[u/mrichter/AliRoot.git] / PWGCF / EBYE / NetParticle / AliAnalysisNetParticleDistribution.h

diff --git a/PWGCF/EBYE/NetParticle/AliAnalysisNetParticleDistribution.h b/PWGCF/EBYE/NetParticle/AliAnalysisNetParticleDistribution.h
index 269135ed0ae4ed32c8545c10bff5eee9387feeb1..1d283119f8a2a650a263bb9bd3a388740a11eb67 100644 (file)
--- a/PWGCF/EBYE/NetParticle/AliAnalysisNetParticleDistribution.h
+++ b/PWGCF/EBYE/NetParticle/AliAnalysisNetParticleDistribution.h
@@ -21,6 +21,7 @@ class AliStack;
 class AliPIDResponse;
 class AliESDInputHandler;
 class AliESDtrackCuts;
+class AliAODInputHandler;
 
 class AliAnalysisNetParticleDistribution : public TNamed {
 
@@ -42,13 +43,13 @@ class AliAnalysisNetParticleDistribution : public TNamed {
    */
 
   /** Initialize */
-  Int_t Initialize(AliAnalysisNetParticleHelper* helper, AliESDtrackCuts* cuts, Bool_t isMC, Float_t *ptRange, Float_t etaMax);
+  Int_t Initialize(AliAnalysisNetParticleHelper* helper, AliESDtrackCuts* cuts, Bool_t isMC, Float_t *ptRange, Float_t etaMax, Int_t trackCutBit);
 
   /** Add histograms to outlist */
   void CreateHistograms(TList *outList);
 
   /** Setup Event */
-  Int_t SetupEvent(AliESDInputHandler *esdHandler, AliMCEvent *mcEvent);
+  Int_t SetupEvent(AliESDInputHandler *esdHandler, AliAODInputHandler *aodHandler, AliMCEvent *mcEvent);
 
   /** Resre Event */
   void ResetEvent();
@@ -71,6 +72,9 @@ class AliAnalysisNetParticleDistribution : public TNamed {
    /** Process ESD tracks and fill histograms */
   Int_t ProcessESDTracks();
 
+   /** Process AOD tracks and fill histograms */
+  Int_t ProcessAODTracks();
+
   /** Process primary particles from the stack and fill histograms */
   Int_t ProcessStackParticles();
 
@@ -103,6 +107,9 @@ class AliAnalysisNetParticleDistribution : public TNamed {
   AliPIDResponse       *fPIDResponse;           //! Ptr to PID response Object
   AliESDEvent          *fESD;                   //! Ptr to ESD event
 
+  AliAODInputHandler   *fAODHandler;            //! Ptr to AOD handler 
+  AliAODEvent          *fAOD;                   //! Ptr to AOD event
+
   Bool_t                fIsMC;                  //  Is MC event
 
   AliMCEvent           *fMCEvent;               //! Ptr to MC event
@@ -112,6 +119,8 @@ class AliAnalysisNetParticleDistribution : public TNamed {
   // -----------------------------------------------------------------------
   Float_t               fEtaMax;                //  Max, absolut eta
   Float_t              *fPtRange;               //  Array of pt [min,max]
+
+  Int_t                 fAODtrackCutBit;        //  Track filter bit for AOD tracks
   // -----------------------------------------------------------------------
   Float_t              *fNp;                    //  Array of particle/anti-particle counts