Added GetPrimary method to retrieve the primary particle associated to the SDigits...

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

diff --git a/PHOS/AliPHOSRecParticle.h b/PHOS/AliPHOSRecParticle.h
index ab35b5a6e6bfffe090013ba3911b8233ccaddb19..32195b88d66715690ff8c65304c31a9c0c116d12 100644 (file)
--- a/PHOS/AliPHOSRecParticle.h
+++ b/PHOS/AliPHOSRecParticle.h
@@ -8,38 +8,51 @@
 //_________________________________________________________________________
 //  A Reconstructed Particle in PHOS    
 //  To become a general class of AliRoot ?        
-//       
+//  why not      
 //*-- Author: Yves Schutz (SUBATECH)
 
 // --- ROOT system ---
 
-#include "TParticle.h"
-#include "TVector3.h"
-
 // --- Standard library ---
 
 // --- AliRoot header files ---
 
-#include "AliPHOSTrackSegment.h"
 #include "AliPHOSFastRecParticle.h"
+#include "AliESDtrack.h" 
+
+class TParticle ;
+#include  "TVector3.h"  
 
 class AliPHOSRecParticle : public AliPHOSFastRecParticle {
 
-public:
+ public:
   
-  AliPHOSRecParticle() {};          // ctor
-  AliPHOSRecParticle(AliPHOSTrackSegment * ts) ;  // ctor
+  AliPHOSRecParticle() ; 
   AliPHOSRecParticle(const AliPHOSRecParticle & rp) ;  // ctor
-  virtual ~AliPHOSRecParticle() ; // dtor
-
-  AliPHOSTrackSegment * GetPHOSTrackSegment() const { return fPHOSTrackSegment ; } 
-  virtual Int_t * GetPrimaries(Int_t & number) ;
-
-private:
+  virtual ~AliPHOSRecParticle(){  }
+
+  Int_t   GetPHOSTSIndex()const {    return fPHOSTrackSegment ;  }
+  virtual const Int_t GetNPrimariesToRecParticles() const ;
+  virtual const Int_t GetNPrimaries() const ;
+  TVector3 GetPos() const { return fPos ; } 
+  virtual const TParticle * GetPrimary(Int_t index) const ;
+  virtual const TParticle * GetPrimary() const ;
+  const Double_t *GetPID();
+  void    SetDebug() { fDebug = kTRUE ; } 
+  void    SetPos(TVector3 pos) { fPos.SetXYZ( pos.X(), pos.Y(), pos.Z() ); } 
+  void    UnsetDebug() { fDebug = kFALSE ; }
+  void    SetTrackSegment(Int_t index){fPHOSTrackSegment = index; }
+
+  typedef TClonesArray RecParticlesList ; 
+  
+ private:
 
-  AliPHOSTrackSegment * fPHOSTrackSegment ; // pointer to the associated track segment in PHOS  
+  Int_t fPHOSTrackSegment ; // pointer to the associated track segment in PHOS  
+  Bool_t fDebug ; // to steer debug output
+  TVector3 fPos ; // position in the global alice coordinate system 
+  Double_t fPID[AliESDtrack::kSPECIESN] ; // PID probability densities
 
-  ClassDef(AliPHOSRecParticle,1)  // Reconstructed Particle
+  ClassDef(AliPHOSRecParticle,3)  // Reconstructed Particle
 };
 
 #endif // AliPHOSRECPARTICLE_H