Cloning PHOS

diff --git a/EMCAL/AliEMCALFastRecParticle.cxx b/EMCAL/AliEMCALFastRecParticle.cxx
index 4bf81c5bde5ca226c67edccaa11555d5895f93ae..e595509c21797ce4acef872b7a3a82bddf6b26c0 100644 (file)
--- a/EMCAL/AliEMCALFastRecParticle.cxx
+++ b/EMCAL/AliEMCALFastRecParticle.cxx
@@ -168,49 +168,285 @@ void AliEMCALFastRecParticle::ExecuteEvent(Int_t event, Int_t /*px*/, Int_t /*py
 }
 
 //____________________________________________________________________________
-TString AliEMCALFastRecParticle::Name()const
+Bool_t AliEMCALFastRecParticle::IsPhoton(TString purity) const
 {
-  // Returns the name of the particle type (only valid if PIDv1 is employed)
+  //FIXME : just a blind copy from PHOS
+  // 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"
 
-  TString  name ; 
-  
-  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
+  purity.ToLower();
+  Bool_t photonLike = kFALSE;
+  if      (purity == "low"   ) photonLike = TestPIDBit(6);
+  else if (purity == "medium") photonLike = TestPIDBit(7);
+  else if (purity == "high"  ) photonLike = TestPIDBit(8);
+  if (photonLike                                   && //  photon by PCA
+      (TestPIDBit(5)||TestPIDBit(4)||TestPIDBit(3))&& //  fast by TOF
+      (TestPIDBit(2)||TestPIDBit(1)||TestPIDBit(0))&& //  neutral by CPV
+      !TestPIDBit(14))                              //  no charged track
+    return kTRUE ;
+  else
+    return kFALSE;
+}
+
+//____________________________________________________________________________
+Bool_t AliEMCALFastRecParticle::IsPi0(TString purity) const
+{
+  //FIXME : just a blind copy from PHOS
+  // 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"
+
+  purity.ToLower();
+  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.Data());
+  if (pi0Like                                      && //  pi0 by PCA
+      (TestPIDBit(5)||TestPIDBit(4)||TestPIDBit(3))&& //  fast by TOF
+      (TestPIDBit(2)||TestPIDBit(1)||TestPIDBit(0))&& //  neutral by CPV
+      !TestPIDBit(14))                              //  no charged track
+    return kTRUE ;
+  else
+    return kFALSE;
+}
+
+//____________________________________________________________________________
+Bool_t AliEMCALFastRecParticle::IsElectron(TString purity) const
+{
+  //FIXME : just a blind copy from PHOS
+  // 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"
+
+  purity.ToLower();
+  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.Data());
   
-  if((fType == 0))
-    name = "CHARGED_SLOW_HADRON" ; //PCA = 000 TOF = 000 CPV = 000
+  if (photonLike                                   && //  photon by PCA
+      (TestPIDBit(5)|| TestPIDBit(4)|| TestPIDBit(3))&& //  fast by TOF
+      (!TestPIDBit(2)||!TestPIDBit(1)||!TestPIDBit(0))&& //  charged by CPV
+      TestPIDBit(14))                                  //  no charged track
+    return kTRUE ;
+  else
+    return kFALSE;
+}
+
+//____________________________________________________________________________
+Bool_t AliEMCALFastRecParticle::IsEleCon(TString purity) const
+{
+  //FIXME : just a blind copy from PHOS
+  // 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"
+
+  purity.ToLower();
+  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.Data());
   
-  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 (photonLike                                   && //  photon by PCA
+      (TestPIDBit(5)|| TestPIDBit(4)|| TestPIDBit(3))&& //  fast by TOF
+      (!TestPIDBit(2)||!TestPIDBit(1)||!TestPIDBit(0))&& //  charged by CPV
+      !TestPIDBit(14))                                  //  no charged track
+    return kTRUE ;
+  else
+    return kFALSE;
+}
+
+//____________________________________________________________________________
+Bool_t AliEMCALFastRecParticle::IsHardPhoton() const
+{
+  //FIXME : just a blind copy from PHOS
+  // Rec.Particle is a hard photon (E > 30 GeV) if its second moment M2x
+  // corresponds to photons
+  if (TestPIDBit(12) && !TestPIDBit(14))
+    return kTRUE;
+  else
+    return kFALSE;
+}
+
+//____________________________________________________________________________
+Bool_t AliEMCALFastRecParticle::IsHardPi0() const
+{
+  //FIXME : just a blind copy from EMCAL
+  // Rec.Particle is a hard pi0 (E > 30 GeV) if its second moment M2x
+  // corresponds to pi0
+  if (TestPIDBit(13)&& !TestPIDBit(14))
+    return kTRUE;
+  else
+    return kFALSE;
+}
+
+//____________________________________________________________________________
+Bool_t AliEMCALFastRecParticle::IsHadron() const
+{
+  //FIXME : just a blind copy from EMCAL
+  // 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 AliEMCALFastRecParticle::IsChargedHadron() const
+{
+  //FIXME : just a blind copy from EMCAL
+  // 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 AliEMCALFastRecParticle::IsNeutralHadron() const
+{
+  //FIXME : just a blind copy from EMCAL
+  // 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 AliEMCALFastRecParticle::IsFastChargedHadron() const
+{
+  //FIXME : just a blind copy from EMCAL
+  // 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 AliEMCALFastRecParticle::IsSlowChargedHadron() const
+{
+  //FIXME : just a blind copy from EMCAL
+  // 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 AliEMCALFastRecParticle::IsFastNeutralHadron() const
+{
+  //FIXME : just a blind copy from EMCAL
+  // 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 AliEMCALFastRecParticle::IsSlowNeutralHadron() const
+{
+  //FIXME : just a blind copy from EMCAL
+  // 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 AliEMCALFastRecParticle::Name()const
+{
+   // Returns the name of the particle type (only valid if PIDv1 is employed)
+
+  TString  name ; 
+
+  name = "Undefined particle" ;
   
+  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 ; 
 }
 
-
 //______________________________________________________________________________
 void AliEMCALFastRecParticle::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)){

diff --git a/EMCAL/AliEMCALFastRecParticle.h b/EMCAL/AliEMCALFastRecParticle.h
index e2d67ef2afbd98b7f7dbc8d997c0dd05e9745e55..db8b731069a6dbe9c4327ece10cb6212491de6c2 100644 (file)
--- a/EMCAL/AliEMCALFastRecParticle.h
+++ b/EMCAL/AliEMCALFastRecParticle.h
@@ -44,7 +44,32 @@ class AliEMCALFastRecParticle : public TParticle {
     // returns the type of the particle
     return fType ; 
   } 
+ 
+  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(Option_t * option = "") const ; 

diff --git a/EMCAL/AliEMCALRecParticle.cxx b/EMCAL/AliEMCALRecParticle.cxx
index 37e49ba2accd62e2bbb709bc7e4c759985056675..a3e7cd322240347559b1341300c63fda076cd930 100644 (file)
--- a/EMCAL/AliEMCALRecParticle.cxx
+++ b/EMCAL/AliEMCALRecParticle.cxx
@@ -36,6 +36,14 @@
 ClassImp(AliEMCALRecParticle)
 
 
+//____________________________________________________________________________
+  AliEMCALRecParticle::AliEMCALRecParticle(): fEMCALRecPoint(0), fDebug(kFALSE)
+{
+  // ctor
+  const Int_t nSPECIES = AliESDtrack::kSPECIESN;
+  for(Int_t i = 0; i<nSPECIES ; i++)
+    fPID[i]=0.;
+}
 //____________________________________________________________________________
 AliEMCALRecParticle::AliEMCALRecParticle(const AliEMCALRecParticle & rp)
   : AliEMCALFastRecParticle(rp)
@@ -66,7 +74,9 @@ AliEMCALRecParticle::AliEMCALRecParticle(const AliEMCALRecParticle & rp)
   fPolarTheta  = rp.fPolarTheta;
   fPolarPhi    = rp.fPolarPhi;
   fParticlePDG = rp.fParticlePDG; 
-  
+  const Int_t nSPECIES = AliESDtrack::kSPECIESN;
+  for(Int_t i = 0; i<nSPECIES ; i++)
+    fPID[i]=rp.fPID[i];  
 }
 
 //____________________________________________________________________________
@@ -104,3 +114,49 @@ const TParticle * AliEMCALRecParticle::GetPrimary(Int_t index) const
   return gime->Primary(primaryindex) ;
 }
 
+//____________________________________________________________________________
+const Double_t * AliEMCALRecParticle::GetPID()
+{
+  // Get the probability densities that this reconstructed particle
+  // has a type of i:
+  // i       particle types
+  // ----------------------
+  // 0       electron
+  // 1       muon
+  // 2       pi+-
+  // 3       K+-
+  // 4       p/pbar
+  // 5       photon
+  // 6       pi0 at high pt
+  // 7       neutron
+  // 8       K0L
+
+ 
+  if (IsElectron()     ) fPID[0] = 1.0;
+  if (IsChargedHadron()) {
+    fPID[1] = 0.25;
+    fPID[2] = 0.25;
+    fPID[3] = 0.25;
+    fPID[4] = 0.25;
+  }
+  if (IsFastChargedHadron()) {
+    fPID[1] = 0.33;
+    fPID[2] = 0.33;
+    fPID[3] = 0.33;
+    fPID[4] = 0.00;
+  }
+  if (IsSlowChargedHadron()) {
+    fPID[1] = 0.00;
+    fPID[2] = 0.00;
+    fPID[3] = 0.00;
+    fPID[4] = 1.00;
+  }
+
+  if (IsPhoton() || IsHardPhoton()) fPID[5] = 1.0;
+  if (IsHardPi0())                  fPID[6] = 1.0;
+  if (IsFastNeutralHadron())        fPID[7] = 1.0;
+  if (IsSlowNeutralHadron())        fPID[8] = 1.0;
+
+  if (IsEleCon()) fPID[9] = 1.0;
+  return fPID;
+}

diff --git a/EMCAL/AliEMCALRecParticle.h b/EMCAL/AliEMCALRecParticle.h
index bc44dd866bcecc452ff41cd36e7ec418e85f3340..f17a1707c7fa1055525b566c87c01aeb348e3b89 100644 (file)
--- a/EMCAL/AliEMCALRecParticle.h
+++ b/EMCAL/AliEMCALRecParticle.h
@@ -17,6 +17,7 @@
 
 // --- AliRoot header files ---
 
+#include "AliESDtrack.h" 
 #include "AliEMCALFastRecParticle.h"
 class TParticle ;
 #include  "TVector3.h"  
@@ -25,7 +26,7 @@ class AliEMCALRecParticle : public AliEMCALFastRecParticle {
 
  public:
   
-  AliEMCALRecParticle() { fEMCALRecPoint = 0 ; fDebug = kFALSE ; } 
+  AliEMCALRecParticle() ;  
   AliEMCALRecParticle(const AliEMCALRecParticle & rp) ;  // ctor
   virtual ~AliEMCALRecParticle(){  }
 
@@ -34,6 +35,7 @@ class AliEMCALRecParticle : public AliEMCALFastRecParticle {
   virtual const Int_t GetNPrimaries() const ;
   TVector3 GetPos() const { return fPos ; } 
   virtual const TParticle * GetPrimary(Int_t index) 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 ; }
@@ -46,8 +48,9 @@ class AliEMCALRecParticle : public AliEMCALFastRecParticle {
   Int_t fEMCALRecPoint ; // pointer to the associated track segment in EMCAL  
   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(AliEMCALRecParticle,2)  // Reconstructed Particle
+  ClassDef(AliEMCALRecParticle,3)  // Reconstructed Particle
 };
 
 #endif // AliEMCALRECPARTICLE_H