start event mixing from 5 events in the pool
[u/mrichter/AliRoot.git] / PWGCF / Correlations / DPhi / AliAnalysisTaskPhiCorrelations.cxx
CommitLineData
e0331fd9 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#include <TROOT.h>
19#include <TChain.h>
20#include <TFile.h>
21#include <TList.h>
22#include <TMath.h>
23#include <TTree.h>
24#include <TH2F.h>
25#include <TRandom.h>
26
27#include "AliAnalysisTaskPhiCorrelations.h"
28#include "AliAnalyseLeadingTrackUE.h"
29#include "AliUEHistograms.h"
30#include "AliUEHist.h"
31
e0331fd9 32#include "AliAnalysisManager.h"
33#include "AliAODHandler.h"
34#include "AliAODInputHandler.h"
35#include "AliAODMCParticle.h"
e0331fd9 36#include "AliInputEventHandler.h"
37#include "AliLog.h"
38#include "AliMCEventHandler.h"
39#include "AliVParticle.h"
2a910c25 40#include "AliCFContainer.h"
e0331fd9 41
42#include "AliESDEvent.h"
43#include "AliESDInputHandler.h"
44#include "AliMultiplicity.h"
2a910c25 45#include "AliCentrality.h"
46#include "AliStack.h"
e0331fd9 47
2a910c25 48#include "AliEventPoolManager.h"
e0331fd9 49
50
51////////////////////////////////////////////////////////////////////////
52//
53// Analysis class for azimuthal correlation studies
54// Based on the UE task from Sara Vallero and Jan Fiete
55//
56// This class needs input AODs.
57// The output is a list of analysis-specific containers.
58//
59// The AOD can be either connected to the InputEventHandler
60// for a chain of AOD files
61// or
62// to the OutputEventHandler
63// for a chain of ESD files,
64// in this case the class should be in the train after the jet-finder
65//
66// Authors:
67// Jan Fiete Grosse-Oetringhaus
68//
69////////////////////////////////////////////////////////////////////////
70
71
72ClassImp( AliAnalysisTaskPhiCorrelations )
a1c31636 73ClassImp( AliDPhiBasicParticle )
e0331fd9 74
75//____________________________________________________________________
76AliAnalysisTaskPhiCorrelations:: AliAnalysisTaskPhiCorrelations(const char* name):
77AliAnalysisTask(name,""),
78// general configuration
79fDebug(0),
80fMode(0),
81fReduceMemoryFootprint(kFALSE),
eed401dc 82fFillMixed(kTRUE),
ac647b0f 83fMixingTracks(50000),
c3294f09 84fCompareCentralities(kFALSE),
1bba939a 85fTwoTrackEfficiencyStudy(kFALSE),
44af28f9 86fUseVtxAxis(kFALSE),
e0331fd9 87// pointers to UE classes
88fAnalyseUE(0x0),
89fHistos(0x0),
90fHistosMixed(0),
91fkTrackingEfficiency(0x0),
92// handlers and events
85bfac17 93fAOD(0x0),
94fESD(0x0),
e0331fd9 95fArrayMC(0x0),
96fInputHandler(0x0),
97fMcEvent(0x0),
98fMcHandler(0x0),
2a910c25 99fPoolMgr(0x0),
e0331fd9 100// histogram settings
101fListOfHistos(0x0),
102// event QA
103fnTracksVertex(1), // QA tracks pointing to principal vertex (= 3 default)
85bfac17 104fZVertex(7.),
2a910c25 105fCentralityMethod("V0M"),
e0331fd9 106// track cuts
107fTrackEtaCut(0.8),
108fPtMin(0.5),
109fFilterBit(0xFF),
110fSelectBit(0),
2a910c25 111fUseChargeHadrons(kFALSE),
c05ff6be 112fSelectCharge(0),
d38fa455 113fTriggerRestrictEta(-1),
c05ff6be 114fFillpT(kFALSE)
e0331fd9 115{
116 // Default constructor
117 // Define input and output slots here
118 // Input slot #0 works with a TChain
119 DefineInput(0, TChain::Class());
120 // Output slot #0 writes into a TList container
121 DefineOutput(0, TList::Class());
e0331fd9 122}
123
124AliAnalysisTaskPhiCorrelations::~AliAnalysisTaskPhiCorrelations()
125{
126 // destructor
127
128 if (fListOfHistos && !AliAnalysisManager::GetAnalysisManager()->IsProofMode())
129 delete fListOfHistos;
130}
131
132//____________________________________________________________________
133void AliAnalysisTaskPhiCorrelations::ConnectInputData(Option_t* /*option*/)
134{
135
136 // Connect the input data
137 if (fDebug > 1) AliInfo("ConnectInputData() ");
138
139 // Since AODs can either be connected to the InputEventHandler
140 // or to the OutputEventHandler ( the AOD is created by a previus task in the train )
141 // we need to get the pointer to the AODEvent correctly.
142
143 // Delta AODs are also accepted.
144
145 TObject* handler = AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler();
146
85bfac17 147 if( handler && handler->InheritsFrom("AliAODInputHandler") )
148 { // input AOD
149 fAOD = ((AliAODInputHandler*)handler)->GetEvent();
150 if (fDebug > 1) AliInfo(" ==== Tracks and Jets from AliAODInputHandler");
151 }
152 else
153 { //output AOD
154 handler = AliAnalysisManager::GetAnalysisManager()->GetOutputEventHandler();
155 if (handler && handler->InheritsFrom("AliAODHandler") )
156 {
157 fAOD = ((AliAODHandler*)handler)->GetAOD();
158 if (fDebug > 1) AliInfo(" ==== Tracks and Jets from AliAODHandler");
159 }
160 else
161 { // no AOD
162 AliWarning("I can't get any AOD Event Handler");
163 }
164 }
165
166 if (handler && handler->InheritsFrom("AliESDInputHandler") )
167 { // input ESD
168 // pointer received per event in ::Exec
169 if (fDebug > 1) AliInfo(" ==== Tracks and Jets from AliESDInputHandler");
170 }
e0331fd9 171
172 // Initialize common pointers
173 Initialize();
e0331fd9 174}
175
176//____________________________________________________________________
177void AliAnalysisTaskPhiCorrelations::CreateOutputObjects()
178{
179 // Create the output container
180
181 if (fDebug > 1) AliInfo("CreateOutputObjects()");
182
183 // Initialize class with main algorithms, event and track selection.
184 fAnalyseUE = new AliAnalyseLeadingTrackUE();
2a910c25 185 fAnalyseUE->SetParticleSelectionCriteria(fFilterBit, fUseChargeHadrons, fTrackEtaCut, fPtMin);
e0331fd9 186 fAnalyseUE->SetDebug(fDebug);
85bfac17 187 fAnalyseUE->DefineESDCuts(fFilterBit);
e0331fd9 188
189 // Initialize output list of containers
190 if (fListOfHistos != NULL){
191 delete fListOfHistos;
192 fListOfHistos = NULL;
193 }
194 if (!fListOfHistos){
195 fListOfHistos = new TList();
196 fListOfHistos->SetOwner(kTRUE);
197 }
198
199 // Initialize class to handle histograms
44af28f9 200 const char* histType = "4";
201 if (fUseVtxAxis)
202 histType = "5";
203 fHistos = new AliUEHistograms("AliUEHistogramsSame", histType);
204 fHistosMixed = new AliUEHistograms("AliUEHistogramsMixed", histType);
e0331fd9 205
2a910c25 206 fHistos->SetSelectCharge(fSelectCharge);
207 fHistosMixed->SetSelectCharge(fSelectCharge);
208
d38fa455 209 fHistos->SetTriggerRestrictEta(fTriggerRestrictEta);
210 fHistosMixed->SetTriggerRestrictEta(fTriggerRestrictEta);
211
e0331fd9 212 // add histograms to list
213 fListOfHistos->Add(fHistos);
214 fListOfHistos->Add(fHistosMixed);
215
2a910c25 216 fListOfHistos->Add(new TH2F("trackletsVsV0Cent", ";L1 clusters;v0 centrality", 100, -0.5, 9999.5, 101, 0, 101));
217 fListOfHistos->Add(new TH2F("processIDs", ";#Delta#phi;process id", 100, -0.5 * TMath::Pi(), 1.5 * TMath::Pi(), kPNoProcess + 1, -0.5, kPNoProcess + 0.5));
c066889a 218 fListOfHistos->Add(new TH1F("eventStat", ";;events", 4, -0.5, 3.5));
219 fListOfHistos->Add(new TH1F("mixedDist", ";tracks;events", 200, 0, fMixingTracks * 1.5));
e0331fd9 220
1a32ecc0 221 PostData(0,fListOfHistos);
222
e0331fd9 223 // Add task configuration to output list
224 AddSettingsTree();
225
2a910c25 226 // event mixing
ac647b0f 227 Int_t trackDepth = fMixingTracks;
a1c31636 228 Int_t poolsize = 1000; // Maximum number of events, ignored in the present implemented of AliEventPool
c066889a 229
2a910c25 230 Int_t nCentralityBins = fHistos->GetUEHist(2)->GetEventHist()->GetNBins(1);
231 Double_t* centralityBins = (Double_t*) fHistos->GetUEHist(2)->GetEventHist()->GetAxis(1, 0)->GetXbins()->GetArray();
232
eb88bdfe 233 Int_t nZvtxBins = 7;
234 Double_t vertexBins[] = { -7, -5, -3, -1, 1, 3, 5, 7 };
235 Double_t* zvtxbin = vertexBins;
c066889a 236
eb88bdfe 237 if (fHistos->GetUEHist(2)->GetEventHist()->GetNVar() > 2)
238 {
239 nZvtxBins = fHistos->GetUEHist(2)->GetEventHist()->GetNBins(2);
240 zvtxbin = (Double_t*) fHistos->GetUEHist(2)->GetEventHist()->GetAxis(2, 0)->GetXbins()->GetArray();
241 }
85bfac17 242
2a910c25 243 fPoolMgr = new AliEventPoolManager(poolsize, trackDepth, nCentralityBins, centralityBins, nZvtxBins, zvtxbin);
e0331fd9 244}
245
246//____________________________________________________________________
247void AliAnalysisTaskPhiCorrelations::Exec(Option_t */*option*/)
248{
249 // array of MC particles
85bfac17 250 if (fMcHandler) {
251 if (fAOD)
252 {
253 fArrayMC = dynamic_cast<TClonesArray*>(fAOD->FindListObject(AliAODMCParticle::StdBranchName()));
254 if (!fArrayMC)
255 AliFatal("No array of MC particles found !!!");
256 }
2a910c25 257 fMcEvent = fMcHandler->MCEvent();
258 }
e0331fd9 259
85bfac17 260 // receive ESD pointer if we are not running AOD analysis
261 if (!fAOD)
262 {
263 AliVEventHandler* handler = AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler();
264 if (handler && handler->InheritsFrom("AliESDInputHandler"))
265 fESD = ((AliESDInputHandler*)handler)->GetEvent();
266 }
267
e0331fd9 268 // Analyse the event
269 if (fMode) AnalyseCorrectionMode();
270 else AnalyseDataMode();
e0331fd9 271}
272
273/******************** ANALYSIS METHODS *****************************/
274
275//____________________________________________________________________
276void AliAnalysisTaskPhiCorrelations::AddSettingsTree()
277{
278 //Write settings to output list
279 TTree *settingsTree = new TTree("UEAnalysisSettings","Analysis Settings in UE estimation");
c05ff6be 280 settingsTree->Branch("fnTracksVertex", &fnTracksVertex,"nTracksVertex/I");
281 settingsTree->Branch("fZVertex", &fZVertex,"ZVertex/D");
282 //settingsTree->Branch("fCentralityMethod", fCentralityMethod.Data(),"CentralityMethod/C");
e0331fd9 283 settingsTree->Branch("fTrackEtaCut", &fTrackEtaCut, "TrackEtaCut/D");
284 settingsTree->Branch("fPtMin", &fPtMin, "PtMin/D");
c05ff6be 285 settingsTree->Branch("fFilterBit", &fFilterBit,"FilterBit/I");
286 settingsTree->Branch("fSelectBit", &fSelectBit,"EventSelectionBit/I");
e0331fd9 287 settingsTree->Branch("fUseChargeHadrons", &fUseChargeHadrons,"UseChHadrons/O");
c05ff6be 288 settingsTree->Branch("fSelectCharge", &fSelectCharge,"SelectCharge/I");
d38fa455 289 settingsTree->Branch("fTriggerRestrictEta", &fTriggerRestrictEta,"SelectCharge/D");
c05ff6be 290 settingsTree->Branch("fFillpT", &fFillpT,"FillpT/O");
e0331fd9 291 settingsTree->Branch("fkTrackingEfficiency", "TH1D", &fkTrackingEfficiency);
ac647b0f 292 settingsTree->Branch("fMixingTracks", &fMixingTracks,"MixingTracks/I");
e0331fd9 293 settingsTree->Fill();
294 fListOfHistos->Add(settingsTree);
295}
296
297//____________________________________________________________________
298void AliAnalysisTaskPhiCorrelations::AnalyseCorrectionMode()
299{
300 // Run the analysis on MC to get the correction maps
301 //
302
2a910c25 303 if ( fDebug > 3 ) AliInfo( " Processing event in Corrections mode ..." );
304
305 Double_t centrality = 0;
306
307 if (fCentralityMethod.Length() > 0)
308 {
85bfac17 309 AliCentrality *centralityObj = 0;
310 if (fAOD)
311 centralityObj = fAOD->GetHeader()->GetCentralityP();
312 else if (fESD)
313 centralityObj = fESD->GetCentrality();
314
2a910c25 315 if (centralityObj)
316 {
317 centrality = centralityObj->GetCentralityPercentileUnchecked(fCentralityMethod);
a8583211 318 AliInfo(Form("Centrality is %f", centrality));
2a910c25 319 }
320 else
321 {
322 Printf("WARNING: Centrality object is 0");
323 centrality = -1;
324 }
325 }
326
85bfac17 327 // Support for ESD and AOD based analysis
328 AliVEvent* inputEvent = fAOD;
329 if (!inputEvent)
330 inputEvent = fESD;
331
332 TObject* mc = fArrayMC;
333 if (!mc)
334 mc = fMcEvent;
335
2a910c25 336 // count all events
337 fHistos->FillEvent(centrality, -1);
338
339 // Only consider MC events within the vtx-z region used also as cut on the reconstructed vertex
340 if (!fAnalyseUE->VertexSelection(fMcEvent, 0, fZVertex))
341 return;
85bfac17 342
343 Float_t zVtx = fMcEvent->GetPrimaryVertex()->GetZ();
c05ff6be 344 Float_t weight = 1;
345 if (fFillpT)
346 weight = -1;
2a910c25 347
348 // Get MC primaries
85bfac17 349 TObjArray* tracksMC = fAnalyseUE->GetAcceptedParticles(mc, 0, kTRUE, -1, kTRUE);
2a910c25 350
351 // (MC-true all particles)
352 // STEP 0
c05ff6be 353 fHistos->FillCorrelations(centrality, zVtx, AliUEHist::kCFStepAll, tracksMC, 0, weight);
2a910c25 354
355 // Trigger selection ************************************************
356 if (fAnalyseUE->TriggerSelection(fInputHandler))
357 {
c32a0ca9 358 // (MC-true all particles)
359 // STEP 1
360 if (!fReduceMemoryFootprint)
c05ff6be 361 fHistos->FillCorrelations(centrality, zVtx, AliUEHist::kCFStepTriggered, tracksMC, 0, weight);
c32a0ca9 362 else
363 fHistos->FillEvent(centrality, AliUEHist::kCFStepTriggered);
364
872dc651 365 if (!inputEvent) {
c32a0ca9 366 AliFatal("UNEXPECTED: inputEvent is 0. Trigger selection should have failed");
872dc651 367 return;
368 }
2a910c25 369
370 // Vertex selection *************************************************
85bfac17 371 if (fAnalyseUE->VertexSelection(inputEvent, fnTracksVertex, fZVertex))
2a910c25 372 {
373 // fill here for tracking efficiency
374 // loop over particle species
375
376 for (Int_t particleSpecies = 0; particleSpecies < 4; particleSpecies++)
377 {
85bfac17 378 TObjArray* primMCParticles = fAnalyseUE->GetAcceptedParticles(mc, 0x0, kTRUE, particleSpecies, kTRUE);
379 TObjArray* primRecoTracksMatched = fAnalyseUE->GetAcceptedParticles(inputEvent, mc, kTRUE, particleSpecies, kTRUE);
380 TObjArray* allRecoTracksMatched = fAnalyseUE->GetAcceptedParticles(inputEvent, mc, kFALSE, particleSpecies, kTRUE);
2a910c25 381
382 fHistos->FillTrackingEfficiency(primMCParticles, primRecoTracksMatched, allRecoTracksMatched, particleSpecies, centrality);
383
eed401dc 384// Printf("%d --> %d %d %d", particleSpecies, primMCParticles->GetEntries(), primRecoTracksMatched->GetEntries(), allRecoTracksMatched->GetEntries());
385
386 delete primMCParticles;
2a910c25 387 delete primRecoTracksMatched;
388 delete allRecoTracksMatched;
389 }
390
391 // (MC-true all particles)
392 // STEP 2
85bfac17 393 if (!fReduceMemoryFootprint)
c05ff6be 394 fHistos->FillCorrelations(centrality, zVtx, AliUEHist::kCFStepVertex, tracksMC, 0, weight);
c32a0ca9 395 else
396 fHistos->FillEvent(centrality, AliUEHist::kCFStepVertex);
2a910c25 397
398 // Get MC primaries that match reconstructed track
85bfac17 399 TObjArray* tracksRecoMatchedPrim = fAnalyseUE->GetAcceptedParticles(inputEvent, mc, kTRUE, -1, kTRUE);
2a910c25 400
401 // (RECO-matched (quantities from MC particle) primary particles)
402 // STEP 4
c05ff6be 403 fHistos->FillCorrelations(centrality, zVtx, AliUEHist::kCFStepTrackedOnlyPrim, tracksRecoMatchedPrim, 0, weight);
2a910c25 404
405 // Get MC primaries + secondaries that match reconstructed track
85bfac17 406 TObjArray* tracksRecoMatchedAll = fAnalyseUE->GetAcceptedParticles(inputEvent, mc, kFALSE, -1, kTRUE);
2a910c25 407
408 // (RECO-matched (quantities from MC particle) all particles)
409 // STEP 5
c05ff6be 410 fHistos->FillCorrelations(centrality, zVtx, AliUEHist::kCFStepTracked, tracksRecoMatchedAll, 0, weight);
2a910c25 411
412 // Get RECO tracks
85bfac17 413 TObjArray* tracks = fAnalyseUE->GetAcceptedParticles(inputEvent, 0, kTRUE, -1, kTRUE);
2a910c25 414
415 // (RECO all tracks)
416 // STEP 6
c05ff6be 417 fHistos->FillCorrelations(centrality, zVtx, AliUEHist::kCFStepReconstructed, tracks, 0, weight);
2a910c25 418
85bfac17 419 if (0 && !fReduceMemoryFootprint)
2a910c25 420 {
421 // make list of secondaries (matched with MC)
422 TObjArray* tracksRecoMatchedSecondaries = new TObjArray;
423 for (Int_t i=0; i<tracksRecoMatchedAll->GetEntries(); i++)
424 if (((AliAODMCParticle*)tracksRecoMatchedAll->At(i))->IsPhysicalPrimary() == kFALSE)
425 tracksRecoMatchedSecondaries->Add(tracksRecoMatchedAll->At(i));
426
427 // Study: Use only secondaries as trigger particles and plot the correlation vs. all particles; store in step 9
c05ff6be 428 fHistos->FillCorrelations(centrality, zVtx, AliUEHist::kCFStepBiasStudy2, tracksRecoMatchedSecondaries, tracksRecoMatchedAll, weight);
2a910c25 429
430 // Study: Use only primaries as trigger particles and plot the correlation vs. secondaries; store in step 8
c05ff6be 431 fHistos->FillCorrelations(centrality, zVtx, AliUEHist::kCFStepBiasStudy, tracksRecoMatchedPrim, tracksRecoMatchedSecondaries, weight);
2a910c25 432
433 // plot delta phi vs process id of secondaries
434 // trigger particles: primaries in 4 < pT < 10
435 // associated particles: secondaries in 1 < pT < 10
436
437 for (Int_t i=0; i<tracksRecoMatchedPrim->GetEntries(); i++)
438 {
439 AliVParticle* triggerParticle = (AliVParticle*) tracksRecoMatchedPrim->At(i);
440
441 if (triggerParticle->Pt() < 4 || triggerParticle->Pt() > 10)
442 continue;
443
444 for (Int_t j=0; j<tracksRecoMatchedSecondaries->GetEntries(); j++)
445 {
446 AliAODMCParticle* particle = (AliAODMCParticle*) tracksRecoMatchedSecondaries->At(j);
447
448 if (particle->Pt() < 1 || particle->Pt() > 10)
449 continue;
450
451 if (particle->Pt() > triggerParticle->Pt())
452 continue;
453
454 Double_t deltaPhi = triggerParticle->Phi() - particle->Phi();
455 if (deltaPhi > 1.5 * TMath::Pi())
456 deltaPhi -= TMath::TwoPi();
457 if (deltaPhi < -0.5 * TMath::Pi())
458 deltaPhi += TMath::TwoPi();
459
460 Int_t processID = fMcEvent->Stack()->Particle(particle->GetLabel())->GetUniqueID();
461
462 ((TH2F*) fListOfHistos->FindObject("processIDs"))->Fill(deltaPhi, processID);
463 }
464 }
465
466 delete tracksRecoMatchedSecondaries;
467 }
468
469 delete tracksRecoMatchedPrim;
470 delete tracksRecoMatchedAll;
471 delete tracks;
472 }
473 }
474
475 delete tracksMC;
e0331fd9 476}
477
478//____________________________________________________________________
479void AliAnalysisTaskPhiCorrelations::AnalyseDataMode()
480{
481
482 // Run the analysis on DATA or MC to get raw distributions
483
e0331fd9 484 if ( fDebug > 3 ) AliInfo( " Processing event in Data mode ..." );
e0331fd9 485
c066889a 486 ((TH1F*) fListOfHistos->FindObject("eventStat"))->Fill(0);
487
2a910c25 488 // skip not selected events here (the AOD is not updated for those)
489 if (!fInputHandler)
490 return;
491
85bfac17 492 if (!(fInputHandler->IsEventSelected() & (AliVEvent::kMB | AliVEvent::kUserDefined)))
2a910c25 493 return;
e0331fd9 494
2a910c25 495 Double_t centrality = 0;
496
c3294f09 497 AliCentrality *centralityObj = 0;
2a910c25 498 if (fCentralityMethod.Length() > 0)
499 {
85bfac17 500 if (fAOD)
501 centralityObj = fAOD->GetHeader()->GetCentralityP();
502 else if (fESD)
503 centralityObj = fESD->GetCentrality();
504
2a910c25 505 if (centralityObj)
506 centrality = centralityObj->GetCentralityPercentile(fCentralityMethod);
507 //centrality = centralityObj->GetCentralityPercentileUnchecked(fCentralityMethod);
508 else
509 centrality = -1;
a6d82f4e 510
511 if (fAOD)
512 {
513 // remove outliers
514 if (centrality == 0)
515 {
516 if (fAOD->GetVZEROData())
517 {
518 Float_t multV0 = 0;
519 for (Int_t i=0; i<64; i++)
520 multV0 += fAOD->GetVZEROData()->GetMultiplicity(i);
521 if (multV0 < 19500)
522 {
523 centrality = -1;
524 AliInfo("Rejecting event due to too small V0 multiplicity");
525 }
526 }
527 }
528 }
529
a8583211 530 AliInfo(Form("Centrality is %f", centrality));
2a910c25 531 }
532
85bfac17 533 // Support for ESD and AOD based analysis
534 AliVEvent* inputEvent = fAOD;
535 if (!inputEvent)
536 inputEvent = fESD;
537
538 fHistos->SetRunNumber(inputEvent->GetRunNumber());
539
2a910c25 540 // Fill the "event-counting-container", it is needed to get the number of events remaining after each event-selection cut
541 fHistos->FillEvent(centrality, AliUEHist::kCFStepAll);
542
e0331fd9 543 // Trigger selection ************************************************
544 if (!fAnalyseUE->TriggerSelection(fInputHandler)) return;
545
546 // Fill the "event-counting-container", it is needed to get the number of events remaining after each event-selection cut
2a910c25 547 fHistos->FillEvent(centrality, AliUEHist::kCFStepTriggered);
e0331fd9 548
549 // Vertex selection *************************************************
85bfac17 550 if(!fAnalyseUE->VertexSelection(inputEvent, fnTracksVertex, fZVertex)) return;
e0331fd9 551
552 // Fill the "event-counting-container", it is needed to get the number of events remaining after each event-selection cut
2a910c25 553 fHistos->FillEvent(centrality, AliUEHist::kCFStepVertex);
e0331fd9 554
c3294f09 555 // optimization
556 if (centrality < 0 && !fCompareCentralities)
2a910c25 557 return;
e0331fd9 558
85bfac17 559 TObjArray* tracks = fAnalyseUE->GetAcceptedParticles(inputEvent, 0, kTRUE, -1, kTRUE);
2a910c25 560 //Printf("Accepted %d tracks", tracks->GetEntries());
e0331fd9 561
85bfac17 562 const AliVVertex* vertex = inputEvent->GetPrimaryVertex();
2a910c25 563 Double_t zVtx = vertex->GetZ();
564
c05ff6be 565 Float_t weight = 1;
566 if (fFillpT)
567 weight = -1;
568
c3294f09 569 // fill two different centralities (syst study)
570 // the zvtx axis is used to distinguish the results of both centralities: configured centrality in zvtx = 0, SPD in zvtx = 2
571 if (fCompareCentralities)
572 zVtx = 0;
2a910c25 573
c3294f09 574 // Fill containers at STEP 6 (reconstructed)
575 if (centrality >= 0)
c066889a 576 {
c05ff6be 577 fHistos->FillCorrelations(centrality, zVtx, AliUEHist::kCFStepReconstructed, tracks, 0, weight);
c066889a 578 ((TH1F*) fListOfHistos->FindObject("eventStat"))->Fill(1);
579 }
c3294f09 580
1bba939a 581 // Two-track effect study
582 if (fTwoTrackEfficiencyStudy)
583 {
6840ec31 584 Float_t bSign = (fESD->GetMagneticField() > 0) ? 1 : -1;
c5c840c5 585 fHistos->TwoTrackEfficiency(tracks, 0, bSign);
1bba939a 586 }
587
c3294f09 588 // fill second time with SPD centrality
589 if (fCompareCentralities && centralityObj)
590 {
591 centrality = centralityObj->GetCentralityPercentile("CL1");
592 if (centrality >= 0)
c05ff6be 593 fHistos->FillCorrelations(centrality, 2, AliUEHist::kCFStepReconstructed, tracks, 0, weight);
c3294f09 594 }
595
eed401dc 596 if (fFillMixed)
e0331fd9 597 {
eed401dc 598 // event mixing
2a910c25 599
eed401dc 600 // 1. First get an event pool corresponding in mult (cent) and
601 // zvertex to the current event. Once initialized, the pool
602 // should contain nMix (reduced) events. This routine does not
603 // pre-scan the chain. The first several events of every chain
604 // will be skipped until the needed pools are filled to the
605 // specified depth. If the pool categories are not too rare, this
606 // should not be a problem. If they are rare, you could lose
607 // statistics.
608
609 // 2. Collect the whole pool's content of tracks into one TObjArray
610 // (bgTracks), which is effectively a single background super-event.
611
612 // 3. The reduced and bgTracks arrays must both be passed into
613 // FillCorrelations(). Also nMix should be passed in, so a weight
614 // of 1./nMix can be applied.
615
616 AliEventPool* pool = fPoolMgr->GetEventPool(centrality, zVtx);
617
618 if (!pool)
619 AliFatal(Form("No pool found for centrality = %f, zVtx = %f", centrality, zVtx));
620
621 //pool->SetDebug(1);
2d7827bf 622
623 if (pool->IsReady() || pool->NTracksInPool() > fMixingTracks / 10 || pool->GetCurrentNEvents() >= 5)
eed401dc 624 {
625
626 Int_t nMix = pool->GetCurrentNEvents();
a1c31636 627// cout << "nMix = " << nMix << " tracks in pool = " << pool->NTracksInPool() << endl;
c066889a 628
629 ((TH1F*) fListOfHistos->FindObject("eventStat"))->Fill(2);
630 ((TH1F*) fListOfHistos->FindObject("mixedDist"))->Fill(pool->NTracksInPool());
631 if (pool->IsReady())
632 ((TH1F*) fListOfHistos->FindObject("eventStat"))->Fill(3);
eed401dc 633
634 // Fill mixed-event histos here
c05ff6be 635 for (Int_t jMix=0; jMix<nMix; jMix++)
636 {
eed401dc 637 TObjArray* bgTracks = pool->GetEvent(jMix);
c5c840c5 638
639 if (fTwoTrackEfficiencyStudy)
640 {
641 Float_t bSign = (fESD->GetMagneticField() > 0) ? 1 : -1;
642 fHistos->TwoTrackEfficiency(tracks, bgTracks, bSign);
643 }
644
acc90d78 645 fHistosMixed->FillCorrelations(centrality, zVtx, AliUEHist::kCFStepReconstructed, tracks, bgTracks, 1.0 / nMix, (jMix == 0));
eed401dc 646 }
e0331fd9 647 }
eed401dc 648
a1c31636 649 // create a list of reduced objects (to reduce memory consumption) and give ownership to event pool
650 TObjArray* tracksClone = new TObjArray;
eed401dc 651 tracksClone->SetOwner(kTRUE);
652
a1c31636 653 for (Int_t i=0; i<tracks->GetEntries(); i++)
654 {
655 AliVParticle* particle = (AliVParticle*) tracks->At(i);
656 tracksClone->Add(new AliDPhiBasicParticle(particle->Eta(), particle->Phi(), particle->Pt(), particle->Charge()));
657 }
658
eed401dc 659 pool->UpdatePool(tracksClone);
660 //pool->PrintInfo();
e0331fd9 661 }
2a910c25 662
e0331fd9 663 delete tracks;
e0331fd9 664}
665
666//____________________________________________________________________
667void AliAnalysisTaskPhiCorrelations::Initialize()
668{
669 // input handler
670 fInputHandler = (AliInputEventHandler*)
671 ((AliAnalysisManager::GetAnalysisManager())->GetInputEventHandler());
672 // MC handler
673 fMcHandler = dynamic_cast<AliMCEventHandler*> (AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler());
674}