fITSModuleIndexInfo is now copied from the AliITStrackV2 to the AliESDtrack (A. Dainese)

[u/mrichter/AliRoot.git] / STEER / AliMCEventHandler.h

diff --git a/STEER/AliMCEventHandler.h b/STEER/AliMCEventHandler.h
index 6ec52b7b1c3dfe77255019c61b252b2d92f83e98..b3e8f0bf4e7c12005bfa1ccaeeb6f1e15eab98fc 100644 (file)
--- a/STEER/AliMCEventHandler.h
+++ b/STEER/AliMCEventHandler.h
@@ -17,6 +17,8 @@
 //-------------------------------------------------------------------------
 #include "AliVEventHandler.h"
 #include "AliHeader.h"
+#include <TExMap.h>
+
 class TFile;
 class TTree;
 class TParticle;
@@ -57,8 +59,16 @@ public:
     TTree*      TreeK()   const {return fTreeK;}
     Int_t       GetParticleAndTR(Int_t i, TParticle*& particle, TClonesArray*& trefs);
     void        DrawCheck(Int_t i, Int_t search=0);
+    Bool_t      InitOk() {return fInitOk;}
+    // Label manipulation
+    void   SelectParticle(Int_t i);
+    Bool_t IsParticleSelected(Int_t i);
+    void   CreateLabelMap();
+    Int_t  GetNewLabel(Int_t i);
+
 private:
     Bool_t      OpenFile(Int_t i);
+    void  VerifySelectedParticles();
     AliMCEventHandler(const AliMCEventHandler& handler);             
     AliMCEventHandler& operator=(const AliMCEventHandler& handler);  
 private:
@@ -71,10 +81,12 @@ private:
     TTree            *fTreeTR;           //! TreeTR (track references tree)
     TDirectoryFile   *fDirK;             //! Directory for Kine Tree
     TDirectoryFile   *fDirTR;            //! Directory for TR Tree
+    TExMap            fParticleSelected; //! List of selected MC particles for t
+    TExMap            fLabelMap;         //! Stores the Map of MC (ESDLabel,AODlabel)  
     Int_t             fNEvent;           //! Number of events
     Int_t             fEvent;            //! Current event
     TString          *fPathName;         //! Input file path 
-    char             *fExtension;        //! File name extension 
+    const Char_t     *fExtension;        //! File name extension 
     Int_t             fFileNumber;       //! Input file number
     Int_t             fEventsPerFile;    //! Number of events per file
     Bool_t            fReadTR;           // determines if TR shall be read