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 "TH3.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   fAxisCent(),
  51   fAxisZ(),
  52   fAxisMEEta(),
  53   fAxisMEPhi(),
  54   fCorrSparse(NULL),
  55   fTrigSparse(NULL),
  56   fTrackSparse(NULL)
  57 {
  58   //Constructor
  59   fAxesList.SetOwner(kFALSE);
  60   fTrackAxisList.SetOwner(kFALSE);
  61   fTrigAxisList.SetOwner(kFALSE);
  62
  63   SetUpDefaultBins();
  64
  65 }
  66
  67
  68 //________________________________________________________________________________
  69 AliAnaConvCorrBase::~AliAnaConvCorrBase() {
  70   ///destructor
  71 }
  72
  73 //________________________________________________________________________________
  74 void AliAnaConvCorrBase::CreateHistograms() {
  75   CreateBaseHistograms();
  76 }
  77
  78 ///________________________________________________________________________________
  79 void AliAnaConvCorrBase::SetUpDefaultBins() {
  80   //Set up default bins
  81   fAxisdEta.Set(32, -1.6, 1.6);
  82   fAxisdEta.SetNameTitle("dEta", "delta eta");
  83
  84   fAxisdPhi.Set(32, -TMath::PiOver2(), 3*TMath::PiOver2());
  85   fAxisdPhi.SetNameTitle("dPhi", "delta Phi");
  86
  87   Double_t tptbins[15] = {2.0, 3.0, 3.5, 4.0, 5.0, 6.0, 8.0, 10.0, 12.5, 15, 20, 25, 30, 50, 100};
  88   fAxistPt.Set(13, tptbins);
  89   fAxistPt.SetNameTitle("tPt", "trigger Pt");
  90
  91   Double_t cptbins[19] = {0.7, 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};
  92   fAxiscPt.Set(18, cptbins);
  93   fAxiscPt.SetNameTitle("cPt", "track Pt");
  94
  95   fAxisCent.SetNameTitle("centrality", "centrality");
  96   fAxisZ.SetNameTitle("vtxz", "vtxz");
  97
  98   fAxisIso.Set(1, -0.5, 2.5);
  99   fAxisIso.SetNameTitle("iso", "isolation");
 100
 101   fAxesList.AddAt(&fAxisdEta, 0);
 102   fAxesList.AddAt(&fAxisdPhi, 1);
 103   fAxesList.AddAt(&fAxistPt, 2);
 104   fAxesList.AddAt(&fAxiscPt, 3);
 105   fAxesList.AddAt(&fAxisCent, 4);
 106   fAxesList.AddAt(&fAxisZ, 5);
 107
 108   fAxisMEEta.Set(320, -0.8, 0.8);
 109   fAxisMEEta.SetNameTitle("eta", "eta");
 110
 111   fTrackAxisList.AddAt(&fAxisMEEta, 0);
 112   fTrackAxisList.AddAt(&fAxistPt, 1);
 113   fTrackAxisList.AddAt(&fAxiscPt, 2);
 114   fTrackAxisList.AddAt(&fAxisCent, 3);
 115   fTrackAxisList.AddAt(&fAxisZ, 4);
 116
 117   fTrigAxisList.AddAt(&fAxisMEEta, 0);
 118   fTrigAxisList.AddAt(&fAxistPt, 1);
 119   fTrigAxisList.AddAt(&fAxisCent, 2);
 120   fTrigAxisList.AddAt(&fAxisZ, 3);
 121
 122
 123 }
 124
 125
 126 //________________________________________________________________________
 127 void AliAnaConvCorrBase::CreateBaseHistograms() {
 128   //Create histograms add, to outputlis
 129
 130   //cout << "Creating histograms for "<< GetName() << endl;
 131
 132   fHistograms = new TList();
 133   fHistograms->SetOwner(kTRUE);
 134   fHistograms->SetName(fName);
 135
 136
 137
 138   fCorrSparse = CreateSparse(GetName(), GetTitle(), &fAxesList);
 139   fHistograms->Add(fCorrSparse);
 140
 141   fTrackSparse = CreateSparse(Form("%s_%s", GetName(), "METrack"), Form("%s %s", GetTitle(), "ME Tracks"), &fTrackAxisList);
 142   fHistograms->Add(fTrackSparse);
 143
 144   fTrigSparse = CreateSparse(Form("%s_%s", GetName(), "METrig"), Form("%s %s", GetTitle(), "ME Triggers"), &fTrigAxisList);
 145   fHistograms->Add(fTrigSparse);
 146
 147 }
 148
 149 ///________________________________________________________________________
 150 THnSparseF * AliAnaConvCorrBase::CreateSparse(TString nameString, TString titleString, TList * axesList) {
 151   //Create sparse
 152   const Int_t dim = axesList->GetSize();
 153
 154   //cout << nameString << " " << titleString << " " <<   "    dimesion: " << dim << endl;
 155
 156   TAxis * axes[dim];
 157   Int_t   bins[dim];
 158   Double_t min[dim];
 159   Double_t max[dim];
 160
 161   for(Int_t i = 0; i<dim; i++) {
 162         TAxis * axis = dynamic_cast<TAxis*>(axesList->At(i));
 163         if(axis) axes[i] = axis;
 164         else {
 165           cout << "AliAnalysisTaskdPhi::CreateSparse: Error error, all the axes are not present in axis list" << endl;
 166           return NULL;
 167         }
 168   }
 169
 170   for(Int_t i = 0; i<dim; i++) {
 171         //cout << axes[i]->GetTitle() << endl;
 172         bins[i] = axes[i]->GetNbins();
 173         min[i] = axes[i]->GetBinLowEdge(1);
 174         max[i] = axes[i]->GetBinUpEdge(axes[i]->GetNbins());
 175   }
 176
 177   THnSparseF * sparse = new THnSparseF(Form("%s", nameString.Data()),
 178                                                                            Form("%s", titleString.Data()),
 179                                                                            dim, bins, min, max);
 180
 181   for(Int_t i = 0; i<dim; i++) {
 182         sparse->GetAxis(i)->SetNameTitle(axes[i]->GetName(), axes[i]->GetTitle() );
 183         if(axes[i]->GetXbins()->GetSize() > 0) {
 184           sparse->SetBinEdges(i, axes[i]->GetXbins()->GetArray() );
 185         }
 186   }
 187   return sparse;
 188 }
 189
 190
 191
 192 //_______________________________________________________________________________
 193 void AliAnaConvCorrBase::FillCounters(TObjArray * particles, TObjArray * tracks, Float_t cent, Float_t vtxz) {
 194   //Fill ME Counters
 195   const Int_t nbins = fAxistPt.GetNbins();
 196   Bool_t tmap[nbins];
 197   for(Int_t ptbin = 0; ptbin < nbins; ptbin++){
 198     tmap[ptbin] = kFALSE;
 199   }
 200
 201
 202   Double_t trackValues[fTrackAxisList.GetSize()];
 203   trackValues[3] = cent;
 204   trackValues[4] = vtxz;
 205
 206   for(Int_t ip = 0; ip < particles->GetEntriesFast(); ip++){
 207     AliAODConversionParticle * particle = static_cast<AliAODConversionParticle*>(particles->At(ip));
 208
 209     Int_t tbin = fAxistPt.FindFixBin(particle->Pt());
 210     if (tbin > 0 && tbin < nbins + 1) {
 211       if(tmap[tbin - 1] == kTRUE) {
 212         continue;
 213       } else {
 214         tmap[tbin -1 ] = kTRUE;
 215
 216         if( fTrackAxisList.GetSize() > 5){
 217           trackValues[5] = particle->M();
 218         }
 219
 220         for(int ij = 0; ij < tracks->GetEntriesFast(); ij++) {
 221           AliVTrack * track = static_cast<AliVTrack*>(tracks->UncheckedAt(ij));
 222           trackValues[0] = track->Eta();
 223           trackValues[1] = particle->Pt();
 224           trackValues[2] = track->Pt();
 225           fTrackSparse->Fill(trackValues);
 226         }
 227       }
 228     }
 229   }
 230 }
 231
 232 //________________________________________________________________
 233 void AliAnaConvCorrBase::CorrelateWithTracks(AliAODConversionParticle * particle, TObjArray * tracks, Int_t const tIDs[4], Float_t cent, Float_t vtxz) {
 234   //Correlate particle with tracks
 235
 236
 237   const Int_t nDim = fAxesList.GetSize();
 238   Double_t dphivalues[nDim];
 239   dphivalues[4] = cent;
 240   dphivalues[5] = vtxz;
 241
 242
 243   Double_t trigValues[fTrigAxisList.GetSize()];
 244   trigValues[0] = particle->Eta();
 245   trigValues[1] = particle->Pt();
 246   trigValues[2] = cent;
 247   trigValues[3] = vtxz;
 248
 249   if(nDim > 6) {
 250     dphivalues[6] = particle->M();
 251     trigValues[4] = particle->M();
 252   }
 253
 254   fTrigSparse->Fill(trigValues);
 255
 256   for(int ij = 0; ij < tracks->GetEntriesFast(); ij++) {
 257     AliVTrack * track = static_cast<AliVTrack*>(tracks->UncheckedAt(ij));
 258
 259     Int_t tid = track->GetID();
 260
 261     if((tid > 0) && (tid == tIDs[0] || tid == tIDs[1] || tid == tIDs[2] || tid == tIDs[3]) ) {
 262       continue;
 263     }
 264
 265     dphivalues[0] = particle->Eta() - track->Eta();
 266     dphivalues[1] = GetDPhi(particle->Phi() - track->Phi());
 267     dphivalues[2] = particle->Pt();
 268     dphivalues[3] = track->Pt();
 269     fCorrSparse->Fill(dphivalues);
 270   }
 271 }
 272