X-Git-Url: http://git.uio.no/git/?a=blobdiff_plain;f=PHOS%2FAliPHOSFastRecParticle.h;h=ce710681976df0d73d789d88b0e6c5b4fbb3a317;hb=88eed77084ec26a9219dbdde7eea6e5f1cf7b066;hp=7ea48297e130628d11a1973181dd10f7cd08497c;hpb=2f04ed65d5949adfed4f99b4cc36732eeca851d9;p=u%2Fmrichter%2FAliRoot.git

diff --git a/PHOS/AliPHOSFastRecParticle.h b/PHOS/AliPHOSFastRecParticle.h
index 7ea48297e13..ce710681976 100644
--- a/PHOS/AliPHOSFastRecParticle.h
+++ b/PHOS/AliPHOSFastRecParticle.h
@@ -7,31 +7,31 @@
 
 //_________________________________________________________________________
 //  A  Particle modified by PHOS response and produced by AliPHOSvFast
+//  This is also a base class for AliPHOSRecParticle produced by AliPHOSPIDv1
+//  Defines the particle type
 //  To become a general class of AliRoot ?    
-//               
+//--               
 //*-- Author: Yves Schutz (SUBATECH)
 
 // --- ROOT system ---
 
+class TClonesArray;
 #include "TParticle.h"
-#include "TVector3.h"
 
 // --- Standard library ---
 
 // --- AliRoot header files ---
 
 class AliPHOSFastRecParticle : public TParticle {
-
+  
  public:
   
-  AliPHOSFastRecParticle() {
-    // ctor 
-  };         
+  AliPHOSFastRecParticle() ;
+  
   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) ;
@@ -39,42 +39,62 @@ class AliPHOSFastRecParticle : public TParticle {
     // returns the index of this in the list
     return fIndexInList ; 
   } 
-  virtual Int_t * GetPrimaries(Int_t & number) ;
-  Int_t GetType() { 
+  virtual const Int_t GetNPrimaries() const {return 0 ;}
+  virtual const TParticle * GetPrimary(Int_t) const  {return 0 ;} 
+  const Int_t GetType() const { 
     // returns the type of the particle
     return fType ; 
   } 
-  TString Name() ; 
-  virtual void Paint(Option_t * option="");
-  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 SetPIDBit(UInt_t fSet) {
+    // Set PID bit number fSet
+    fType |= (1<<fSet) ; 
   } 
+  
+  Bool_t TestPIDBit(UInt_t fTest) const {
+    // Check PID bit number fTest
+    if (fType & (1<<fTest) ) return  kTRUE ;	
+    else return kFALSE ;
+  }
+  
+  Bool_t IsPhoton           (TString purity = "low") const;
+  Bool_t IsPi0              (TString purity = "low") const;
+  Bool_t IsElectron         (TString purity = "low") const;
+  Bool_t IsHardPhoton       () const;
+  Bool_t IsHardPi0          () const;
+  Bool_t IsHadron           () const;
+  Bool_t IsChargedHadron    () const;
+  Bool_t IsNeutralHadron    () const;
+  Bool_t IsFastChargedHadron() const;
+  Bool_t IsSlowChargedHadron() const;
+  Bool_t IsFastNeutralHadron() const;
+  Bool_t IsSlowNeutralHadron() const;
+  Bool_t IsEleCon(TString purity = "low") const; 
+
+  TString Name() const ; 
+  virtual void Paint(Option_t * option="");
+  virtual void Print() const ; 
+  
+  void SetType(Int_t 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 } ; 
-
+  }
+  //This has to disappear
+  enum EParticleType { kUNDEFINED=-1, 
+		       kNEUTRALEMFAST, kNEUTRALHAFAST,  kNEUTRALEMSLOW, kNEUTRALHASLOW, 
+		       kCHARGEDEMFAST, kCHARGEDHAFAST,  kCHARGEDEMSLOW, kCHARGEDHASLOW } ; 
+  
   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 ;     // (unique) primary particle index 
   Int_t fType ;        // particle type obtained by "virtual" reconstruction
-
  private:
 
-
-  ClassDef(AliPHOSFastRecParticle,1)  // Reconstructed Particle produced by the fast simulation 
+  ClassDef(AliPHOSFastRecParticle,2)  // Reconstructed Particle produced by the fast simulation 
 
 };