fix for event mixing

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

diff --git a/ANALYSIS/AliAnalysisTaskMCParticleFilter.h b/ANALYSIS/AliAnalysisTaskMCParticleFilter.h
index 4a9442a26b04f66dfd5f9bce8e35182450b83742..d47f2844d0fb5a5a728b89ad8d5f263cf565092b 100644 (file)
--- a/ANALYSIS/AliAnalysisTaskMCParticleFilter.h
+++ b/ANALYSIS/AliAnalysisTaskMCParticleFilter.h
@@ -9,34 +9,45 @@
 //  Fill AOD tracks from Kinematic stack
 //
 
-#include <TList.h> 
 #include "AliAnalysisTaskSE.h"
 
 class AliAnalysisFilter;
 class TString;
+class TList;
+class AliAODMCHeader;
+class AliMCParticle;
+class TClonesArray;
 
 class AliAnalysisTaskMCParticleFilter : public AliAnalysisTaskSE
 {
  public:
                                   AliAnalysisTaskMCParticleFilter();
                                   AliAnalysisTaskMCParticleFilter( const char* name );
-                                  AliAnalysisTaskMCParticleFilter(const AliAnalysisTaskMCParticleFilter& obj);
     virtual                      ~AliAnalysisTaskMCParticleFilter();
-     AliAnalysisTaskMCParticleFilter&   operator=(const AliAnalysisTaskMCParticleFilter& other);
     
     // Implementation of interface methods
     virtual                void   UserCreateOutputObjects();
     virtual                void   UserExec( Option_t *option );
-    
+    virtual                Bool_t Notify();
+    virtual                void   Terminate( Option_t *option );
     // Setters
     virtual                void   SetTrackFilterMother(AliAnalysisFilter* trackF) { fTrackFilterMother = trackF; }
     
  private:
     Bool_t Select(TParticle* part, Float_t rv, Float_t zv);                
+    
+    // pivate c'tors to prevent misuse
+    AliAnalysisTaskMCParticleFilter&   operator=(const AliAnalysisTaskMCParticleFilter& other);
+    AliAnalysisTaskMCParticleFilter(const AliAnalysisTaskMCParticleFilter& obj);
+    void PrintMCParticle(const AliMCParticle *mcp,Int_t np);
+
 
     AliAnalysisFilter*  fTrackFilterMother;   //  Track Filter
-                
-    ClassDef( AliAnalysisTaskMCParticleFilter, 1 ); // Analysis task for Kinematic filtering
+    AliAODMCHeader *fAODMcHeader;                 // ! MC Header
+    TClonesArray *fAODMcParticles;            // ! tclones array of MC Particles
+    TList *fHistList;                         // list to store e histograms, only as exchange
+
+    ClassDef( AliAnalysisTaskMCParticleFilter, 4 ); // Analysis task for Kinematic filtering
 };
  
 #endif