1)AliCaloPID: Posibility to recalculate PID bayesian in EMCAL
authorgconesab <gconesab@f7af4fe6-9843-0410-8265-dc069ae4e863>
Tue, 1 Sep 2009 18:49:36 +0000 (18:49 +0000)
committergconesab <gconesab@f7af4fe6-9843-0410-8265-dc069ae4e863>
Tue, 1 Sep 2009 18:49:36 +0000 (18:49 +0000)
2)AliAnalysisTaskEMCALPi0CalibSelection: Corrected number of supermodules and towers
Few coding violations corrected.

PWG4/CaloCalib/AliAnalysisTaskEMCALPi0CalibSelection.cxx
PWG4/CaloCalib/AliAnalysisTaskEMCALPi0CalibSelection.h
PWG4/PartCorrBase/AliCaloPID.cxx
PWG4/PartCorrBase/AliCaloPID.h

index b6c154370fa5dbf0b185577d08b8db68b9deb97a..36c0e53e088ed58fdd82b84f03af068e6bb5e470 100644 (file)
@@ -52,9 +52,9 @@ fEMCALGeoName("EMCAL_COMPLETE"), fOldData(kFALSE),fOutputContainer(0x0),fHmgg(0x
 {
   //Default constructor.
 
-  for(Int_t iMod=0; iMod<5; iMod++) {
-    for(Int_t iX=0; iX<64; iX++) {
-      for(Int_t iZ=0; iZ<56; iZ++) {
+  for(Int_t iMod=0; iMod < 12; iMod++) {
+    for(Int_t iX=0; iX<24; iX++) {
+      for(Int_t iZ=0; iZ<48; iZ++) {
        fHmpi0[iMod][iX][iZ]=0;
       }
     } 
@@ -71,9 +71,9 @@ AliAnalysisTaskEMCALPi0CalibSelection::AliAnalysisTaskEMCALPi0CalibSelection(con
   
   DefineOutput(1,TList::Class());
   
-  for(Int_t iMod=0; iMod<5; iMod++) {
-    for(Int_t iX=0; iX<64; iX++) {
-      for(Int_t iZ=0; iZ<56; iZ++) {
+  for(Int_t iMod=0; iMod < 12; iMod++) {
+    for(Int_t iX=0; iX<24; iX++) {
+      for(Int_t iZ=0; iZ<48; iZ++) {
        fHmpi0[iMod][iX][iZ]=0;
       }
     } 
@@ -431,7 +431,7 @@ void AliAnalysisTaskEMCALPi0CalibSelection::UserExec(Option_t* /* option */)
   else{        
     if(DebugLevel() > 1) printf("AliAnalysisTaskEMCALPi0CalibSelection Load Misaligned matrices. \n");
     AliESDEvent* esd = dynamic_cast<AliESDEvent*>(InputEvent()) ;
-    for(Int_t mod=0; mod<5; mod++){ 
+    for(Int_t mod=0; mod < 12; mod++){ 
       if(esd->GetEMCALMatrix(mod)) fEMCALGeo->SetMisalMatrix(esd->GetEMCALMatrix(mod),mod) ;
     }
   }
@@ -561,7 +561,7 @@ void AliAnalysisTaskEMCALPi0CalibSelection::UserExec(Option_t* /* option */)
 }
 
 //__________________________________________________
-void AliAnalysisTaskEMCALPi0CalibSelection::MaxEnergyCellPos(AliAODCaloCells *cells, AliAODCaloCluster* clu, Int_t& maxId)
+void AliAnalysisTaskEMCALPi0CalibSelection::MaxEnergyCellPos(AliAODCaloCells* const cells, AliAODCaloCluster* const clu, Int_t& maxId)
 {
   //For a given CaloCluster calculates the absId of the cell 
   //with maximum energy deposit.
@@ -580,7 +580,7 @@ void AliAnalysisTaskEMCALPi0CalibSelection::MaxEnergyCellPos(AliAODCaloCells *ce
 }
 
 //__________________________________________________
-void AliAnalysisTaskEMCALPi0CalibSelection::SetCalibCorrections(AliEMCALCalibData* cdata)
+void AliAnalysisTaskEMCALPi0CalibSelection::SetCalibCorrections(AliEMCALCalibData* const cdata)
 {
   //Set new correction factors (~1) to calibration coefficients, delete previous.
 
index 847670158aed73979fa309ac624b4256699219de..12b4e4332194cbac6e31c081efdf2b5572ee94f4 100644 (file)
@@ -36,7 +36,7 @@ public:
   
   void SetClusterMinEnergy(Float_t emin) {fEmin=emin;}
   void SetLogWeight(Float_t weight) {fLogWeight=weight;}
-  void SetCalibCorrections(AliEMCALCalibData* cdata);
+  void SetCalibCorrections(AliEMCALCalibData* const cdata);
   void CreateAODFromESD();
   void CreateAODFromAOD();     
 
@@ -51,7 +51,7 @@ public:
        
 private:
 
-  void MaxEnergyCellPos(AliAODCaloCells *cells, AliAODCaloCluster* clu, Int_t& maxId);
+  void MaxEnergyCellPos(AliAODCaloCells* const cells, AliAODCaloCluster* const clu, Int_t& maxId);
 
 private:
 
index d1209209b354e868089f7a0b49648a8d6897ca0d..7bb34f6527fe27cfe0013d6b8fbdd59eef945fa8 100755 (executable)
 /* $Id: AliCaloPID.cxx 21839 2007-10-29 13:49:42Z gustavo $ */
 
 //_________________________________________________________________________
-// Class for track/cluster acceptance selection
-// Selection in Central barrel, EMCAL and PHOS
-//                
+// Class for PID selection with calorimeters
+// The Output of the 2 main methods GetPdg is a PDG number identifying the cluster, 
+// being kPhoton, kElectron, kPi0 ... as defined in the header file
+//   - GetPdg(const TString calo, const Double_t * pid, const Float_t energy)
+//      Reads the PID weights array of the ESDs and depending on its magnitude identifies the particle
+//   - GetPdg(const TString calo,const TLorentzVector mom, const AliAODCaloCluster * cluster)
+//      Recalcultes PID, the bayesian or any new one to be implemented in the future
+//      Right now only the possibility to recalculate EMCAL with bayesian and simple PID.
+//      In order to recalculate Bayesian, it is necessary to load the EMCALUtils library
+//      and do SwitchOnBayesianRecalculation().
+//      To change the PID parameters from Low to High like the ones by default, use the constructor 
+//      AliCaloPID(flux)
+//      where flux is AliCaloPID::kLow or AliCaloPID::kHigh
+//      If it is necessary to change the parameters use the constructor 
+//      AliCaloPID(AliEMCALPIDUtils *utils) and set the parameters before.
+//   - SetPIDBits: Simple PID, depending on the thresholds fDispCut fTOFCut and even the
+//     result of the PID bayesian a different PID bit is set. 
+//
+//  All these methods can be called in the analysis you are interested.
+//
 //*-- Author: Gustavo Conesa (LNF-INFN) 
 //////////////////////////////////////////////////////////////////////////////
 
@@ -31,7 +48,9 @@
 #include "AliCaloPID.h"
 #include "AliAODCaloCluster.h"
 #include "AliAODPWG4Particle.h"
-
+#ifdef __EMCALUTIL__
+#include "AliEMCALPIDUtils.h"
+#endif
 ClassImp(AliCaloPID)
 
 
@@ -44,7 +63,11 @@ fPHOSPhotonWeight(0.), fPHOSPi0Weight(0.),
 fPHOSElectronWeight(0.), fPHOSChargeWeight(0.) , 
 fPHOSNeutralWeight(0.), fPHOSWeightFormula(0), 
 fPHOSPhotonWeightFormula(0x0), fPHOSPi0WeightFormula(0x0),
-fDispCut(0.),fTOFCut(0.), fDebug(-1)
+fDispCut(0.),fTOFCut(0.), fDebug(-1), 
+fRecalculateBayesian(kFALSE), fParticleFlux(kLow)
+#ifdef __EMCALUTIL__
+, fEMCALPIDUtils(new AliEMCALPIDUtils)
+#endif
 {
   //Ctor
   
@@ -52,6 +75,48 @@ fDispCut(0.),fTOFCut(0.), fDebug(-1)
   InitParameters();
 }
 
+//________________________________________________
+AliCaloPID::AliCaloPID(const Int_t flux) : 
+TObject(), fEMCALPhotonWeight(0.), fEMCALPi0Weight(0.),  
+fEMCALElectronWeight(0.),  fEMCALChargeWeight(0.),
+fEMCALNeutralWeight(0.),
+fPHOSPhotonWeight(0.), fPHOSPi0Weight(0.),  
+fPHOSElectronWeight(0.), fPHOSChargeWeight(0.) , 
+fPHOSNeutralWeight(0.), fPHOSWeightFormula(0), 
+fPHOSPhotonWeightFormula(0x0), fPHOSPi0WeightFormula(0x0),
+fDispCut(0.),fTOFCut(0.), fDebug(-1), 
+fRecalculateBayesian(kTRUE), fParticleFlux(flux)
+#ifdef __EMCALUTIL__
+, fEMCALPIDUtils(new AliEMCALPIDUtils)
+#endif
+{
+       //Ctor
+       
+       //Initialize parameters
+       InitParameters();
+}
+
+//________________________________________________
+AliCaloPID::AliCaloPID(const TTask * emcalpid) : 
+TObject(), fEMCALPhotonWeight(0.), fEMCALPi0Weight(0.),  
+fEMCALElectronWeight(0.),  fEMCALChargeWeight(0.),
+fEMCALNeutralWeight(0.),
+fPHOSPhotonWeight(0.), fPHOSPi0Weight(0.),  
+fPHOSElectronWeight(0.), fPHOSChargeWeight(0.) , 
+fPHOSNeutralWeight(0.), fPHOSWeightFormula(0), 
+fPHOSPhotonWeightFormula(0x0), fPHOSPi0WeightFormula(0x0),
+fDispCut(0.),fTOFCut(0.), fDebug(-1), 
+fRecalculateBayesian(kTRUE), fParticleFlux(-1)
+#ifdef __EMCALUTIL__
+, fEMCALPIDUtils( (AliEMCALPIDUtils*) emcalpid)
+#endif
+{
+       //Ctor
+       
+       //Initialize parameters
+       InitParameters();
+}
+
 //____________________________________________________________________________
 AliCaloPID::AliCaloPID(const AliCaloPID & pid) :   
 TObject(pid), fEMCALPhotonWeight(pid.fEMCALPhotonWeight), 
@@ -68,7 +133,11 @@ fPHOSWeightFormula(pid.fPHOSWeightFormula),
 fPHOSPhotonWeightFormula(pid.fPHOSPhotonWeightFormula), 
 fPHOSPi0WeightFormula(pid.fPHOSPi0WeightFormula), 
 fDispCut(pid.fDispCut),fTOFCut(pid.fTOFCut),
-fDebug(pid.fDebug)
+fDebug(pid.fDebug), fRecalculateBayesian(pid.fRecalculateBayesian),
+fParticleFlux(pid.fParticleFlux)
+#ifdef __EMCALUTIL__
+, fEMCALPIDUtils(pid.fEMCALPIDUtils)
+#endif
 {
   // cpy ctor
        
@@ -101,6 +170,13 @@ AliCaloPID & AliCaloPID::operator = (const AliCaloPID & pid)
   fTOFCut   = pid.fTOFCut;
   fDebug    = pid.fDebug;
   
+  fRecalculateBayesian = pid.fRecalculateBayesian;
+  fParticleFlux        = pid.fParticleFlux;
+#ifdef __EMCALUTIL__
+  fEMCALPIDUtils       = pid.fEMCALPIDUtils;
+#endif
+       
+       
   return *this;
   
 }
@@ -111,7 +187,9 @@ AliCaloPID::~AliCaloPID() {
   
   if(fPHOSPhotonWeightFormula) delete  fPHOSPhotonWeightFormula ;
   if(fPHOSPi0WeightFormula) delete  fPHOSPi0WeightFormula ;
-  
+#ifdef __EMCALUTIL__
+       if(fEMCALPIDUtils) delete fEMCALPIDUtils ;
+#endif
 }
 
 
@@ -142,6 +220,19 @@ void AliCaloPID::InitParameters()
   fDispCut  = 1.5;
   fTOFCut   = 5.e-9;
   fDebug = -1;
+       
+  if(fRecalculateBayesian){
+#ifdef __EMCALUTIL__
+       if(fParticleFlux == kLow){
+               printf("AliCaloPID::Init() - SetLOWFluxParam\n");
+               fEMCALPIDUtils->SetLowFluxParam() ;
+       }
+       else if (fParticleFlux == kHigh){
+               printf("AliCaloPID::Init() - SetHIGHFluxParam\n");
+               fEMCALPIDUtils->SetHighFluxParam() ;
+       }
+#endif
+  }
 }
 
 //_______________________________________________________________
@@ -202,6 +293,7 @@ Int_t AliCaloPID::GetPdg(const TString calo, const Double_t * pid, const Float_t
       pdg = kNeutralUnknown ;
   }
   else{//EMCAL
+
     if(pid[AliAODCluster::kPhoton]+pid[AliAODCluster::kElectron]  > wPh) pdg = kPhoton ; //temporal sollution until track matching for electrons is considered
     //if(pid[AliAODCluster::kPhoton]  > wPh) pdg = kPhoton ;
     else if(pid[AliAODCluster::kPi0] > wPi0) pdg = kPi0 ; 
@@ -221,12 +313,38 @@ Int_t AliCaloPID::GetPdg(const TString calo, const Double_t * pid, const Float_t
 //_______________________________________________________________
 Int_t AliCaloPID::GetPdg(const TString calo,const TLorentzVector mom, const AliAODCaloCluster * cluster) const {
   //Recalculated PID with all parameters
-  if(fDebug > 0)printf("AliCaloPID::GetPdg: Calorimeter %s, E %3.2f, l0 %3.2f, l1 %3.2f, disp %3.2f, tof %1.11f, distCPV %3.2f, distToBC %1.1f, NMax %d\n",
-                      calo.Data(),mom.E(),cluster->GetM02(),cluster->GetM20(),cluster->GetDispersion(),cluster->GetTOF(), 
-                      cluster->GetEmcCpvDistance(), cluster->GetDistToBadChannel(),cluster->GetNExMax());
-  
+  Float_t lambda0 = cluster->GetM02();
+  Float_t energy = mom.E();    
+
+  if(fDebug > 0) printf("AliCaloPID::GetPdg: Calorimeter %s, E %3.2f, l0 %3.2f, l1 %3.2f, disp %3.2f, tof %1.11f, distCPV %3.2f, distToBC %1.1f, NMax %d\n",
+                                               calo.Data(),energy,lambda0,cluster->GetM20(),cluster->GetDispersion(),cluster->GetTOF(), 
+                                               cluster->GetEmcCpvDistance(), cluster->GetDistToBadChannel(),cluster->GetNExMax());
+
   if(calo == "EMCAL") {
-    if(cluster->GetM02() < 0.25) return kPhoton ;
+         //Recalculate Bayesian
+         if(fRecalculateBayesian){       
+#ifdef __EMCALUTIL__
+                
+                 if(fDebug > 0)  {
+                         const Double_t  *pid0 = cluster->PID();
+                         printf("AliCaloPID::GetPdg: BEFORE calo %s, ph %0.2f, pi0 %0.2f, el %0.2f, conv el %0.2f, hadrons: pion %0.2f, kaon %0.2f, proton %0.2f , neutron %0.2f, kaon %0.2f \n",
+                                                                calo.Data(),pid0[AliAODCluster::kPhoton], pid0[AliAODCluster::kPi0],
+                                                                pid0[AliAODCluster::kElectron], pid0[AliAODCluster::kEleCon],
+                                                                pid0[AliAODCluster::kPion], pid0[AliAODCluster::kKaon], pid0[AliAODCluster::kProton],
+                                                                pid0[AliAODCluster::kNeutron], pid0[AliAODCluster::kKaon0]);
+                 }
+                 
+                fEMCALPIDUtils->ComputePID(energy, lambda0);
+                Double_t pid[AliPID::kSPECIESN];
+                for(Int_t i = 0; i < AliPID::kSPECIESN; i++) pid[i] = fEMCALPIDUtils->GetPIDFinal(i);
+                return GetPdg(calo, pid, energy);
+                 
+#endif
+         
+       }
+         
+       // If no use of bayesian, simple PID  
+    if(lambda0 < 0.25) return kPhoton ;
     //else return  kNeutralHadron ; 
        else return  kPi0 ;
   }
@@ -310,7 +428,8 @@ void AliCaloPID::Print(const Option_t * opt) const
   printf("TOF cut        = %e\n",fTOFCut);
   printf("Dispersion cut = %2.2f\n",fDispCut);
   printf("Debug level    = %d\n",fDebug);
-  
+  printf("Recalculate Bayesian?    = %d\n",fRecalculateBayesian);
+ if(fRecalculateBayesian) printf("Particle Flux?    = %d\n",fParticleFlux);
   printf(" \n");
   
 } 
index e190c38a1b9cd986874be0f160aeed70b1512efb..75699cf3a432126bf86ce1ede2e38b7da15a4d63 100755 (executable)
@@ -5,8 +5,25 @@
 /* $Id:  $ */
 
 //_________________________________________________________________________
-// Class for track/cluster acceptance selection
-// Selection in Central barrel, EMCAL and PHOS
+// Class for PID selection with calorimeters
+// The Output of the 2 main methods GetPdg is a PDG number identifying the cluster, 
+// being kPhoton, kElectron, kPi0 ... as defined in the header file
+//   - GetPdg(const TString calo, const Double_t * pid, const Float_t energy)
+//      Reads the PID weights array of the ESDs and depending on its magnitude identifies the particle
+//   - GetPdg(const TString calo,const TLorentzVector mom, const AliAODCaloCluster * cluster)
+//      Recalcultes PID, the bayesian or any new one to be implemented in the future
+//      Right now only the possibility to recalculate EMCAL with bayesian and simple PID.
+//      In order to recalculate Bayesian, it is necessary to load the EMCALUtils library
+//      and do SwitchOnBayesianRecalculation().
+//      To change the PID parameters from Low to High like the ones by default, use the constructor 
+//      AliCaloPID(flux)
+//      where flux is AliCaloPID::kLow or AliCaloPID::kHigh
+//      If it is necessary to change the parameters use the constructor 
+//      AliCaloPID(AliEMCALPIDUtils *utils) and set the parameters before.
+//   - SetPIDBits: Simple PID, depending on the thresholds fDispCut fTOFCut and even the
+//     result of the PID bayesian a different PID bit is set. 
+//
+//  All these methods can be called in the analysis you are interested.
 //
 //*-- Author: Gustavo Conesa (INFN-LNF)
 
 class TString ;
 class TLorentzVector ;
 class TFormula ;
+class TTask;
 
 //--- AliRoot system ---
 class AliAODCaloCluster;
 class AliAODPWG4Particle;
+#ifdef __EMCALUTIL__
+class AliEMCALPIDUtils;
+#endif
 
 class AliCaloPID : public TObject {
        
  public: 
   
   AliCaloPID() ; // ctor
+  AliCaloPID(const Int_t particleFlux) ; // ctor, to be used when recalculating bayesian PID
+  AliCaloPID(const TTask * emcalpid) ; // ctor, to be used when recalculating bayesian PID and need different parameters
   AliCaloPID(const AliCaloPID & g) ; // cpy ctor
   AliCaloPID & operator = (const AliCaloPID & g) ;//cpy assignment
   virtual ~AliCaloPID() ;//virtual dtor
@@ -84,8 +107,8 @@ class AliCaloPID : public TObject {
   void SetPHOSNeutralWeight(Float_t  w){  fPHOSNeutralWeight = w ; }
   
   void UsePHOSPIDWeightFormula(Bool_t par)  { fPHOSWeightFormula  = par; } 
-  void SetPHOSPhotonWeightFormula(TFormula * photon)    {  fPHOSPhotonWeightFormula  = photon; } 
-  void SetPHOSPi0WeightFormula(TFormula * pi0)   {  fPHOSPi0WeightFormula  = pi0; }
+  void SetPHOSPhotonWeightFormula(TFormula * const photon)    {  fPHOSPhotonWeightFormula  = photon; } 
+  void SetPHOSPi0WeightFormula(TFormula * const pi0)   {  fPHOSPi0WeightFormula  = pi0; }
   
   //PID bits setters and getters
   void SetDispersionCut(Float_t dcut ) {fDispCut = dcut; }
@@ -97,6 +120,12 @@ class AliCaloPID : public TObject {
   void SetDebug(Int_t deb) {fDebug=deb;}
   Int_t GetDebug() const {return fDebug;}      
   
+  void SwitchOnBayesianRecalculation()  {fRecalculateBayesian = kTRUE ; }
+  void SwitchOffBayesianRecalculation() {fRecalculateBayesian = kFALSE; }
+  enum eventType{kLow,kHigh};
+  void SetLowParticleFlux()  {fParticleFlux = kLow;}
+  void SetHighParticleFlux() {fParticleFlux = kHigh;}
+
  private:
   
   Float_t      fEMCALPhotonWeight; //Bayesian PID weight for photons in EMCAL 
@@ -118,8 +147,15 @@ class AliCaloPID : public TObject {
   Float_t fTOFCut;     //Cut on TOF, used in PID evaluation
   
   Int_t         fDebug; //Debug level
-  
-  ClassDef(AliCaloPID,3)
+       
+  Bool_t fRecalculateBayesian; // Recalculate PID bayesian or use simple PID?
+  Int_t  fParticleFlux;        // Particle flux for setting PID parameters
+#ifdef __EMCALUTIL__
+       AliEMCALPIDUtils * fEMCALPIDUtils; //Pointer to EMCALPID to redo the PID Bayesian calculation
+#endif
+       
+       
+  ClassDef(AliCaloPID,4)
     } ;