PWG4/JetTasks/AliUEHistograms.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: AliUEHistograms.cxx 20164 2007-08-14 15:31:50Z morsch $ */
  17
  18 //
  19 //
  20 // encapsulates several AliUEHist objects for a full UE analysis plus additional control histograms
  21 //
  22 //
  23 // Author: Jan Fiete Grosse-Oetringhaus, Sara Vallero
  24
  25 #include "AliUEHistograms.h"
  26
  27 #include "AliCFContainer.h"
  28 #include "AliVParticle.h"
  29
  30 #include "TList.h"
  31 #include "TH2F.h"
  32 #include "TH1F.h"
  33 #include "TH3F.h"
  34 #include "TMath.h"
  35
  36 ClassImp(AliUEHistograms)
  37
  38 AliUEHistograms::AliUEHistograms() :
  39   TObject(),
  40   fNumberDensitypT(0),
  41   fSumpT(0),
  42   fNumberDensityPhi(0),
  43   fCorrelationpT(0),
  44   fCorrelationEta(0),
  45   fCorrelationPhi(0),
  46   fCorrelationR(0),
  47   fCorrelationLeading2Phi(0),
  48   fCorrelationMultiplicity(0),
  49   fEventCount(0),
  50   fEventCountDifferential(0),
  51   fVertexContributors(0)
  52 {
  53   // Constructor
  54
  55   fNumberDensitypT = new AliUEHist("NumberDensitypT");
  56   fSumpT = new AliUEHist("SumpT");
  57   fNumberDensityPhi = new AliUEHist("NumberDensityPhi");
  58
  59   // do not add this hists to the directory
  60   Bool_t oldStatus = TH1::AddDirectoryStatus();
  61   TH1::AddDirectory(kFALSE);
  62
  63   fCorrelationpT  = new TH2F("fCorrelationpT", ";p_{T,lead} (MC);p_{T,lead} (RECO)", 200, 0, 50, 200, 0, 50);
  64   fCorrelationEta = new TH2F("fCorrelationEta", ";#eta_{T,lead} (MC);#eta_{T,lead} (RECO)", 200, -1, 1, 200, -1, 1);
  65   fCorrelationPhi = new TH2F("fCorrelationPhi", ";#phi_{T,lead} (MC);#phi_{T,lead} (RECO)", 200, 0, TMath::TwoPi(), 200, 0, TMath::TwoPi());
  66   fCorrelationR =   new TH2F("fCorrelationR", ";R;p_{T,lead} (MC)", 200, 0, 2, 200, 0, 50);
  67   fCorrelationLeading2Phi = new TH2F("fCorrelationLeading2Phi", ";#Delta #phi;p_{T,lead} (MC)", 200, -TMath::Pi(), TMath::Pi(), 200, 0, 50);
  68   fCorrelationMultiplicity = new TH2F("fCorrelationMultiplicity", ";MC tracks;Reco tracks", 100, -0.5, 99.5, 100, -0.5, 99.5);
  69
  70   fEventCount = new TH2F("fEventCount", ";step;event type;count", AliUEHist::fgkCFSteps+2, -2.5, -0.5 + AliUEHist::fgkCFSteps, 3, -0.5, 2.5);
  71   fEventCount->GetYaxis()->SetBinLabel(1, "ND");
  72   fEventCount->GetYaxis()->SetBinLabel(2, "SD");
  73   fEventCount->GetYaxis()->SetBinLabel(3, "DD");
  74
  75   fEventCountDifferential = new TH3F("fEventCountDifferential", ";p_{T,lead};step;event type", 100, 0, 50, AliUEHist::fgkCFSteps, -0.5, -0.5 + AliUEHist::fgkCFSteps, 3, -0.5, 2.5);
  76   fEventCountDifferential->GetZaxis()->SetBinLabel(1, "ND");
  77   fEventCountDifferential->GetZaxis()->SetBinLabel(2, "SD");
  78   fEventCountDifferential->GetZaxis()->SetBinLabel(3, "DD");
  79
  80   fVertexContributors = new TH1F("fVertexContributors", ";contributors;count", 100, -0.5, 99.5);
  81
  82   TH1::AddDirectory(oldStatus);
  83 }
  84
  85 //_____________________________________________________________________________
  86 AliUEHistograms::AliUEHistograms(const AliUEHistograms &c) :
  87   TObject(),
  88   fNumberDensitypT(0),
  89   fSumpT(0),
  90   fNumberDensityPhi(0),
  91   fCorrelationpT(0),
  92   fCorrelationEta(0),
  93   fCorrelationPhi(0),
  94   fCorrelationR(0),
  95   fCorrelationLeading2Phi(0),
  96   fCorrelationMultiplicity(0),
  97   fEventCount(0),
  98   fEventCountDifferential(0),
  99   fVertexContributors(0)
 100 {
 101   //
 102   // AliUEHistograms copy constructor
 103   //
 104
 105   ((AliUEHistograms &) c).Copy(*this);
 106 }
 107
 108 //____________________________________________________________________
 109 AliUEHistograms::~AliUEHistograms()
 110 {
 111   // Destructor
 112
 113   if (fNumberDensitypT)
 114   {
 115     delete fNumberDensitypT;
 116     fNumberDensitypT = 0;
 117   }
 118
 119   if (fSumpT)
 120   {
 121     delete fSumpT;
 122     fSumpT = 0;
 123   }
 124
 125   if (fNumberDensityPhi)
 126   {
 127     delete fNumberDensityPhi;
 128     fNumberDensityPhi = 0;
 129   }
 130
 131   if (fCorrelationpT)
 132   {
 133     delete fCorrelationpT;
 134     fCorrelationpT = 0;
 135   }
 136
 137   if (fCorrelationEta)
 138   {
 139     delete fCorrelationEta;
 140     fCorrelationEta = 0;
 141   }
 142
 143   if (fCorrelationPhi)
 144   {
 145     delete fCorrelationPhi;
 146     fCorrelationPhi = 0;
 147   }
 148
 149   if (fCorrelationR)
 150   {
 151     delete fCorrelationR;
 152     fCorrelationR = 0;
 153   }
 154
 155   if (fCorrelationLeading2Phi)
 156   {
 157     delete fCorrelationLeading2Phi;
 158     fCorrelationLeading2Phi = 0;
 159   }
 160
 161   if (fCorrelationMultiplicity)
 162   {
 163     delete fCorrelationMultiplicity;
 164     fCorrelationMultiplicity = 0;
 165   }
 166
 167   if (fEventCount)
 168   {
 169     delete fEventCount;
 170     fEventCount = 0;
 171   }
 172
 173   if (fEventCountDifferential)
 174   {
 175     delete fEventCountDifferential;
 176     fEventCountDifferential = 0;
 177   }
 178
 179   if (fVertexContributors)
 180   {
 181     delete fVertexContributors;
 182     fVertexContributors = 0;
 183   }
 184 }
 185
 186 //____________________________________________________________________
 187 Int_t AliUEHistograms::CountParticles(TList* list, Float_t ptMin)
 188 {
 189   // counts the number of particles in the list with a pT above ptMin
 190   // TODO eta cut needed here?
 191
 192   Int_t count = 0;
 193   for (Int_t j=0; j<list->GetEntries(); j++)
 194     if (((AliVParticle*) list->At(j))->Pt() > ptMin)
 195       count++;
 196
 197   return count;
 198 }
 199
 200 //____________________________________________________________________
 201 void AliUEHistograms::Fill(Int_t eventType, AliUEHist::CFStep step, AliVParticle* leading, TList* toward, TList* away, TList* min, TList* max)
 202 {
 203   // fills the UE event histograms
 204   //
 205   // this function needs the leading (track or jet or ...) and four lists of AliVParticles which contain the particles/tracks to be filled in the four regions
 206
 207   // if leading is not set, just fill event statistics
 208   if (leading)
 209   {
 210     Int_t multiplicity = 0;
 211
 212     // TODO configurable?
 213     Float_t ptMin = 0.15;
 214     if (leading->Pt() > ptMin)
 215       multiplicity++;
 216
 217     multiplicity += CountParticles(toward, ptMin);
 218     multiplicity += CountParticles(away, ptMin);
 219     multiplicity += CountParticles(min, ptMin);
 220     multiplicity += CountParticles(max, ptMin);
 221
 222     FillRegion(AliUEHist::kToward, step, leading, toward, multiplicity);
 223     FillRegion(AliUEHist::kAway,   step, leading, away, multiplicity);
 224     FillRegion(AliUEHist::kMin,    step, leading, min, multiplicity);
 225     FillRegion(AliUEHist::kMax,    step, leading, max, multiplicity);
 226
 227     Double_t vars[2];
 228     vars[0] = leading->Pt();
 229     vars[1] = multiplicity;
 230     fNumberDensitypT->GetEventHist()->Fill(vars, step);
 231     fSumpT->GetEventHist()->Fill(vars, step);
 232     fNumberDensityPhi->GetEventHist()->Fill(vars, step);
 233
 234     fEventCountDifferential->Fill(leading->Pt(), step, eventType);
 235   }
 236
 237   FillEvent(eventType, step);
 238 }
 239
 240 //____________________________________________________________________
 241 void AliUEHistograms::FillRegion(AliUEHist::Region region, AliUEHist::CFStep step, AliVParticle* leading, TList* list, Int_t multiplicity)
 242 {
 243   // loops over AliVParticles in list and fills the given region at the given step
 244   //
 245   // See also Fill(...)
 246
 247   for (Int_t i=0; i<list->GetEntries(); i++)
 248   {
 249     AliVParticle* particle = (AliVParticle*) list->At(i);
 250
 251     Double_t vars[5];
 252     vars[0] = particle->Eta();
 253     vars[1] = particle->Pt();
 254     vars[2] = leading->Pt();
 255     vars[3] = multiplicity;
 256     vars[4] = leading->Phi() - particle->Phi();
 257     if (vars[4]>TMath::Pi()/2.)vars[4]-=TMath::TwoPi();
 258     if (vars[4]< - 3.*TMath::Pi()/2.)vars[4]+=TMath::TwoPi();
 259
 260     fNumberDensitypT->GetTrackHist(region)->Fill(vars, step);
 261     fSumpT->GetTrackHist(region)->Fill(vars, step, particle->Pt());
 262
 263     // fill all in toward region (is anyway as function of delta phi!)
 264     fNumberDensityPhi->GetTrackHist(AliUEHist::kToward)->Fill(vars, step);
 265   }
 266 }
 267
 268 //____________________________________________________________________
 269 void AliUEHistograms::Fill(AliVParticle* leadingMC, AliVParticle* leadingReco)
 270 {
 271   // fills the correlation histograms
 272
 273   if (leadingMC)
 274   {
 275     fCorrelationpT->Fill(leadingMC->Pt(), leadingReco->Pt());
 276     if (leadingMC->Pt() > 0.5)
 277     {
 278       fCorrelationEta->Fill(leadingMC->Eta(), leadingReco->Eta());
 279       fCorrelationPhi->Fill(leadingMC->Phi(), leadingReco->Phi());
 280     }
 281
 282     Float_t phiDiff = leadingMC->Phi() - leadingReco->Phi();
 283     if (phiDiff > TMath::Pi())
 284       phiDiff -= TMath::TwoPi();
 285     if (phiDiff < -TMath::Pi())
 286       phiDiff += TMath::TwoPi();
 287
 288     Float_t etaDiff = leadingMC->Eta() - leadingReco->Eta();
 289
 290     fCorrelationR->Fill(TMath::Sqrt(phiDiff * phiDiff + etaDiff * etaDiff), leadingMC->Pt());
 291     fCorrelationLeading2Phi->Fill(phiDiff, leadingMC->Pt());
 292   }
 293   else
 294   {
 295     fCorrelationpT->Fill(1.0, leadingReco->Pt());
 296     if (leadingReco->Pt() > 0.5)
 297     {
 298       fCorrelationEta->Fill(0.0, leadingReco->Eta());
 299       fCorrelationPhi->Fill(0.0, leadingReco->Phi());
 300     }
 301   }
 302 }
 303
 304 //____________________________________________________________________
 305 void AliUEHistograms::FillTrackingEfficiency(TObjArray* mc, TObjArray* recoPrim, TObjArray* recoAll, Int_t particleType)
 306 {
 307   // fills the tracking efficiency objects
 308   //
 309   // mc: all primary MC particles
 310   // recoPrim: reconstructed primaries (again MC particles)
 311   // recoAll: reconstructed (again MC particles)
 312   // particleType is: 0 for pion, 1 for kaon, 2 for proton, 3 for others
 313
 314   for (Int_t step=0; step<3; step++)
 315   {
 316     TObjArray* list = mc;
 317     if (step == 1)
 318       list = recoPrim;
 319     else if (step == 2)
 320       list = recoAll;
 321
 322     for (Int_t i=0; i<list->GetEntries(); i++)
 323     {
 324       AliVParticle* particle = (AliVParticle*) list->At(i);
 325       Double_t vars[2];
 326       vars[0] = particle->Eta();
 327       vars[1] = particle->Pt();
 328       vars[2] = particleType;
 329
 330       fNumberDensitypT->GetTrackHistEfficiency()->Fill(vars, step);
 331       fSumpT->GetTrackHistEfficiency()->Fill(vars, step);
 332       fNumberDensityPhi->GetTrackHistEfficiency()->Fill(vars, step);
 333     }
 334   }
 335 }
 336
 337 //____________________________________________________________________
 338 void AliUEHistograms::FillEvent(Int_t eventType, Int_t step)
 339 {
 340   // fills the number of events at the given step and the given enty type
 341   //
 342   // WARNING: This function is called from Fill, so only call it for steps where Fill is not called
 343
 344   fEventCount->Fill(step, eventType);
 345 }
 346
 347 //____________________________________________________________________
 348 void AliUEHistograms::SetEtaRange(Float_t etaMin, Float_t etaMax)
 349 {
 350   // sets eta min and max for all contained AliUEHist classes
 351
 352   fNumberDensitypT->SetEtaRange(etaMin, etaMax);
 353   fSumpT->SetEtaRange(etaMin, etaMax);
 354   fNumberDensityPhi->SetEtaRange(etaMin, etaMax);
 355 }
 356
 357 //____________________________________________________________________
 358 void AliUEHistograms::SetPtRange(Float_t ptMin, Float_t ptMax)
 359 {
 360   // sets pT min and max for all contained AliUEHist classes
 361
 362   fNumberDensitypT->SetPtRange(ptMin, ptMax);
 363   fSumpT->SetPtRange(ptMin, ptMax);
 364   fNumberDensityPhi->SetPtRange(ptMin, ptMax);
 365 }
 366
 367 //____________________________________________________________________
 368 void AliUEHistograms::SetContaminationEnhancement(TH1F* hist)
 369 {
 370   // sets the contamination enhancement histogram in all contained AliUEHist classes
 371
 372   fNumberDensitypT->SetContaminationEnhancement(hist);
 373   fSumpT->SetContaminationEnhancement(hist);
 374   fNumberDensityPhi->SetContaminationEnhancement(hist);
 375 }
 376
 377 //____________________________________________________________________
 378 void AliUEHistograms::SetCombineMinMax(Bool_t flag)
 379 {
 380   // sets pT min and max for all contained AliUEHist classes
 381
 382   fNumberDensitypT->SetCombineMinMax(flag);
 383   fSumpT->SetCombineMinMax(flag);
 384   fNumberDensityPhi->SetCombineMinMax(flag);
 385 }
 386
 387 //____________________________________________________________________
 388 void AliUEHistograms::Correct(AliUEHistograms* corrections)
 389 {
 390   // corrects the contained histograms by calling AliUEHist::Correct
 391
 392   fNumberDensitypT->Correct(corrections->fNumberDensitypT);
 393   fSumpT->Correct(corrections->fSumpT);
 394   //fNumberDensityPhi->Correct(corrections->fNumberDensityPhi);
 395 }
 396
 397 //____________________________________________________________________
 398 AliUEHistograms &AliUEHistograms::operator=(const AliUEHistograms &c)
 399 {
 400   // assigment operator
 401
 402   if (this != &c)
 403     ((AliUEHistograms &) c).Copy(*this);
 404
 405   return *this;
 406 }
 407
 408 //____________________________________________________________________
 409 void AliUEHistograms::Copy(TObject& c) const
 410 {
 411   // copy function
 412
 413   AliUEHistograms& target = (AliUEHistograms &) c;
 414
 415   if (fNumberDensitypT)
 416     target.fNumberDensitypT = dynamic_cast<AliUEHist*> (fNumberDensitypT->Clone());
 417
 418   if (fSumpT)
 419     target.fSumpT = dynamic_cast<AliUEHist*> (fSumpT->Clone());
 420
 421   if (fNumberDensityPhi)
 422     target.fNumberDensityPhi = dynamic_cast<AliUEHist*> (fNumberDensityPhi->Clone());
 423
 424   if (fCorrelationpT)
 425     target.fCorrelationpT = dynamic_cast<TH2F*> (fCorrelationpT->Clone());
 426
 427   if (fCorrelationEta)
 428     target.fCorrelationEta = dynamic_cast<TH2F*> (fCorrelationEta->Clone());
 429
 430   if (fCorrelationPhi)
 431     target.fCorrelationPhi = dynamic_cast<TH2F*> (fCorrelationPhi->Clone());
 432
 433   if (fCorrelationR)
 434     target.fCorrelationR = dynamic_cast<TH2F*> (fCorrelationR->Clone());
 435
 436   if (fCorrelationLeading2Phi)
 437     target.fCorrelationLeading2Phi = dynamic_cast<TH2F*> (fCorrelationLeading2Phi->Clone());
 438
 439   if (fCorrelationMultiplicity)
 440     target.fCorrelationMultiplicity = dynamic_cast<TH2F*> (fCorrelationMultiplicity->Clone());
 441
 442   if (fEventCount)
 443     target.fEventCount = dynamic_cast<TH2F*> (fEventCount->Clone());
 444
 445   if (fEventCountDifferential)
 446     target.fEventCountDifferential = dynamic_cast<TH3F*> (fEventCountDifferential->Clone());
 447
 448   if (fVertexContributors)
 449     target.fVertexContributors = dynamic_cast<TH1F*> (fVertexContributors->Clone());
 450 }
 451
 452 //____________________________________________________________________
 453 Long64_t AliUEHistograms::Merge(TCollection* list)
 454 {
 455   // Merge a list of AliUEHistograms objects with this (needed for
 456   // PROOF).
 457   // Returns the number of merged objects (including this).
 458
 459   if (!list)
 460     return 0;
 461
 462   if (list->IsEmpty())
 463     return 1;
 464
 465   TIterator* iter = list->MakeIterator();
 466   TObject* obj;
 467
 468   // collections of objects
 469   const Int_t kMaxLists = 12;
 470   TList* lists[kMaxLists];
 471
 472   for (Int_t i=0; i<kMaxLists; i++)
 473     lists[i] = new TList;
 474
 475   Int_t count = 0;
 476   while ((obj = iter->Next())) {
 477
 478     AliUEHistograms* entry = dynamic_cast<AliUEHistograms*> (obj);
 479     if (entry == 0)
 480       continue;
 481
 482     lists[0]->Add(entry->fNumberDensitypT);
 483     lists[1]->Add(entry->fSumpT);
 484     lists[2]->Add(entry->fNumberDensityPhi);
 485     lists[3]->Add(entry->fCorrelationpT);
 486     lists[4]->Add(entry->fCorrelationEta);
 487     lists[5]->Add(entry->fCorrelationPhi);
 488     lists[6]->Add(entry->fCorrelationR);
 489     lists[7]->Add(entry->fCorrelationLeading2Phi);
 490     lists[8]->Add(entry->fCorrelationMultiplicity);
 491     lists[9]->Add(entry->fEventCount);
 492     lists[10]->Add(entry->fEventCountDifferential);
 493     lists[11]->Add(entry->fVertexContributors);
 494
 495     count++;
 496   }
 497
 498   fNumberDensitypT->Merge(lists[0]);
 499   fSumpT->Merge(lists[1]);
 500   fNumberDensityPhi->Merge(lists[2]);
 501   fCorrelationpT->Merge(lists[3]);
 502   fCorrelationEta->Merge(lists[4]);
 503   fCorrelationPhi->Merge(lists[5]);
 504   fCorrelationR->Merge(lists[6]);
 505   fCorrelationLeading2Phi->Merge(lists[7]);
 506   fCorrelationMultiplicity->Merge(lists[8]);
 507   fEventCount->Merge(lists[9]);
 508   fEventCountDifferential->Merge(lists[10]);
 509   fVertexContributors->Merge(lists[11]);
 510
 511   for (Int_t i=0; i<kMaxLists; i++)
 512     delete lists[i];
 513
 514   return count+1;
 515 }
 516
 517 void AliUEHistograms::CopyReconstructedData(AliUEHistograms* from)
 518 {
 519   // copies those histograms extracted from ESD to this object
 520
 521   fNumberDensitypT->CopyReconstructedData(from->fNumberDensitypT);
 522   fSumpT->CopyReconstructedData(from->fSumpT);
 523   fNumberDensityPhi->CopyReconstructedData(from->fNumberDensityPhi);
 524 }