PWGGA/GammaConv/AliAnalysisTaskdPhi.cxx

   1 /**************************************************************************
   2  * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
   3  *                                                                                                                                        *
   4  * Authors: Svein Lindal                                                  *
   5  * Version 1.0                                                                    *
   6  *                                                                                                                                        *
   7  *                                                                                                                                        *
   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  **************************************************************************/
  16
  17 ////////////////////////////////////////////////
  18 //---------------------------------------------
  19 // Class doing conversion gamma dPhi correlations
  20 // Gamma Conversion analysis
  21 //---------------------------------------------
  22 ////////////////////////////////////////////////
  23
  24 #include "AliAnalysisTaskdPhi.h"
  25
  26 #include <TH2I.h>
  27 #include <TList.h>
  28 #include <TChain.h>
  29
  30 #include <AliAnalysisManager.h>
  31 #include <AliInputEventHandler.h>
  32 #include <AliESDInputHandler.h>
  33 #include <AliAODInputHandler.h>
  34 #include <AliAnalysisFilter.h>
  35
  36 #include "AliConversionTrackCuts.h"
  37 #include "AliConversionCuts.h"
  38 #include "AliAODConversionPhoton.h"
  39 #include "AliAODConversionMother.h"
  40 #include "AliAnaConvCorrPhoton.h"
  41 #include "AliAnaConvCorrPion.h"
  42 #include "AliAnaConvIsolation.h"
  43 // Author Svein Lindal <slindal@fys.uio.no>
  44 using namespace std;
  45
  46 ClassImp(AliAnalysisTaskdPhi)
  47
  48
  49 //________________________________________________________________________
  50 AliAnalysisTaskdPhi::AliAnalysisTaskdPhi(const char *name) : AliAnalysisTaskSE(name),
  51   fHistograms(NULL),
  52   fHistoGamma(NULL),
  53   fHistoPion(NULL),
  54   fV0Filter(NULL),
  55   fTrackCuts(NULL),
  56   fGammas(NULL),
  57   fPions(NULL),
  58   hMETracks(NULL),
  59   hMEPhotons(NULL),
  60   hMEPions(NULL),
  61   hMEvents(NULL),
  62   fPhotonCorr(NULL),
  63   fPionCorr(NULL),
  64   fIsoAna(NULL),
  65   fDeltaAODBranchName("AliAODGammaConversion_gamma"),
  66   fAxistPt(),
  67   fAxiscPt(),
  68   fAxisEta(),
  69   fAxisPhi(),
  70   fAxisCent(),
  71   fAxisZ(),
  72   fAxisPiM()
  73 {
  74   //constructor
  75   fAxistPt.SetNameTitle("tPtAxis", "tPt");
  76   fAxistPt.Set(20, 0, 100);
  77
  78   fAxiscPt.SetNameTitle("cPtAxis", "cPt");
  79   fAxiscPt.Set(20, 0, 100);
  80
  81   fAxisEta.SetNameTitle("EtaAxis", "Eta");
  82   fAxisEta.Set(160, -0.8, 0.8);
  83
  84   fAxisPhi.SetNameTitle("PhiAxis", "Phi");
  85   fAxisPhi.Set(128, 0, TMath::TwoPi());
  86
  87   fAxisZ.SetNameTitle("ZAxis", "Z");
  88   fAxisZ.Set(4, -10, 10);
  89
  90   fAxisCent.SetNameTitle("CentAxis", "Cent");
  91   Double_t centbins[5] = {0, 10, 30, 60, 100.1};
  92   fAxisCent.Set(4, centbins);
  93
  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);
  97
  98   fGammas = new TObjArray();
  99   fGammas->SetOwner(kFALSE);
 100
 101   fPions = new TObjArray();
 102   fPions->SetOwner(kFALSE);
 103
 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());
 110 }
 111
 112
 113
 114 //________________________________________________________________________
 115 AliAnalysisTaskdPhi::~AliAnalysisTaskdPhi(){
 116   //destructor
 117   if(fPions)
 118         delete fPions;
 119   fPions = NULL;
 120
 121   if(fGammas)
 122         delete fGammas;
 123   fGammas = NULL;
 124
 125   if(fIsoAna)
 126         delete fIsoAna;
 127   fIsoAna = NULL;
 128
 129   if(fV0Filter)
 130         delete fV0Filter;
 131   fV0Filter = NULL;
 132
 133   if(fHistograms)
 134         delete fHistograms;
 135   fHistograms = NULL;
 136
 137   if(fHistoPion)
 138         delete fHistoPion;
 139   fHistoPion = NULL;
 140
 141   if(fHistoGamma)
 142         delete fHistoGamma;
 143   fHistoGamma = NULL;
 144
 145 }
 146
 147 ///________________________________________________________________________
 148 void AliAnalysisTaskdPhi::SetUpCorrObjects() {
 149   ///Creat corr obj
 150   fIsoAna = new AliAnaConvIsolation();
 151
 152
 153   fPhotonCorr = new TObjArray();
 154   fPionCorr = new TObjArray();
 155
 156   TList * hPhoton = new TList();
 157   hPhoton->SetName("hPhotonCorr");
 158   hPhoton->SetOwner(kTRUE);
 159   fHistoGamma->Add(hPhoton);
 160
 161   TList * hPion = new TList();
 162   hPion->SetName("hPionCorr");
 163   hPion->SetOwner(kTRUE);
 164   fHistoPion->Add(hPion);
 165
 166
 167   for(Int_t iz = 0; iz < fAxisZ.GetNbins(); iz++) {
 168         TObjArray * photonArray = new TObjArray();
 169         photonArray->SetOwner(kTRUE);
 170         fPhotonCorr->AddAt(photonArray, iz);
 171
 172         TObjArray * pionArray = new TObjArray();
 173         pionArray->SetOwner(kTRUE);
 174         fPionCorr->AddAt(pionArray, iz);
 175
 176         TList * photonList = new TList();
 177         photonList->SetName(Form("photon_%d", iz));
 178         photonList->SetOwner(kTRUE);
 179         hPhoton->AddAt(photonList, iz);
 180
 181         TList * pionList = new TList();
 182         pionList->SetName(Form("pion_%d", iz));
 183         pionList->SetOwner(kTRUE);
 184         hPion->AddAt(pionList, iz);
 185
 186         for(Int_t ic = 0; ic < fAxisCent.GetNbins(); ic++) {
 187
 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));
 192
 193
 194
 195           AliAnaConvCorrPhoton * photonCorr = new AliAnaConvCorrPhoton(Form("PhotonCorr_%s", nameString.Data()), Form("photon %s", titleString.Data()));
 196           photonArray->AddAt(photonCorr, ic);
 197           photonCorr->GetAxistPt().Set(fAxistPt.GetNbins(), fAxistPt.GetXbins()->GetArray());
 198           photonCorr->GetAxiscPt().Set(fAxiscPt.GetNbins(), fAxiscPt.GetXbins()->GetArray());
 199           photonCorr->CreateHistograms();
 200           photonList->Add(photonCorr->GetHistograms());
 201
 202           AliAnaConvCorrPion * pionCorr = new AliAnaConvCorrPion(Form("PionCorr_%s", nameString.Data()), Form("pion %s", titleString.Data()));
 203           pionArray->AddAt(pionCorr, ic);
 204           pionCorr->GetAxistPt().Set(fAxistPt.GetNbins(), fAxistPt.GetXbins()->GetArray());
 205           pionCorr->GetAxiscPt().Set(fAxiscPt.GetNbins(), fAxiscPt.GetXbins()->GetArray());
 206           pionCorr->GetAxisM().Set(fAxisPiM.GetNbins(), fAxisPiM.GetXbins()->GetArray());
 207           pionCorr->CreateHistograms();
 208           pionList->Add(pionCorr->GetHistograms());
 209         }
 210   }
 211 }
 212
 213 //________________________________________________________________________
 214 void AliAnalysisTaskdPhi::UserCreateOutputObjects() {
 215   // Create histograms
 216
 217   fHistograms = new TList();
 218   fHistograms->SetName("dPhi_histograms");
 219   fHistograms->SetOwner(kTRUE);
 220
 221   fHistoGamma = new TList();
 222   fHistoGamma->SetName("Gamma_histo");
 223   fHistoGamma->SetOwner(kTRUE);
 224
 225   fHistoPion = new TList();
 226   fHistoPion->SetName("Pion_histo");
 227   fHistoPion->SetOwner(kTRUE);
 228
 229
 230   if(fV0Filter) {
 231         fV0Filter->InitCutHistograms();
 232         fHistograms->Add(fV0Filter->GetCutHistograms());
 233   }
 234
 235
 236   AliConversionTrackCuts * tc = dynamic_cast<AliConversionTrackCuts*>(fTrackCuts);
 237   if(tc) fHistograms->Add(tc->CreateHistograms());
 238
 239   SetUpCorrObjects();
 240
 241
 242   ///Set up ME histograms
 243   TList * MEHistograms = new TList();
 244   MEHistograms->SetName("MEHistograms");
 245   MEHistograms->SetOwner(kTRUE);
 246   fHistograms->Add(MEHistograms);
 247
 248
 249   hMEvents = new TH2I("hMEvents", "Nevents vs centrality vertexz",
 250                                           fAxisZ.GetNbins(), fAxisZ.GetBinLowEdge(1), fAxisZ.GetBinUpEdge(fAxisZ.GetNbins()),
 251                                           fAxisCent.GetNbins(), fAxisCent.GetBinLowEdge(1), fAxisCent.GetBinUpEdge(fAxisCent.GetNbins()));
 252   hMEvents->GetYaxis()->Set(fAxisCent.GetNbins(), fAxisCent.GetXbins()->GetArray());
 253   MEHistograms->Add(hMEvents);
 254
 255   TList * outAxesList = new TList();
 256   outAxesList->Add(&fAxisCent);
 257   outAxesList->Add(&fAxisZ);
 258   MEHistograms->Add(outAxesList);
 259
 260   PostData(1, fHistograms);
 261   PostData(2, fHistoGamma);
 262   PostData(3, fHistoPion);
 263
 264 }
 265
 266 ///________________________________________________________________________
 267 THnSparseF * AliAnalysisTaskdPhi::CreateSparse(TString nameString, TString titleString, TList * axesList) {
 268   ///Creat sparse
 269   const Int_t dim = axesList->GetSize();
 270
 271   TAxis * axes[dim];
 272   Int_t bins[dim];
 273   Double_t min[dim];
 274   Double_t max[dim];
 275
 276   for(Int_t i = 0; i<dim; i++) {
 277         TAxis * axis = dynamic_cast<TAxis*>(axesList->At(i));
 278         if(axis) {
 279           axes[i] = axis;
 280         } else {
 281           cout << "AliAnalysisTaskdPhi::CreateSparse: Error error, all the axes are not present in axis list" << endl;
 282           return NULL;
 283         }
 284   }
 285
 286   for(Int_t i = 0; i<dim; i++) {
 287         bins[i] = axes[i]->GetNbins();
 288         min[i] = axes[i]->GetBinLowEdge(1);
 289         max[i] = axes[i]->GetBinUpEdge(axes[i]->GetNbins());
 290   }
 291
 292   THnSparseF * sparse = new THnSparseF(Form("METracks_%s", nameString.Data()),
 293                                                                                            Form("tracks %s", titleString.Data()),
 294                                                                                            dim, bins, min, max);
 295
 296   for(Int_t i = 0; i<dim; i++) {
 297         sparse->GetAxis(i)->SetNameTitle(axes[i]->GetName(), axes[i]->GetTitle() );
 298         if(axes[i]->GetXbins()->GetSize() > 0) {
 299           sparse->SetBinEdges(i, axes[i]->GetXbins()->GetArray() );
 300         }
 301   }
 302
 303   return sparse;
 304 }
 305
 306 //________________________________________________________________________
 307 void AliAnalysisTaskdPhi::UserExec(Option_t *) {
 308   ///User exec.
 309
 310   //if(! fV0Filter->EventIsSelected(fInputEvent)) return;
 311
 312
 313   AliAnalysisManager *man=AliAnalysisManager::GetAnalysisManager();
 314   Bool_t isAOD=man->GetInputEventHandler()->IsA()==AliAODInputHandler::Class();
 315
 316   AliInputEventHandler* inputHandler = (AliInputEventHandler*) (man->GetInputEventHandler());
 317   if (!inputHandler) {
 318         cout << "cout no input event handler"<<endl;
 319         return;
 320   }
 321
 322
 323   if ( fV0Filter && !fV0Filter->GetPIDResponse() ) {
 324         if ( inputHandler->GetPIDResponse() ){
 325           fV0Filter->SetPIDResponse( inputHandler->GetPIDResponse() );
 326         } else {
 327
 328           //AOD case
 329           if (isAOD){
 330                 if (!fV0Filter->GetPIDResponse()){
 331                   fV0Filter->InitAODpidUtil(1);
 332                 }
 333           }
 334         }
 335   }
 336
 337   Double_t centrality = 0.0;
 338   Double_t eventPlane = 0.0;
 339   Double_t vertexz = fInputEvent->GetPrimaryVertex()->GetZ();
 340   if(isAOD) {
 341     AliAODHeader * header = static_cast<AliAODHeader*>(fInputEvent->GetHeader());
 342         centrality = header->GetCentrality();
 343         eventPlane = header->GetEventplane();
 344   } else {
 345         centrality = static_cast<AliESDEvent*>(fInputEvent)->GetCentrality()->GetCentralityPercentile("V0M");
 346         eventPlane = fInputEvent->GetEventplane()->GetEventplane("Q");
 347   }
 348
 349
 350   const Int_t centBin = GetBin(fAxisCent, centrality);
 351   const Int_t vertexBin = GetBin(fAxisZ, vertexz);
 352
 353
 354   if(DebugLevel () > 4) {
 355         cout << "centrality: " << centrality <<  " " << GetBin(fAxisCent, centrality) << endl;
 356         cout << "vertexz: " << vertexz <<  " " << GetBin(fAxisZ, vertexz) << endl;
 357         cout << "eventPlane: " << eventPlane <<  " " << endl;
 358   }
 359
 360
 361   if(centBin < 0 || vertexBin < 0) {
 362         AliError("bin out of range");
 363 //      cout << "bad bin"<<endl;
 364         return;
 365   }
 366
 367   fGammas->Clear();
 368   fPions->Clear();
 369
 370   TClonesArray * aodGammas = GetConversionGammas(isAOD);
 371   if(!aodGammas) {
 372         AliError("no aod gammas found!");
 373         return;
 374   }
 375
 376   if(DebugLevel() > 1) printf("Number of conversion gammas %d \n", aodGammas->GetEntriesFast());
 377   for(Int_t ig = 0; ig < aodGammas->GetEntriesFast(); ig++) {
 378     AliAODConversionPhoton * photon = dynamic_cast<AliAODConversionPhoton*>(aodGammas->At(ig));
 379
 380     if(!photon) continue;
 381     if(!fV0Filter || fV0Filter->PhotonIsSelected(static_cast<AliConversionPhotonBase*>(photon), fInputEvent)) {
 382       fGammas->Add(static_cast<TObject*>(photon));
 383     }
 384   }
 385
 386   if(DebugLevel() > 4) printf("Number of accepted gammas %d \n", fGammas->GetEntriesFast());
 387   hMEvents->Fill(vertexz, centrality);
 388
 389
 390
 391   ///create track array
 392   TObjArray tracks;
 393   const Double_t etalim[2] = { fAxisEta.GetBinLowEdge(1), fAxisEta.GetBinUpEdge(fAxisEta.GetNbins())};
 394   for(Int_t iTrack = 0; iTrack < fInputEvent->GetNumberOfTracks(); iTrack++) {
 395
 396         AliVTrack * track = static_cast<AliVTrack*>(fInputEvent->GetTrack(iTrack));
 397         if(track->Pt() < fAxiscPt.GetBinLowEdge(1) ) continue;
 398         if(track->Eta() < etalim[0] || track->Eta() > etalim[1]) continue;
 399
 400
 401         if(!fTrackCuts || fTrackCuts->IsSelected((track))) {
 402           tracks.Add(track);
 403         }
 404   }
 405
 406   Process(fGammas, &tracks, vertexBin, centBin);
 407
 408   PostData(1, fHistograms);
 409   PostData(2, fHistoGamma);
 410   PostData(3, fHistoPion);
 411
 412 }
 413
 414
 415 //________________________________________________________________________
 416 void AliAnalysisTaskdPhi::Process(TObjArray * gammas, TObjArray * tracks, Int_t vertexBin, Int_t centBin) {
 417   ///Process stuff
 418
 419   if(DebugLevel() > 4) printf("Number of accepted gammas, tracks %d  %d \n", gammas->GetEntriesFast(), tracks->GetEntriesFast());
 420
 421   AliAnaConvCorrBase * gCorr = GetCorrObject(vertexBin, centBin, fPhotonCorr);
 422   AliAnaConvCorrPion * piCorr = dynamic_cast<AliAnaConvCorrPion*>(GetCorrObject(vertexBin, centBin, fPionCorr));
 423
 424   if(!gCorr || !piCorr) {
 425         AliError("corr object missing");
 426         return;
 427   }
 428
 429   for(Int_t i1 = 0; i1 < gammas->GetEntriesFast(); i1++) {
 430         AliAODConversionPhoton * ph1 = static_cast<AliAODConversionPhoton*>(gammas->UncheckedAt(i1));
 431         Int_t tIDs[4] = {ph1->GetLabel(0), ph1->GetLabel(1), -1, -1};
 432
 433         Int_t leading = fIsoAna->IsLeading(static_cast<AliAODConversionParticle*>(ph1), tracks, tIDs);
 434         if(ph1->Pt() > fAxistPt.GetBinLowEdge(1)) {
 435           gCorr->CorrelateWithTracks( static_cast<AliAODConversionParticle*>(ph1), tracks, tIDs, leading);
 436         }
 437         for(Int_t i2 = 0; i2 < i1; i2++) {
 438           AliAODConversionPhoton * ph2 = static_cast<AliAODConversionPhoton*>(gammas->UncheckedAt(i2));
 439
 440           if( ph2->GetTrackLabelPositive()==ph1->GetTrackLabelPositive()
 441                   || ph2->GetTrackLabelNegative()==ph1->GetTrackLabelNegative()
 442                   || ph2->GetTrackLabelNegative()==ph1->GetTrackLabelPositive()
 443                   || ph2->GetTrackLabelPositive()==ph1->GetTrackLabelNegative()) {
 444                 continue;
 445           }
 446
 447           AliAODConversionMother * pion = new AliAODConversionMother(ph1, ph2);
 448           pion->SetLabels(i1, i2);
 449
 450
 451           if(!fV0Filter || fV0Filter->MesonIsSelected(pion, kTRUE) ) {
 452
 453                 Int_t leadingpi = fIsoAna->IsLeading(static_cast<AliAODConversionParticle*>(pion), tracks, tIDs);
 454                 piCorr->FillTriggerCounters(pion, leadingpi);
 455
 456                 tIDs[2] = ph2->GetLabel(0);
 457                 tIDs[3] = ph2->GetLabel(1);
 458                 if(pion->Pt() > fAxistPt.GetBinLowEdge(1) &&
 459                    pion->M() > fAxisPiM.GetBinLowEdge(1) &&
 460                    pion->M() < fAxisPiM.GetBinUpEdge(fAxisPiM.GetNbins())) {
 461                   piCorr->CorrelateWithTracks(pion, tracks, tIDs, leadingpi);
 462                 }
 463           }
 464         }
 465   }
 466 }
 467
 468
 469 //________________________________________________________________________
 470 void AliAnalysisTaskdPhi::Terminate(Option_t *) {
 471
 472   // Draw result to the screen
 473   // Called once at the end of the query
 474 }
 475
 476 //________________________________________________________________________
 477 TClonesArray * AliAnalysisTaskdPhi::GetConversionGammas(Bool_t isAOD) {
 478
 479
 480   if(isAOD) {
 481
 482         TClonesArray * gammas = dynamic_cast<TClonesArray*>(fInputEvent->FindListObject(fDeltaAODBranchName.Data()));
 483         if(gammas) {
 484           return gammas;
 485         }
 486
 487         FindDeltaAODBranchName(fInputEvent);
 488         gammas = dynamic_cast<TClonesArray*>(fInputEvent->FindListObject(fDeltaAODBranchName.Data()));
 489         return gammas;
 490
 491   } else {
 492         TClonesArray * gammas = dynamic_cast<TClonesArray*>(GetInputData(1));
 493         return gammas;
 494   }
 495
 496 }
 497
 498 //________________________________________________________________________
 499 void AliAnalysisTaskdPhi::FindDeltaAODBranchName(AliVEvent * event){
 500   ///Find aod branch
 501   TList *list=event->GetList();
 502   for(Int_t ii=0;ii<list->GetEntries();ii++){
 503         TString name((list->At(ii))->GetName());
 504         if(name.BeginsWith("GammaConv")&&name.EndsWith("gamma")){
 505           fDeltaAODBranchName=name;
 506           AliInfo(Form("Set DeltaAOD BranchName to: %s",fDeltaAODBranchName.Data()));
 507           return;
 508         }
 509   }
 510 }
 511
 512