PWGCF/Correlations/DPhi/AliAnalysisTaskTwoPlusOne.cxx

   1 #include <TROOT.h>
   2 #include <TChain.h>
   3 #include <TFile.h>
   4 #include <TList.h>
   5 #include <TMath.h>
   6 #include <TTree.h>
   7 //#include <TParameter.h>
   8
   9 #include "AliAnalysisTaskTwoPlusOne.h"
  10 #include "AliCFParticle.h"
  11 #include "AliAnalyseLeadingTrackUE.h"
  12 #include "AliTwoPlusOneContainer.h"
  13 #include "AliUEHist.h"
  14
  15 #include "AliAODHandler.h"
  16 #include "AliAODInputHandler.h"
  17 #include "AliVParticle.h"
  18 #include "AliCFContainer.h"
  19
  20 #include "AliEventPoolManager.h"
  21
  22
  23 ClassImp( AliAnalysisTaskTwoPlusOne )
  24
  25 //________________________________________________________________________
  26 AliAnalysisTaskTwoPlusOne::AliAnalysisTaskTwoPlusOne(const char *name)
  27 : AliAnalysisTaskSE(name),
  28   fMixingTracks(10000),
  29   fAnalyseUE(0x0),
  30 // pointers to UE classes
  31   fHistos(0x0),
  32 // handlers and events
  33   fAOD(0x0),
  34   fPoolMgr(0x0),
  35 // histogram settings
  36   fListOfHistos(0x0),
  37 // event QA
  38   fnTracksVertex(1),  // QA tracks pointing to principal vertex (= 3 default)
  39   fZVertex(7.),
  40   fCentralityMethod("V0M"),
  41 // track cuts
  42   fTrackEtaCut(0.9),
  43   fTrackEtaCutMin(-1.),
  44   fPtMin(0.5),
  45   fDCAXYCut(0),
  46   fSharedClusterCut(-1),
  47   fCrossedRowsCut(-1),
  48   fFoundFractionCut(-1),
  49   fFilterBit(0xFF),
  50   fTrackStatus(0),
  51   fThreeParticleMixed(0),
  52   fCustomBinning(),
  53   fAlpha(0.2)
  54 {
  55
  56   DefineOutput(1, TList::Class());
  57 }
  58
  59 //________________________________________________________________________
  60 AliAnalysisTaskTwoPlusOne::~AliAnalysisTaskTwoPlusOne()
  61 {
  62     // Destructor
  63
  64 }
  65
  66 //________________________________________________________________________
  67 void AliAnalysisTaskTwoPlusOne::UserCreateOutputObjects()
  68 {
  69   // Initialize class with main algorithms, event and track selection.
  70   fAnalyseUE = new AliAnalyseLeadingTrackUE();
  71   fAnalyseUE->SetParticleSelectionCriteria(fFilterBit, kFALSE, fTrackEtaCut, fTrackEtaCutMin, fPtMin);
  72   fAnalyseUE->SetDCAXYCut(fDCAXYCut);
  73   fAnalyseUE->SetSharedClusterCut(fSharedClusterCut);
  74   fAnalyseUE->SetCrossedRowsCut(fCrossedRowsCut);
  75   fAnalyseUE->SetFoundFractionCut(fFoundFractionCut);
  76   fAnalyseUE->SetTrackStatus(fTrackStatus);
  77   fAnalyseUE->SetDebug(fDebug);
  78   fAnalyseUE->DefineESDCuts(fFilterBit);
  79
  80   fListOfHistos = new TList();
  81   fListOfHistos->SetOwner(kTRUE);
  82
  83   fHistos = new AliTwoPlusOneContainer("AliTwoPlusOneContainer", fCustomBinning, fAlpha);
  84   fHistos->GetData()->SetTrackEtaCut(fTrackEtaCut);
  85
  86   fListOfHistos->Add(fHistos);
  87
  88   fListOfHistos->Add(new TH1F("eventStat", ";;events", 4, -0.5, 3.5));
  89   fListOfHistos->Add(new TH2F("mixedDist", ";centrality;tracks;events", 101, 0, 101, 200, 0, fMixingTracks * 1.5));
  90
  91   PostData(1,fListOfHistos);
  92
  93   // Add task configuration to output list
  94   AddSettingsTree();
  95
  96   // event mixing
  97   Int_t poolsize   = 1000;  // Maximum number of events, ignored in the present implemention of AliEventPoolManager
  98
  99   Int_t nCentralityBins  = fHistos->GetData()->GetTrackHist(AliUEHist::kToward)->GetNBins(3);
 100
 101
 102   Double_t* centralityBins = (Double_t*) fHistos->GetData()->GetTrackHist(AliUEHist::kToward)->GetAxis(3, 0)->GetXbins()->GetArray();
 103
 104   Int_t nZvtxBins = fHistos->GetData()->GetTrackHist(AliUEHist::kToward)->GetNBins(5);
 105
 106   Double_t* zvtxbin = (Double_t*) fHistos->GetData()->GetTrackHist(AliUEHist::kToward)->GetAxis(5, 0)->GetXbins()->GetArray();
 107
 108   fPoolMgr = new AliEventPoolManager(poolsize, fMixingTracks, nCentralityBins, centralityBins, nZvtxBins, zvtxbin);
 109   fPoolMgr->SetTargetValues(fMixingTracks, 0.1, 5);
 110 }
 111
 112 //________________________________________________________________________
 113 void AliAnalysisTaskTwoPlusOne::UserExec(Option_t *)
 114 {
 115   // exec (per event)
 116   fAnalyseUE->NextEvent();
 117
 118   ((TH1F*) fListOfHistos->FindObject("eventStat"))->Fill(0);
 119
 120   // Support for AOD based analysis
 121   fAOD = dynamic_cast<AliAODEvent*> (InputEvent());
 122
 123   Double_t centrality = 0;
 124   AliCentrality *centralityObj = 0;
 125
 126   centralityObj = fAOD->GetHeader()->GetCentralityP();
 127
 128
 129  if (centralityObj)
 130    centrality = centralityObj->GetCentralityPercentile(fCentralityMethod);
 131  else
 132    centrality = -1;
 133
 134   if (fAOD){
 135     // remove outliers
 136     if (centrality == 0)
 137       {
 138         if (fAOD->GetVZEROData())
 139           {
 140             Float_t multV0 = 0;
 141             for (Int_t i=0; i<64; i++)
 142               multV0 += fAOD->GetVZEROData()->GetMultiplicity(i);
 143             if (multV0 < 19500)
 144               {
 145                 centrality = -1;
 146                 AliInfo("Rejecting event due to too small V0 multiplicity");
 147               }
 148           }
 149       }
 150   }
 151
 152
 153   AliInfo(Form("Centrality is %f", centrality));
 154
 155
 156   // Vertex selection *************************************************
 157   if(!fAnalyseUE->VertexSelection(InputEvent(), fnTracksVertex, fZVertex)) return;
 158
 159   // optimization
 160   if (centrality < 0)
 161     return;
 162
 163   TObjArray* tracks = fAnalyseUE->GetAcceptedParticles(InputEvent(), 0, kTRUE, -1, kTRUE);
 164   // create a list of reduced objects. This speeds up processing and reduces memory consumption for the event pool
 165   TObjArray* tracksClone = CloneAndReduceTrackList(tracks);
 166   delete tracks;
 167
 168   const AliVVertex* vertex = InputEvent()->GetPrimaryVertex();
 169   Double_t zVtx = vertex->GetZ();
 170
 171   fHistos->FillCorrelations(centrality, zVtx, AliTwoPlusOneContainer::kSameNS, tracksClone, tracksClone, tracksClone, tracksClone, 1.0);//same event for near and away side
 172
 173   ((TH1F*) fListOfHistos->FindObject("eventStat"))->Fill(1);
 174
 175   // event mixing
 176
 177   // 1. First get an event pool corresponding in mult (cent) and
 178   //    zvertex to the current event. Once initialized, the pool
 179   //    should contain nMix (reduced) events. This routine does not
 180   //    pre-scan the chain. The first several events of every chain
 181   //    will be skipped until the needed pools are filled to the
 182   //    specified depth. If the pool categories are not too rare, this
 183   //    should not be a problem. If they are rare, you could lose
 184   //    statistics.
 185
 186   // 2. Collect the whole pool's content of tracks into one TObjArray
 187   //    (bgTracks), which is effectively a single background super-event.
 188
 189   // 3. The reduced and bgTracks arrays must both be passed into
 190   //    FillCorrelations(). Also nMix should be passed in, so a weight
 191   //    of 1./nMix can be applied.
 192
 193   AliEventPool* pool = fPoolMgr->GetEventPool(centrality, zVtx);
 194
 195   if (!pool)
 196     AliFatal(Form("No pool found for centrality = %f, zVtx = %f", centrality, zVtx));
 197   if (pool->IsReady()){
 198     ((TH2F*) fListOfHistos->FindObject("mixedDist"))->Fill(centrality, pool->NTracksInPool());
 199     Int_t nMix = pool->GetCurrentNEvents();
 200
 201     ((TH1F*) fListOfHistos->FindObject("eventStat"))->Fill(2);
 202
 203     // Fill mixed-event histos here
 204     for (Int_t jMix=0; jMix<nMix; jMix++){
 205       TObjArray* bgTracks = pool->GetEvent(jMix);
 206
 207       //standard mixed event
 208       if(!fThreeParticleMixed)
 209         fHistos->FillCorrelations(centrality, zVtx, AliTwoPlusOneContainer::kMixedNS, tracksClone, tracksClone, bgTracks, bgTracks, 1.0 / nMix);
 210
 211       //mixed combinatorics
 212       fHistos->FillCorrelations(centrality, zVtx, AliTwoPlusOneContainer::kMixedCombNS, tracksClone, bgTracks, tracksClone, bgTracks, 1.0 / nMix);
 213     }
 214
 215
 216     // use 3 particle mixed event
 217     if(fThreeParticleMixed && nMix>1){
 218       TObjArray* tracks_t2 = pool->GetEvent(0);
 219       for (Int_t jMix=1; jMix<nMix; jMix++){
 220
 221         TObjArray* bgTracks = pool->GetEvent(jMix);
 222
 223         fHistos->FillCorrelations(centrality, zVtx, AliTwoPlusOneContainer::kMixedNS, tracksClone, tracks_t2, bgTracks, bgTracks, 1.0 / (nMix-1));
 224       }
 225     }
 226
 227
 228   }
 229
 230   // ownership is with the pool now
 231   pool->UpdatePool(tracksClone);
 232 }
 233
 234 //________________________________________________________________________
 235 void AliAnalysisTaskTwoPlusOne::Terminate(Option_t*)
 236 {
 237     //terminate function is called at the end
 238     //can be used to draw histograms etc.
 239
 240
 241 }
 242
 243
 244 TObjArray* AliAnalysisTaskTwoPlusOne::CloneAndReduceTrackList(TObjArray* tracks)
 245 {
 246   // clones a track list by using AliCFParticle which uses much less memory (used for event mixing)
 247
 248   TObjArray* tracksClone = new TObjArray;
 249   tracksClone->SetOwner(kTRUE);
 250
 251   for (Int_t i=0; i<tracks->GetEntriesFast(); i++)
 252   {
 253     AliVParticle* particle = (AliVParticle*) tracks->UncheckedAt(i);
 254     AliCFParticle* copy = new AliCFParticle(particle->Pt(), particle->Eta(), particle->Phi(), particle->Charge(), 0);
 255     copy->SetUniqueID(particle->GetUniqueID());
 256     tracksClone->Add(copy);
 257   }
 258
 259   return tracksClone;
 260 }
 261
 262
 263 //____________________________________________________________________
 264 void  AliAnalysisTaskTwoPlusOne::AddSettingsTree()
 265 {
 266   //Write settings to output list
 267   TTree *settingsTree   = new TTree("UEAnalysisSettings","Analysis Settings in UE estimation");
 268   settingsTree->Branch("fnTracksVertex", &fnTracksVertex,"nTracksVertex/I");
 269   settingsTree->Branch("fZVertex", &fZVertex,"ZVertex/D");
 270   //settingsTree->Branch("fCentralityMethod", fCentralityMethod.Data(),"CentralityMethod/C");
 271   settingsTree->Branch("fTrackEtaCut", &fTrackEtaCut, "TrackEtaCut/D");
 272   settingsTree->Branch("fTrackEtaCutMin", &fTrackEtaCutMin, "TrackEtaCutMin/D");
 273   settingsTree->Branch("fPtMin", &fPtMin, "PtMin/D");
 274   settingsTree->Branch("fFilterBit", &fFilterBit,"FilterBit/I");
 275   settingsTree->Branch("fSharedClusterCut", &fSharedClusterCut,"SharedClusterCut/D");
 276   settingsTree->Branch("fCrossedRowsCut", &fCrossedRowsCut,"CrossedRowsCut/I");
 277   settingsTree->Branch("fFoundFractionCut", &fFoundFractionCut,"FoundFractionCut/D");
 278   settingsTree->Branch("fTrackStatus", &fTrackStatus,"TrackStatus/I");
 279   settingsTree->Branch("fThreeParticleMixed", &fThreeParticleMixed,"fThreeParticleMixed/I");
 280   settingsTree->Branch("fMixingTracks", &fMixingTracks,"MixingTracks/I");
 281
 282   settingsTree->Fill();
 283   fListOfHistos->Add(settingsTree);
 284 }