1 /**************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * Authors: Svein Lindal *
8 * Permission to use, copy, modify and distribute this software and its *
9 * documentation strictly for non-commercial purposes is hereby granted *
10 * without fee, provided that the above copyright notice appears in all *
11 * copies and that both the copyright notice and this permission notice *
12 * appear in the supporting documentation. The authors make no claims *
13 * about the suitability of this software for any purpose. It is *
14 * provided "as is" without express or implied warranty. *
15 **************************************************************************/
17 ////////////////////////////////////////////////
18 //---------------------------------------------
19 // Class doing conversion gamma dPhi correlations
20 // Gamma Conversion analysis
21 //---------------------------------------------
22 ////////////////////////////////////////////////
24 #include "AliAnalysisTaskdPhi.h"
30 #include <AliAnalysisManager.h>
31 #include <AliInputEventHandler.h>
32 #include <AliESDInputHandler.h>
33 #include <AliAODInputHandler.h>
34 #include <AliAnalysisFilter.h>
36 #include "AliConversionCuts.h"
37 #include "AliAODConversionPhoton.h"
38 #include "AliAODConversionMother.h"
39 #include "AliAnaConvCorrPhoton.h"
40 #include "AliAnaConvCorrPion.h"
41 #include "AliAnaConvIsolation.h"
42 // Author Svein Lindal <slindal@fys.uio.no>
45 ClassImp(AliAnalysisTaskdPhi)
48 //________________________________________________________________________
49 AliAnalysisTaskdPhi::AliAnalysisTaskdPhi(const char *name) : AliAnalysisTaskSE(name),
53 fDielV0TrackFilter(NULL),
56 fDielTrackFilter(NULL),
69 fDeltaAODBranchName("AliAODGammaConversion_gamma"),
80 fAxisPt.SetNameTitle("PtAxis", "Pt");
81 fAxisPt.Set(20, 0, 100);
82 fAxisEta.SetNameTitle("EtaAxis", "Eta");
83 fAxisEta.Set(160, -0.8, 0.8);
84 fAxisPhi.SetNameTitle("PhiAxis", "Phi");
85 fAxisPhi.Set(128, 0, TMath::TwoPi());
86 fAxisZ.SetNameTitle("ZAxis", "Z");
87 fAxisZ.Set(4, -10, 10);
88 fAxisCent.SetNameTitle("CentAxis", "Cent");
90 Double_t centbins[5] = {0, 10, 30, 60, 100.1};
91 fAxisCent.Set(4, centbins);
94 Double_t mbins[7] = {0.1, 0.11, 0.12, 0.15, 0.16, 0.18, 0.2};
95 fAxisPiM.SetNameTitle("InvMassPi0", "Invariant mass");
96 fAxisPiM.Set(6, mbins);
98 fGammas = new TObjArray();
99 fGammas->SetOwner(kFALSE);
101 fPions = new TObjArray();
102 fPions->SetOwner(kFALSE);
104 // Define input and output slots here
105 DefineInput(0, TChain::Class());
106 DefineInput(1, TClonesArray::Class());
107 DefineOutput(1, TList::Class());
108 DefineOutput(2, TList::Class());
109 DefineOutput(3, TList::Class());
114 //________________________________________________________________________
115 AliAnalysisTaskdPhi::~AliAnalysisTaskdPhi(){
147 ///________________________________________________________________________
148 void AliAnalysisTaskdPhi::SetUpCorrObjects() {
150 fIsoAna = new AliAnaConvIsolation();
153 fPhotonCorr = new TObjArray();
154 fPionCorr = new TObjArray();
156 TList * hPhoton = new TList();
157 hPhoton->SetName("hPhotonCorr");
158 hPhoton->SetOwner(kTRUE);
159 fHistoGamma->Add(hPhoton);
161 TList * hPion = new TList();
162 hPion->SetName("hPionCorr");
163 hPion->SetOwner(kTRUE);
164 fHistoPion->Add(hPion);
167 for(Int_t iz = 0; iz < fAxisZ.GetNbins(); iz++) {
168 TObjArray * photonArray = new TObjArray();
169 photonArray->SetOwner(kTRUE);
170 fPhotonCorr->AddAt(photonArray, iz);
172 TObjArray * pionArray = new TObjArray();
173 pionArray->SetOwner(kTRUE);
174 fPionCorr->AddAt(pionArray, iz);
176 TList * photonList = new TList();
177 photonList->SetName(Form("photon_%d", iz));
178 photonList->SetOwner(kTRUE);
179 hPhoton->AddAt(photonList, iz);
181 TList * pionList = new TList();
182 pionList->SetName(Form("pion_%d", iz));
183 pionList->SetOwner(kTRUE);
184 hPion->AddAt(pionList, iz);
186 for(Int_t ic = 0; ic < fAxisCent.GetNbins(); ic++) {
188 TString nameString = Form("%d_%d", iz, ic);
189 TString titleString = Form("%f < Z < %f ... %f cent %f",
190 fAxisZ.GetBinLowEdge(iz+1), fAxisZ.GetBinUpEdge(iz+1),
191 fAxisCent.GetBinLowEdge(ic+1), fAxisCent.GetBinUpEdge(ic+1));
195 AliAnaConvCorrPhoton * photonCorr = new AliAnaConvCorrPhoton(Form("PhotonCorr_%s", nameString.Data()), Form("photon %s", titleString.Data()));
196 photonArray->AddAt(photonCorr, ic);
197 photonCorr->GetAxistPt().Set(fAxisPt.GetNbins(), fAxisPt.GetXbins()->GetArray());
198 photonCorr->GetAxiscPt().Set(fAxisPt.GetNbins(), fAxisPt.GetXbins()->GetArray());
199 photonCorr->CreateHistograms();
200 photonList->Add(photonCorr->GetHistograms());
202 AliAnaConvCorrPion * pionCorr = new AliAnaConvCorrPion(Form("PionCorr_%s", nameString.Data()), Form("pion %s", titleString.Data()));
203 pionArray->AddAt(pionCorr, ic);
204 pionCorr->GetAxistPt().Set(fAxisPt.GetNbins(), fAxisPt.GetXbins()->GetArray());
205 pionCorr->GetAxiscPt().Set(fAxisPt.GetNbins(), fAxisPt.GetXbins()->GetArray());
206 pionCorr->GetAxisM().Set(fAxisPiM.GetNbins(), fAxisPiM.GetXbins()->GetArray());
207 pionCorr->CreateHistograms();
208 pionList->Add(pionCorr->GetHistograms());
213 //________________________________________________________________________
214 void AliAnalysisTaskdPhi::UserCreateOutputObjects() {
217 fHistograms = new TList();
218 fHistograms->SetName("dPhi_histograms");
219 fHistograms->SetOwner(kTRUE);
221 fHistoGamma = new TList();
222 fHistoGamma->SetName("Gamma_histo");
223 fHistoGamma->SetOwner(kTRUE);
225 fHistoPion = new TList();
226 fHistoPion->SetName("Pion_histo");
227 fHistoPion->SetOwner(kTRUE);
231 fV0Filter->InitCutHistograms();
232 fHistograms->Add(fV0Filter->GetCutHistograms());
239 ///Set up ME histograms
240 TList * MEHistograms = new TList();
241 MEHistograms->SetName("MEHistograms");
242 MEHistograms->SetOwner(kTRUE);
243 fHistograms->Add(MEHistograms);
245 hMETracks = new TObjArray();
246 hMETracks->SetName("TrackArray");
247 hMETracks->SetOwner(kTRUE);
248 hMEPhotons = new TObjArray();
249 hMEPhotons->SetName("PhotonArray");
250 hMEPhotons->SetOwner(kTRUE);
251 hMEPions = new TObjArray();
252 hMEPions->SetName("PionArray");
253 hMEPions->SetOwner(kTRUE);
255 MEHistograms->Add(hMETracks);
256 MEHistograms->Add(hMEPions);
257 MEHistograms->Add(hMEPhotons);
259 hMEvents = new TH2I("hMEvents", "Nevents vs centrality vertexz",
260 fAxisZ.GetNbins(), fAxisZ.GetBinLowEdge(1), fAxisZ.GetBinUpEdge(fAxisZ.GetNbins()),
261 fAxisCent.GetNbins(), fAxisCent.GetBinLowEdge(1), fAxisCent.GetBinUpEdge(fAxisCent.GetNbins()));
262 hMEvents->GetYaxis()->Set(fAxisCent.GetNbins(), fAxisCent.GetXbins()->GetArray());
263 MEHistograms->Add(hMEvents);
267 axesList.AddAt(&GetAxisEta(), 0);
268 axesList.AddAt(&GetAxisPhi(), 1);
269 axesList.AddAt(&GetAxisPt(), 2);
270 axesList.SetOwner(kFALSE);
273 piAxesList.AddAt(&GetAxisEta(), 0);
274 piAxesList.AddAt(&GetAxisPhi(), 1);
275 piAxesList.AddAt(&GetAxisPt(), 2);
276 piAxesList.AddAt(&GetAxisPiMass(), 3);
277 piAxesList.SetOwner(kFALSE);
280 TList * outAxesList = new TList();
281 outAxesList->Add(&fAxisCent);
282 outAxesList->Add(&fAxisZ);
284 for(Int_t iz = 0; iz < fAxisZ.GetNbins(); iz++) {
285 TObjArray * trackArray = new TObjArray();
286 trackArray->SetName(Form("METracks_%d", iz));
287 trackArray->SetOwner(kTRUE);
288 TObjArray * photonArray = new TObjArray();
289 photonArray->SetName(Form("MEPhotons_%d", iz));
290 photonArray->SetOwner(kTRUE);
291 TObjArray * pionArray = new TObjArray();
292 pionArray->SetName(Form("MEPions_%d", iz));
293 pionArray->SetOwner(kTRUE);
296 hMEPions->AddAt(pionArray, iz);
297 hMETracks->AddAt(trackArray, iz);
298 hMEPhotons->AddAt(photonArray, iz);
300 for(Int_t ic = 0; ic < fAxisCent.GetNbins(); ic++) {
302 TString nameString = Form("%d_%d", iz, ic);
303 TString titleString = Form("%f < Z < %f ... %f cent %f",
304 fAxisZ.GetBinLowEdge(iz+1), fAxisZ.GetBinUpEdge(iz+1),
305 fAxisCent.GetBinLowEdge(ic+1), fAxisCent.GetBinUpEdge(ic+1));
308 THnSparseF * trackHistogram = CreateSparse(Form("tracks_%s", nameString.Data()),
309 Form("tracks %s", titleString.Data()), &axesList );
310 trackArray->AddAt(trackHistogram, ic);
312 THnSparseF * photonHistogram = CreateSparse(Form("photons_%s", nameString.Data()),
313 Form("photons %s", titleString.Data()), &axesList );
314 photonArray->AddAt(photonHistogram, ic);
316 THnSparseF * pionHistogram = CreateSparse(Form("pions_%s", nameString.Data()),
317 Form("pions %s", titleString.Data()), &piAxesList );
318 pionArray->AddAt(pionHistogram, ic);
322 PostData(1, fHistograms);
323 PostData(2, fHistoGamma);
324 PostData(3, fHistoPion);
328 ///________________________________________________________________________
329 THnSparseF * AliAnalysisTaskdPhi::CreateSparse(TString nameString, TString titleString, TList * axesList) {
331 const Int_t dim = axesList->GetSize();
338 for(Int_t i = 0; i<dim; i++) {
339 TAxis * axis = dynamic_cast<TAxis*>(axesList->At(i));
343 cout << "AliAnalysisTaskdPhi::CreateSparse: Error error, all the axes are not present in axis list" << endl;
348 for(Int_t i = 0; i<dim; i++) {
349 bins[i] = axes[i]->GetNbins();
350 min[i] = axes[i]->GetBinLowEdge(1);
351 max[i] = axes[i]->GetBinUpEdge(axes[i]->GetNbins());
354 THnSparseF * sparse = new THnSparseF(Form("METracks_%s", nameString.Data()),
355 Form("tracks %s", titleString.Data()),
356 dim, bins, min, max);
358 for(Int_t i = 0; i<dim; i++) {
359 sparse->GetAxis(i)->SetNameTitle(axes[i]->GetName(), axes[i]->GetTitle() );
360 if(axes[i]->GetXbins()->GetSize() > 0) {
361 sparse->SetBinEdges(i, axes[i]->GetXbins()->GetArray() );
368 //________________________________________________________________________
369 void AliAnalysisTaskdPhi::UserExec(Option_t *) {
372 //if(! fV0Filter->EventIsSelected(fInputEvent)) return;
375 AliAnalysisManager *man=AliAnalysisManager::GetAnalysisManager();
376 Bool_t isAOD=man->GetInputEventHandler()->IsA()==AliAODInputHandler::Class();
378 AliInputEventHandler* inputHandler = (AliInputEventHandler*) (man->GetInputEventHandler());
380 cout << "cout no input event handler"<<endl;
385 if ( fV0Filter && !fV0Filter->GetPIDResponse() ) {
386 if ( inputHandler->GetPIDResponse() ){
387 fV0Filter->SetPIDResponse( inputHandler->GetPIDResponse() );
392 if (!fV0Filter->GetPIDResponse()){
393 fV0Filter->InitAODpidUtil(1);
399 Double_t centrality = 0.0;
400 Double_t eventPlane = 0.0;
401 Double_t vertexz = fInputEvent->GetPrimaryVertex()->GetZ();
403 AliAODHeader * header = static_cast<AliAODHeader*>(fInputEvent->GetHeader());
404 centrality = header->GetCentrality();
405 eventPlane = header->GetEventplane();
407 centrality = static_cast<AliESDEvent*>(fInputEvent)->GetCentrality()->GetCentralityPercentile("kV0M");
408 eventPlane = fInputEvent->GetEventplane()->GetEventplane("Q");
411 if(DebugLevel () > 15) {
412 cout << "centrality: " << centrality << " " << GetBin(fAxisCent, centrality) << endl;
413 cout << "vertexz: " << vertexz << " " << GetBin(fAxisZ, vertexz) << endl;
414 cout << "eventPlane: " << eventPlane << " " << endl;
417 const Int_t centBin = GetBin(fAxisCent, centrality);
418 const Int_t vertexBin = GetBin(fAxisZ, vertexz);
420 if(centBin < 0 || vertexBin < 0) {
421 AliError("bin out of range");
428 TClonesArray * aodGammas = GetConversionGammas(isAOD);
430 AliError("no aod gammas found!");
434 if(aodGammas->GetEntriesFast() > 0) {
435 if( static_cast<AliAODConversionParticle*>(aodGammas->At(0))->GetLabel(0) == fL1 &&
436 static_cast<AliAODConversionParticle*>(aodGammas->At(0))->GetLabel(1) == fL2
440 fL1 = static_cast<AliAODConversionParticle*>(aodGammas->At(0))->GetLabel(0);
441 fL2 = static_cast<AliAODConversionParticle*>(aodGammas->At(0))->GetLabel(1);
442 //cout << aodGammas->GetEntriesFast() << " " << fInputEvent->GetNumberOfTracks() << "c" << endl;
445 if(DebugLevel() > 1) printf("Number of conversion gammas %d \n", aodGammas->GetEntriesFast());
446 for(Int_t ig = 0; ig < aodGammas->GetEntriesFast(); ig++) {
447 AliAODConversionPhoton * photon = dynamic_cast<AliAODConversionPhoton*>(aodGammas->At(ig));
449 if(!photon) continue;
450 if(!fV0Filter || fV0Filter->PhotonIsSelected(static_cast<AliConversionPhotonBase*>(photon), fInputEvent)) {
451 fGammas->Add(static_cast<TObject*>(photon));
455 if(DebugLevel() > 4) printf("Number of accepted gammas %d \n", fGammas->GetEntriesFast());
457 THnSparseF * trackMehist = GetMEHistogram(vertexBin, centBin, hMETracks);
458 hMEvents->Fill(vertexz, centrality);
460 ///Add tracks to array
462 for(Int_t iTrack = 0; iTrack < fInputEvent->GetNumberOfTracks(); iTrack++) {
464 AliVTrack * track = static_cast<AliVTrack*>(fInputEvent->GetTrack(iTrack));
465 if(track->Pt() < 0.5) continue;
466 if(TMath::Abs(track->Eta()) > 0.8) continue;
469 Double_t hValues[3] = { track->Eta(), track->Phi(), track->Pt() };
470 if(trackMehist) trackMehist->Fill(hValues);
473 Process(fGammas, &tracks, vertexBin, centBin);
475 PostData(1, fHistograms);
476 PostData(2, fHistoGamma);
477 PostData(3, fHistoPion);
482 //________________________________________________________________________
483 void AliAnalysisTaskdPhi::Process(TObjArray * gammas, TObjArray * tracks, Int_t vertexBin, Int_t centBin) {
486 if(DebugLevel() > 4) printf("Number of accepted tracks %d \n", tracks->GetEntriesFast());
488 THnSparseF * gHisto = GetMEHistogram(vertexBin, centBin, hMEPhotons);
489 THnSparseF * piHisto = GetMEHistogram(vertexBin, centBin, hMEPions);
491 AliAnaConvCorrBase * gCorr = GetCorrObject(vertexBin, centBin, fPhotonCorr);
492 AliAnaConvCorrBase * piCorr = GetCorrObject(vertexBin, centBin, fPionCorr);
494 if(!gCorr || !piCorr) {
495 AliError("corr object missing");
501 for(Int_t i1 = 0; i1 < gammas->GetEntriesFast(); i1++) {
502 AliAODConversionPhoton * ph1 = static_cast<AliAODConversionPhoton*>(gammas->UncheckedAt(i1));
503 Int_t tIDs[4] = {ph1->GetLabel(0), ph1->GetLabel(1), -1, -1};
505 // AliESDv0 * v0 = ((AliESDEvent*) (fInputEvent))->GetV0(ph1->GetV0Index());
507 // cout << v0->GetNindex() << " " << v0->GetPindex() << endl;
508 // cout << tIDs[1] << " " << tIDs[0] << endl;
512 Bool_t leading = fIsoAna->IsLeading(static_cast<AliAODConversionParticle*>(ph1), tracks, tIDs);
515 gCorr->CorrelateWithTracks( static_cast<AliAODConversionParticle*>(ph1), tracks, tIDs, kFALSE);
516 Double_t phval[3] = {ph1->Eta(), ph1->Phi(), ph1->Pt()};
520 for(Int_t i2 = 0; i2 < i1; i2++) {
521 AliAODConversionPhoton * ph2 = static_cast<AliAODConversionPhoton*>(gammas->UncheckedAt(i2));
523 if( ph2->GetTrackLabelPositive()==ph1->GetTrackLabelPositive()
524 || ph2->GetTrackLabelNegative()==ph1->GetTrackLabelNegative()
525 || ph2->GetTrackLabelNegative()==ph1->GetTrackLabelPositive()
526 || ph2->GetTrackLabelPositive()==ph1->GetTrackLabelNegative()) {
530 AliAODConversionMother * pion = new AliAODConversionMother(ph1, ph2);
531 pion->SetLabels(i1, i2);
533 if(!fV0Filter || fV0Filter->MesonIsSelected(pion, kTRUE) ) {
535 Bool_t leadingpi = fIsoAna->IsLeading(static_cast<AliAODConversionParticle*>(pion), tracks, tIDs);
538 //piCorr->FillTriggerCounters(pion);
540 Double_t pival[4] = { pion->Eta(), pion->Phi(), pion->Pt(), pion->M() };
541 piHisto->Fill(pival);
543 if(pion->M() > fAxisPiM.GetXmin() && pion->M() < fAxisPiM.GetXmax()) {
544 tIDs[2] = ph2->GetLabel(0);
545 tIDs[3] = ph2->GetLabel(1);
546 piCorr->CorrelateWithTracks(pion, tracks, tIDs, kFALSE);
555 //________________________________________________________________________
556 void AliAnalysisTaskdPhi::Terminate(Option_t *) {
558 // Draw result to the screen
559 // Called once at the end of the query
562 //________________________________________________________________________
563 TClonesArray * AliAnalysisTaskdPhi::GetConversionGammas(Bool_t isAOD) {
568 TClonesArray * gammas = dynamic_cast<TClonesArray*>(fInputEvent->FindListObject(fDeltaAODBranchName.Data()));
572 //If not found try to locate branch
573 FindDeltaAODBranchName(AODEvent());
574 //gammas = dynamic_cast<TClonesArray*>(fInputEvent->FindListObject(fDeltaAODBranchName.Data()));
577 gammas = dynamic_cast<TClonesArray*>(AODEvent()->FindListObject(fDeltaAODBranchName.Data()));
581 TClonesArray * gammas = dynamic_cast<TClonesArray*>(GetInputData(1));
587 //________________________________________________________________________
588 void AliAnalysisTaskdPhi::FindDeltaAODBranchName(AliAODEvent * event){
590 TList *list=event->GetList();
591 for(Int_t ii=0;ii<list->GetEntries();ii++){
592 TString name((list->At(ii))->GetName());
593 if(name.BeginsWith("GammaConv")&&name.EndsWith("gamma")){
594 fDeltaAODBranchName=name;
595 AliInfo(Form("Set DeltaAOD BranchName to: %s",fDeltaAODBranchName.Data()));