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 #include <iostream>\r
  17 \r
  18 #include "TFile.h"\r
  19 #include "TCint.h"\r
  20 #include "TH1.h"\r
  21 #include "TH2.h"\r
  22 \r
  23 #include "AliHeader.h"  \r
  24 #include "AliGenEventHeader.h"  \r
  25 #include "AliStack.h"  \r
  26 #include "AliESDEvent.h"  \r
  27 #include "AliMCEvent.h"  \r
  28 #include "AliESDtrackCuts.h"  \r
  29 #include "AliLog.h" \r
  30 \r
  31 #include "AlidNdPtEventCuts.h"\r
  32 #include "AlidNdPtAcceptanceCuts.h"\r
  33 \r
  34 #include "AliPWG0Helper.h"\r
  35 #include "AlidNdPtHelper.h"\r
  36 #include "AlidNdPtCutAnalysis.h"\r
  37 \r
  38 using namespace std;\r
  39 \r
  40 ClassImp(AlidNdPtCutAnalysis)\r
  41 \r
  42 //_____________________________________________________________________________\r
  43   AlidNdPtCutAnalysis::AlidNdPtCutAnalysis(): AlidNdPt(),\r
  44   fAnalysisFolder(0),\r
  45   fEventCount(0),\r
  46   fRecEventHist(0),\r
  47   fMCEventHist(0),\r
  48   fRecMCEventHist(0),\r
  49   fRecMCTrackHist(0)\r
  50 {\r
  51   // default constructor\r
  52   Init();\r
  53 }\r
  54 \r
  55 //_____________________________________________________________________________\r
  56 AlidNdPtCutAnalysis::AlidNdPtCutAnalysis(Char_t* name, Char_t* title): AlidNdPt(name,title),\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   // constructor\r
  65   Init();\r
  66 }\r
  67 \r
  68 //_____________________________________________________________________________\r
  69 AlidNdPtCutAnalysis::~AlidNdPtCutAnalysis() {\r
  70   // \r
  71   if(fEventCount) delete fEventCount; fEventCount=0;\r
  72   if(fRecEventHist) delete fRecEventHist; fRecEventHist=0;\r
  73   if(fMCEventHist) delete fMCEventHist; fMCEventHist=0;\r
  74   if(fRecMCEventHist) delete fRecMCEventHist; fRecMCEventHist=0;\r
  75   if(fRecMCTrackHist) delete fRecMCTrackHist; fRecMCTrackHist=0;\r
  76 \r
  77   if(fAnalysisFolder) delete fAnalysisFolder; fAnalysisFolder=0;\r
  78 }\r
  79 \r
  80 //_____________________________________________________________________________\r
  81 void AlidNdPtCutAnalysis::Init(){\r
  82   //\r
  83   // Init histograms\r
  84   //\r
  85   const Int_t ptNbins = 56; \r
  86   const Double_t ptMin = 0.; \r
  87   const Double_t ptMax = 16.; \r
  88 \r
  89   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
  90 \r
  91   // \r
  92   Int_t binsEventCount[2]={2,2};\r
  93   Double_t minEventCount[2]={0,0}; \r
  94   Double_t maxEventCount[2]={2,2}; \r
  95   fEventCount = new THnSparseF("fEventCount","trig vs trig+vertex",2,binsEventCount,minEventCount,maxEventCount);\r
  96   fEventCount->GetAxis(0)->SetTitle("trig");\r
  97   fEventCount->GetAxis(1)->SetTitle("trig+vert");\r
  98   fEventCount->Sumw2();\r
  99 \r
 100   //Xv:Yv:Zv:ResZv:Mult\r
 101   Int_t binsRecEventHist[5]={100,100,140,100,150};\r
 102   Double_t minRecEventHist[5]={-3.,-3.,-35.,0.,0.}; \r
 103   Double_t maxRecEventHist[5]={3.,3.,35.,10.,150.}; \r
 104   fRecEventHist = new THnSparseF("fRecEventHist","Xv:Yv:Zv:ResZv:Mult",5,binsRecEventHist,minRecEventHist,maxRecEventHist);\r
 105   fRecEventHist->GetAxis(0)->SetTitle("Xv (cm)");\r
 106   fRecEventHist->GetAxis(1)->SetTitle("Yv (cm)");\r
 107   fRecEventHist->GetAxis(2)->SetTitle("Zv (cm)");\r
 108   fRecEventHist->GetAxis(3)->SetTitle("ResZv (cm)");\r
 109   fRecEventHist->GetAxis(4)->SetTitle("Mult");\r
 110   fRecEventHist->Sumw2();\r
 111 \r
 112   //Xv:Yv:Zv\r
 113   Int_t binsMCEventHist[3]={100,100,140};\r
 114   Double_t minMCEventHist[3]={-0.1,-0.1,-35.}; \r
 115   Double_t maxMCEventHist[3]={0.1,0.1,35.}; \r
 116   fMCEventHist = new THnSparseF("fMCEventHist","mcXv:mcYv:mcZv",3,binsMCEventHist,minMCEventHist,maxMCEventHist);\r
 117   fMCEventHist->GetAxis(0)->SetTitle("mcXv (cm)");\r
 118   fMCEventHist->GetAxis(1)->SetTitle("mcYv (cm)");\r
 119   fMCEventHist->GetAxis(2)->SetTitle("mcZv (cm)");\r
 120   fMCEventHist->Sumw2();\r
 121 \r
 122   //Xv-mcXv:Yv-mcYv:Zv-mcZv:Mult\r
 123   Int_t binsRecMCEventHist[4]={100,100,100,150};\r
 124   Double_t minRecMCEventHist[4]={-10.0,-10.0,-10.0,0.}; \r
 125   Double_t maxRecMCEventHist[4]={10.0,10.0,10.0,150.}; \r
 126   fRecMCEventHist = new THnSparseF("fRecMCEventHist","mcXv-Xv:mcYv-Yv:mcZv-Zv:Mult",4,binsRecMCEventHist,minRecMCEventHist,maxRecMCEventHist);\r
 127   fRecMCEventHist->GetAxis(0)->SetTitle("mcXv-Xv (cm)");\r
 128   fRecMCEventHist->GetAxis(1)->SetTitle("mcYv-Yv (cm)");\r
 129   fRecMCEventHist->GetAxis(2)->SetTitle("mcZv-Zv (cm)");\r
 130   fRecMCEventHist->GetAxis(3)->SetTitle("Mult");\r
 131   fRecMCEventHist->Sumw2();\r
 132 \r
 133   //\r
 134   // THnSparse track histograms\r
 135   //\r
 136 \r
 137   //nClust:chi2PerClust:nClust/nFindableClust:DCAy:DCAz:eta:phi:pt:isKink:isPrim\r
 138   Int_t binsRecMCTrackHist[10]={160,100,100,100,100,30,90,ptNbins, 2,2};\r
 139   Double_t minRecMCTrackHist[10]={0., 0., 0., -10.,-10.,-1.5, 0., ptMin, 0., 0.};\r
 140   Double_t maxRecMCTrackHist[10]={160.,10.,1.2, 10.,10.,1.5, 2.*TMath::Pi(), ptMax, 2.,2.};\r
 141 \r
 142   fRecMCTrackHist = new THnSparseF("fRecMCTrackHist","nClust:chi2PerClust:nClust/nFindableClust:DCAy:DCAz:eta:phi:pt:isKink:isPrim",10,binsRecMCTrackHist,minRecMCTrackHist,maxRecMCTrackHist);\r
 143   fRecMCTrackHist->SetBinEdges(7,binsPt);\r
 144 \r
 145   fRecMCTrackHist->GetAxis(0)->SetTitle("nClust");\r
 146   fRecMCTrackHist->GetAxis(1)->SetTitle("chi2PerClust");\r
 147   fRecMCTrackHist->GetAxis(2)->SetTitle("nClust/nFindableClust");\r
 148   fRecMCTrackHist->GetAxis(3)->SetTitle("DCAy (cm)");\r
 149   fRecMCTrackHist->GetAxis(4)->SetTitle("DCAz (cm)");\r
 150   fRecMCTrackHist->GetAxis(5)->SetTitle("#eta");\r
 151   fRecMCTrackHist->GetAxis(6)->SetTitle("#phi (rad)");\r
 152   fRecMCTrackHist->GetAxis(7)->SetTitle("p_{T} (GeV/c)");\r
 153   fRecMCTrackHist->GetAxis(8)->SetTitle("isKink");\r
 154   fRecMCTrackHist->GetAxis(9)->SetTitle("isPrim");\r
 155   fRecMCTrackHist->Sumw2();\r
 156 \r
 157   // init output folder\r
 158   fAnalysisFolder = CreateFolder("folderdNdPt","Analysis dNdPt Folder");\r
 159 \r
 160 }\r
 161 \r
 162 //_____________________________________________________________________________\r
 163 void AlidNdPtCutAnalysis::Process(AliESDEvent *const esdEvent, AliMCEvent * const mcEvent)\r
 164 {\r
 165   //\r
 166   // Process real and/or simulated events\r
 167   //\r
 168   if(!esdEvent) {\r
 169     AliDebug(AliLog::kError, "esdEvent not available");\r
 170     return;\r
 171   }\r
 172 \r
 173   // get selection cuts\r
 174   AlidNdPtEventCuts *evtCuts = GetEventCuts(); \r
 175   AlidNdPtAcceptanceCuts *accCuts = GetAcceptanceCuts(); \r
 176   AliESDtrackCuts *esdTrackCuts = GetTrackCuts(); \r
 177 \r
 178   if(!evtCuts || !accCuts  || !esdTrackCuts) {\r
 179     AliDebug(AliLog::kError, "cuts not available");\r
 180     return;\r
 181   }\r
 182 \r
 183   // trigger selection\r
 184   Bool_t isEventTriggered = kTRUE;\r
 185   if(evtCuts->IsTriggerRequired())  {\r
 186     //static AliTriggerAnalysis* triggerAnalysis = new AliTriggerAnalysis;\r
 187     //isEventTriggered = triggerAnalysis->IsTriggerFired(esdEvent, GetTrigger());\r
 188     isEventTriggered = esdEvent->IsTriggerClassFired(GetTriggerClass());\r
 189   }\r
 190 \r
 191   // use MC information\r
 192   AliHeader* header = 0;\r
 193   AliGenEventHeader* genHeader = 0;\r
 194   AliStack* stack = 0;\r
 195   TArrayF vtxMC(3);\r
 196   //AlidNdPtHelper::MCProcessType evtType = AlidNdPtHelper::kInvalidProcess;\r
 197   AliPWG0Helper::MCProcessType evtType = AliPWG0Helper::kInvalidProcess;\r
 198 \r
 199   if(IsUseMCInfo())\r
 200   {\r
 201     if(!mcEvent) {\r
 202       AliDebug(AliLog::kError, "mcEvent not available");\r
 203       return;\r
 204     }\r
 205 \r
 206     // get MC event header\r
 207     header = mcEvent->Header();\r
 208     if (!header) {\r
 209       AliDebug(AliLog::kError, "Header not available");\r
 210       return;\r
 211     }\r
 212 \r
 213     // MC particle stack\r
 214     stack = mcEvent->Stack();\r
 215     if (!stack) {\r
 216       AliDebug(AliLog::kError, "Stack not available");\r
 217       return;\r
 218     }\r
 219 \r
 220     // get event type (ND=0x1, DD=0x2, SD=0x4)\r
 221     evtType = AliPWG0Helper::GetEventProcessType(header);\r
 222     AliDebug(AliLog::kDebug+1, Form("Found process type %d", evtType));\r
 223 \r
 224     // get MC vertex\r
 225     genHeader = header->GenEventHeader();\r
 226     if (!genHeader) {\r
 227       AliDebug(AliLog::kError, "Could not retrieve genHeader from Header");\r
 228       return;\r
 229     }\r
 230     genHeader->PrimaryVertex(vtxMC);\r
 231 \r
 232     // Fill MC event histogram\r
 233     Double_t vMCEventHist[3]={vtxMC[0],vtxMC[1],vtxMC[2]};\r
 234     fMCEventHist->Fill(vMCEventHist);\r
 235 \r
 236   } // end bUseMC\r
 237 \r
 238   // get reconstructed vertex  \r
 239   Bool_t bRedoTPCVertex = evtCuts->IsRedoTPCVertex();\r
 240   Bool_t bUseConstraints = evtCuts->IsUseBeamSpotConstraint();\r
 241   const AliESDVertex* vtxESD = AlidNdPtHelper::GetVertex(esdEvent,evtCuts,accCuts,esdTrackCuts,GetAnalysisMode(),kFALSE,bRedoTPCVertex,bUseConstraints); \r
 242   if(!vtxESD) return; \r
 243 \r
 244   Bool_t isRecVertex = AlidNdPtHelper::TestRecVertex(vtxESD, GetAnalysisMode(), kFALSE);\r
 245   Bool_t isEventOK = evtCuts->AcceptEvent(esdEvent,mcEvent,vtxESD) && isRecVertex;\r
 246 \r
 247   TObjArray *allChargedTracks=0;\r
 248   Int_t multAll=0;\r
 249   \r
 250   //\r
 251   // event counter\r
 252   // \r
 253   //printf("isEventOK %d, isEventTriggered %d \n",isEventOK,isEventTriggered);\r
 254 \r
 255   Bool_t isTrigAndVertex = isEventTriggered && isEventOK;\r
 256   Double_t vEventCount[2] = { isEventTriggered, isTrigAndVertex};\r
 257   fEventCount->Fill(vEventCount);\r
 258 \r
 259   // check event cuts\r
 260   if(isEventOK && isEventTriggered)\r
 261   {\r
 262     // get all charged tracks\r
 263     allChargedTracks = AlidNdPtHelper::GetAllChargedTracks(esdEvent,GetAnalysisMode());\r
 264     if(!allChargedTracks) return;\r
 265 \r
 266     Int_t entries = allChargedTracks->GetEntries();\r
 267     for(Int_t i=0; i<entries;++i) \r
 268     {\r
 269       AliESDtrack *track = (AliESDtrack*)allChargedTracks->At(i);\r
 270       if(!track) continue;\r
 271 \r
 272       FillHistograms(track, stack);\r
 273       multAll++;\r
 274     }\r
 275 \r
 276   Double_t vRecEventHist[5] = {vtxESD->GetXv(),vtxESD->GetYv(),vtxESD->GetZv(),vtxESD->GetZRes(),multAll};\r
 277   fRecEventHist->Fill(vRecEventHist);\r
 278 \r
 279   if(IsUseMCInfo()) {\r
 280     Double_t vRecMCEventHist[5] = {vtxESD->GetXv()-vtxMC[0],vtxESD->GetYv()-vtxMC[1],vtxESD->GetZv()-vtxMC[2],multAll};\r
 281     fRecMCEventHist->Fill(vRecMCEventHist);\r
 282   }\r
 283   }\r
 284 \r
 285   if(allChargedTracks) delete allChargedTracks; allChargedTracks = 0;\r
 286 }\r
 287 \r
 288 //_____________________________________________________________________________\r
 289 void AlidNdPtCutAnalysis::FillHistograms(AliESDtrack *const esdTrack, AliStack *const stack)\r
 290 {\r
 291   //\r
 292   // Fill ESD track and MC histograms \r
 293   //\r
 294   if(!esdTrack) return;\r
 295   if(esdTrack->Charge() == 0.) return;\r
 296 \r
 297   Float_t pt = esdTrack->Pt();\r
 298   Float_t eta = esdTrack->Eta();\r
 299   Float_t phi = esdTrack->Phi();\r
 300   Int_t nClust = esdTrack->GetTPCclusters(0);\r
 301   Int_t nFindableClust = esdTrack->GetTPCNclsF();\r
 302 \r
 303   Float_t chi2PerCluster = 0.;\r
 304   if(nClust>0.) chi2PerCluster = esdTrack->GetTPCchi2()/Float_t(nClust);\r
 305 \r
 306   Float_t clustPerFindClust = 0.;\r
 307   if(nFindableClust>0.) clustPerFindClust = Float_t(nClust)/nFindableClust;\r
 308 \r
 309   Float_t b[2], bCov[3];\r
 310   esdTrack->GetImpactParameters(b,bCov);\r
 311 \r
 312   // kink daughter\r
 313   Bool_t isKink = kFALSE;\r
 314   if(esdTrack->GetKinkIndex(0)>0) isKink=kTRUE;\r
 315 \r
 316   //\r
 317   // Fill rec vs MC information\r
 318   //\r
 319 \r
 320   Bool_t isPrim = kTRUE;\r
 321 \r
 322   if(IsUseMCInfo()) {\r
 323     if(!stack) return;\r
 324     Int_t label = TMath::Abs(esdTrack->GetLabel()); \r
 325     TParticle* particle = stack->Particle(label);\r
 326     if(!particle) return;\r
 327     if(particle->GetPDG() && particle->GetPDG()->Charge()==0.) return;\r
 328     isPrim = stack->IsPhysicalPrimary(label);\r
 329   }\r
 330 \r
 331   // fill histo\r
 332   Double_t vRecMCTrackHist[10] = {nClust,chi2PerCluster,clustPerFindClust,b[0],b[1],eta,phi,pt,isKink,isPrim}; \r
 333   fRecMCTrackHist->Fill(vRecMCTrackHist);\r
 334 }\r
 335 \r
 336 //_____________________________________________________________________________\r
 337 Long64_t AlidNdPtCutAnalysis::Merge(TCollection* list) \r
 338 {\r
 339   // Merge list of objects (needed by PROOF)\r
 340 \r
 341   if (!list)\r
 342   return 0;\r
 343 \r
 344   if (list->IsEmpty())\r
 345   return 1;\r
 346 \r
 347   TIterator* iter = list->MakeIterator();\r
 348   TObject* obj = 0;\r
 349 \r
 350   // collection of generated histograms\r
 351   Int_t count=0;\r
 352   while((obj = iter->Next()) != 0) {\r
 353     AlidNdPtCutAnalysis* entry = dynamic_cast<AlidNdPtCutAnalysis*>(obj);\r
 354     if (entry == 0) continue; \r
 355   \r
 356     // event histo\r
 357     fEventCount->Add(entry->fEventCount);\r
 358     fRecEventHist->Add(entry->fRecEventHist);\r
 359     fRecMCEventHist->Add(entry->fRecMCEventHist);\r
 360     fMCEventHist->Add(entry->fMCEventHist);\r
 361 \r
 362     // track histo\r
 363     fRecMCTrackHist->Add(entry->fRecMCTrackHist);\r
 364     \r
 365   count++;\r
 366   }\r
 367 \r
 368 return count;\r
 369 }\r
 370 \r
 371 //_____________________________________________________________________________\r
 372 void AlidNdPtCutAnalysis::Analyse() \r
 373 {\r
 374   //\r
 375   // Analyse histograms\r
 376   //\r
 377   TH1::AddDirectory(kFALSE);\r
 378   TObjArray *aFolderObj = new TObjArray;\r
 379   TH1D *h1D = 0; \r
 380   TH2D *h2D = 0; \r
 381 \r
 382 \r
 383   //\r
 384   // get cuts\r
 385   //\r
 386   AlidNdPtEventCuts *evtCuts = GetEventCuts(); \r
 387   AlidNdPtAcceptanceCuts *accCuts = GetAcceptanceCuts(); \r
 388   AliESDtrackCuts *esdTrackCuts = GetTrackCuts(); \r
 389 \r
 390   if(!evtCuts || !accCuts || !esdTrackCuts) {\r
 391     Error("AlidNdPtCutAnalysis::Analyse()", "cuts not available");\r
 392     return;\r
 393   }\r
 394 \r
 395   //\r
 396   // set min and max values\r
 397   //\r
 398   Double_t minPt = accCuts->GetMinPt();\r
 399   Double_t maxPt = accCuts->GetMaxPt();\r
 400   Double_t minEta = accCuts->GetMinEta();\r
 401   Double_t maxEta = accCuts->GetMaxEta()-0.00001;\r
 402 \r
 403   Double_t maxDCAr = accCuts->GetMaxDCAr();\r
 404 \r
 405   //\r
 406   // Event counters\r
 407   //\r
 408   h2D = (TH2D*)fEventCount->Projection(0,1);\r
 409   h2D->SetName("trig_vs_trigANDvertex");\r
 410   aFolderObj->Add(h2D);\r
 411 \r
 412   fEventCount->GetAxis(0)->SetRange(1,1); // triggered\r
 413   h1D = (TH2D*)fEventCount->Projection(1);\r
 414   h1D->SetTitle("rec. vertex for triggered events");\r
 415   h1D->SetName("trigANDvertex");\r
 416   aFolderObj->Add(h1D);\r
 417 \r
 418   //\r
 419   // Create rec. event histograms\r
 420   //\r
 421   h1D = (TH1D *)fRecEventHist->Projection(0);\r
 422   h1D->SetName("rec_xv");\r
 423   aFolderObj->Add(h1D);\r
 424 \r
 425   h1D = (TH1D *)fRecEventHist->Projection(1);\r
 426   h1D->SetName("rec_yv");\r
 427   aFolderObj->Add(h1D);\r
 428 \r
 429   h1D = (TH1D *)fRecEventHist->Projection(2);\r
 430   h1D->SetName("rec_zv");\r
 431   aFolderObj->Add(h1D);\r
 432 \r
 433   h2D = (TH2D *)fRecEventHist->Projection(3,4);\r
 434   h2D->SetName("rec_resZv_vs_Mult");\r
 435   aFolderObj->Add(h2D);\r
 436 \r
 437   h2D = (TH2D *)fRecEventHist->Projection(0,1);\r
 438   h2D->SetName("rec_xv_vs_yv");\r
 439   aFolderObj->Add(h2D);\r
 440 \r
 441   h2D = (TH2D *)fRecEventHist->Projection(0,2);\r
 442   h2D->SetName("rec_xv_vs_zv");\r
 443   aFolderObj->Add(h2D);\r
 444 \r
 445   h2D = (TH2D *)fRecEventHist->Projection(3,4);\r
 446   h2D->SetName("rec_resZv_vs_Mult");\r
 447   aFolderObj->Add(h2D);\r
 448 \r
 449   //\r
 450   // MC available\r
 451   //\r
 452   if(IsUseMCInfo()) {\r
 453 \r
 454   //\r
 455   // Create mc event histograms\r
 456   //\r
 457   h2D = (TH2D *)fMCEventHist->Projection(0,1);\r
 458   h2D->SetName("mc_xv_vs_yv");\r
 459   aFolderObj->Add(h2D);\r
 460 \r
 461   h2D = (TH2D *)fMCEventHist->Projection(0,2);\r
 462   h2D->SetName("mc_xv_vs_zv");\r
 463   aFolderObj->Add(h2D);\r
 464 \r
 465   //\r
 466   // Create rec-mc event histograms\r
 467   //\r
 468   h2D = (TH2D *)fRecMCEventHist->Projection(0,3);\r
 469   h2D->SetName("rec_mc_deltaXv_vs_mult");\r
 470   aFolderObj->Add(h2D);\r
 471 \r
 472   h2D = (TH2D *)fRecMCEventHist->Projection(1,3);\r
 473   h2D->SetName("rec_mc_deltaYv_vs_mult");\r
 474   aFolderObj->Add(h2D);\r
 475 \r
 476   h2D = (TH2D *)fRecMCEventHist->Projection(2,3);\r
 477   h2D->SetName("rec_mc_deltaZv_vs_mult");\r
 478   aFolderObj->Add(h2D);\r
 479 \r
 480   } // end use MC info \r
 481 \r
 482 \r
 483 \r
 484   //\r
 485   // Create rec-mc track track histograms \r
 486   //\r
 487 \r
 488   // DCA cuts\r
 489   fRecMCTrackHist->GetAxis(3)->SetRangeUser(-maxDCAr,maxDCAr);\r
 490   fRecMCTrackHist->GetAxis(4)->SetRangeUser(-maxDCAr,maxDCAr);\r
 491 \r
 492   h2D = (TH2D *)fRecMCTrackHist->Projection(7,5);\r
 493   h2D->SetName("pt_vs_eta");\r
 494   aFolderObj->Add(h2D);\r
 495 \r
 496   fRecMCTrackHist->GetAxis(7)->SetRangeUser(minPt,maxPt);  \r
 497 \r
 498   h2D = (TH2D *)fRecMCTrackHist->Projection(0,5);\r
 499   h2D->SetName("nClust_vs_eta");\r
 500   aFolderObj->Add(h2D);\r
 501 \r
 502   h2D = (TH2D *)fRecMCTrackHist->Projection(1,5);\r
 503   h2D->SetName("chi2PerClust_vs_eta");\r
 504   aFolderObj->Add(h2D);\r
 505 \r
 506   h2D = (TH2D *)fRecMCTrackHist->Projection(2,5);\r
 507   h2D->SetName("ratio_nClust_nFindableClust_vs_eta");\r
 508   aFolderObj->Add(h2D);\r
 509 \r
 510   h2D = (TH2D *)fRecMCTrackHist->Projection(5,6);\r
 511   h2D->SetName("eta_vs_phi");\r
 512   aFolderObj->Add(h2D);\r
 513 \r
 514   //\r
 515   fRecMCTrackHist->GetAxis(5)->SetRangeUser(minEta,maxEta);  \r
 516 \r
 517   h2D = (TH2D *)fRecMCTrackHist->Projection(0,6);\r
 518   h2D->SetName("nClust_vs_phi");\r
 519   aFolderObj->Add(h2D);\r
 520 \r
 521   h2D = (TH2D *)fRecMCTrackHist->Projection(1,6);\r
 522   h2D->SetName("chi2PerClust_vs_phi");\r
 523   aFolderObj->Add(h2D);\r
 524 \r
 525   h2D = (TH2D *)fRecMCTrackHist->Projection(2,6);\r
 526   h2D->SetName("ratio_nClust_nFindableClust_vs_phi");\r
 527   aFolderObj->Add(h2D);\r
 528 \r
 529   //\r
 530   fRecMCTrackHist->GetAxis(7)->SetRangeUser(0.0,maxPt);  \r
 531 \r
 532   h2D = (TH2D *)fRecMCTrackHist->Projection(0,7);\r
 533   h2D->SetName("nClust_vs_pt");\r
 534   aFolderObj->Add(h2D);\r
 535 \r
 536   h2D = (TH2D *)fRecMCTrackHist->Projection(1,7);\r
 537   h2D->SetName("chi2PerClust_vs_pt");\r
 538   aFolderObj->Add(h2D);\r
 539 \r
 540   h2D = (TH2D *)fRecMCTrackHist->Projection(2,7);\r
 541   h2D->SetName("ratio_nClust_nFindableClust_vs_pt");\r
 542   aFolderObj->Add(h2D);\r
 543 \r
 544   h2D = (TH2D *)fRecMCTrackHist->Projection(6,7);\r
 545   h2D->SetName("phi_vs_pt");\r
 546   aFolderObj->Add(h2D);\r
 547 \r
 548 \r
 549   // fiducial volume\r
 550   fRecMCTrackHist->GetAxis(5)->SetRangeUser(minEta,maxEta);  \r
 551   fRecMCTrackHist->GetAxis(7)->SetRangeUser(minPt,maxPt);  \r
 552 \r
 553   // DCA cuts\r
 554   fRecMCTrackHist->GetAxis(3)->SetRangeUser(-maxDCAr,maxDCAr);\r
 555   fRecMCTrackHist->GetAxis(4)->SetRangeUser(-maxDCAr,maxDCAr);\r
 556 \r
 557   h2D = (TH2D *)fRecMCTrackHist->Projection(0,1);\r
 558   h2D->SetName("nClust_vs_chi2PerClust");\r
 559   aFolderObj->Add(h2D);\r
 560 \r
 561   h2D = (TH2D *)fRecMCTrackHist->Projection(0,2);\r
 562   h2D->SetName("nClust_vs_ratio_nClust_nFindableClust");\r
 563   aFolderObj->Add(h2D);\r
 564 \r
 565   //\r
 566   // DCAy cuts\r
 567   //\r
 568   fRecMCTrackHist->GetAxis(0)->SetRange(50,160); // nClust/track > 50\r
 569   fRecMCTrackHist->GetAxis(1)->SetRangeUser(0.,3.9999); // chi2/cluster < 4.0\r
 570   fRecMCTrackHist->GetAxis(3)->SetRange(1,fRecMCTrackHist->GetAxis(3)->GetNbins());\r
 571   //fRecMCTrackHist->GetAxis(4)->SetRangeUser(-1.0,1.0);\r
 572   fRecMCTrackHist->GetAxis(4)->SetRange(1,fRecMCTrackHist->GetAxis(4)->GetNbins());\r
 573 \r
 574   // sec\r
 575   fRecMCTrackHist->GetAxis(9)->SetRange(1,1);\r
 576   h1D = (TH1D *)fRecMCTrackHist->Projection(3);\r
 577   h1D->SetName("dcay_sec");\r
 578   aFolderObj->Add(h1D);\r
 579 \r
 580   // prim\r
 581   fRecMCTrackHist->GetAxis(9)->SetRange(2,2);\r
 582   h1D = (TH1D *)fRecMCTrackHist->Projection(3);\r
 583   h1D->SetName("dcay_prim");\r
 584   aFolderObj->Add(h1D);\r
 585 \r
 586   // DCAz cuts\r
 587   //fRecMCTrackHist->GetAxis(3)->SetRangeUser(-1.0,1.0);\r
 588   fRecMCTrackHist->GetAxis(4)->SetRange(1,fRecMCTrackHist->GetAxis(4)->GetNbins());\r
 589 \r
 590   // sec\r
 591   fRecMCTrackHist->GetAxis(9)->SetRange(1,1);\r
 592   h1D = (TH1D *)fRecMCTrackHist->Projection(4);\r
 593   h1D->SetName("dcaz_sec");\r
 594   aFolderObj->Add(h1D);\r
 595 \r
 596   // prim\r
 597   fRecMCTrackHist->GetAxis(9)->SetRange(2,2);\r
 598   h1D = (TH1D *)fRecMCTrackHist->Projection(4);\r
 599   h1D->SetName("dcaz_prim");\r
 600   aFolderObj->Add(h1D);\r
 601 \r
 602 \r
 603   // export objects to analysis folder\r
 604   fAnalysisFolder = ExportToFolder(aFolderObj);\r
 605 \r
 606   // delete only TObjArray\r
 607   if(aFolderObj) delete aFolderObj;\r
 608 }\r
 609 \r
 610 //_____________________________________________________________________________\r
 611 TFolder* AlidNdPtCutAnalysis::ExportToFolder(TObjArray * array) \r
 612 {\r
 613   // recreate folder avery time and export objects to new one\r
 614   //\r
 615   AlidNdPtCutAnalysis * comp=this;\r
 616   TFolder *folder = comp->GetAnalysisFolder();\r
 617 \r
 618   TString name, title;\r
 619   TFolder *newFolder = 0;\r
 620   Int_t i = 0;\r
 621   Int_t size = array->GetSize();\r
 622 \r
 623   if(folder) { \r
 624      // get name and title from old folder\r
 625      name = folder->GetName();  \r
 626      title = folder->GetTitle();  \r
 627 \r
 628          // delete old one\r
 629      delete folder;\r
 630 \r
 631          // create new one\r
 632      newFolder = CreateFolder(name.Data(),title.Data());\r
 633      newFolder->SetOwner();\r
 634 \r
 635          // add objects to folder\r
 636      while(i < size) {\r
 637            newFolder->Add(array->At(i));\r
 638            i++;\r
 639          }\r
 640   }\r
 641 \r
 642 return newFolder;\r
 643 }\r
 644 \r
 645 //_____________________________________________________________________________\r
 646 TFolder* AlidNdPtCutAnalysis::CreateFolder(TString name,TString title) { \r
 647 // create folder for analysed histograms\r
 648 //\r
 649 TFolder *folder = 0;\r
 650   folder = new TFolder(name.Data(),title.Data());\r
 651 \r
 652   return folder;\r
 653 }\r