Partial redesign of identification bits and coding rule corrections
authorschutz <schutz@f7af4fe6-9843-0410-8265-dc069ae4e863>
Thu, 20 Mar 2003 17:50:50 +0000 (17:50 +0000)
committerschutz <schutz@f7af4fe6-9843-0410-8265-dc069ae4e863>
Thu, 20 Mar 2003 17:50:50 +0000 (17:50 +0000)
PHOS/AliPHOSFastRecParticle.cxx
PHOS/AliPHOSFastRecParticle.h

index e35c942..a88c883 100644 (file)
 //_________________________________________________________________________
 //  A  Particle modified by PHOS response and produced by AliPHOSvFast
 //  To become a general class of AliRoot ?    
-//               
+//--
+//  This is also a base class for AliPHOSRecParticle produced by AliPHOSPIDv1
+//  The rec.particle type is to be defined by AliPHOSvFast or AliPHOSPIDv1
+//--
 //*-- Author: Yves Schutz (SUBATECH)
 
 // --- ROOT system ---
@@ -164,42 +167,209 @@ void AliPHOSFastRecParticle::ExecuteEvent(Int_t event, Int_t px, Int_t py)
 }
 
 //____________________________________________________________________________
-TString AliPHOSFastRecParticle::Name()const
+Bool_t AliPHOSFastRecParticle::IsPhoton(const char* purity) const
+{
+  // Rec.Particle is a photon if it has a photon-like shape, fast and neutral
+  // photon-like shape is defined with a purity "low", "medium" or "high"
+
+  Bool_t photonLike = kFALSE;
+  if      (purity == "low"   ) photonLike = TestPIDBit(6);
+  else if (purity == "medium") photonLike = TestPIDBit(7);
+  else if (purity == "high"  ) photonLike = TestPIDBit(8);
+  else Error("IsPhoton","Wrong purity type: %s",purity);
+  if (photonLike                                   && //  photon by PCA
+      (TestPIDBit(5)||TestPIDBit(4)||TestPIDBit(3))&& //  fast by TOF
+      (TestPIDBit(2)||TestPIDBit(1)||TestPIDBit(0)))  //  neutral by CPV
+    return kTRUE ;
+  else
+    return kFALSE;
+}
+
+//____________________________________________________________________________
+Bool_t AliPHOSFastRecParticle::IsPi0(const char* purity) const
+{
+  // Rec.Particle is a pi0 if it has a pi0-like shape, fast and neutral
+  // pi0-like shape is defined with a purity "low", "medium" or "high"
+
+  Bool_t pi0Like = kFALSE;
+  if      (purity == "low"   ) pi0Like = TestPIDBit(9);
+  else if (purity == "medium") pi0Like = TestPIDBit(10);
+  else if (purity == "high"  ) pi0Like = TestPIDBit(11);
+  else Error("IsPi0","Wrong purity type: %s",purity);
+  if (pi0Like                                      && //  pi0 by PCA
+      (TestPIDBit(5)||TestPIDBit(4)||TestPIDBit(3))&& //  fast by TOF
+      (TestPIDBit(2)||TestPIDBit(1)||TestPIDBit(0)))  //  neutral by CPV
+    return kTRUE ;
+  else
+    return kFALSE;
+}
+
+//____________________________________________________________________________
+Bool_t AliPHOSFastRecParticle::IsElectron(const char* purity) const
+{
+  // Rec.Particle is an electron if it has a photon-like shape, fast and charged
+  // photon-like shape is defined with a purity "low", "medium" or "high"
+
+  Bool_t photonLike = kFALSE;
+  if      (purity == "low"   ) photonLike = TestPIDBit(6);
+  else if (purity == "medium") photonLike = TestPIDBit(7);
+  else if (purity == "high"  ) photonLike = TestPIDBit(8);
+  else Error("IsElectron","Wrong purity type: %s",purity);
+  if (photonLike                                   && //  photon by PCA
+      (TestPIDBit(5)||TestPIDBit(4)||TestPIDBit(3))&& //  fast by TOF
+     !(TestPIDBit(2)||TestPIDBit(1)||TestPIDBit(0)))  //  charged by CPV
+    return kTRUE ;
+  else
+    return kFALSE;
+}
+
+//____________________________________________________________________________
+Bool_t AliPHOSFastRecParticle::IsHadron() const
+{
+  // Rec.Particle is an hadron if it does not look like
+  // a low-purity photon nor low-purity pi0
+
+  if ( !TestPIDBit(6) && !TestPIDBit(9) )             // not photon nor pi0
+    return kTRUE ;
+  else
+    return kFALSE;
+}
+
+//____________________________________________________________________________
+Bool_t AliPHOSFastRecParticle::IsChargedHadron() const
+{
+  // Rec.Particle is a charged hadron if it does not look like
+  // a low-purity photon nor low-purity pi0 and is low-purity charged
+
+  if ( !TestPIDBit(6) && !TestPIDBit(9) &&            // not photon nor pi0
+       !TestPIDBit(2))                                // charged by CPV
+    return kTRUE ;
+  else
+    return kFALSE;
+}
+
+//____________________________________________________________________________
+Bool_t AliPHOSFastRecParticle::IsNeutralHadron() const
+{
+  // Rec.Particle is a neutral hadron if it does not look like
+  // a low-purity photon nor low-purity pi0 and is high-purity neutral
+
+  if ( !TestPIDBit(6) && !TestPIDBit(9) &&            // not photon nor pi0
+        TestPIDBit(2))                                // neutral by CPV
+    return kTRUE ;
+  else
+    return kFALSE;
+}
+
+//____________________________________________________________________________
+Bool_t AliPHOSFastRecParticle::IsFastChargedHadron() const
+{
+  // Rec.Particle is a fast charged hadron if it does not look like
+  // a low-purity photon nor low-purity pi0, is low-purity charged
+  // and is high-purity fast
+
+  if ( !TestPIDBit(6) && !TestPIDBit(9) &&            // not photon nor pi0
+       !TestPIDBit(2) &&                              // charged by CPV
+        TestPIDBit(5))                                // fast by TOF
+    return kTRUE ;
+  else
+    return kFALSE;
+}
+
+//____________________________________________________________________________
+Bool_t AliPHOSFastRecParticle::IsSlowChargedHadron() const
+{
+  // Rec.Particle is a slow neutral hadron if it does not look like
+  // a low-purity photon nor low-purity pi0, is high-purity neutral
+  // and is not high-purity fast
+
+  if ( !TestPIDBit(6) && !TestPIDBit(9) &&            // not photon nor pi0
+       !TestPIDBit(2) &&                              // charged by CPV
+       !TestPIDBit(5))                                // slow by TOF
+    return kTRUE ;
+  else
+    return kFALSE;
+
+}
+
+//____________________________________________________________________________
+Bool_t AliPHOSFastRecParticle::IsFastNeutralHadron() const
+{
+  // Rec.Particle is a fast neutral hadron if it does not look like
+  // a low-purity photon nor low-purity pi0, is high-purity neutral
+  // and is high-purity fast
+
+  if ( !TestPIDBit(6) && !TestPIDBit(9) &&            // not photon nor pi0
+        TestPIDBit(2) &&                              // neutral by CPV
+        TestPIDBit(5))                                // fast by TOF
+    return kTRUE ;
+  else
+    return kFALSE;
+}
+
+//____________________________________________________________________________
+Bool_t AliPHOSFastRecParticle::IsSlowNeutralHadron() const
+{
+  // Rec.Particle is a slow neutral hadron if it does not look like
+  // a low-purity photon nor low-purity pi0, is high-purity neutral
+  // and is not high-purity fast
+
+  if ( !TestPIDBit(6) && !TestPIDBit(9) &&            // not photon nor pi0
+        TestPIDBit(2) &&                              // neutral by CPV
+       !TestPIDBit(5))                                // slow by TOF
+    return kTRUE ;
+  else
+    return kFALSE;
+}
+
+//____________________________________________________________________________
+TString AliPHOSFastRecParticle::Name() const
 {
   // Returns the name of the particle type (only valid if PIDv1 is employed)
 
   TString  name ; 
+
+  name = "Undefined particle" ;
   
-  if(fType == 127)
-    name = "PHOTON_LOPU_HIEF" ;    //PCA = 001 TOF = 111 CPV = 111
-  
-  if(fType == 511)
-    name = "PHOTON_HIPU_LOEF" ;    //PCA = 011 TOF = 111 CPV = 111
-  
-  if(fType == 255)
-       name = "PHOTON_MED_PU_EF" ;    //PCA = 111 TOF = 111 CPV = 111
-  
-  if((fType == 383)||(fType == 447)) 
-    name = "PHOTON_STRANGE" ;      //PCA = 101 or 110 TOF = 111 CPV = 111
-  
-  if(fType == 63)
-    name = "NEUTRAL_FAST_HADRON" ; //PCA = 000 TOF = 111 CPV = 111
-  
-  if((fType == 504) || (fType == 505) ||(fType == 248)||(fType == 249)||(fType == 120)||(fType == 121))
-    name = "CHARGED_FAST_EM" ;     //PCA = 111, 011 or 001 TOF =111 CPV = 000 or 001  
-  
-  if((fType == 56)||(fType == 57))
-    name = "CHARGED_FAST_HADRON" ; //PCA = 000 TOF = 111 CPV = 000 or 001 
-  
-  if((fType < 8)&&(fType > 0))
-    name = "NEUTRAL_SLOW_HADRON" ; //PCA = 000 TOF = 000 CPV = 001 or 011 or 111
-  
-  if((fType == 0))
-    name = "CHARGED_SLOW_HADRON" ; //PCA = 000 TOF = 000 CPV = 000
-  
-  if((fType == 448) || (fType == 449) ||(fType == 192)||(fType == 193)||(fType == 64)||(fType == 64))
-    name = "CHARGED_SLOW_EM" ;    //PCA = 111, 011 or 001 TOF =000 CPV = 000 or 001  
-  
+  if      (IsPhoton("low"))
+    name = "Photon low purity ";
+  else if (IsPhoton("medium"))
+    name = "Photon medium purity";
+  else if (IsPhoton("high"))
+    name = "Photon high purity ";
+
+  if      (IsPi0("low"))
+    name += "Pi0 low purity ";
+  else if (IsPi0("medium"))
+    name += "Pi0 medium purity ";
+  else if (IsPi0("high"))
+    name += "Pi0 high purity ";
+
+  if      (IsElectron("low"))
+    name += "Electron low purity ";
+  else if (IsElectron("medium"))
+    name += "Electron medium purity ";
+  else if (IsElectron("high"))
+    name += "Electron high purity ";
+
+  if     (IsHadron()) {
+    name = "hadron";
+    if      (IsChargedHadron()) {
+      name.Prepend("charged ");
+      if      (IsFastChargedHadron())
+       name.Prepend("fast ");
+      else if (IsSlowChargedHadron())
+       name.Prepend("slow ");
+    }
+    else if (IsNeutralHadron()) {
+      name.Prepend("neutral ");
+      if      (IsFastNeutralHadron())
+       name.Prepend("fast ");
+      else if (IsSlowNeutralHadron())
+       name.Prepend("slow ");
+    }
+  }
+
   return name ; 
 }
 
@@ -207,6 +377,12 @@ TString AliPHOSFastRecParticle::Name()const
 //______________________________________________________________________________
 void AliPHOSFastRecParticle::SetType(Int_t type) { 
   // sets the particle type 
+  // bit-mask of the particle type means the following:
+  // bits 0,1,2   - neutral particle with low, medium and high purity
+  // bits 3.4,5   - fast particle with low, medium and high purity
+  // bits 6.7,8   - photon shower with low, medium and high purity
+  // bits 9,10,11 - hard-pi0 shower with low, medium and high purity
+
   fType = type ; 
   
   if((type == 127) || (fType == 511) || (fType == 255) ||(fType == 383)||(fType == 447)){
@@ -262,14 +438,18 @@ void AliPHOSFastRecParticle::Paint(Option_t *)
 void AliPHOSFastRecParticle::Print(Option_t * opt)const
 {
   // Print the type, energy and momentum of the reconstructed particle
-  
+
   TString message ; 
-  message  = "AliPHOSFastRecParticle > type is  %s\n" ; 
-  message += "                         Energy = %e\n" ; 
-  message += "                         Px     = %e\n" ; 
-  message += "                         Py     = %e\n" ;
-  message += "                         Pz     = %e\n" ;
+  message  = "\n   PID bits are %d%d%d %d%d%d %d%d%d %d%d%d" ; 
+  message += ", type is \"%s\"\n" ; 
+  message += "   (E,Px,Py,Pz) = (% .3e, % .3e, % .3e, % .3e) GeV\n" ; 
   Info("Print", message.Data(), 
+       TestPIDBit(0),TestPIDBit(1),
+       TestPIDBit(2),TestPIDBit(3),
+       TestPIDBit(4),TestPIDBit(5),
+       TestPIDBit(6),TestPIDBit(7),
+       TestPIDBit(8),TestPIDBit(9),
+       TestPIDBit(10),TestPIDBit(11),
        Name().Data(), 
        Energy(), 
        Px(), 
index ceaf3c9..f3c2735 100644 (file)
@@ -7,14 +7,15 @@
 
 //_________________________________________________________________________
 //  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 ---
 
 #include "TParticle.h"
-#include "TVector3.h"
 
 // --- Standard library ---
 
@@ -28,9 +29,8 @@ class AliPHOSFastRecParticle : public TParticle {
   
   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) ;
@@ -45,95 +45,31 @@ class AliPHOSFastRecParticle : public TParticle {
     return fType ; 
   } 
   
-  void SetPIDBit(UInt_t fSet)
-    {
-      fType |= (1<<fSet) ; 
-    } 
-  
-  Bool_t TestPIDBit(UInt_t fTest){
-    if (fType & (1<<fTest) )
-      return  kTRUE ;  
-    else
-      return kFALSE ;
-  }
-  
-  Bool_t IsPhotonHiPuLoEf()  {
-    Bool_t pid=kFALSE ;
-    if(TestPIDBit(8)&& //PCA
-       TestPIDBit(5)&& //TOF
-       TestPIDBit(2))  //RCPV
-      pid = kTRUE;
-    return pid ;
-  }
-  
-  Bool_t IsPhotonMedPuEf(){
-    Bool_t pid=kFALSE ;
-    if(TestPIDBit(7)&& //PCA
-       TestPIDBit(4)&& //TOF
-       TestPIDBit(1))  //RCPV
-      pid = kTRUE ;
-    return pid ;
-  }
+  void SetPIDBit(UInt_t fSet) {
+    // Set PID bit number fSet
+    fType |= (1<<fSet) ; 
+  } 
   
-  Bool_t IsPhotonHiEfLoPu()  {
-    Bool_t pid=kFALSE ;
-    if(TestPIDBit(6)&& //PCA
-       TestPIDBit(3)&& //TOF
-       TestPIDBit(0))  //RCPV
-      pid = kTRUE ;
-    return pid ;
-  }
-  Bool_t IsPhoton()  {
-    Bool_t pid=kFALSE ;
-    if(IsPhotonHiEfLoPu()) pid = kTRUE ;
-    return pid ;
+  Bool_t TestPIDBit(UInt_t fTest) const {
+    // Check PID bit number fTest
+    if (fType & (1<<fTest) ) return  kTRUE ;   
+    else return kFALSE ;
   }
   
-  Bool_t IsFastChargedHadron()  {
-    Bool_t pid=kFALSE ;
-    if(TestPIDBit(5)&&TestPIDBit(4)&&TestPIDBit(3)) //TOF
-      pid = kTRUE ;
-    return pid ;
-  }
-  Bool_t IsSlowChargedHadron()  {
-    Bool_t pid=kFALSE ;
-    if(TestPIDBit(1)||TestPIDBit(0)) //CPV
-      pid = kTRUE ;
-    return pid ;
-  }
-  Bool_t IsFastNeutralHadron()  {
-    Bool_t pid=kFALSE ;
-    if(TestPIDBit(5)&&TestPIDBit(4)&&TestPIDBit(3)&& //TOF
-       TestPIDBit(2)&&TestPIDBit(1)&&TestPIDBit(0))//RCPV
-      pid = kTRUE ;
-    return pid ;
-  }
-  Bool_t IsSlowNeutralHadron()  {
-    Bool_t pid=kFALSE ;
-    if(TestPIDBit(2)&&TestPIDBit(1)&&TestPIDBit(0))//RCPV
-      pid = kTRUE ;
-    return pid ;
-  }
+  Bool_t IsPhoton           (const char* purity = "low") const;
+  Bool_t IsPi0              (const char* purity = "low") const;
+  Bool_t IsElectron         (const char* purity = "low") 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 IsFastChargedEM()  {
-    Bool_t pid=kFALSE ;
-    if( (TestPIDBit(8)||TestPIDBit(7)||TestPIDBit(6))&& //  PCA
-        (TestPIDBit(5)||TestPIDBit(4)||TestPIDBit(3))&& //  TOF
-       !(TestPIDBit(2)||TestPIDBit(1)||TestPIDBit(0)))  // !CPV
-      pid = kTRUE ;
-    return pid ;
-  }
-  
-  Bool_t IsSlowChargedEM()  {
-    Bool_t pid=kFALSE ;
-    if(TestPIDBit(8)||TestPIDBit(7)||TestPIDBit(6))
-      pid = kTRUE ;
-    return pid ;
-  }
-  
   TString Name() const ; 
   virtual void Paint(Option_t * option="");
-  virtual void Print(Option_t * option = "") const ; 
+  virtual void Print(Option_t * option="") const ; 
   
   void SetType(Int_t type) ;
   
@@ -146,7 +82,6 @@ class AliPHOSFastRecParticle : public TParticle {
                       kNEUTRALEMFAST, kNEUTRALHAFAST,  kNEUTRALEMSLOW, kNEUTRALHASLOW, 
                       kCHARGEDEMFAST, kCHARGEDHAFAST,  kCHARGEDEMSLOW, kCHARGEDHASLOW } ; 
   
-  
   typedef TClonesArray  FastRecParticlesList ; 
   
  protected:
@@ -155,7 +90,6 @@ class AliPHOSFastRecParticle : public TParticle {
   Int_t fType ;        // particle type obtained by "virtual" reconstruction
  private:
 
-
   ClassDef(AliPHOSFastRecParticle,2)  // Reconstructed Particle produced by the fast simulation 
 
 };