ESDCheck/AliEMCALQATask.cxx

   1 /**************************************************************************
   2  * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   3  *                                                                        *
   4  * Author: The ALICE Off-line Project.                                    *
   5  * Contributors are mentioned in the code where appropriate.              *
   6  *                                                                        *
   7  * Permission to use, copy, modify and distribute this software and its   *
   8  * documentation strictly for non-commercial purposes is hereby granted   *
   9  * without fee, provided that the above copyright notice appears in all   *
  10  * copies and that both the copyright notice and this permission notice   *
  11  * appear in the supporting documentation. The authors make no claims     *
  12  * about the suitability of this software for any purpose. It is          *
  13  * provided "as is" without express or implied warranty.                  *
  14  **************************************************************************/
  15
  16 /* $Id$ */
  17
  18 //_________________________________________________________________________
  19 // An analysis task to check the EMCAL photon data in simulated data
  20 // An analysis task to check the EMCAL photon data in simulated data
  21 // An analysis task to check the EMCAL photon data in simulated data
  22 // An analysis task to check the EMCAL photon data in simulated data
  23 // An analysis task to check the EMCAL photon data in simulated data
  24 //
  25 //*-- Yves Schutz
  26 //////////////////////////////////////////////////////////////////////////////
  27
  28 #include <TCanvas.h>
  29 #include <TChain.h>
  30 #include <TFile.h>
  31 #include <TH1.h>
  32 #include <TNtuple.h>
  33 #include <TROOT.h>
  34 #include <TVector3.h>
  35 #include <TString.h>
  36
  37 #include "AliEMCALQATask.h"
  38 #include "AliESD.h"
  39 #include "AliLog.h"
  40
  41 //______________________________________________________________________________
  42 AliEMCALQATask::AliEMCALQATask(const char *name) :
  43   AliAnalysisTask(name,""),
  44   fChain(0),
  45   fESD(0),
  46   fOutputContainer(0),
  47   fhEMCALPos(0),
  48   fhEMCAL(0),
  49   fhEMCALEnergy(0),
  50   fhEMCALDigits(0),
  51   fhEMCALRecParticles(0),
  52   fhEMCALPhotons(0),
  53   fhEMCALInvariantMass(0),
  54   fhEMCALDigitsEvent(0)
  55 {
  56   // Constructor.
  57   // Input slot #0 works with an Ntuple
  58   DefineInput(0, TChain::Class());
  59   // Output slot #0 writes into a TH1 container
  60   DefineOutput(0,  TObjArray::Class()) ;
  61 }
  62
  63 //____________________________________________________________________________
  64 AliEMCALQATask::AliEMCALQATask(const AliEMCALQATask& ta) :
  65   AliAnalysisTask(ta.GetName(),""),
  66   fChain(ta.fChain),
  67   fESD(ta.fESD),
  68   fOutputContainer(ta.fOutputContainer),
  69   fhEMCALPos(ta.fhEMCALPos),
  70   fhEMCAL(ta.fhEMCAL),
  71   fhEMCALEnergy(ta.fhEMCALEnergy),
  72   fhEMCALDigits(ta.fhEMCALDigits),
  73   fhEMCALRecParticles(ta.fhEMCALRecParticles),
  74   fhEMCALPhotons(ta.fhEMCALPhotons),
  75   fhEMCALInvariantMass(ta.fhEMCALInvariantMass),
  76   fhEMCALDigitsEvent(ta.fhEMCALDigitsEvent)
  77 {
  78   // cpy ctor
  79 }
  80
  81 //_____________________________________________________________________________
  82 AliEMCALQATask& AliEMCALQATask::operator = (const AliEMCALQATask& ap)
  83 {
  84 // assignment operator
  85
  86   this->~AliEMCALQATask();
  87   new(this) AliEMCALQATask(ap);
  88   return *this;
  89 }
  90
  91 //______________________________________________________________________________
  92 AliEMCALQATask::~AliEMCALQATask()
  93 {
  94   // dtor
  95   fOutputContainer->Clear() ;
  96   delete fOutputContainer ;
  97
  98   delete fhEMCALPos ;
  99   delete fhEMCAL ;
 100   delete fhEMCALEnergy ;
 101   delete fhEMCALDigits ;
 102   delete fhEMCALRecParticles ;
 103   delete fhEMCALPhotons ;
 104   delete fhEMCALInvariantMass ;
 105   delete fhEMCALDigitsEvent ;
 106 }
 107
 108 //______________________________________________________________________________
 109 void AliEMCALQATask::ConnectInputData(const Option_t*)
 110 {
 111   // Initialisation of branch container and histograms
 112
 113   AliInfo(Form("*** Initialization of %s", GetName())) ;
 114
 115   // Get input data
 116   fChain = dynamic_cast<TChain *>(GetInputData(0)) ;
 117   if (!fChain) {
 118     AliError(Form("Input 0 for %s not found\n", GetName()));
 119     return ;
 120   }
 121
 122   // One should first check if the branch address was taken by some other task
 123   char ** address = (char **)GetBranchAddress(0, "ESD");
 124   if (address) {
 125     fESD = (AliESD*)(*address);
 126   } else {
 127     fESD = new AliESD();
 128     SetBranchAddress(0, "ESD", &fESD);
 129   }
 130 }
 131
 132 //______________________________________________________________________________
 133 void AliEMCALQATask::CreateOutputObjects()
 134 {
 135 // create histograms
 136
 137   OpenFile(0) ;
 138
 139   fhEMCALPos           = new TNtuple("EMCALPos"        , "Position in EMCAL" , "x:y:z");
 140   fhEMCAL              = new TNtuple("EMCAL"           , "EMCAL" , "event:digits:clusters:photons");
 141   fhEMCALEnergy        = new TH1D("EMCALEnergy"        , "EMCALEnergy"       , 1000, 0., 10. ) ;
 142   fhEMCALDigits        = new TH1I("EMCALDigitsCluster" , "EMCALDigits"       , 20 , 0 , 20  ) ;
 143   fhEMCALRecParticles  = new TH1D("EMCALRecParticles"  , "EMCALRecParticles", 20 , 0., 20. ) ;
 144   fhEMCALPhotons       = new TH1I("EMCALPhotons"       , "EMCALPhotons"      , 20 , 0 , 20  ) ;
 145   fhEMCALInvariantMass = new TH1D("EMCALInvariantMass" , "EMCALInvariantMass", 400, 0., 400.) ;
 146   fhEMCALDigitsEvent   = new TH1I("EMCALDigitsEvent"   , "EMCALDigitsEvent"  , 30 , 0 , 30  ) ;
 147
 148   // create output container
 149
 150   fOutputContainer = new TObjArray(8) ;
 151   fOutputContainer->SetName(GetName()) ;
 152
 153   fOutputContainer->AddAt(fhEMCALPos,            0) ;
 154   fOutputContainer->AddAt(fhEMCAL,               1) ;
 155   fOutputContainer->AddAt(fhEMCALEnergy,         2) ;
 156   fOutputContainer->AddAt(fhEMCALDigits,         3) ;
 157   fOutputContainer->AddAt(fhEMCALRecParticles,   4) ;
 158   fOutputContainer->AddAt(fhEMCALPhotons,        5) ;
 159   fOutputContainer->AddAt(fhEMCALInvariantMass,  6) ;
 160   fOutputContainer->AddAt(fhEMCALDigitsEvent,    7) ;
 161 }
 162
 163 //______________________________________________________________________________
 164 void AliEMCALQATask::Exec(Option_t *)
 165 {
 166   // Processing of one event
 167
 168   Long64_t entry = fChain->GetReadEntry() ;
 169
 170   if (!fESD) {
 171     AliError("fESD is not connected to the input!") ;
 172     return ;
 173   }
 174
 175   if ( !((entry-1)%100) )
 176     AliInfo(Form("%s ----> Processing event # %lld",  (dynamic_cast<TChain *>(fChain))->GetFile()->GetName(), entry)) ;
 177
 178   //************************  EMCAL *************************************
 179   Int_t       firstEmcalCluster       = fESD->GetFirstEMCALCluster() ;
 180   const Int_t kNumberOfEmcalClusters   = fESD->GetNumberOfEMCALClusters() ;
 181
 182   TVector3 ** emcalVector        = new TVector3*[kNumberOfEmcalClusters] ;
 183   Float_t   * emcalPhotonsEnergy = new Float_t[kNumberOfEmcalClusters] ;
 184   Int_t      emcalCluster ;
 185   Int_t      numberOfEmcalClustersv1 = 0 ;
 186   Int_t      numberOfPhotonsInEmcal  = 0 ;
 187   Int_t      numberOfDigitsInEmcal   = 0 ;
 188
 189
 190   // loop over all the EMCAL Cluster
 191   for(emcalCluster = firstEmcalCluster ; emcalCluster < firstEmcalCluster + kNumberOfEmcalClusters ; emcalCluster++) {
 192     AliESDCaloCluster * caloCluster = fESD->GetCaloCluster(emcalCluster) ;
 193     if (caloCluster) {
 194       Float_t pos[3] ;
 195       if(caloCluster->GetClusterType() == AliESDCaloCluster::kEMCALClusterv1) {
 196         caloCluster->GetPosition(pos) ;
 197         fhEMCALPos->Fill(pos[0],pos[1],pos[2]) ;
 198         fhEMCALEnergy->Fill(caloCluster->E()) ;
 199         fhEMCALDigits->Fill(entry, caloCluster->GetNumberOfDigits()) ;
 200         numberOfEmcalClustersv1++ ;
 201         numberOfDigitsInEmcal += caloCluster->GetNumberOfDigits() ;
 202         // Float_t * pid = clus->GetPid() ;
 203         // if(pid[AliPID::kPhoton]>0.9){
 204         emcalVector[numberOfPhotonsInEmcal] = new TVector3(pos[0],pos[1],pos[2]) ;
 205         emcalPhotonsEnergy[numberOfPhotonsInEmcal] = caloCluster->E() ;
 206         numberOfPhotonsInEmcal++ ;
 207       }
 208     }
 209   } // EMCAL clusters loop
 210
 211   fhEMCALRecParticles->Fill(numberOfEmcalClustersv1);
 212   fhEMCALPhotons->Fill(numberOfPhotonsInEmcal);
 213   fhEMCALDigitsEvent->Fill(numberOfDigitsInEmcal);
 214   fhEMCAL->Fill(entry, numberOfDigitsInEmcal, numberOfEmcalClustersv1, numberOfPhotonsInEmcal) ;
 215
 216   // invariant Mass
 217   if (numberOfPhotonsInEmcal > 1 ) {
 218     Int_t emcalPhoton1, emcalPhoton2 ;
 219     for(emcalPhoton1 = 0 ; emcalPhoton1 < numberOfPhotonsInEmcal ; emcalPhoton1++) {
 220       for(emcalPhoton2 = emcalPhoton1 + 1 ; emcalPhoton2 < numberOfPhotonsInEmcal ; emcalPhoton2++) {
 221         Float_t tempMass = TMath::Sqrt( 2 * emcalPhotonsEnergy[emcalPhoton1] * emcalPhotonsEnergy[emcalPhoton2] *
 222                                         ( 1 - TMath::Cos( emcalVector[emcalPhoton1]->Angle(*emcalVector[emcalPhoton2]))
 223                                           )
 224                                         );
 225         fhEMCALInvariantMass->Fill(tempMass*1000.);
 226       }
 227     }
 228   }
 229
 230   PostData(0, fOutputContainer);
 231
 232   delete [] emcalVector ;
 233   delete [] emcalPhotonsEnergy ;
 234 }
 235
 236 //______________________________________________________________________________
 237 void AliEMCALQATask::Terminate(Option_t *)
 238 {
 239   // Processing when the event loop is ended
 240   fOutputContainer = (TObjArray*)GetOutputData(0);
 241   fhEMCALEnergy = (TH1D*)fOutputContainer->At(2);
 242   fhEMCALDigits = (TH1I*)fOutputContainer->At(3);
 243   fhEMCALRecParticles = (TH1D*)fOutputContainer->At(4);
 244   fhEMCALPhotons = (TH1I*)fOutputContainer->At(5);
 245   fhEMCALInvariantMass = (TH1D*)fOutputContainer->At(6);
 246   fhEMCALDigitsEvent = (TH1I*)fOutputContainer->At(7);
 247
 248   Bool_t problem = kFALSE ;
 249   AliInfo(Form(" *** %s Report:", GetName())) ;
 250   printf("        EMCALEnergy Mean        : %5.3f , RMS : %5.3f \n", fhEMCALEnergy->GetMean(),        fhEMCALEnergy->GetRMS()        ) ;
 251   printf("        EMCALDigits Mean        : %5.3f , RMS : %5.3f \n", fhEMCALDigits->GetMean(),        fhEMCALDigits->GetRMS()        ) ;
 252   printf("        EMCALRecParticles Mean  : %5.3f , RMS : %5.3f \n", fhEMCALRecParticles->GetMean(),  fhEMCALRecParticles->GetRMS()  ) ;
 253   printf("        EMCALPhotons Mean       : %5.3f , RMS : %5.3f \n", fhEMCALPhotons->GetMean(),       fhEMCALPhotons->GetRMS()       ) ;
 254   printf("        EMCALInvariantMass Mean : %5.3f , RMS : %5.3f \n", fhEMCALInvariantMass->GetMean(), fhEMCALInvariantMass->GetRMS() ) ;
 255   printf("        EMCALDigitsEvent Mean   : %5.3f , RMS : %5.3f \n", fhEMCALDigitsEvent->GetMean(),   fhEMCALDigitsEvent->GetRMS()   ) ;
 256
 257   TCanvas  * cEMCAL = new TCanvas("EMCAL", "EMCAL ESD Test", 400, 10, 600, 700);
 258   cEMCAL->Divide(3, 2) ;
 259
 260   cEMCAL->cd(1) ;
 261   if ( fhEMCALEnergy->GetMaximum() > 0. )
 262     gPad->SetLogy();
 263   fhEMCALEnergy->SetAxisRange(0, 25.);
 264   fhEMCALEnergy->SetXTitle("Energy (GeV)");
 265   fhEMCALEnergy->Draw();
 266
 267   cEMCAL->cd(2) ;
 268   if ( fhEMCALDigits->GetMaximum() > 0. )
 269     gPad->SetLogy();
 270   fhEMCALDigits->SetAxisRange(0, 25.);
 271   fhEMCALDigits->SetXTitle("DigitsPerCluster");
 272   fhEMCALDigits->Draw();
 273
 274   cEMCAL->cd(3) ;
 275   if ( fhEMCALRecParticles->GetMaximum() > 0. )
 276      gPad->SetLogy();
 277   fhEMCALRecParticles->SetAxisRange(0, 25.);
 278   fhEMCALRecParticles->SetXTitle("RecParticles");
 279   fhEMCALRecParticles->Draw();
 280
 281   cEMCAL->cd(4) ;
 282   if ( fhEMCALPhotons->GetMaximum() > 0. )
 283     gPad->SetLogy();
 284   fhEMCALPhotons->SetAxisRange(0, 25.);
 285   fhEMCALPhotons->SetXTitle("Photons");
 286   fhEMCALPhotons->Draw();
 287
 288   cEMCAL->cd(5) ;
 289   fhEMCALInvariantMass->SetXTitle("InvariantMass (MeV/c²)");
 290   fhEMCALInvariantMass->Draw();
 291
 292   cEMCAL->cd(6) ;
 293   if ( fhEMCALDigitsEvent->GetMaximum() > 0. )
 294     gPad->SetLogy();
 295   fhEMCALDigitsEvent->SetAxisRange(0, 40.);
 296   fhEMCALDigitsEvent->SetXTitle("DigitsPerEvent");
 297   fhEMCALDigitsEvent->Draw();
 298
 299   cEMCAL->Print("EMCAL.eps");
 300
 301   char line[1024] ;
 302   sprintf(line, ".!tar -zcf %s.tar.gz *.eps", GetName()) ;
 303   gROOT->ProcessLine(line);
 304   sprintf(line, ".!rm -fR *.eps");
 305   gROOT->ProcessLine(line);
 306
 307   AliInfo(Form("!!! All the eps files are in %s.tar.gz !!!", GetName())) ;
 308
 309   TString report ;
 310   if(problem)
 311     report="Problems found, please check!!!";
 312   else
 313     report="OK";
 314
 315   AliInfo(Form("*** %s Summary Report: %s \n",GetName(), report.Data())) ;
 316 }