QA ref defaut storage setter in sim and rec
[u/mrichter/AliRoot.git] / 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 }