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