PWGGA/GammaConv/AliAnaConvCorrBase.cxx

   1 /**************************************************************************
   2  * This file is property of and copyright by the ALICE HLT Project        *
   3  * ALICE Experiment at CERN, All rights reserved.                         *
   4  *                                                                        *
   5  * Primary Author: Svein Lindal <slindal@fys.uio.no>                      *
   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 /// @file   AliAnaConvCorrBase.cxx
  17 /// @author Svein Lindal
  18 /// @brief  Base class for analysation of conversion particle - track correlations
  19
  20
  21 #include "AliAnaConvCorrBase.h"
  22 #include "AliAODTrack.h"
  23
  24 #include "TClonesArray.h"
  25 #include "TH1F.h"
  26 #include "TH2F.h"
  27 #include "TList.h"
  28 #include "AliAODConversionParticle.h"
  29
  30 #include <iostream>
  31
  32 // Gamma - jet correlation analysis task
  33 // Authors: Svein Lindal
  34
  35
  36 using namespace std;
  37 ClassImp(AliAnaConvCorrBase)
  38
  39 //________________________________________________________________________
  40 AliAnaConvCorrBase::AliAnaConvCorrBase(TString name, TString title = "title") : TNamed(name, title),
  41   fHistograms(NULL),
  42   fAxesList(),
  43   fTrigAxisList(),
  44   fTrackAxisList(),
  45   fAxistPt(),
  46   fAxiscPt(),
  47   fAxisdEta(),
  48   fAxisdPhi(),
  49   fAxisIso(),
  50   fAxisMEEta(),
  51   fAxisMEPhi(),
  52   fCorrSparse(NULL),
  53   fTrigSparse(NULL),
  54   fTrackSparse(NULL)
  55 {
  56   //Constructor
  57
  58   SetUpDefaultBins();
  59
  60   fAxesList.SetOwner(kTRUE);
  61   fTrackAxisList.SetOwner(kTRUE);
  62   fTrigAxisList.SetOwner(kTRUE);
  63
  64 }
  65
  66
  67 //________________________________________________________________________________
  68 AliAnaConvCorrBase::~AliAnaConvCorrBase() {
  69   ///destructor
  70 }
  71
  72 //________________________________________________________________________________
  73 void AliAnaConvCorrBase::CreateHistograms() {
  74   CreateBaseHistograms();
  75 }
  76
  77 ///________________________________________________________________________________
  78 void AliAnaConvCorrBase::SetUpDefaultBins() {
  79   //Set up default bins
  80   fAxisdEta.Set(40, -1.6, 1.6);
  81   fAxisdEta.SetNameTitle("dEta", "delta eta");
  82
  83   fAxisdPhi.Set(32, -TMath::PiOver2(), 3*TMath::PiOver2());
  84   fAxisdPhi.SetNameTitle("dPhi", "delta Phi");
  85
  86   Double_t tptbins[14] = {3.0, 3.5, 4.0, 5.0, 6.0, 8.0, 10.0, 12.5, 15, 20, 25, 30, 50, 100};
  87   fAxistPt.Set(13, tptbins);
  88   fAxistPt.SetNameTitle("tPt", "trigger Pt");
  89
  90   Double_t cptbins[18] = {1.0, 1.5, 2.0, 2.5, 3.0, 3.5, 4.0, 5.0, 6.0, 8.0, 10.0, 12.5, 15, 20, 25, 30, 50, 100};
  91   fAxiscPt.Set(17, cptbins);
  92   fAxiscPt.SetNameTitle("cPt", "track Pt");
  93
  94   fAxisIso.Set(3, -0.5, 2.5);
  95   fAxisIso.SetNameTitle("iso", "isolation");
  96
  97   fAxesList.AddAt(&fAxisdEta, 0);
  98   fAxesList.AddAt(&fAxisdPhi, 1);
  99   fAxesList.AddAt(&fAxistPt, 2);
 100   fAxesList.AddAt(&fAxiscPt, 3);
 101   fAxesList.AddAt(&fAxisIso, 4);
 102
 103   fAxisMEEta.Set(320, -0.8, 0.8);
 104   fAxisMEEta.SetNameTitle("eta", "eta");
 105
 106   fAxisMEPhi.Set(256, 0, TMath::TwoPi());
 107   fAxisMEPhi.SetNameTitle("phi", "phi");
 108
 109   fTrackAxisList.AddAt(&fAxisMEEta, 0);
 110   fTrackAxisList.AddAt(&fAxisMEPhi, 1);
 111   fTrackAxisList.AddAt(&fAxistPt, 2);
 112   fTrackAxisList.AddAt(&fAxiscPt, 3);
 113   fTrackAxisList.AddAt(&fAxisIso, 4);
 114
 115   fTrigAxisList.AddAt(&fAxisMEEta, 0);
 116   fTrigAxisList.AddAt(&fAxisMEPhi, 1);
 117   fTrigAxisList.AddAt(&fAxistPt, 2);
 118   fTrigAxisList.AddAt(&fAxisIso, 3);
 119
 120 }
 121
 122
 123 //________________________________________________________________________
 124 void AliAnaConvCorrBase::CreateBaseHistograms() {
 125   //Create histograms add, to outputlis
 126
 127   //cout << "Creating histograms for "<< GetName() << endl;
 128
 129   fHistograms = new TList();
 130   fHistograms->SetOwner(kTRUE);
 131   fHistograms->SetName(fName);
 132
 133
 134   fCorrSparse = CreateSparse(GetName(), GetTitle(), &fAxesList);
 135   fHistograms->Add(fCorrSparse);
 136
 137   fTrackSparse = CreateSparse(Form("%s_%s", GetName(), "METrack"), Form("%s %s", GetTitle(), "ME Tracks"), &fTrackAxisList);
 138   fHistograms->Add(fTrackSparse);
 139
 140   fTrigSparse = CreateSparse(Form("%s_%s", GetName(), "METrig"), Form("%s %s", GetTitle(), "ME Triggers"), &fTrigAxisList);
 141   fHistograms->Add(fTrigSparse);
 142
 143 }
 144
 145 ///________________________________________________________________________
 146 THnSparseF * AliAnaConvCorrBase::CreateSparse(TString nameString, TString titleString, TList * axesList) {
 147   //Create sparse
 148   const Int_t dim = axesList->GetSize();
 149
 150   //cout << nameString << " " << titleString << " " <<   "    dimesion: " << dim << endl;
 151
 152   TAxis * axes[dim];
 153   Int_t   bins[dim];
 154   Double_t min[dim];
 155   Double_t max[dim];
 156
 157   for(Int_t i = 0; i<dim; i++) {
 158         TAxis * axis = dynamic_cast<TAxis*>(axesList->At(i));
 159         if(axis) axes[i] = axis;
 160         else {
 161           cout << "AliAnalysisTaskdPhi::CreateSparse: Error error, all the axes are not present in axis list" << endl;
 162           return NULL;
 163         }
 164   }
 165
 166   for(Int_t i = 0; i<dim; i++) {
 167         //cout << axes[i]->GetTitle() << endl;
 168         bins[i] = axes[i]->GetNbins();
 169         min[i] = axes[i]->GetBinLowEdge(1);
 170         max[i] = axes[i]->GetBinUpEdge(axes[i]->GetNbins());
 171   }
 172
 173   THnSparseF * sparse = new THnSparseF(Form("%s", nameString.Data()),
 174                                                                            Form("%s", titleString.Data()),
 175                                                                            dim, bins, min, max);
 176
 177   for(Int_t i = 0; i<dim; i++) {
 178         sparse->GetAxis(i)->SetNameTitle(axes[i]->GetName(), axes[i]->GetTitle() );
 179         if(axes[i]->GetXbins()->GetSize() > 0) {
 180           sparse->SetBinEdges(i, axes[i]->GetXbins()->GetArray() );
 181         }
 182   }
 183   return sparse;
 184 }
 185
 186
 187
 188 //________________________________________________________________
 189 void AliAnaConvCorrBase::CorrelateWithTracks(AliAODConversionParticle * particle, TObjArray * tracks, Int_t const tIDs[4], Int_t isolated = 0) {
 190   //Correlate particle with tracks
 191
 192
 193
 194   Int_t nDim = fAxesList.GetSize();
 195   Double_t dphivalues[nDim];
 196   Double_t trackValues[nDim];
 197   Double_t trigValues[nDim - 1];
 198
 199   trigValues[0] = particle->Eta();
 200   trigValues[1] = particle->Phi();
 201   trigValues[2] = particle->Pt();
 202   trigValues[3] = isolated;
 203
 204   if(nDim > 4) {
 205         dphivalues[5] = particle->M();
 206         trackValues[5] = particle->M();
 207         trigValues[4] = particle->M();
 208   }
 209
 210   fTrigSparse->Fill(trigValues);
 211
 212
 213   for(int ij = 0; ij < tracks->GetEntriesFast(); ij++) {
 214         AliVTrack * track = static_cast<AliVTrack*>(tracks->UncheckedAt(ij));
 215         Int_t tid = track->GetID();
 216
 217         if((tid > 0) && (tid == tIDs[0] || tid == tIDs[1] || tid == tIDs[2] || tid == tIDs[3]) ) {
 218           continue;
 219         }
 220
 221         dphivalues[0] = particle->Eta() - track->Eta();
 222         dphivalues[1] = GetDPhi(particle->Phi() - track->Phi());
 223         dphivalues[2] = particle->Pt();
 224         dphivalues[3] = track->Pt();
 225         dphivalues[4] = isolated;
 226
 227         trackValues[0] = track->Eta();
 228         trackValues[1] = track->Phi();
 229         trackValues[2] = particle->Pt();
 230         trackValues[3] = track->Pt();
 231         trackValues[4] = isolated;
 232
 233
 234
 235         fCorrSparse->Fill(dphivalues);
 236         fTrackSparse->Fill(trackValues);
 237   }
 238 }
 239