PWG0/dNdPt/AlidNdPtCutAnalysis.cxx

   1 /**************************************************************************\r
   2  * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *\r
   3  *                                                                        *\r
   4  * Author: The ALICE Off-line Project.                                    *\r
   5  * Contributors are mentioned in the code where appropriate.              *\r
   6  *                                                                        *\r
   7  * Permission to use, copy, modify and distribute this software and its   *\r
   8  * documentation strictly for non-commercial purposes is hereby granted   *\r
   9  * without fee, provided that the above copyright notice appears in all   *\r
  10  * copies and that both the copyright notice and this permission notice   *\r
  11  * appear in the supporting documentation. The authors make no claims     *\r
  12  * about the suitability of this software for any purpose. It is          *\r
  13  * provided "as is" without express or implied warranty.                  *\r
  14  **************************************************************************/\r
  15 //------------------------------------------------------------------------------\r
  16 // AlidNdPtCutAnalysis class. \r
  17 //\r
  18 // a. functionality:\r
  19 // - fills generic cut histograms\r
  20 // - generates cuts (selection criteria)\r
  21 //\r
  22 // b. data members:\r
  23 // - generic cut histograms\r
  24 // - control histograms\r
  25 //\r
  26 // Author: J.Otwinowski 04/11/2008 \r
  27 //------------------------------------------------------------------------------\r
  28 #include "TH1.h"\r
  29 #include "TH2.h"\r
  30 \r
  31 #include "AliHeader.h"  \r
  32 #include "AliGenEventHeader.h"  \r
  33 #include "AliInputEventHandler.h"  \r
  34 #include "AliStack.h"  \r
  35 #include "AliESDEvent.h"  \r
  36 #include "AliMCEvent.h"  \r
  37 #include "AliESDtrackCuts.h"  \r
  38 #include "AliLog.h" \r
  39 #include "AliTracker.h" \r
  40 \r
  41 #include "AlidNdPtEventCuts.h"\r
  42 #include "AlidNdPtAcceptanceCuts.h"\r
  43 #include "AlidNdPtBackgroundCuts.h"\r
  44 #include "AlidNdPtAnalysis.h"\r
  45 #include "AliPhysicsSelection.h"\r
  46 \r
  47 #include "AliPWG0Helper.h"\r
  48 #include "AlidNdPtHelper.h"\r
  49 #include "AlidNdPtCutAnalysis.h"\r
  50 \r
  51 using namespace std;\r
  52 \r
  53 ClassImp(AlidNdPtCutAnalysis)\r
  54 \r
  55 //_____________________________________________________________________________\r
  56   AlidNdPtCutAnalysis::AlidNdPtCutAnalysis(): AlidNdPt(),\r
  57   fAnalysisFolder(0),\r
  58   fEventCount(0),\r
  59   fRecEventHist(0),\r
  60   fMCEventHist(0),\r
  61   fRecMCEventHist(0),\r
  62   fRecMCTrackHist(0)\r
  63 {\r
  64   // default constructor\r
  65   Init();\r
  66 }\r
  67 \r
  68 //_____________________________________________________________________________\r
  69 AlidNdPtCutAnalysis::AlidNdPtCutAnalysis(Char_t* name, Char_t* title): AlidNdPt(name,title),\r
  70   fAnalysisFolder(0),\r
  71   fEventCount(0),\r
  72   fRecEventHist(0),\r
  73   fMCEventHist(0),\r
  74   fRecMCEventHist(0),\r
  75   fRecMCTrackHist(0)\r
  76 {\r
  77   // constructor\r
  78   Init();\r
  79 }\r
  80 \r
  81 //_____________________________________________________________________________\r
  82 AlidNdPtCutAnalysis::~AlidNdPtCutAnalysis() {\r
  83   // \r
  84   if(fEventCount) delete fEventCount; fEventCount=0;\r
  85   if(fRecEventHist) delete fRecEventHist; fRecEventHist=0;\r
  86   if(fMCEventHist) delete fMCEventHist; fMCEventHist=0;\r
  87   if(fRecMCEventHist) delete fRecMCEventHist; fRecMCEventHist=0;\r
  88   if(fRecMCTrackHist) delete fRecMCTrackHist; fRecMCTrackHist=0;\r
  89 \r
  90   if(fAnalysisFolder) delete fAnalysisFolder; fAnalysisFolder=0;\r
  91 }\r
  92 \r
  93 //_____________________________________________________________________________\r
  94 void AlidNdPtCutAnalysis::Init(){\r
  95   //\r
  96   // Init histograms\r
  97   //\r
  98   /*\r
  99   const Int_t ptNbins = 56; \r
 100   const Double_t ptMin = 0.; \r
 101   const Double_t ptMax = 16.; \r
 102   Double_t binsPt[ptNbins+1] = {0.,0.05,0.1,0.15,0.2,0.25,0.3,0.35,0.4,0.45,0.5,0.55,0.6,0.65,0.7,0.75,0.8,0.85,0.9,0.95,1.0,1.1,1.2,1.3,1.4,1.5,1.6,1.7,1.8,1.9,2.0,2.2,2.4,2.6,2.8,3.0,3.2,3.4,3.6,3.8,4.0,4.5,5.0,5.5,6.0,6.5,7.0,7.5,8.0,9.0,10.0,11.0,12.0,13.0,14.0,15.0,16.0};\r
 103   */\r
 104   // set pt bins\r
 105   const Int_t ptNbins = 50;\r
 106   const Double_t ptMin = 1.e-2, ptMax = 50.;\r
 107   Double_t *binsPt = CreateLogAxis(ptNbins,ptMin,ptMax);\r
 108 \r
 109   // \r
 110   Int_t binsEventCount[2]={2,2};\r
 111   Double_t minEventCount[2]={0,0}; \r
 112   Double_t maxEventCount[2]={2,2}; \r
 113   fEventCount = new THnSparseF("fEventCount","trig vs trig+vertex",2,binsEventCount,minEventCount,maxEventCount);\r
 114   fEventCount->GetAxis(0)->SetTitle("trig");\r
 115   fEventCount->GetAxis(1)->SetTitle("trig+vert");\r
 116   fEventCount->Sumw2();\r
 117 \r
 118   //Xv:Yv:Zv:ResZv:Mult\r
 119   Double_t kFact = 1.0;\r
 120 \r
 121   Int_t binsRecEventHist[5]={80,80,100,80,150};\r
 122   Double_t minRecEventHist[5]={-3.*kFact,-3.*kFact,-35.,0.,0.}; \r
 123   Double_t maxRecEventHist[5]={3.*kFact,3.*kFact,35.,10.,150.}; \r
 124   fRecEventHist = new THnSparseF("fRecEventHist","Xv:Yv:Zv:ResZv:Mult",5,binsRecEventHist,minRecEventHist,maxRecEventHist);\r
 125   fRecEventHist->GetAxis(0)->SetTitle("Xv (cm)");\r
 126   fRecEventHist->GetAxis(1)->SetTitle("Yv (cm)");\r
 127   fRecEventHist->GetAxis(2)->SetTitle("Zv (cm)");\r
 128   fRecEventHist->GetAxis(3)->SetTitle("ResZv (cm)");\r
 129   fRecEventHist->GetAxis(4)->SetTitle("Mult");\r
 130   fRecEventHist->Sumw2();\r
 131 \r
 132   //Xv:Yv:Zv\r
 133   Int_t binsMCEventHist[3]={80,80,100};\r
 134   Double_t minMCEventHist[3]={-0.1,-0.1,-35.}; \r
 135   Double_t maxMCEventHist[3]={0.1,0.1,35.}; \r
 136   fMCEventHist = new THnSparseF("fMCEventHist","mcXv:mcYv:mcZv",3,binsMCEventHist,minMCEventHist,maxMCEventHist);\r
 137   fMCEventHist->GetAxis(0)->SetTitle("mcXv (cm)");\r
 138   fMCEventHist->GetAxis(1)->SetTitle("mcYv (cm)");\r
 139   fMCEventHist->GetAxis(2)->SetTitle("mcZv (cm)");\r
 140   fMCEventHist->Sumw2();\r
 141 \r
 142   //Xv-mcXv:Yv-mcYv:Zv-mcZv:Mult\r
 143   Int_t binsRecMCEventHist[4]={100,100,100,150};\r
 144   Double_t minRecMCEventHist[4]={-1.0*kFact,-1.0*kFact,-1.0*kFact,0.}; \r
 145   Double_t maxRecMCEventHist[4]={1.0*kFact,1.0*kFact,1.0*kFact,150.}; \r
 146   fRecMCEventHist = new THnSparseF("fRecMCEventHist","Xv-mcXv:Yv-mcYv:Zv-mcZv:Mult",4,binsRecMCEventHist,minRecMCEventHist,maxRecMCEventHist);\r
 147   fRecMCEventHist->GetAxis(0)->SetTitle("Xv-mcXv (cm)");\r
 148   fRecMCEventHist->GetAxis(1)->SetTitle("Yv-mcYv (cm)");\r
 149   fRecMCEventHist->GetAxis(2)->SetTitle("Zv-mcZv (cm)");\r
 150   fRecMCEventHist->GetAxis(3)->SetTitle("Mult");\r
 151   fRecMCEventHist->Sumw2();\r
 152 \r
 153   //\r
 154   // THnSparse track histograms\r
 155   //\r
 156 \r
 157  //nCrossRows:chi2PerClust:nCrossRows/nFindableClust:fracSharedClust:DCAy:DCAz:eta:phi:pt:hasStrangeMother:isFromMaterial:isPrim:charge\r
 158   Int_t binsRecMCTrackHist[13]=  {160, 10, 20, 20, 50,  50,  20,  90,             ptNbins, 2,  2,  2,  3};\r
 159   Double_t minRecMCTrackHist[13]={0.,  0., 0., 0.,-0.5,-0.5,-1.0, 0.,             ptMin,   0., 0., 0.,-1.};\r
 160   Double_t maxRecMCTrackHist[13]={160.,10.,1.,  1., 0.5, 0.5, 1.0, 2.*TMath::Pi(), ptMax,   2., 2., 2., 2.};\r
 161 \r
 162   fRecMCTrackHist = new THnSparseF("fRecMCTrackHist","nCrossRows:chi2PerClust:nCrossRows/nFindableClust:fracSharedClust:DCAy:DCAz:eta:phi:pt:hasStrangeMother:isFromMaterial:isPrim:charge",13,binsRecMCTrackHist,minRecMCTrackHist,maxRecMCTrackHist);\r
 163   fRecMCTrackHist->SetBinEdges(8,binsPt);\r
 164   fRecMCTrackHist->GetAxis(0)->SetTitle("nCrossRows");\r
 165   fRecMCTrackHist->GetAxis(1)->SetTitle("chi2PerClust");\r
 166   fRecMCTrackHist->GetAxis(2)->SetTitle("nCrossRows/nFindableClust");\r
 167   fRecMCTrackHist->GetAxis(3)->SetTitle("fracSharedClust");\r
 168   fRecMCTrackHist->GetAxis(4)->SetTitle("DCAy (cm)");\r
 169   fRecMCTrackHist->GetAxis(5)->SetTitle("DCAz (cm)");\r
 170   fRecMCTrackHist->GetAxis(6)->SetTitle("#eta");\r
 171   fRecMCTrackHist->GetAxis(7)->SetTitle("#phi (rad)");\r
 172   fRecMCTrackHist->GetAxis(8)->SetTitle("p_{T} (GeV/c)");\r
 173   fRecMCTrackHist->GetAxis(9)->SetTitle("hasStrangeMother");\r
 174   fRecMCTrackHist->GetAxis(10)->SetTitle("isFromMaterial");\r
 175   fRecMCTrackHist->GetAxis(11)->SetTitle("isPrim");\r
 176   fRecMCTrackHist->GetAxis(12)->SetTitle("charge");\r
 177   fRecMCTrackHist->Sumw2();\r
 178 \r
 179   //nClust:chi2PerClust:nClust/nFindableClust:DCAy:DCAz:eta:phi:pt:kinkIdx:isPrim:polarity\r
 180   /*\r
 181   Int_t binsRecMCTrackHist[11]={160,80,80,100,100,90,90,ptNbins, 3, 2, 2};\r
 182   Double_t minRecMCTrackHist[11]={0., 0., 0., -1.,-1.,-1.5, 0., ptMin, -1., 0., 0.};\r
 183   Double_t maxRecMCTrackHist[11]={160.,10.,1.2, 1.,1.,1.5, 2.*TMath::Pi(), ptMax, 2., 2., 2.};\r
 184 \r
 185   fRecMCTrackHist = new THnSparseF("fRecMCTrackHist","nClust:chi2PerClust:nClust/nFindableClust:DCAy:DCAz:eta:phi:pt:kinkIdx:isPrim:polarity",11,binsRecMCTrackHist,minRecMCTrackHist,maxRecMCTrackHist);\r
 186   fRecMCTrackHist->SetBinEdges(7,binsPt);\r
 187 \r
 188   fRecMCTrackHist->GetAxis(0)->SetTitle("nClust");\r
 189   fRecMCTrackHist->GetAxis(1)->SetTitle("chi2PerClust");\r
 190   fRecMCTrackHist->GetAxis(2)->SetTitle("nClust/nFindableClust");\r
 191   fRecMCTrackHist->GetAxis(3)->SetTitle("DCAy (cm)");\r
 192   fRecMCTrackHist->GetAxis(4)->SetTitle("DCAz (cm)");\r
 193   fRecMCTrackHist->GetAxis(5)->SetTitle("#eta");\r
 194   fRecMCTrackHist->GetAxis(6)->SetTitle("#phi (rad)");\r
 195   fRecMCTrackHist->GetAxis(7)->SetTitle("p_{T} (GeV/c)");\r
 196   fRecMCTrackHist->GetAxis(8)->SetTitle("kinkIdx"); // 0 - no kink, -1 - kink mother, 1 - kink daugther \r
 197   fRecMCTrackHist->GetAxis(9)->SetTitle("isPrim");\r
 198   fRecMCTrackHist->GetAxis(10)->SetTitle("polarity");\r
 199   fRecMCTrackHist->Sumw2();\r
 200   */\r
 201 \r
 202   // init output folder\r
 203   fAnalysisFolder = CreateFolder("folderdNdPt","Analysis dNdPt Folder");\r
 204 \r
 205 }\r
 206 \r
 207 //_____________________________________________________________________________\r
 208 void AlidNdPtCutAnalysis::Process(AliESDEvent *const esdEvent, AliMCEvent * const mcEvent)\r
 209 {\r
 210   //\r
 211   // Process real and/or simulated events\r
 212   //\r
 213   if(!esdEvent) {\r
 214     AliDebug(AliLog::kError, "esdEvent not available");\r
 215     return;\r
 216   }\r
 217 \r
 218   // get selection cuts\r
 219   AlidNdPtEventCuts *evtCuts = GetEventCuts(); \r
 220   AlidNdPtAcceptanceCuts *accCuts = GetAcceptanceCuts(); \r
 221   AliESDtrackCuts *esdTrackCuts = GetTrackCuts(); \r
 222 \r
 223   if(!evtCuts || !accCuts  || !esdTrackCuts) {\r
 224     AliDebug(AliLog::kError, "cuts not available");\r
 225     return;\r
 226   }\r
 227 \r
 228   // trigger selection\r
 229 \r
 230   Bool_t isEventTriggered = kTRUE;\r
 231   AliPhysicsSelection *physicsSelection = NULL;\r
 232   AliTriggerAnalysis* triggerAnalysis = NULL;\r
 233 \r
 234   // \r
 235   AliInputEventHandler* inputHandler = (AliInputEventHandler*) AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler();\r
 236   if (!inputHandler)\r
 237   {\r
 238     Printf("ERROR: Could not receive input handler");\r
 239     return;\r
 240   }\r
 241 \r
 242   if(evtCuts->IsTriggerRequired())  \r
 243   {\r
 244     // always MB\r
 245     isEventTriggered = inputHandler->IsEventSelected() & AliVEvent::kMB;\r
 246 \r
 247     physicsSelection = static_cast<AliPhysicsSelection*> (inputHandler->GetEventSelection());\r
 248     if(!physicsSelection) return;\r
 249     //SetPhysicsTriggerSelection(physicsSelection);\r
 250 \r
 251     if (isEventTriggered && (GetTrigger() == AliTriggerAnalysis::kV0AND)) {\r
 252       // set trigger (V0AND)\r
 253       triggerAnalysis = physicsSelection->GetTriggerAnalysis();\r
 254       if(!triggerAnalysis) return;\r
 255       isEventTriggered = triggerAnalysis->IsOfflineTriggerFired(esdEvent, GetTrigger());\r
 256     }\r
 257   }\r
 258 \r
 259   // use MC information\r
 260   AliHeader* header = 0;\r
 261   AliGenEventHeader* genHeader = 0;\r
 262   AliStack* stack = 0;\r
 263   TArrayF vtxMC(3);\r
 264   AliPWG0Helper::MCProcessType evtType = AliPWG0Helper::kInvalidProcess;\r
 265 \r
 266   if(IsUseMCInfo())\r
 267   {\r
 268     if(!mcEvent) {\r
 269       AliDebug(AliLog::kError, "mcEvent not available");\r
 270       return;\r
 271     }\r
 272 \r
 273     // get MC event header\r
 274     header = mcEvent->Header();\r
 275     if (!header) {\r
 276       AliDebug(AliLog::kError, "Header not available");\r
 277       return;\r
 278     }\r
 279 \r
 280     // MC particle stack\r
 281     stack = mcEvent->Stack();\r
 282     if (!stack) {\r
 283       AliDebug(AliLog::kError, "Stack not available");\r
 284       return;\r
 285     }\r
 286 \r
 287     // get event type (ND=0x1, DD=0x2, SD=0x4)\r
 288     evtType = AliPWG0Helper::GetEventProcessType(header);\r
 289     AliDebug(AliLog::kDebug+1, Form("Found process type %d", evtType));\r
 290 \r
 291     // get MC vertex\r
 292     genHeader = header->GenEventHeader();\r
 293     if (!genHeader) {\r
 294       AliDebug(AliLog::kError, "Could not retrieve genHeader from Header");\r
 295       return;\r
 296     }\r
 297     genHeader->PrimaryVertex(vtxMC);\r
 298 \r
 299     // Fill MC event histogram\r
 300     Double_t vMCEventHist[3]={vtxMC[0],vtxMC[1],vtxMC[2]};\r
 301     fMCEventHist->Fill(vMCEventHist);\r
 302 \r
 303   } // end bUseMC\r
 304 \r
 305   // get reconstructed vertex  \r
 306   Bool_t bRedoTPCVertex = evtCuts->IsRedoTPCVertex();\r
 307   Bool_t bUseConstraints = evtCuts->IsUseBeamSpotConstraint();\r
 308   const AliESDVertex* vtxESD = AlidNdPtHelper::GetVertex(esdEvent,evtCuts,accCuts,esdTrackCuts,GetAnalysisMode(),kFALSE,bRedoTPCVertex,bUseConstraints); \r
 309   Bool_t isRecVertex = AlidNdPtHelper::TestRecVertex(vtxESD, esdEvent->GetPrimaryVertexSPD(), GetAnalysisMode(), kFALSE);\r
 310   Bool_t isEventOK = evtCuts->AcceptEvent(esdEvent,mcEvent,vtxESD) && isRecVertex;\r
 311 \r
 312   TObjArray *allChargedTracks=0;\r
 313   Int_t multAll=0;\r
 314   \r
 315   //\r
 316   // event counter\r
 317   // \r
 318   //printf("isEventOK %d, isEventTriggered %d \n",isEventOK,isEventTriggered);\r
 319 \r
 320   Bool_t isTrigAndVertex = isEventTriggered && isEventOK;\r
 321   Double_t vEventCount[2] = { isEventTriggered, isTrigAndVertex};\r
 322   fEventCount->Fill(vEventCount);\r
 323 \r
 324   //\r
 325   // cosmic background and splitted tracks\r
 326   //\r
 327   if(GetParticleMode() == AlidNdPtHelper::kBackgroundTrack) \r
 328   {\r
 329     AlidNdPtBackgroundCuts *backCuts = GetBackgroundCuts(); \r
 330     if(!backCuts) return;\r
 331 \r
 332     for (Int_t iTrack = 0; iTrack < esdEvent->GetNumberOfTracks(); iTrack++) \r
 333     {\r
 334       AliESDtrack *track1 = esdEvent->GetTrack(iTrack);\r
 335       if(!track1) continue; \r
 336       if(track1->Charge()==0) continue; \r
 337 \r
 338       for (Int_t jTrack = iTrack+1; jTrack < esdEvent->GetNumberOfTracks(); jTrack++) \r
 339       {\r
 340         AliESDtrack *track2 = esdEvent->GetTrack(jTrack);\r
 341         if(!track2) continue; \r
 342         if(track2->Charge()==0) continue; \r
 343 \r
 344         //printf("track2->Charge() %d\n",track2->Charge());\r
 345 \r
 346         backCuts->IsBackgroundTrack(track1, track2);\r
 347       }\r
 348     }\r
 349   }\r
 350 \r
 351   // check event cuts\r
 352   if(isEventOK && isEventTriggered)\r
 353   {\r
 354     // get all charged tracks\r
 355     allChargedTracks = AlidNdPtHelper::GetAllChargedTracks(esdEvent,GetAnalysisMode());\r
 356     if(!allChargedTracks) return;\r
 357 \r
 358     Int_t entries = allChargedTracks->GetEntries();\r
 359     for(Int_t i=0; i<entries;++i) \r
 360     {\r
 361       AliESDtrack *track = (AliESDtrack*)allChargedTracks->At(i);\r
 362       if(!track) continue;\r
 363 \r
 364       if(!esdTrackCuts->AcceptTrack(track)) continue;\r
 365 \r
 366       //\r
 367       Bool_t isOK = kFALSE;\r
 368       Double_t x[3]; track->GetXYZ(x);\r
 369       Double_t b[3]; AliTracker::GetBxByBz(x,b);\r
 370 \r
 371       //\r
 372       // if TPC-ITS hybrid tracking (kTPCITSHybrid)\r
 373       // replace track parameters with TPC-ony track parameters\r
 374       //\r
 375       if( GetAnalysisMode() == AlidNdPtHelper::kTPCITSHybrid || GetAnalysisMode() == AlidNdPtHelper::kTPCITSHybridTrackSPDvtx || GetAnalysisMode() == AlidNdPtHelper::kTPCITSHybridTrackSPDvtxDCArPt) \r
 376       {\r
 377         // Relate TPC-only tracks to SPD vertex\r
 378         isOK = track->RelateToVertexTPCBxByBz(vtxESD, b, kVeryBig);\r
 379         if(!isOK) continue;\r
 380 \r
 381         // replace esd track parameters with TPCinner\r
 382         AliExternalTrackParam  *tpcTrack  = new AliExternalTrackParam(*(track->GetTPCInnerParam()));\r
 383         if (!tpcTrack) return;\r
 384         track->Set(tpcTrack->GetX(),tpcTrack->GetAlpha(),tpcTrack->GetParameter(),tpcTrack->GetCovariance());\r
 385 \r
 386         if(tpcTrack) delete tpcTrack; \r
 387       } \r
 388 \r
 389       //\r
 390       if (GetAnalysisMode()==AlidNdPtHelper::kTPCSPDvtxUpdate || GetAnalysisMode() == AlidNdPtHelper::kTPCTrackSPDvtxUpdate) \r
 391       {\r
 392         //\r
 393         // update track parameters\r
 394         //\r
 395         AliExternalTrackParam cParam;\r
 396         isOK = track->RelateToVertexTPCBxByBz(vtxESD, b, kVeryBig, &cParam);\r
 397         if(!isOK) continue;\r
 398 \r
 399         track->Set(cParam.GetX(),cParam.GetAlpha(),cParam.GetParameter(),cParam.GetCovariance());\r
 400       }\r
 401 \r
 402       FillHistograms(track, stack);\r
 403       multAll++;\r
 404     }\r
 405 \r
 406     Double_t vRecEventHist[5] = {vtxESD->GetXv(),vtxESD->GetYv(),vtxESD->GetZv(),vtxESD->GetZRes(),multAll};\r
 407     fRecEventHist->Fill(vRecEventHist);\r
 408 \r
 409     if(IsUseMCInfo()) {\r
 410       Double_t vRecMCEventHist[5] = {vtxESD->GetXv()-vtxMC[0],vtxESD->GetYv()-vtxMC[1],vtxESD->GetZv()-vtxMC[2],multAll};\r
 411       fRecMCEventHist->Fill(vRecMCEventHist);\r
 412     }\r
 413   }\r
 414 \r
 415   if(allChargedTracks) delete allChargedTracks; allChargedTracks = 0;\r
 416 \r
 417 }\r
 418 \r
 419 //_____________________________________________________________________________\r
 420 void AlidNdPtCutAnalysis::FillHistograms(AliESDtrack *const esdTrack, AliStack *const stack) const\r
 421 {\r
 422   //\r
 423   // Fill ESD track and MC histograms \r
 424   //\r
 425   if(!esdTrack) return;\r
 426   if(esdTrack->Charge() == 0.) return;\r
 427 \r
 428   Float_t pt = esdTrack->Pt();\r
 429   Float_t eta = esdTrack->Eta();\r
 430   Float_t phi = esdTrack->Phi();\r
 431 \r
 432 \r
 433   Int_t nClust = 0;\r
 434   if(GetAnalysisMode() == AlidNdPtHelper::kTPC) { \r
 435     nClust = esdTrack->GetTPCNclsIter1();\r
 436   } else {\r
 437     nClust = esdTrack->GetTPCclusters(0);\r
 438   }\r
 439 \r
 440   Float_t chi2PerCluster = 0.;\r
 441   if(GetAnalysisMode() == AlidNdPtHelper::kTPC) { \r
 442     if(nClust>0.) chi2PerCluster = esdTrack->GetTPCchi2Iter1()/Float_t(nClust);\r
 443   } else {\r
 444     chi2PerCluster = esdTrack->GetTPCchi2()/Float_t(nClust);\r
 445   }\r
 446 \r
 447   Int_t nFindableClust = esdTrack->GetTPCNclsF();\r
 448 \r
 449   \r
 450   Float_t clustPerFindClust = 0.;\r
 451   if(nFindableClust>0.) clustPerFindClust = Float_t(nClust)/nFindableClust;\r
 452 \r
 453   Float_t b[2], bCov[3];\r
 454   esdTrack->GetImpactParameters(b,bCov);\r
 455 \r
 456   // \r
 457   Float_t nCrossedRowsTPC = esdTrack->GetTPCClusterInfo(2,1);\r
 458 \r
 459   Float_t  ratioCrossedRowsOverFindableClustersTPC = 1.0;\r
 460   if (esdTrack->GetTPCNclsF()>0) {\r
 461      ratioCrossedRowsOverFindableClustersTPC = esdTrack->GetTPCClusterInfo(2,1)/esdTrack->GetTPCNclsF();\r
 462   }\r
 463 \r
 464   //\r
 465   Int_t nClustersTPCShared = esdTrack->GetTPCnclsS();\r
 466   Float_t fracClustersTPCShared = -1.;\r
 467   fracClustersTPCShared = Float_t(nClustersTPCShared)/Float_t(nClust);\r
 468 \r
 469 \r
 470   // kink idx\r
 471   Int_t kinkIdx = 0;\r
 472   //if(esdTrack->GetKinkIndex(0) > 0.)   isKink  = kTRUE;\r
 473   if(esdTrack->GetKinkIndex(0) > 0)      kinkIdx = 1;   // kink daughter\r
 474   else if(esdTrack->GetKinkIndex(0) < 0) kinkIdx = -1;  // kink mother\r
 475   else kinkIdx = 0; // not kink\r
 476 \r
 477   //printf("esdTrack->GetKinkIndex(0) %d \n", esdTrack->GetKinkIndex(0));\r
 478   //printf("esdTrack->GetKinkIndex(1) %d \n", esdTrack->GetKinkIndex(1));\r
 479   //printf("esdTrack->GetKinkIndex(2) %d \n", esdTrack->GetKinkIndex(2));\r
 480   //printf("kinkIdx %d \n", kinkIdx);\r
 481 \r
 482   //\r
 483   // Fill rec vs MC information\r
 484   //\r
 485   Bool_t isPrim = kTRUE;\r
 486   Bool_t hasStrangeMother = kFALSE;\r
 487   Bool_t isFromMaterial = kFALSE;\r
 488 \r
 489   if(IsUseMCInfo()) {\r
 490     if(!stack) return;\r
 491     Int_t label = TMath::Abs(esdTrack->GetLabel()); \r
 492     TParticle* particle = stack->Particle(label);\r
 493     if(!particle) return;\r
 494     if(particle->GetPDG() && particle->GetPDG()->Charge()==0.) return;\r
 495     isPrim = stack->IsPhysicalPrimary(label);\r
 496 \r
 497     // check whether has stange mother\r
 498     //\r
 499     Int_t motherPdg = -1; \r
 500     TParticle* mother = 0; \r
 501        \r
 502     Int_t motherLabel = particle->GetMother(0);  \r
 503     if(motherLabel>0) mother = stack->Particle(motherLabel); \r
 504     if(mother) motherPdg = TMath::Abs(mother->GetPdgCode()); // take abs for visualisation only \r
 505     Int_t mech = particle->GetUniqueID(); // production mechanism \r
 506 \r
 507     if( (motherPdg == 3122) || (motherPdg == -3122) || (motherPdg == 310)) // lambda, antilambda, k0s\r
 508     {\r
 509       if( (mech == 4) || (mech == 5) ) hasStrangeMother = kTRUE;\r
 510     } \r
 511     else {\r
 512       //if(isPrim==0 && mech == 13)   \r
 513       //printf("mech %d \n", mech);\r
 514       if(!isPrim) isFromMaterial = kTRUE; \r
 515     }\r
 516 \r
 517     //if(isPrim && pt > 1.5 && kinkIdx == -1) printf("nClust  %d \n", nClust);\r
 518   }\r
 519   \r
 520   // fill histo\r
 521   Int_t charge = esdTrack->Charge(); \r
 522 \r
 523   //Double_t vRecMCTrackHist[11] = { nClust,chi2PerCluster,clustPerFindClust,b[0],b[1],eta,phi,pt,kinkIdx,isPrim, polarity }; \r
 524   //fRecMCTrackHist->Fill(vRecMCTrackHist);\r
 525 \r
 526   Double_t vRecMCTrackHist[13] = { nCrossedRowsTPC, chi2PerCluster, ratioCrossedRowsOverFindableClustersTPC, fracClustersTPCShared , b[0], b[1], eta, phi, pt, hasStrangeMother, isFromMaterial, isPrim, charge }; \r
 527   fRecMCTrackHist->Fill(vRecMCTrackHist);\r
 528 \r
 529 }\r
 530 \r
 531 //_____________________________________________________________________________\r
 532 Long64_t AlidNdPtCutAnalysis::Merge(TCollection* const list) \r
 533 {\r
 534   // Merge list of objects (needed by PROOF)\r
 535 \r
 536   if (!list)\r
 537   return 0;\r
 538 \r
 539   if (list->IsEmpty())\r
 540   return 1;\r
 541 \r
 542   TIterator* iter = list->MakeIterator();\r
 543   TObject* obj = 0;\r
 544 \r
 545   //TList *collPhysSelection = new TList;\r
 546 \r
 547   // collection of generated histograms\r
 548   Int_t count=0;\r
 549   while((obj = iter->Next()) != 0) {\r
 550     AlidNdPtCutAnalysis* entry = dynamic_cast<AlidNdPtCutAnalysis*>(obj);\r
 551     if (entry == 0) continue; \r
 552   \r
 553     // event histo\r
 554     fEventCount->Add(entry->fEventCount);\r
 555     fRecEventHist->Add(entry->fRecEventHist);\r
 556     fRecMCEventHist->Add(entry->fRecMCEventHist);\r
 557     fMCEventHist->Add(entry->fMCEventHist);\r
 558 \r
 559     // track histo\r
 560     fRecMCTrackHist->Add(entry->fRecMCTrackHist);\r
 561 \r
 562     // physics selection\r
 563     //collPhysSelection->Add(entry->GetPhysicsTriggerSelection());\r
 564     \r
 565   count++;\r
 566   }\r
 567 \r
 568   //AliPhysicsSelection *trigSelection = GetPhysicsTriggerSelection();\r
 569   //trigSelection->Merge(collPhysSelection);\r
 570 \r
 571   //if(collPhysSelection) delete collPhysSelection;\r
 572 \r
 573 return count;\r
 574 }\r
 575 \r
 576 //_____________________________________________________________________________\r
 577 void AlidNdPtCutAnalysis::Analyse() \r
 578 {\r
 579   //\r
 580   // Analyse histograms\r
 581   //\r
 582   TH1::AddDirectory(kFALSE);\r
 583   TObjArray *aFolderObj = new TObjArray;\r
 584   TH1D *h1D = 0; \r
 585   TH2D *h2D = 0; \r
 586 \r
 587 \r
 588   //\r
 589   // get cuts\r
 590   //\r
 591   AlidNdPtEventCuts *evtCuts = GetEventCuts(); \r
 592   AlidNdPtAcceptanceCuts *accCuts = GetAcceptanceCuts(); \r
 593   AliESDtrackCuts *esdTrackCuts = GetTrackCuts(); \r
 594 \r
 595   if(!evtCuts || !accCuts || !esdTrackCuts) {\r
 596     Error("AlidNdPtCutAnalysis::Analyse()", "cuts not available");\r
 597     return;\r
 598   }\r
 599 \r
 600   //\r
 601   // set min and max values\r
 602   //\r
 603   Double_t minPt = accCuts->GetMinPt();\r
 604   Double_t maxPt = accCuts->GetMaxPt();\r
 605   Double_t minEta = accCuts->GetMinEta();\r
 606   Double_t maxEta = accCuts->GetMaxEta()-0.00001;\r
 607 \r
 608   Double_t maxDCAr = accCuts->GetMaxDCAr();\r
 609 \r
 610   //\r
 611   // Event counters\r
 612   //\r
 613   h2D = (TH2D*)fEventCount->Projection(0,1);\r
 614   h2D->SetName("trig_vs_trigANDvertex");\r
 615   aFolderObj->Add(h2D);\r
 616 \r
 617   fEventCount->GetAxis(0)->SetRange(2,2); // triggered\r
 618   h1D = (TH1D*)fEventCount->Projection(1);\r
 619   h1D->SetTitle("rec. vertex for triggered events");\r
 620   h1D->SetName("trigANDvertex");\r
 621   aFolderObj->Add(h1D);\r
 622 \r
 623   //\r
 624   // Create rec. event histograms\r
 625   //\r
 626   h1D = (TH1D *)fRecEventHist->Projection(0);\r
 627   h1D->SetName("rec_xv");\r
 628   aFolderObj->Add(h1D);\r
 629 \r
 630   h1D = (TH1D *)fRecEventHist->Projection(1);\r
 631   h1D->SetName("rec_yv");\r
 632   aFolderObj->Add(h1D);\r
 633 \r
 634   h1D = (TH1D *)fRecEventHist->Projection(2);\r
 635   h1D->SetName("rec_zv");\r
 636   aFolderObj->Add(h1D);\r
 637 \r
 638   h2D = (TH2D *)fRecEventHist->Projection(3,4);\r
 639   h2D->SetName("rec_resZv_vs_Mult");\r
 640   aFolderObj->Add(h2D);\r
 641 \r
 642   h2D = (TH2D *)fRecEventHist->Projection(0,1);\r
 643   h2D->SetName("rec_xv_vs_yv");\r
 644   aFolderObj->Add(h2D);\r
 645 \r
 646   h2D = (TH2D *)fRecEventHist->Projection(0,2);\r
 647   h2D->SetName("rec_xv_vs_zv");\r
 648   aFolderObj->Add(h2D);\r
 649 \r
 650   h2D = (TH2D *)fRecEventHist->Projection(3,4);\r
 651   h2D->SetName("rec_resZv_vs_Mult");\r
 652   aFolderObj->Add(h2D);\r
 653 \r
 654   //\r
 655   // MC available\r
 656   //\r
 657   if(IsUseMCInfo()) {\r
 658 \r
 659   //\r
 660   // Create mc event histograms\r
 661   //\r
 662   h2D = (TH2D *)fMCEventHist->Projection(0,1);\r
 663   h2D->SetName("mc_xv_vs_yv");\r
 664   aFolderObj->Add(h2D);\r
 665 \r
 666   h2D = (TH2D *)fMCEventHist->Projection(0,2);\r
 667   h2D->SetName("mc_xv_vs_zv");\r
 668   aFolderObj->Add(h2D);\r
 669 \r
 670   //\r
 671   // Create rec-mc event histograms\r
 672   //\r
 673   h2D = (TH2D *)fRecMCEventHist->Projection(0,3);\r
 674   h2D->SetName("rec_mc_deltaXv_vs_mult");\r
 675   aFolderObj->Add(h2D);\r
 676 \r
 677   h2D = (TH2D *)fRecMCEventHist->Projection(1,3);\r
 678   h2D->SetName("rec_mc_deltaYv_vs_mult");\r
 679   aFolderObj->Add(h2D);\r
 680 \r
 681   h2D = (TH2D *)fRecMCEventHist->Projection(2,3);\r
 682   h2D->SetName("rec_mc_deltaZv_vs_mult");\r
 683   aFolderObj->Add(h2D);\r
 684 \r
 685   } // end use MC info \r
 686 \r
 687 \r
 688 \r
 689   //\r
 690   // Create rec-mc track track histograms \r
 691   //\r
 692 \r
 693   // DCA cuts\r
 694   fRecMCTrackHist->GetAxis(3)->SetRangeUser(-maxDCAr,maxDCAr);\r
 695   fRecMCTrackHist->GetAxis(4)->SetRangeUser(-maxDCAr,maxDCAr);\r
 696 \r
 697   h2D = (TH2D *)fRecMCTrackHist->Projection(7,5);\r
 698   h2D->SetName("pt_vs_eta");\r
 699   aFolderObj->Add(h2D);\r
 700 \r
 701   fRecMCTrackHist->GetAxis(7)->SetRangeUser(minPt,maxPt);  \r
 702 \r
 703   h2D = (TH2D *)fRecMCTrackHist->Projection(0,5);\r
 704   h2D->SetName("nClust_vs_eta");\r
 705   aFolderObj->Add(h2D);\r
 706 \r
 707   h2D = (TH2D *)fRecMCTrackHist->Projection(1,5);\r
 708   h2D->SetName("chi2PerClust_vs_eta");\r
 709   aFolderObj->Add(h2D);\r
 710 \r
 711   h2D = (TH2D *)fRecMCTrackHist->Projection(2,5);\r
 712   h2D->SetName("ratio_nClust_nFindableClust_vs_eta");\r
 713   aFolderObj->Add(h2D);\r
 714 \r
 715   h2D = (TH2D *)fRecMCTrackHist->Projection(5,6);\r
 716   h2D->SetName("eta_vs_phi");\r
 717   aFolderObj->Add(h2D);\r
 718 \r
 719   //\r
 720   fRecMCTrackHist->GetAxis(5)->SetRangeUser(minEta,maxEta);  \r
 721 \r
 722   h2D = (TH2D *)fRecMCTrackHist->Projection(0,6);\r
 723   h2D->SetName("nClust_vs_phi");\r
 724   aFolderObj->Add(h2D);\r
 725 \r
 726   h2D = (TH2D *)fRecMCTrackHist->Projection(1,6);\r
 727   h2D->SetName("chi2PerClust_vs_phi");\r
 728   aFolderObj->Add(h2D);\r
 729 \r
 730   h2D = (TH2D *)fRecMCTrackHist->Projection(2,6);\r
 731   h2D->SetName("ratio_nClust_nFindableClust_vs_phi");\r
 732   aFolderObj->Add(h2D);\r
 733 \r
 734   //\r
 735   fRecMCTrackHist->GetAxis(7)->SetRangeUser(0.0,maxPt);  \r
 736 \r
 737   h2D = (TH2D *)fRecMCTrackHist->Projection(0,7);\r
 738   h2D->SetName("nClust_vs_pt");\r
 739   aFolderObj->Add(h2D);\r
 740 \r
 741   h2D = (TH2D *)fRecMCTrackHist->Projection(1,7);\r
 742   h2D->SetName("chi2PerClust_vs_pt");\r
 743   aFolderObj->Add(h2D);\r
 744 \r
 745   h2D = (TH2D *)fRecMCTrackHist->Projection(2,7);\r
 746   h2D->SetName("ratio_nClust_nFindableClust_vs_pt");\r
 747   aFolderObj->Add(h2D);\r
 748 \r
 749   h2D = (TH2D *)fRecMCTrackHist->Projection(6,7);\r
 750   h2D->SetName("phi_vs_pt");\r
 751   aFolderObj->Add(h2D);\r
 752 \r
 753 \r
 754   // fiducial volume\r
 755   fRecMCTrackHist->GetAxis(5)->SetRangeUser(minEta,maxEta);  \r
 756   fRecMCTrackHist->GetAxis(7)->SetRangeUser(minPt,maxPt);  \r
 757 \r
 758   // DCA cuts\r
 759   fRecMCTrackHist->GetAxis(3)->SetRangeUser(-maxDCAr,maxDCAr);\r
 760   fRecMCTrackHist->GetAxis(4)->SetRangeUser(-maxDCAr,maxDCAr);\r
 761 \r
 762   h2D = (TH2D *)fRecMCTrackHist->Projection(0,1);\r
 763   h2D->SetName("nClust_vs_chi2PerClust");\r
 764   aFolderObj->Add(h2D);\r
 765 \r
 766   h2D = (TH2D *)fRecMCTrackHist->Projection(0,2);\r
 767   h2D->SetName("nClust_vs_ratio_nClust_nFindableClust");\r
 768   aFolderObj->Add(h2D);\r
 769 \r
 770   //\r
 771   // DCAy cuts\r
 772   //\r
 773   fRecMCTrackHist->GetAxis(0)->SetRange(50,160); // nClust/track > 50\r
 774   fRecMCTrackHist->GetAxis(1)->SetRangeUser(0.,3.9999); // chi2/cluster < 4.0\r
 775   fRecMCTrackHist->GetAxis(3)->SetRange(1,fRecMCTrackHist->GetAxis(3)->GetNbins());\r
 776   //fRecMCTrackHist->GetAxis(4)->SetRangeUser(-1.0,1.0);\r
 777   fRecMCTrackHist->GetAxis(4)->SetRange(1,fRecMCTrackHist->GetAxis(4)->GetNbins());\r
 778 \r
 779   // sec\r
 780   fRecMCTrackHist->GetAxis(9)->SetRange(1,1);\r
 781   h1D = (TH1D *)fRecMCTrackHist->Projection(3);\r
 782   h1D->SetName("dcay_sec");\r
 783   aFolderObj->Add(h1D);\r
 784 \r
 785   // prim\r
 786   fRecMCTrackHist->GetAxis(9)->SetRange(2,2);\r
 787   h1D = (TH1D *)fRecMCTrackHist->Projection(3);\r
 788   h1D->SetName("dcay_prim");\r
 789   aFolderObj->Add(h1D);\r
 790 \r
 791   // DCAz cuts\r
 792   //fRecMCTrackHist->GetAxis(3)->SetRangeUser(-1.0,1.0);\r
 793   fRecMCTrackHist->GetAxis(4)->SetRange(1,fRecMCTrackHist->GetAxis(4)->GetNbins());\r
 794 \r
 795   // sec\r
 796   fRecMCTrackHist->GetAxis(9)->SetRange(1,1);\r
 797   h1D = (TH1D *)fRecMCTrackHist->Projection(4);\r
 798   h1D->SetName("dcaz_sec");\r
 799   aFolderObj->Add(h1D);\r
 800 \r
 801   // prim\r
 802   fRecMCTrackHist->GetAxis(9)->SetRange(2,2);\r
 803   h1D = (TH1D *)fRecMCTrackHist->Projection(4);\r
 804   h1D->SetName("dcaz_prim");\r
 805   aFolderObj->Add(h1D);\r
 806 \r
 807 \r
 808   // export objects to analysis folder\r
 809   fAnalysisFolder = ExportToFolder(aFolderObj);\r
 810 \r
 811   // delete only TObjArray\r
 812   if(aFolderObj) delete aFolderObj;\r
 813 }\r
 814 \r
 815 //_____________________________________________________________________________\r
 816 TFolder* AlidNdPtCutAnalysis::ExportToFolder(TObjArray * const array) \r
 817 {\r
 818   // recreate folder avery time and export objects to new one\r
 819   //\r
 820   AlidNdPtCutAnalysis * comp=this;\r
 821   TFolder *folder = comp->GetAnalysisFolder();\r
 822 \r
 823   TString name, title;\r
 824   TFolder *newFolder = 0;\r
 825   Int_t i = 0;\r
 826   Int_t size = array->GetSize();\r
 827 \r
 828   if(folder) { \r
 829      // get name and title from old folder\r
 830      name = folder->GetName();  \r
 831      title = folder->GetTitle();  \r
 832 \r
 833          // delete old one\r
 834      delete folder;\r
 835 \r
 836          // create new one\r
 837      newFolder = CreateFolder(name.Data(),title.Data());\r
 838      newFolder->SetOwner();\r
 839 \r
 840          // add objects to folder\r
 841      while(i < size) {\r
 842            newFolder->Add(array->At(i));\r
 843            i++;\r
 844          }\r
 845   }\r
 846 \r
 847 return newFolder;\r
 848 }\r
 849 \r
 850 //_____________________________________________________________________________\r
 851 TFolder* AlidNdPtCutAnalysis::CreateFolder(TString name,TString title) { \r
 852 // create folder for analysed histograms\r
 853 //\r
 854 TFolder *folder = 0;\r
 855   folder = new TFolder(name.Data(),title.Data());\r
 856 \r
 857   return folder;\r
 858 }\r