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 ic = 0; ic < fAxisCent.GetNbins(); ic++) {
 168         TObjArray * photonArray = new TObjArray();
 169         photonArray->SetOwner(kTRUE);
 170         fPhotonCorr->AddAt(photonArray, ic);
 171
 172         TObjArray * pionArray = new TObjArray();
 173         pionArray->SetOwner(kTRUE);
 174         fPionCorr->AddAt(pionArray, ic);
 175
 176         TList * photonList = new TList();
 177         photonList->SetName(Form("photon_%d", ic));
 178         photonList->SetOwner(kTRUE);
 179         hPhoton->AddAt(photonList, ic);
 180
 181         TList * pionList = new TList();
 182         pionList->SetName(Form("pion_%d", ic));
 183         pionList->SetOwner(kTRUE);
 184         hPion->AddAt(pionList, ic);
 185
 186
 187         for(Int_t iz = 0; iz < fAxisZ.GetNbins(); iz++) {
 188           TString nameString = Form("%d_%d", ic, iz);
 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, iz);
 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, iz);
 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   fHistograms->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   ///Create 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   cout <<"a"<<endl;
 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   cout <<"a"<<endl;
 338   Double_t centrality = 0.0;
 339   Double_t eventPlane = 0.0;
 340   Double_t vertexz = fInputEvent->GetPrimaryVertex()->GetZ();
 341   if(isAOD) {
 342     AliAODHeader * header = static_cast<AliAODHeader*>(fInputEvent->GetHeader());
 343         centrality = header->GetCentrality();
 344         eventPlane = header->GetEventplane();
 345   } else {
 346         centrality = static_cast<AliESDEvent*>(fInputEvent)->GetCentrality()->GetCentralityPercentile("V0M");
 347         eventPlane = fInputEvent->GetEventplane()->GetEventplane("Q");
 348   }
 349
 350   cout <<"a"<<endl;
 351
 352   const Int_t centBin = GetBin(fAxisCent, centrality);
 353   const Int_t vertexBin = GetBin(fAxisZ, vertexz);
 354
 355
 356   if(DebugLevel () > 4) {
 357         cout << "centrality: " << centrality <<  " " << GetBin(fAxisCent, centrality) << endl;
 358         cout << "vertexz: " << vertexz <<  " " << GetBin(fAxisZ, vertexz) << endl;
 359         cout << "eventPlane: " << eventPlane <<  " " << endl;
 360   }
 361
 362   cout <<"a"<<endl;
 363
 364   if(centBin < 0 || vertexBin < 0) {
 365         AliError("bin out of range");
 366         cout << "bad bin"<<endl;
 367         return;
 368   }
 369
 370   cout <<"a"<<endl;
 371   fGammas->Clear();
 372   fPions->Clear();
 373
 374   TClonesArray * aodGammas = GetConversionGammas(isAOD);
 375   if(!aodGammas) {
 376         AliError("no aod gammas found!");
 377         return;
 378   }
 379
 380   if(DebugLevel() > 1) printf("Number of conversion gammas %d \n", aodGammas->GetEntriesFast());
 381   for(Int_t ig = 0; ig < aodGammas->GetEntriesFast(); ig++) {
 382     AliAODConversionPhoton * photon = dynamic_cast<AliAODConversionPhoton*>(aodGammas->At(ig));
 383
 384     if(!photon) continue;
 385     if(!fV0Filter || fV0Filter->PhotonIsSelected(static_cast<AliConversionPhotonBase*>(photon), fInputEvent)) {
 386       fGammas->Add(static_cast<TObject*>(photon));
 387     }
 388   }
 389
 390   if(DebugLevel() > 4) printf("Number of accepted gammas %d \n", fGammas->GetEntriesFast());
 391   hMEvents->Fill(vertexz, centrality);
 392
 393
 394
 395   ///create track array
 396   TObjArray tracks;
 397   const Double_t etalim[2] = { fAxisEta.GetBinLowEdge(1), fAxisEta.GetBinUpEdge(fAxisEta.GetNbins())};
 398   for(Int_t iTrack = 0; iTrack < fInputEvent->GetNumberOfTracks(); iTrack++) {
 399
 400         AliVTrack * track = static_cast<AliVTrack*>(fInputEvent->GetTrack(iTrack));
 401         if(track->Pt() < fAxiscPt.GetBinLowEdge(1) ) continue;
 402         if(track->Eta() < etalim[0] || track->Eta() > etalim[1]) continue;
 403
 404
 405         if(!fTrackCuts || fTrackCuts->IsSelected((track))) {
 406           tracks.Add(track);
 407         }
 408   }
 409
 410   Process(fGammas, &tracks, vertexBin, centBin);
 411
 412   PostData(1, fHistograms);
 413   PostData(2, fHistoGamma);
 414   PostData(3, fHistoPion);
 415
 416 }
 417
 418
 419 //________________________________________________________________________
 420 void AliAnalysisTaskdPhi::Process(TObjArray * gammas, TObjArray * tracks, Int_t vertexBin, Int_t centBin) {
 421   ///Process stuff
 422
 423   if(DebugLevel() > 4) printf("Number of accepted gammas, tracks %d  %d \n", gammas->GetEntriesFast(), tracks->GetEntriesFast());
 424
 425   AliAnaConvCorrBase * gCorr = GetCorrObject(vertexBin, centBin, fPhotonCorr);
 426   AliAnaConvCorrPion * piCorr = dynamic_cast<AliAnaConvCorrPion*>(GetCorrObject(vertexBin, centBin, fPionCorr));
 427
 428   if(!gCorr || !piCorr) {
 429         AliError("corr object missing");
 430         return;
 431   }
 432
 433   for(Int_t i1 = 0; i1 < gammas->GetEntriesFast(); i1++) {
 434         AliAODConversionPhoton * ph1 = static_cast<AliAODConversionPhoton*>(gammas->UncheckedAt(i1));
 435         Int_t tIDs[4] = {ph1->GetLabel(0), ph1->GetLabel(1), -1, -1};
 436
 437         Int_t leading = fIsoAna->IsLeading(static_cast<AliAODConversionParticle*>(ph1), tracks, tIDs);
 438         if(ph1->Pt() > fAxistPt.GetBinLowEdge(1)) {
 439           gCorr->CorrelateWithTracks( static_cast<AliAODConversionParticle*>(ph1), tracks, tIDs, leading);
 440         }
 441         for(Int_t i2 = 0; i2 < i1; i2++) {
 442           AliAODConversionPhoton * ph2 = static_cast<AliAODConversionPhoton*>(gammas->UncheckedAt(i2));
 443
 444           if( ph2->GetTrackLabelPositive()==ph1->GetTrackLabelPositive()
 445                   || ph2->GetTrackLabelNegative()==ph1->GetTrackLabelNegative()
 446                   || ph2->GetTrackLabelNegative()==ph1->GetTrackLabelPositive()
 447                   || ph2->GetTrackLabelPositive()==ph1->GetTrackLabelNegative()) {
 448                 continue;
 449           }
 450
 451           AliAODConversionMother * pion = new AliAODConversionMother(ph1, ph2);
 452           pion->SetLabels(i1, i2);
 453
 454           if(!fV0Filter || fV0Filter->MesonIsSelected(pion, kTRUE) ) {
 455
 456                 Int_t leadingpi = fIsoAna->IsLeading(static_cast<AliAODConversionParticle*>(pion), tracks, tIDs);
 457                 piCorr->FillTriggerCounters(pion, leadingpi);
 458                 tIDs[2] = ph2->GetLabel(0);
 459                 tIDs[3] = ph2->GetLabel(1);
 460                 if(pion->Pt() > fAxistPt.GetBinLowEdge(1) &&
 461                    pion->M() > fAxisPiM.GetBinLowEdge(1) &&
 462                    pion->M() < fAxisPiM.GetBinUpEdge(fAxisPiM.GetNbins())) {
 463                   piCorr->CorrelateWithTracks(pion, tracks, tIDs, leadingpi);
 464                 }
 465           }
 466         }
 467   }
 468 }
 469
 470 //________________________________________________________________________
 471 void AliAnalysisTaskdPhi::Terminate(Option_t *) {
 472
 473   // Draw result to the screen
 474   // Called once at the end of the query
 475 }
 476
 477 //________________________________________________________________________
 478 TClonesArray * AliAnalysisTaskdPhi::GetConversionGammas(Bool_t isAOD) {
 479
 480
 481   if(isAOD) {
 482
 483         TClonesArray * gammas = dynamic_cast<TClonesArray*>(fInputEvent->FindListObject(fDeltaAODBranchName.Data()));
 484         if(gammas) {
 485           return gammas;
 486         }
 487
 488         FindDeltaAODBranchName(fInputEvent);
 489         gammas = dynamic_cast<TClonesArray*>(fInputEvent->FindListObject(fDeltaAODBranchName.Data()));
 490         return gammas;
 491
 492   } else {
 493         TClonesArray * gammas = dynamic_cast<TClonesArray*>(GetInputData(1));
 494         return gammas;
 495   }
 496
 497 }
 498
 499 //________________________________________________________________________
 500 void AliAnalysisTaskdPhi::FindDeltaAODBranchName(AliVEvent * event){
 501   ///Find aod branch
 502   TList *list=event->GetList();
 503   for(Int_t ii=0;ii<list->GetEntries();ii++){
 504         TString name((list->At(ii))->GetName());
 505         if(name.BeginsWith("GammaConv")&&name.EndsWith("gamma")){
 506           fDeltaAODBranchName=name;
 507           AliInfo(Form("Set DeltaAOD BranchName to: %s",fDeltaAODBranchName.Data()));
 508           return;
 509         }
 510   }
 511 }
 512
 513