Add option to write MC particles to the AOD, new lightweight class AliAODMCParticle...

[u/mrichter/AliRoot.git] / ANALYSIS / AliFlowAnalysis.h

diff --git a/ANALYSIS/AliFlowAnalysis.h b/ANALYSIS/AliFlowAnalysis.h
index 27c296447702a71e4fd5d7e42d682ee2f72e44aa..ea6c003a40b24fe98c23062c8e1395936506f7a7 100644 (file)
--- a/ANALYSIS/AliFlowAnalysis.h
+++ b/ANALYSIS/AliFlowAnalysis.h
@@ -1,39 +1,44 @@
 #ifndef ALIFLOWANALYSIS_H
 #define ALIFLOWANALYSIS_H
-//________________________________
-///////////////////////////////////////////////////////////
-//
+
+//*********************************************************
 // class AliFlowAnalysis
-//
 // Flow Analysis
-//
-//
 // S.Radomski@gsi.de
 // Piotr.Skowronski@cern.ch
-//
-///////////////////////////////////////////////////////////
+//*********************************************************
 
 #include "AliAnalysis.h"
 
 class AliESD;
+class AliAOD;
 class AliStack;
-
+class AliAODParticleCut;
+    
 class AliFlowAnalysis: public AliAnalysis
 { 
   public: 
-     AliFlowAnalysis(){}
-     ~AliFlowAnalysis(){}
+     AliFlowAnalysis();
+     virtual ~AliFlowAnalysis();
 
     Int_t Init();
-    Int_t ProcessEvent(AliESD* esd, AliStack* stack = 0x0);
+    Int_t ProcessEvent(AliAOD* aodrec, AliAOD* aodsim = 0x0);
     Int_t Finish();
    
+    void SetParticleCut(AliAODParticleCut* pcut){fPartCut = pcut;}
     static Double_t GetEventPlane(AliESD* esd);
-       
-  protected:
+    static void     GetFlow(AliESD* esd,Double_t& v2,Double_t& psi);
 
+    Double_t        GetEventPlane(AliAOD* aod);
+    void            GetFlow(AliAOD* aod,Double_t& v2,Double_t& psi);
+  protected:
+    
   private:
- 
+    AliAODParticleCut* fPartCut;//Particle Cut
+
+    AliFlowAnalysis(const AliFlowAnalysis&); // Not implemented
+    AliFlowAnalysis& operator=(const AliFlowAnalysis&); // Not implemented
+
     ClassDef(AliFlowAnalysis,1)
 };