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+1, -1.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
 258     fNumberDensitypT->GetTrackHist(region)->Fill(vars, step);
 259     fSumpT->GetTrackHist(region)->Fill(vars, step, particle->Pt());
 260
 261     // fill all in toward region (is anyway as function of delta phi!)
 262     fNumberDensityPhi->GetTrackHist(AliUEHist::kToward)->Fill(vars, step);
 263   }
 264 }
 265
 266 //____________________________________________________________________
 267 void AliUEHistograms::Fill(AliVParticle* leadingMC, AliVParticle* leadingReco)
 268 {
 269   // fills the correlation histograms
 270
 271   if (leadingMC)
 272   {
 273     fCorrelationpT->Fill(leadingMC->Pt(), leadingReco->Pt());
 274     if (leadingMC->Pt() > 0.5)
 275     {
 276       fCorrelationEta->Fill(leadingMC->Eta(), leadingReco->Eta());
 277       fCorrelationPhi->Fill(leadingMC->Phi(), leadingReco->Phi());
 278     }
 279
 280     Float_t phiDiff = leadingMC->Phi() - leadingReco->Phi();
 281     if (phiDiff > TMath::Pi())
 282       phiDiff -= TMath::TwoPi();
 283     if (phiDiff < -TMath::Pi())
 284       phiDiff += TMath::TwoPi();
 285
 286     Float_t etaDiff = leadingMC->Eta() - leadingReco->Eta();
 287
 288     fCorrelationR->Fill(TMath::Sqrt(phiDiff * phiDiff + etaDiff * etaDiff), leadingMC->Pt());
 289     fCorrelationLeading2Phi->Fill(phiDiff, leadingMC->Pt());
 290   }
 291   else
 292   {
 293     fCorrelationpT->Fill(1.0, leadingReco->Pt());
 294     if (leadingReco->Pt() > 0.5)
 295     {
 296       fCorrelationEta->Fill(0.0, leadingReco->Eta());
 297       fCorrelationPhi->Fill(0.0, leadingReco->Phi());
 298     }
 299   }
 300 }
 301
 302 //____________________________________________________________________
 303 void AliUEHistograms::FillEvent(Int_t eventType, Int_t step)
 304 {
 305   // fills the number of events at the given step and the given enty type
 306   //
 307   // WARNING: This function is called from Fill, so only call it for steps where Fill is not called
 308
 309   fEventCount->Fill(step, eventType);
 310 }
 311
 312 //____________________________________________________________________
 313 void AliUEHistograms::SetEtaRange(Float_t etaMin, Float_t etaMax)
 314 {
 315   // sets eta min and max for all contained AliUEHist classes
 316
 317   fNumberDensitypT->SetEtaRange(etaMin, etaMax);
 318   fSumpT->SetEtaRange(etaMin, etaMax);
 319   fNumberDensityPhi->SetEtaRange(etaMin, etaMax);
 320 }
 321
 322 //____________________________________________________________________
 323 void AliUEHistograms::SetPtRange(Float_t ptMin, Float_t ptMax)
 324 {
 325   // sets pT min and max for all contained AliUEHist classes
 326
 327   fNumberDensitypT->SetPtRange(ptMin, ptMax);
 328   fSumpT->SetPtRange(ptMin, ptMax);
 329   fNumberDensityPhi->SetPtRange(ptMin, ptMax);
 330 }
 331
 332 //____________________________________________________________________
 333 void AliUEHistograms::SetCombineMinMax(Bool_t flag)
 334 {
 335   // sets pT min and max for all contained AliUEHist classes
 336
 337   fNumberDensitypT->SetCombineMinMax(flag);
 338   fSumpT->SetCombineMinMax(flag);
 339   fNumberDensityPhi->SetCombineMinMax(flag);
 340 }
 341
 342 //____________________________________________________________________
 343 void AliUEHistograms::Correct(AliUEHistograms* corrections)
 344 {
 345   // corrects the contained histograms by calling AliUEHist::Correct
 346
 347   fNumberDensitypT->Correct(corrections->fNumberDensitypT);
 348   fSumpT->Correct(corrections->fSumpT);
 349   fNumberDensityPhi->Correct(corrections->fNumberDensityPhi);
 350 }
 351
 352 //____________________________________________________________________
 353 AliUEHistograms &AliUEHistograms::operator=(const AliUEHistograms &c)
 354 {
 355   // assigment operator
 356
 357   if (this != &c)
 358     ((AliUEHistograms &) c).Copy(*this);
 359
 360   return *this;
 361 }
 362
 363 //____________________________________________________________________
 364 void AliUEHistograms::Copy(TObject& c) const
 365 {
 366   // copy function
 367
 368   AliUEHistograms& target = (AliUEHistograms &) c;
 369
 370   if (fNumberDensitypT)
 371     target.fNumberDensitypT = dynamic_cast<AliUEHist*> (fNumberDensitypT->Clone());
 372
 373   if (fSumpT)
 374     target.fSumpT = dynamic_cast<AliUEHist*> (fSumpT->Clone());
 375
 376   if (fNumberDensityPhi)
 377     target.fNumberDensityPhi = dynamic_cast<AliUEHist*> (fNumberDensityPhi->Clone());
 378
 379   if (fCorrelationpT)
 380     target.fCorrelationpT = dynamic_cast<TH2F*> (fCorrelationpT->Clone());
 381
 382   if (fCorrelationEta)
 383     target.fCorrelationEta = dynamic_cast<TH2F*> (fCorrelationEta->Clone());
 384
 385   if (fCorrelationPhi)
 386     target.fCorrelationPhi = dynamic_cast<TH2F*> (fCorrelationPhi->Clone());
 387
 388   if (fCorrelationR)
 389     target.fCorrelationR = dynamic_cast<TH2F*> (fCorrelationR->Clone());
 390
 391   if (fCorrelationLeading2Phi)
 392     target.fCorrelationLeading2Phi = dynamic_cast<TH2F*> (fCorrelationLeading2Phi->Clone());
 393
 394   if (fCorrelationMultiplicity)
 395     target.fCorrelationMultiplicity = dynamic_cast<TH2F*> (fCorrelationMultiplicity->Clone());
 396
 397   if (fEventCount)
 398     target.fEventCount = dynamic_cast<TH2F*> (fEventCount->Clone());
 399
 400   if (fEventCountDifferential)
 401     target.fEventCountDifferential = dynamic_cast<TH3F*> (fEventCountDifferential->Clone());
 402
 403   if (fVertexContributors)
 404     target.fVertexContributors = dynamic_cast<TH1F*> (fVertexContributors->Clone());
 405 }
 406
 407 //____________________________________________________________________
 408 Long64_t AliUEHistograms::Merge(TCollection* list)
 409 {
 410   // Merge a list of AliUEHistograms objects with this (needed for
 411   // PROOF).
 412   // Returns the number of merged objects (including this).
 413
 414   if (!list)
 415     return 0;
 416
 417   if (list->IsEmpty())
 418     return 1;
 419
 420   TIterator* iter = list->MakeIterator();
 421   TObject* obj;
 422
 423   // collections of objects
 424   const Int_t kMaxLists = 12;
 425   TList* lists[kMaxLists];
 426
 427   for (Int_t i=0; i<kMaxLists; i++)
 428     lists[i] = new TList;
 429
 430   Int_t count = 0;
 431   while ((obj = iter->Next())) {
 432
 433     AliUEHistograms* entry = dynamic_cast<AliUEHistograms*> (obj);
 434     if (entry == 0)
 435       continue;
 436
 437     lists[0]->Add(entry->fNumberDensitypT);
 438     lists[1]->Add(entry->fSumpT);
 439     lists[2]->Add(entry->fNumberDensityPhi);
 440     lists[3]->Add(entry->fCorrelationpT);
 441     lists[4]->Add(entry->fCorrelationEta);
 442     lists[5]->Add(entry->fCorrelationPhi);
 443     lists[6]->Add(entry->fCorrelationR);
 444     lists[7]->Add(entry->fCorrelationLeading2Phi);
 445     lists[8]->Add(entry->fCorrelationMultiplicity);
 446     lists[9]->Add(entry->fEventCount);
 447     lists[10]->Add(entry->fEventCountDifferential);
 448     lists[11]->Add(entry->fVertexContributors);
 449
 450     count++;
 451   }
 452
 453   fNumberDensitypT->Merge(lists[0]);
 454   fSumpT->Merge(lists[1]);
 455   fNumberDensityPhi->Merge(lists[2]);
 456   fCorrelationpT->Merge(lists[3]);
 457   fCorrelationEta->Merge(lists[4]);
 458   fCorrelationPhi->Merge(lists[5]);
 459   fCorrelationR->Merge(lists[6]);
 460   fCorrelationLeading2Phi->Merge(lists[7]);
 461   fCorrelationMultiplicity->Merge(lists[8]);
 462   fEventCount->Merge(lists[9]);
 463   fEventCountDifferential->Merge(lists[10]);
 464   fVertexContributors->Merge(lists[11]);
 465
 466   for (Int_t i=0; i<kMaxLists; i++)
 467     delete lists[i];
 468
 469   return count+1;
 470 }