PWG1/AliComparisonDCA.cxx

   1 //------------------------------------------------------------------------------
   2 // Implementation of AliComparisonDCA class. It keeps information from
   3 // comparison of reconstructed and MC particle tracks. In addtion,
   4 // it keeps selection cuts used during comparison. The comparison
   5 // information is stored in the ROOT histograms. Analysis of these
   6 // histograms can be done by using Analyse() class function. The result of
   7 // the analysis (histograms/graphs) are stored in the folder
   8 // which is a data member of AliComparisonDCA.
   9 //
  10 // Author: J.Otwinowski 04/02/2008
  11 //------------------------------------------------------------------------------
  12
  13 /*
  14
  15   // after running comparison task, read the file, and get component
  16   gROOT->LoadMacro("$ALICE_ROOT/PWG1/Macros/LoadMyLibs.C");
  17   LoadMyLibs();
  18   TFile f("Output.root");
  19   //AliComparisonDCA * compObj = (AliComparisonDCA*)f.Get("AliComparisonDCA");
  20   AliComparisonDCA * compObj = (AliComparisonDCA*)cOutput->FindObject("AliComparisonDCA");
  21
  22   // Analyse comparison data
  23   compObj->Analyse();
  24
  25   // the output histograms/graphs will be stored in the folder "folderDCA"
  26   compObj->GetAnalysisFolder()->ls("*");
  27
  28   // user can save whole comparison object (or only folder with anlysed histograms)
  29   // in the seperate output file (e.g.)
  30   TFile fout("Analysed_DCA.root","recreate");
  31   compObj->Write(); // compObj->GetAnalysisFolder()->Write();
  32   fout.Close();
  33
  34 */
  35
  36 #include <TAxis.h>
  37 #include <TCanvas.h>
  38 #include <TGraph.h>
  39 #include <TGraph2D.h>
  40 #include <TH1.h>
  41
  42 #include "AliComparisonDCA.h"
  43 #include "AliESDEvent.h"
  44 #include "AliESDRecInfo.h"
  45 #include "AliESDVertex.h"
  46 #include "AliLog.h"
  47 #include "AliMCInfo.h"
  48 #include "AliMCInfoCuts.h"
  49 #include "AliMathBase.h"
  50 #include "AliRecInfoCuts.h"
  51 #include "AliTracker.h"
  52
  53 using namespace std;
  54
  55 ClassImp(AliComparisonDCA)
  56
  57 //_____________________________________________________________________________
  58 AliComparisonDCA::AliComparisonDCA():
  59   AliComparisonObject("AliComparisonDCA"),
  60
  61   // DCA histograms
  62   fDCAHisto(0),
  63   /*
  64   fD0TanSPtTPCITS(0),
  65   fD1TanSPtTPCITS(0),
  66   fD0TanSPt(0),
  67   fD1TanSPt(0),
  68   fD0TanSPtTPC(0),
  69   fD1TanSPtTPC(0),
  70   */
  71
  72   // Cuts
  73   fCutsRC(0),
  74   fCutsMC(0),
  75
  76   // histogram folder
  77   fAnalysisFolder(0)
  78 {
  79   // default constructor
  80 }
  81
  82 //_____________________________________________________________________________
  83 AliComparisonDCA::AliComparisonDCA(Char_t* name="AliComparisonDCA", Char_t* title="AliComparisonDCA",Int_t analysisMode=0, Bool_t hptGenerator=kFALSE):
  84   AliComparisonObject(name,title),
  85
  86   // DCA histograms
  87   fDCAHisto(0),
  88   /*
  89   fD0TanSPtTPCITS(0),
  90   fD1TanSPtTPCITS(0),
  91   fD0TanSPt(0),
  92   fD1TanSPt(0),
  93   fD0TanSPtTPC(0),
  94   fD1TanSPtTPC(0),
  95   */
  96
  97   // Cuts
  98   fCutsRC(0),
  99   fCutsMC(0),
 100
 101   // histogram folder
 102   fAnalysisFolder(0)
 103 {
 104   // named constructor
 105
 106   SetAnalysisMode(analysisMode);
 107   SetHptGenerator(hptGenerator);
 108   Init();
 109 }
 110
 111
 112 //_____________________________________________________________________________
 113 AliComparisonDCA::~AliComparisonDCA()
 114 {
 115   // destructor
 116   if(fDCAHisto)  delete fDCAHisto; fDCAHisto=0;
 117   /*
 118   if(fD0TanSPtTPCITS) delete fD0TanSPtTPCITS; fD0TanSPtTPCITS=0;
 119   if(fD1TanSPtTPCITS) delete fD1TanSPtTPCITS; fD1TanSPtTPCITS=0;
 120   if(fD0TanSPt) delete fD0TanSPt; fD0TanSPt=0;
 121   if(fD1TanSPt) delete fD1TanSPt; fD1TanSPt=0;
 122   if(fD0TanSPtTPC) delete fD0TanSPtTPC; fD0TanSPtTPC=0;
 123   if(fD1TanSPtTPC) delete fD1TanSPtTPC; fD1TanSPtTPC=0;
 124   */
 125   if(fAnalysisFolder) delete fAnalysisFolder; fAnalysisFolder=0;
 126
 127 }
 128
 129 //_____________________________________________________________________________
 130 void AliComparisonDCA::Init()
 131 {
 132   // DCA histograms
 133
 134  Int_t nPBins = 31;
 135     Double_t binsP[32] = {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.7,0.8,0.9,1.0,1.2,1.4,1.6,1.8,2.0,2.25,2.5,2.75,3.,3.5,4.,5.,6.,8.,10.};
 136     Double_t pMin = 0., pMax = 10.;
 137
 138     if(IsHptGenerator() == kTRUE) {
 139       nPBins = 100;
 140       pMin = 0.; pMax = 100.;
 141     }
 142
 143    //dca_r, dca_z, eta, pt
 144    Int_t binsQA[4]    = {100,100,20,nPBins};
 145    Double_t xminQA[4] = {-10.,-10.,-1., pMin};
 146    Double_t xmaxQA[4] = {10.,10.,1., pMax};
 147
 148    fDCAHisto = new THnSparseF("fDCAHisto","dca_r:dca_z:eta:pt",4,binsQA,xminQA,xmaxQA);
 149    if(!IsHptGenerator()) fDCAHisto->SetBinEdges(3,binsP);
 150
 151    fDCAHisto->GetAxis(0)->SetTitle("dca_r (cm)");
 152    fDCAHisto->GetAxis(1)->SetTitle("dca_z (cm)");
 153    fDCAHisto->GetAxis(2)->SetTitle("eta");
 154    fDCAHisto->GetAxis(3)->SetTitle("pt (GeV/c)");
 155    fDCAHisto->Sumw2();
 156
 157   /*
 158   fD0TanSPtTPCITS = new TH3F("DCAyTanSPtTPCITS","DCAyTanSPt",40,-2,2, 10,0.3,3, 100,-1,1);
 159   fD0TanSPtTPCITS->SetXTitle("tan(#theta)");
 160   fD0TanSPtTPCITS->SetYTitle("#sqrt{p_{t}(GeV)}");
 161   fD0TanSPtTPCITS->SetZTitle("DCA_{xy}");
 162
 163   fD1TanSPtTPCITS = new TH3F("DCAzTanSPtTPCITS","DCAzTanSPt",40,-2,2, 10,0.3,3, 100,-1,1);
 164   fD1TanSPtTPCITS->SetXTitle("tan(#theta)");
 165   fD1TanSPtTPCITS->SetYTitle("#sqrt(p_{t}(GeV))");
 166   fD1TanSPtTPCITS->SetZTitle("DCA_{z}");
 167
 168   fD0TanSPt = new TH3F("DCAyTanSPt","DCAyTanSPt",40,-2,2, 10,0.3,3, 100,-1,1);
 169   fD0TanSPt->SetXTitle("tan(#theta)");
 170   fD0TanSPt->SetYTitle("#sqrt{p_{t}(GeV)}");
 171   fD0TanSPt->SetZTitle("DCA_{xy}");
 172
 173   fD1TanSPt = new TH3F("DCAzTanSPt","DCAzTanSPt",40,-2,2, 10,0.3,3, 100, -1,1);
 174   fD1TanSPt->SetXTitle("tan(#theta)");
 175   fD1TanSPt->SetYTitle("#sqrt{p_{t}(GeV)}");
 176   fD1TanSPt->SetZTitle("DCA_{z}");
 177
 178   fD0TanSPtTPC = new TH3F("DCAyTanSPtTPC","DCAyTanSPt",40,-2,2, 10,0.3,3, 100,-1,1);
 179   fD0TanSPtTPC->SetXTitle("tan(#theta)");
 180   fD0TanSPtTPC->SetYTitle("#sqrt{p_{t}(GeV)}");
 181   fD0TanSPtTPC->SetZTitle("DCA_{xy}");
 182
 183   fD1TanSPtTPC = new TH3F("DCAzTanSPtTPC","DCAzTanSPt",40,-2,2, 10,0.3,3, 100, -1,1);
 184   fD1TanSPtTPC->SetXTitle("tan(#theta)");
 185   fD1TanSPtTPC->SetYTitle("#sqrt{p_{t}(GeV)}");
 186   fD1TanSPtTPC->SetZTitle("DCA_{z}");
 187   */
 188
 189   // init cuts
 190   if(!fCutsMC)
 191     AliDebug(AliLog::kError, "ERROR: Cannot find AliMCInfoCuts object");
 192   if(!fCutsRC)
 193     AliDebug(AliLog::kError, "ERROR: Cannot find AliRecInfoCuts object");
 194
 195   // init folder
 196   fAnalysisFolder = CreateFolder("folderDCA","Analysis DCA Folder");
 197 }
 198
 199 //_____________________________________________________________________________
 200 void AliComparisonDCA::ProcessTPC(AliMCInfo* const infoMC, AliESDRecInfo * const infoRC)
 201 {
 202   // Fill DCA comparison information
 203   AliExternalTrackParam *track = 0;
 204   Double_t field      = AliTracker::GetBz(); // nominal Bz field [kG]
 205   Double_t kMaxD      = 123456.0; // max distance
 206
 207   Double_t dca[2], cov[3]; // dca_xy, dca_z, sigma_xy, sigma_xy_z, sigma_z
 208
 209   Float_t mcpt = infoMC->GetParticle().Pt();
 210   Float_t mceta = infoMC->GetParticle().Eta();
 211   //Float_t spt = TMath::Sqrt(mcpt);
 212
 213   // distance to Prim. vertex
 214   const Double_t* dv = infoMC->GetVDist();
 215
 216   Bool_t isPrim = TMath::Sqrt(dv[0]*dv[0] + dv[1]*dv[1])<fCutsMC->GetMaxR() && TMath::Abs(dv[2])<fCutsMC->GetMaxVz();
 217
 218   // Check selection cuts
 219   if (fCutsMC->IsPdgParticle(TMath::Abs(infoMC->GetParticle().GetPdgCode())) == kFALSE) return;
 220   if (!isPrim) return;
 221   if (infoRC->GetStatus(1)!=3) return;
 222   if (!infoRC->GetESDtrack()) return;
 223   if (infoRC->GetESDtrack()->GetTPCNcls()<fCutsRC->GetMinNClustersTPC()) return;
 224   //if (!infoRC->GetESDtrack()->GetConstrainedParam()) return;
 225
 226   // calculate and set prim. vertex
 227   AliESDVertex vertexMC;
 228   vertexMC.SetXv( infoMC->GetParticle().Vx() - dv[0] );
 229   vertexMC.SetYv( infoMC->GetParticle().Vy() - dv[1] );
 230   vertexMC.SetZv( infoMC->GetParticle().Vz() - dv[2] );
 231
 232   // calculate track parameters at vertex
 233   if (infoRC->GetESDtrack()->GetTPCInnerParam())
 234   {
 235     if ((track = new AliExternalTrackParam(*infoRC->GetESDtrack()->GetTPCInnerParam())) != 0 )
 236     {
 237       Bool_t bDCAStatus = track->PropagateToDCA(&vertexMC,field,kMaxD,dca,cov);
 238
 239       if(bDCAStatus) {
 240          Double_t vDCAHisto[4]={dca[0],dca[1],mceta,mcpt};
 241          fDCAHisto->Fill(vDCAHisto);
 242       }
 243     delete track;
 244     }
 245   }
 246 }
 247
 248 //_____________________________________________________________________________
 249 void AliComparisonDCA::ProcessTPCITS(AliMCInfo* const infoMC, AliESDRecInfo * const infoRC)
 250 {
 251   // Fill DCA comparison information
 252   Int_t clusterITS[200];
 253   Float_t dca[2], cov[3]; // dca_xy, dca_z, sigma_xy, sigma_xy_z, sigma_z
 254
 255   Float_t mcpt = infoMC->GetParticle().Pt();
 256   Float_t mceta = infoMC->GetParticle().Eta();
 257   //Float_t spt = TMath::Sqrt(mcpt);
 258
 259   // distance to Prim. vertex
 260   const Double_t* dv = infoMC->GetVDist();
 261   Bool_t isPrim = TMath::Sqrt(dv[0]*dv[0] + dv[1]*dv[1])<fCutsMC->GetMaxR() && TMath::Abs(dv[2])<fCutsMC->GetMaxVz();
 262
 263   // Check selection cuts
 264   if (fCutsMC->IsPdgParticle(TMath::Abs(infoMC->GetParticle().GetPdgCode())) == kFALSE) return;
 265   if (!isPrim) return;
 266   if (infoRC->GetStatus(1)!=3) return;
 267   if (!infoRC->GetESDtrack()) return;
 268   if (infoRC->GetESDtrack()->GetTPCNcls()<fCutsRC->GetMinNClustersTPC()) return;
 269   //if (!infoRC->GetESDtrack()->GetConstrainedParam()) return;
 270
 271   infoRC->GetESDtrack()->GetImpactParameters(dca,cov);
 272
 273   // ITS + TPC
 274   if(infoRC->GetESDtrack()->GetITSclusters(clusterITS)>fCutsRC->GetMinNClustersITS())
 275   {
 276     Double_t vDCAHisto[4]={dca[0],dca[1],mceta,mcpt};
 277     fDCAHisto->Fill(vDCAHisto);
 278   }
 279 }
 280
 281 void AliComparisonDCA::ProcessConstrained(AliMCInfo* const /*infoMC*/, AliESDRecInfo * const /*infoRC*/)
 282 {
 283   // Fill DCA comparison information
 284
 285   AliDebug(AliLog::kWarning, "Warning: Not implemented");
 286 }
 287
 288 //_____________________________________________________________________________
 289 Long64_t AliComparisonDCA::Merge(TCollection* const list)
 290 {
 291   // Merge list of objects (needed by PROOF)
 292
 293   if (!list)
 294   return 0;
 295
 296   if (list->IsEmpty())
 297   return 1;
 298
 299   TIterator* iter = list->MakeIterator();
 300   TObject* obj = 0;
 301
 302   // collection of generated histograms
 303   Int_t count=0;
 304   while((obj = iter->Next()) != 0)
 305   {
 306     AliComparisonDCA* entry = dynamic_cast<AliComparisonDCA*>(obj);
 307     if (entry == 0) continue;
 308
 309     fDCAHisto->Add(entry->fDCAHisto);
 310     /*
 311     fD0TanSPtTPCITS->Add(entry->fD0TanSPtTPCITS);
 312     fD1TanSPtTPCITS->Add(entry->fD1TanSPtTPCITS);
 313     fD0TanSPt->Add(entry->fD0TanSPt);
 314     fD1TanSPt->Add(entry->fD1TanSPt);
 315     fD0TanSPtTPC->Add(entry->fD0TanSPtTPC);
 316     fD1TanSPtTPC->Add(entry->fD1TanSPtTPC);
 317     */
 318
 319     count++;
 320   }
 321
 322 return count;
 323 }
 324
 325 //_____________________________________________________________________________
 326 void AliComparisonDCA::Exec(AliMCInfo* const infoMC, AliESDRecInfo * const infoRC)
 327 {
 328   // Process comparison information
 329   if(GetAnalysisMode() == 0) ProcessTPC(infoMC,infoRC);
 330   else if(GetAnalysisMode() == 1) ProcessTPCITS(infoMC,infoRC);
 331   else if(GetAnalysisMode() == 2) ProcessConstrained(infoMC,infoRC);
 332   else {
 333     printf("ERROR: AnalysisMode %d \n",fAnalysisMode);
 334     return;
 335   }
 336 }
 337
 338 //_____________________________________________________________________________
 339 void AliComparisonDCA::Analyse()
 340 {
 341   //
 342   // Analyse comparison information and store output histograms
 343   // in the analysis folder "folderDCA"
 344   //
 345
 346   TH1::AddDirectory(kFALSE);
 347   TObjArray *aFolderObj = new TObjArray;
 348
 349   /*
 350   TGraph * gr[8]= { 0,0,0,0,0,0,0,0 };
 351   TGraph2D *gr2[8]= { 0,0,0,0,0,0,0,0};
 352   AliComparisonDCA * comp=this;
 353
 354   // write results in the folder
 355   // Canvas to draw analysed histograms
 356   TCanvas * c = new TCanvas("canDCA","DCA resolution");
 357   c->Divide(2,4);
 358   //
 359   // DCA resolution
 360   //
 361   c->cd(1);
 362   gr[0] = AliMathBase::MakeStat1D(comp->fD0TanSPtTPC,2,5);
 363   gr[0]->GetXaxis()->SetTitle("Tan(#theta)");
 364   gr[0]->GetYaxis()->SetTitle("#sigmaDCA_xy (cm)");
 365   gr[0]->SetName("DCAXYResolTanTPC");
 366   gr[0]->SetTitle("resol. DCA_xy (TPC only)");
 367   gr[0]->Draw("Al*");
 368
 369   aFolderObj->Add(gr[0]);
 370
 371   c->cd(2);
 372   gr[1] = AliMathBase::MakeStat1D(comp->fD1TanSPtTPC,2,5);
 373   gr[1]->GetXaxis()->SetTitle("Tan(#theta)");
 374   gr[1]->GetYaxis()->SetTitle("#sigmaDCA_z (cm)");
 375   gr[1]->SetName("DCAZResolTanTPC");
 376   gr[1]->SetTitle("resol. DCA_z (TPC only)");
 377   gr[1]->Draw("Al*");
 378
 379   aFolderObj->Add(gr[1]);
 380
 381   c->cd(3);
 382   gr[2] = AliMathBase::MakeStat1D(comp->fD0TanSPtTPCITS,2,5);
 383   gr[2]->GetXaxis()->SetTitle("Tan(#theta)");
 384   gr[2]->GetYaxis()->SetTitle("#sigmaDCA_xy (cm)");
 385   gr[2]->SetName("DCAXYResolTanTPCITS");
 386   gr[2]->SetTitle("resol. DCA_xy (TPC+ITS)");
 387   gr[2]->Draw("Al*");
 388
 389   aFolderObj->Add(gr[2]);
 390
 391   c->cd(4);
 392   gr[3] = AliMathBase::MakeStat1D(comp->fD1TanSPtTPCITS,2,5);
 393   gr[3]->GetXaxis()->SetTitle("Tan(#theta)");
 394   gr[3]->GetYaxis()->SetTitle("#sigmaDCA_z (cm)");
 395   gr[3]->SetName("DCAZResolTanTPCITS");
 396   gr[3]->SetTitle("resol. DCA_z (TPC+ITS)");
 397   gr[3]->Draw("Al*");
 398
 399   aFolderObj->Add(gr[3]);
 400
 401   //
 402   // DCA mean value
 403   //
 404   c->cd(5);
 405   gr[4] = AliMathBase::MakeStat1D(comp->fD0TanSPtTPC,2,4);
 406   gr[4]->GetXaxis()->SetTitle("Tan(#theta)");
 407   gr[4]->GetYaxis()->SetTitle("mean DCA_xy (cm)");
 408   gr[4]->SetName("DCAXYMeanTanTPC");
 409   gr[4]->SetTitle("mean DCA_xy (TPC only)");
 410   gr[4]->Draw("Al*");
 411
 412   aFolderObj->Add(gr[4]);
 413
 414   c->cd(6);
 415   gr[5] = AliMathBase::MakeStat1D(comp->fD1TanSPtTPC,2,4);
 416   gr[5]->GetXaxis()->SetTitle("Tan(#theta)");
 417   gr[5]->GetYaxis()->SetTitle("mean DCA_z (cm)");
 418   gr[5]->SetName("DCAZMeanTanTPC");
 419   gr[5]->SetTitle("mean DCA_z (TPC only)");
 420   gr[5]->Draw("Al*");
 421
 422   aFolderObj->Add(gr[5]);
 423
 424   c->cd(7);
 425   gr[6] = AliMathBase::MakeStat1D(comp->fD0TanSPtTPCITS,2,4);
 426   gr[6]->GetXaxis()->SetTitle("Tan(#theta)");
 427   gr[6]->GetYaxis()->SetTitle("mean DCA_xy (cm)");
 428   gr[6]->SetName("DCAXYMeanTanTPCITS");
 429   gr[6]->SetTitle("mean DCA_xy (TPC+ITS)");
 430   gr[6]->Draw("Al*");
 431
 432   aFolderObj->Add(gr[6]);
 433
 434   c->cd(8);
 435   gr[7] = AliMathBase::MakeStat1D(comp->fD1TanSPtTPCITS,2,4);
 436   gr[7]->GetXaxis()->SetTitle("Tan(#theta)");
 437   gr[7]->GetYaxis()->SetTitle("mean DCA_z (cm)");
 438   gr[7]->SetName("DCAZMeanTanTPCITS");
 439   gr[7]->SetTitle("mean DCA_z (TPC+ITS)");
 440   gr[7]->Draw("Al*");
 441
 442   aFolderObj->Add(gr[7]);
 443
 444   // 2D DCA resolution
 445   TCanvas * c1 = new TCanvas("canDCA1","2D DCA resolution");
 446   c1->Divide(2,4);
 447
 448   // TPC only
 449   c1->cd(1);
 450   gr2[0] = AliMathBase::MakeStat2D(comp->fD0TanSPtTPC,4,2,5);
 451   gr2[0]->GetXaxis()->SetTitle("Tan(#theta)");
 452   gr2[0]->GetYaxis()->SetTitle("#sqrt{p_{t}(GeV)}");
 453   gr2[0]->GetZaxis()->SetTitle("#sigmaDCA_xy (cm)");
 454   gr2[0]->SetName("DCAXYResolSPTTanTPC");
 455   gr2[0]->SetTitle("#sigma DCA_xy (TPC only)");
 456   gr2[0]->GetHistogram()->Draw("colz");
 457
 458   gr2[0]->GetHistogram()->SetName("DCAXYResolSPTTanTPC");
 459   aFolderObj->Add(gr2[0]->GetHistogram());
 460
 461   c1->cd(2);
 462   gr2[1] = AliMathBase::MakeStat2D(comp->fD1TanSPtTPC,4,2,5);
 463   gr2[1]->GetXaxis()->SetTitle("Tan(#theta)");
 464   gr2[1]->GetYaxis()->SetTitle("#sqrt{p_{t}(GeV)}");
 465   gr2[1]->GetZaxis()->SetTitle("#sigmaDCA_z (cm)");
 466   gr2[1]->SetName("DCAZResolSPTTanTPC");
 467   gr2[1]->SetTitle("#sigma DCA_z (TPC only)");
 468   gr2[1]->GetHistogram()->Draw("colz");
 469
 470   gr2[1]->GetHistogram()->SetName("DCAZResolSPTTanTPC");
 471   aFolderObj->Add(gr2[1]->GetHistogram());
 472
 473   // TPC+ITS
 474   c1->cd(3);
 475   gr2[2] = AliMathBase::MakeStat2D(comp->fD0TanSPtTPCITS,4,2,5);
 476   gr2[2]->GetXaxis()->SetTitle("Tan(#theta)");
 477   gr2[2]->GetYaxis()->SetTitle("#sqrt{p_{t}(GeV)}");
 478   gr2[2]->GetZaxis()->SetTitle("#sigmaDCA_xy (cm)");
 479   gr2[2]->SetName("DCAXYResolSPTTanTPCITS");
 480   gr2[2]->SetTitle("#sigma DCA_xy (TPC+ITS)");
 481   gr2[2]->GetHistogram()->Draw("colz");
 482
 483   gr2[2]->GetHistogram()->SetName("DCAXYResolSPTTanTPCITS");
 484   aFolderObj->Add(gr2[2]->GetHistogram());
 485
 486   c1->cd(4);
 487   gr2[3] = AliMathBase::MakeStat2D(comp->fD1TanSPtTPCITS,4,2,5);
 488   gr2[3]->GetXaxis()->SetTitle("Tan(#theta)");
 489   gr2[3]->GetYaxis()->SetTitle("#sqrt{p_{t}(GeV)}");
 490   gr2[3]->GetZaxis()->SetTitle("#sigmaDCA_z (cm)");
 491   gr2[3]->SetName("DCAZResolSPTTanTPCITS");
 492   gr2[3]->SetTitle("#sigma DCA_z (TPC+ITS)");
 493   gr2[3]->GetHistogram()->Draw("colz");
 494
 495   gr2[3]->GetHistogram()->SetName("DCAZResolSPTTanTPCITS");
 496   aFolderObj->Add(gr2[3]->GetHistogram());
 497
 498   // 2D DCA mean value
 499   c1->cd(5);
 500   gr2[4] = AliMathBase::MakeStat2D(comp->fD0TanSPtTPC,4,2,4);
 501   gr2[4]->GetXaxis()->SetTitle("Tan(#theta)");
 502   gr2[4]->GetYaxis()->SetTitle("#sqrt{p_{t}(GeV)}");
 503   gr2[4]->GetZaxis()->SetTitle("mean DCA_xy (cm)");
 504   gr2[4]->SetName("DCAXYMeanSPTTanTPC");
 505   gr2[4]->SetTitle("mean DCA_xy (TPC only)");
 506   gr2[4]->GetHistogram()->Draw("colz");
 507
 508   gr2[4]->GetHistogram()->SetName("DCAXYMeanSPTTanTPC");
 509   aFolderObj->Add(gr2[4]->GetHistogram());
 510
 511   c1->cd(6);
 512   gr2[5] = AliMathBase::MakeStat2D(comp->fD1TanSPtTPC,4,2,4);
 513   gr2[5]->GetXaxis()->SetTitle("Tan(#theta)");
 514   gr2[5]->GetYaxis()->SetTitle("#sqrt{p_{t}(GeV)}");
 515   gr2[5]->GetZaxis()->SetTitle("mean DCA_z (cm)");
 516   gr2[5]->SetName("DCAZMeanSPTTanTPC");
 517   gr2[5]->SetTitle("mean DCA_z (TPC only)");
 518   gr2[5]->GetHistogram()->Draw("colz");
 519
 520   gr2[5]->GetHistogram()->SetName("DCAZMeanSPTTanTPC");
 521   aFolderObj->Add(gr2[5]->GetHistogram());
 522
 523   c1->cd(7);
 524   gr2[6] = AliMathBase::MakeStat2D(comp->fD0TanSPtTPCITS,4,2,4);
 525   gr2[6]->GetXaxis()->SetTitle("Tan(#theta)");
 526   gr2[6]->GetYaxis()->SetTitle("#sqrt{p_{t}(GeV)}");
 527   gr2[6]->GetZaxis()->SetTitle("mean DCA_xy (cm)");
 528   gr2[6]->SetName("DCAXYMeanSPTTanTPCITS");
 529   gr2[6]->SetTitle("mean DCA_xy (TPC+ITS)");
 530   gr2[6]->GetHistogram()->Draw("colz");
 531
 532   gr2[6]->GetHistogram()->SetName("DCAXYMeanSPTTanTPCITS");
 533   aFolderObj->Add(gr2[6]->GetHistogram());
 534
 535   c1->cd(8);
 536   gr2[7] = AliMathBase::MakeStat2D(comp->fD1TanSPtTPCITS,4,2,4);
 537   gr2[7]->GetXaxis()->SetTitle("Tan(#theta)");
 538   gr2[7]->GetYaxis()->SetTitle("#sqrt{p_{t}(GeV)}");
 539   gr2[7]->GetZaxis()->SetTitle("mean DCA_z (cm)");
 540   gr2[7]->SetName("DCAZMeanSPTTanTPCITS");
 541   gr2[7]->SetTitle("mean DCA_z (TPC+ITS)");
 542   gr2[7]->GetHistogram()->Draw("colz");
 543
 544   gr2[7]->GetHistogram()->SetName("DCAZMeanSPTTanTPCITS");
 545   aFolderObj->Add(gr2[7]->GetHistogram());
 546
 547   */
 548   // export objects to analysis folder
 549   fAnalysisFolder = ExportToFolder(aFolderObj);
 550
 551   // delete only TObjArray
 552   if(aFolderObj) delete aFolderObj;
 553 }
 554
 555 //_____________________________________________________________________________
 556 TFolder* AliComparisonDCA::ExportToFolder(TObjArray * array)
 557 {
 558   // recreate folder avery time and export objects to new one
 559   //
 560   AliComparisonDCA * comp=this;
 561   TFolder *folder = comp->GetAnalysisFolder();
 562
 563   TString name, title;
 564   TFolder *newFolder = 0;
 565   Int_t i = 0;
 566   Int_t size = array->GetSize();
 567
 568   if(folder) {
 569      // get name and title from old folder
 570      name = folder->GetName();
 571      title = folder->GetTitle();
 572
 573          // delete old one
 574      delete folder;
 575
 576          // create new one
 577      newFolder = CreateFolder(name.Data(),title.Data());
 578      newFolder->SetOwner();
 579
 580          // add objects to folder
 581      while(i < size) {
 582            newFolder->Add(array->At(i));
 583            i++;
 584          }
 585   }
 586
 587 return newFolder;
 588 }
 589
 590
 591 //_____________________________________________________________________________
 592 TFolder* AliComparisonDCA::CreateFolder(TString name,TString title) {
 593 // create folder for analysed histograms
 594 TFolder *folder = 0;
 595   folder = new TFolder(name.Data(),title.Data());
 596
 597   return folder;
 598 }