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 // An analysis task to check the EMCAL photon data in simulated data
  17 //
  18 //*-- Yves Schutz
  19 //////////////////////////////////////////////////////////////////////////////
  20
  21 #include <TChain.h>
  22 #include <TH1.h>
  23 #include <TH1F.h>
  24 #include <TH1I.h>
  25 #include <TNtuple.h>
  26 #include <TCanvas.h>
  27 #include <TLegend.h>
  28 #include <TVector3.h>
  29 #include <TFile.h>
  30
  31 #include "AliEMCALQATask.h"
  32 #include "AliESD.h"
  33 #include "AliLog.h"
  34
  35 //______________________________________________________________________________
  36 AliEMCALQATask::AliEMCALQATask(const char *name) :
  37   AliAnalysisTask(name,""),
  38   fChain(0),
  39   fESD(0),
  40   fhEMCALPos(0),
  41   fhEMCAL(0),
  42   fhEMCALEnergy(0),
  43   fhEMCALDigits(0),
  44   fhEMCALRecParticles(0),
  45   fhEMCALPhotons(0),
  46   fhEMCALInvariantMass(0),
  47   fhEMCALDigitsEvent(0)
  48 {
  49   // Constructor.
  50   // Input slot #0 works with an Ntuple
  51   DefineInput(0, TChain::Class());
  52   // Output slot #0 writes into a TH1 container
  53   DefineOutput(0,  TObjArray::Class()) ;
  54 }
  55
  56 //______________________________________________________________________________
  57 AliEMCALQATask::~AliEMCALQATask()
  58 {
  59   // dtor
  60   fOutputContainer->Clear() ;
  61   delete fOutputContainer ;
  62
  63   delete fhEMCALPos ;
  64   delete fhEMCAL ;
  65   delete fhEMCALEnergy ;
  66   delete fhEMCALDigits ;
  67   delete fhEMCALRecParticles ;
  68   delete fhEMCALPhotons ;
  69   delete fhEMCALInvariantMass ;
  70   delete fhEMCALDigitsEvent ;
  71 }
  72
  73 //______________________________________________________________________________
  74 void AliEMCALQATask::Init(const Option_t*)
  75 {
  76   // Initialisation of branch container and histograms
  77
  78   AliInfo(Form("*** Initialization of %s", GetName())) ;
  79
  80   // Get input data
  81   fChain = dynamic_cast<TChain *>(GetInputData(0)) ;
  82   if (!fChain) {
  83     AliError(Form("Input 0 for %s not found\n", GetName()));
  84     return ;
  85   }
  86
  87   if (!fESD) {
  88     // One should first check if the branch address was taken by some other task
  89     char ** address = (char **)GetBranchAddress(0, "ESD") ;
  90     if (address)
  91       fESD = (AliESD *)(*address) ;
  92     if (!fESD)
  93       fChain->SetBranchAddress("ESD", &fESD) ;
  94   }
  95   // The output objects will be written to
  96   TDirectory * cdir = gDirectory ;
  97   // Open a file for output #0
  98   char outputName[1024] ;
  99   sprintf(outputName, "%s.root", GetName() ) ;
 100   OpenFile(0, outputName , "RECREATE") ;
 101   if (cdir)
 102     cdir->cd() ;
 103
 104   // create histograms
 105
 106   fhEMCALPos           = new TNtuple("EMCALPos"        , "Position in EMCAL" , "x:y:z");
 107   fhEMCAL              = new TNtuple("EMCAL"           , "EMCAL" , "event:digits:clusters:photons");
 108   fhEMCALEnergy        = new TH1D("EMCALEnergy"        , "EMCALEnergy"       , 1000, 0., 10. ) ;
 109   fhEMCALDigits        = new TH1I("EMCALDigitsCluster" , "EMCALDigits"       , 20 , 0 , 20  ) ;
 110   fhEMCALRecParticles  = new TH1D("EMCALRecParticles"  , "EMCALRecParticles", 20 , 0., 20. ) ;
 111   fhEMCALPhotons       = new TH1I("EMCALPhotons"       , "EMCALPhotons"      , 20 , 0 , 20  ) ;
 112   fhEMCALInvariantMass = new TH1D("EMCALInvariantMass" , "EMCALInvariantMass", 400, 0., 400.) ;
 113   fhEMCALDigitsEvent   = new TH1I("EMCALDigitsEvent"   , "EMCALDigitsEvent"  , 30 , 0 , 30  ) ;
 114
 115   // create output container
 116
 117   fOutputContainer = new TObjArray(8) ;
 118   fOutputContainer->SetName(GetName()) ;
 119
 120   fOutputContainer->AddAt(fhEMCALPos,            0) ;
 121   fOutputContainer->AddAt(fhEMCAL,               1) ;
 122   fOutputContainer->AddAt(fhEMCALEnergy,         2) ;
 123   fOutputContainer->AddAt(fhEMCALDigits,         3) ;
 124   fOutputContainer->AddAt(fhEMCALRecParticles,   4) ;
 125   fOutputContainer->AddAt(fhEMCALPhotons,        5) ;
 126   fOutputContainer->AddAt(fhEMCALInvariantMass,  6) ;
 127   fOutputContainer->AddAt(fhEMCALDigitsEvent,    7) ;
 128 }
 129
 130 //______________________________________________________________________________
 131 void AliEMCALQATask::Exec(Option_t *)
 132 {
 133   // Processing of one event
 134
 135   Long64_t entry = fChain->GetReadEntry() ;
 136
 137   if (!fESD) {
 138     AliError("fESD is not connected to the input!") ;
 139     return ;
 140   }
 141
 142   if ( !((entry-1)%100) )
 143     AliInfo(Form("%s ----> Processing event # %lld",  (dynamic_cast<TChain *>(fChain))->GetFile()->GetName(), entry)) ;
 144
 145   //************************  EMCAL *************************************
 146   Int_t       firstEmcalCluster       = fESD->GetFirstEMCALCluster() ;
 147   const Int_t mumberOfEmcalClusters   = fESD->GetNumberOfEMCALClusters() ;
 148
 149   TVector3 ** emcalVector        = new TVector3*[mumberOfEmcalClusters] ;
 150   Float_t   * emcalPhotonsEnergy = new Float_t[mumberOfEmcalClusters] ;
 151   Int_t      emcalCluster ;
 152   Int_t      numberOfEmcalClustersv1 = 0 ;
 153   Int_t      numberOfPhotonsInEmcal  = 0 ;
 154   Int_t      numberOfDigitsInEmcal   = 0 ;
 155
 156
 157   // loop over all the EMCAL Cluster
 158   for(emcalCluster = firstEmcalCluster ; emcalCluster < firstEmcalCluster + mumberOfEmcalClusters ; emcalCluster++) {
 159     AliESDCaloCluster * caloCluster = fESD->GetCaloCluster(emcalCluster) ;
 160     if (caloCluster) {
 161       Float_t pos[3] ;
 162       fhEMCALPos->Fill(pos[0],pos[1],pos[2]) ;
 163       if(caloCluster->GetClusterType() == AliESDCaloCluster::kClusterv1) {
 164         caloCluster->GetGlobalPosition(pos) ;
 165         fhEMCALEnergy->Fill(caloCluster->GetClusterEnergy()) ;
 166         fhEMCALDigits->Fill(entry, caloCluster->GetNumberOfDigits()) ;
 167         numberOfEmcalClustersv1++ ;
 168         numberOfDigitsInEmcal += caloCluster->GetNumberOfDigits() ;
 169         // Float_t * pid = clus->GetPid() ;
 170         // if(pid[AliPID::kPhoton]>0.9){
 171         emcalVector[numberOfPhotonsInEmcal] = new TVector3(pos[0],pos[1],pos[2]) ;
 172         emcalPhotonsEnergy[numberOfPhotonsInEmcal] = caloCluster->GetClusterEnergy() ;
 173         numberOfPhotonsInEmcal++ ;
 174       }
 175     }
 176   } // EMCAL clusters loop
 177
 178   fhEMCALRecParticles->Fill(numberOfEmcalClustersv1);
 179   fhEMCALPhotons->Fill(numberOfPhotonsInEmcal);
 180   fhEMCALDigitsEvent->Fill(numberOfDigitsInEmcal);
 181   fhEMCAL->Fill(entry, numberOfDigitsInEmcal, numberOfEmcalClustersv1, numberOfPhotonsInEmcal) ;
 182
 183   // invariant Mass
 184   if (numberOfPhotonsInEmcal > 1 ) {
 185     Int_t emcalPhoton1, emcalPhoton2 ;
 186     for(emcalPhoton1 = 0 ; emcalPhoton1 < numberOfPhotonsInEmcal ; emcalPhoton1++) {
 187       for(emcalPhoton2 = emcalPhoton1 + 1 ; emcalPhoton2 < numberOfPhotonsInEmcal ; emcalPhoton2++) {
 188         Float_t tempMass = TMath::Sqrt( 2 * emcalPhotonsEnergy[emcalPhoton1] * emcalPhotonsEnergy[emcalPhoton2] *
 189                                         ( 1 - TMath::Cos( emcalVector[emcalPhoton1]->Angle(*emcalVector[emcalPhoton2]))
 190                                           )
 191                                         );
 192         fhEMCALInvariantMass->Fill(tempMass*1000.);
 193       }
 194     }
 195   }
 196
 197   PostData(0, fOutputContainer);
 198
 199   delete [] emcalVector ;
 200   delete [] emcalPhotonsEnergy ;
 201 }
 202
 203 //______________________________________________________________________________
 204 void AliEMCALQATask::Terminate(Option_t *)
 205 {
 206   // Processing when the event loop is ended
 207
 208   printf("EMCALEnergy Mean        : %5.3f , RMS : %5.3f \n", fhEMCALEnergy->GetMean(),        fhEMCALEnergy->GetRMS()        ) ;
 209   printf("EMCALDigits Mean        : %5.3f , RMS : %5.3f \n", fhEMCALDigits->GetMean(),        fhEMCALDigits->GetRMS()        ) ;
 210   printf("EMCALRecParticles Mean  : %5.3f , RMS : %5.3f \n", fhEMCALRecParticles->GetMean(),  fhEMCALRecParticles->GetRMS()  ) ;
 211   printf("EMCALPhotons Mean       : %5.3f , RMS : %5.3f \n", fhEMCALPhotons->GetMean(),       fhEMCALPhotons->GetRMS()       ) ;
 212   printf("EMCALInvariantMass Mean : %5.3f , RMS : %5.3f \n", fhEMCALInvariantMass->GetMean(), fhEMCALInvariantMass->GetRMS() ) ;
 213   printf("EMCALDigitsEvent Mean   : %5.3f , RMS : %5.3f \n", fhEMCALDigitsEvent->GetMean(),   fhEMCALDigitsEvent->GetRMS()   ) ;
 214
 215   TCanvas  * cEMCAL = new TCanvas("EMCAL", "EMCAL ESD Test", 400, 10, 600, 700);
 216   cEMCAL->Divide(3, 2) ;
 217
 218   cEMCAL->cd(1) ;
 219   gPad->SetLogy();
 220   fhEMCALEnergy->SetAxisRange(0, 25.);
 221   fhEMCALEnergy->SetXTitle("Energy (GeV)");
 222   fhEMCALEnergy->Draw();
 223
 224   cEMCAL->cd(2) ;
 225   gPad->SetLogy();
 226   fhEMCALDigits->SetAxisRange(0, 25.);
 227   fhEMCALDigits->SetXTitle("DigitsPerCluster");
 228   fhEMCALDigits->Draw();
 229
 230   cEMCAL->cd(3) ;
 231   gPad->SetLogy();
 232   fhEMCALRecParticles->SetAxisRange(0, 25.);
 233   fhEMCALRecParticles->SetXTitle("RecParticles");
 234   fhEMCALRecParticles->Draw();
 235
 236   cEMCAL->cd(4) ;
 237   gPad->SetLogy();
 238   fhEMCALPhotons->SetAxisRange(0, 25.);
 239   fhEMCALPhotons->SetXTitle("Photons");
 240   fhEMCALPhotons->Draw();
 241
 242   cEMCAL->cd(5) ;
 243   fhEMCALInvariantMass->SetXTitle("InvariantMass (MeV/c²)");
 244   fhEMCALInvariantMass->Draw();
 245
 246   cEMCAL->cd(6) ;
 247   gPad->SetLogy();
 248   fhEMCALDigitsEvent->SetAxisRange(0, 40.);
 249   fhEMCALDigitsEvent->SetXTitle("DigitsPerEvent");
 250   fhEMCALDigitsEvent->Draw();
 251
 252   cEMCAL->Print("EMCAL.eps");
 253
 254   char line[1024] ;
 255   sprintf(line, ".!tar -zcvf %s.tar.gz *.eps", GetName()) ;
 256   gROOT->ProcessLine(line);
 257   sprintf(line, ".!rm -fR *.eps");
 258   gROOT->ProcessLine(line);
 259
 260   AliInfo(Form("!!! All the eps files are in %s.tar.gz !!! \n", GetName())) ;
 261 }