GetPrimaries modified

[u/mrichter/AliRoot.git] / PHOS / AliPHOSFastRecParticle.h

diff --git a/PHOS/AliPHOSFastRecParticle.h b/PHOS/AliPHOSFastRecParticle.h
index e4e026ef978b0aab555209cc4f75af7a21bd1c1e..e6fa46d4932a2710edff95fdbdd432287b6b79e0 100644 (file)
--- a/PHOS/AliPHOSFastRecParticle.h
+++ b/PHOS/AliPHOSFastRecParticle.h
@@ -5,12 +5,11 @@
 
 /* $Id$ */
 
-////////////////////////////////////////////////
-//  A  Particle modified by PHOS response     //
-//  Yves Schutz SUBATECH                      //
-//  To become a general class of AliRoot ?    //  
-//                                            //
-////////////////////////////////////////////////
+//_________________________________________________________________________
+//  A  Particle modified by PHOS response and produced by AliPHOSvFast
+//  To become a general class of AliRoot ?    
+//               
+//*-- Author: Yves Schutz (SUBATECH)
 
 // --- ROOT system ---
 
@@ -21,43 +20,61 @@
 
 // --- AliRoot header files ---
 
-//#include "AliPHOSRecParticle.h" 
-
-typedef TClonesArray  FastRecParticlesList ; 
-const static Int_t kUNDEFINED     = -1; 
-const static Int_t kGAMMA         = 0 ; 
-const static Int_t kELECTRON      = 1 ;
-const static Int_t kNEUTRAL       = 2 ;  
-const static Int_t kCHARGED       = 3 ;  
-const static Int_t kCHARGEDHADRON = 4 ;  
-const static Int_t kNEUTRALHADRON = 5 ;  
-const static Int_t kNEUTRALEM     = 6 ;  
-const static Int_t kGAMMAHADRON   = 7 ; 
-
-
 class AliPHOSFastRecParticle : public TParticle {
 
-public:
+ public:
   
-  AliPHOSFastRecParticle() {};          // ctor
+  AliPHOSFastRecParticle() {
+    // ctor 
+  };         
+
   AliPHOSFastRecParticle(const AliPHOSFastRecParticle & rp) ;  // ctor
   AliPHOSFastRecParticle(const TParticle & p) ;  // ctor
-  virtual ~AliPHOSFastRecParticle() ; // dtor
-
+  virtual ~AliPHOSFastRecParticle(){
+    // dtor
+  }
   virtual Int_t DistancetoPrimitive(Int_t px, Int_t py) ; 
   virtual void Draw(Option_t *option) ;  
-  virtual void ExecuteEvent(Int_t event, Int_t px, Int_t py) ; 
-  Int_t GetType() { return fType ; } 
+  virtual void ExecuteEvent(Int_t event, Int_t px, Int_t py) ;
+  Int_t GetIndexInList() const { 
+    // returns the index of this in the list
+    return fIndexInList ; 
+  } 
+  Int_t GetPrimary(){return fPrimary;}
+  Int_t GetType() { 
+    // returns the type of the particle
+    return fType ; 
+  } 
   TString Name() ; 
   virtual void Paint(Option_t * option="");
-  void Print() ; 
-  void SetType(Int_t type) { fType = type ; } 
+  virtual void Print(const char * opt) ; 
+  void SetPrimary(Int_t index) { // sets the primary particle index
+    fPrimary = index ; 
+  }
+  void SetType(Int_t type) { 
+    // sets the particle type 
+    fType = type ; 
+  } 
+  void SetIndexInList(Int_t val) { 
+    // sets the value of the index in the list 
+    fIndexInList = val ; 
+  } 
+
+  enum EParticleType { kUNDEFINED=-1, kNEUTRALEM,  kNEUTRALHA,  kGAMMA , kGAMMAHA , 
+                      kABSURDEM,  kABSURDHA ,  kELECTRON, kCHARGEDHA } ; 
+
+  typedef TClonesArray  FastRecParticlesList ; 
+
+ protected:
+
+  Int_t fIndexInList ; // the index of this RecParticle in the list stored in TreeR (to be set by analysis)
+  Int_t fPrimary ;  //  primary particle index 
+  Int_t fType ;        // particle type obtained by "virtual" reconstruction
 
-protected:
+ private:
 
-  Int_t fType ;   // identified particle type
 
-  ClassDef(AliPHOSFastRecParticle,1)  // Reconstructed Particle, version 1
+  ClassDef(AliPHOSFastRecParticle,1)  // Reconstructed Particle produced by the fast simulation 
 
 };