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