]> git.uio.no Git - u/mrichter/AliRoot.git/blame - TPC/AnalyzeESDtracks.C
git://git.uio.no / u / mrichter / AliRoot.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
Install a new script
[u/mrichter/AliRoot.git] / TPC / AnalyzeESDtracks.C
CommitLineData
eaad8c01 1//
2// Process ESD tracks -
3// Extract TPC tracks - write them to tree
4//
5/*
6 .L AnalyzeESDtracks.C+
7 .L AliGenInfo.C+
a2411279 8 AnalyzeESDtracks(567); // process tracks
eaad8c01 9 // Tracks are written to the file "TPCtracks.root"
10 // Now yo can analyze it
11 TFile fesd("AliESDs.root");
12 TTree * treeE = (TTree*)fesd.Get("esdTree");
13 TFile f("TPCtracks.root")
14 TTree * tree =(TTree*)f.Get("Tracks");
15 AliComparisonDraw comp;
16 comp->fTree = tree;
17
18*/
19
20//
21
22/*
23.L AnalyzeESDtracks.C+
24.L AliGenInfo.C+
25
a2411279 26//AnalyzeESDtracks(567);
eaad8c01 27
28
29TFile fesd("AliESDs.root");
30TTree * treeE = (TTree*)fesd.Get("esdTree");
31TFile f("TPCtracks.root")
32TTree * tree =(TTree*)f.Get("Tracks");
33AliComparisonDraw comp;
34comp->fTree = tree;
35
36TFile fs("TPCsignal.root");
37TTree *treeB =(TTree*)fs.Get("SignalB");
38TTree *treeN =(TTree*)fs.Get("SignalN");
39TTree *treeS =(TTree*)fs.Get("Signal");
40TTree *treef =(TTree*)fs.Get("Fit");
41
42
43FitSignals(treeB,"Max-Median>100&&RMS06<2.5")
44TFile ffit("FitSignal.root");
45TTree * treeF = (TTree*)ffit->Get("Fit");
46
47
48TChain chaincl("TreeR","TreeR")
49chaincl.Add("TPC.RecPoints.root/Event0/TreeR")
50chaincl.Add("TPC.RecPoints1.root/Event1/TreeR")
51chaincl.Add("TPC.RecPoints2.root/Event2/TreeR")
52chaincl.Add("TPC.RecPoints3.root/Event3/TreeR")
53
54*/
55
56
57
58#include "TObject.h"
59#include "TFile.h"
60#include "TTree.h"
a82f6396 61#include "TChain.h"
eaad8c01 62#include "TBranch.h"
63#include "TTreeStream.h"
64#include "TEventList.h"
65#include "TCut.h"
66#include "TFitter.h"
67#include "TMatrixD.h"
663296cb 68#include "TRobustEstimator.h"
b78c817f 69#include "TTimeStamp.h"
eaad8c01 70
71#include "AliLog.h"
72#include "AliMagF.h"
73
74#include "AliESD.h"
75#include "AliESDfriend.h"
76#include "AliESDtrack.h"
77#include "AliTracker.h"
78#include "AliTPCseed.h"
79#include "AliTPCclusterMI.h"
663296cb 80#include "AliTPCParamSR.h"
81#include "AliTPCROC.h"
82
83
eaad8c01 84#include "TTreeStream.h"
85#include "TF1.h"
86#include "TGraph.h"
87#include "AliSignalProcesor.h"
88#include "TCanvas.h"
89
90
663296cb 91void FitSignals(TTree * treeB, TCut cut="Max-Median>150&&RMS06<2&&abs(Median-Mean09)<0.5", Int_t maxS=100);
ba6ac751 92void LaserCalib(TTreeSRedirector & cstream, TTree * chain, Float_t tmin, Float_t tmax, Float_t fraction=0.7);
de31eda2 93
a82f6396 94void AnalyzeESDtracks(Int_t run){
eaad8c01 95 //
96 // output redirect
a2411279 97 TTreeSRedirector * pcstream = new TTreeSRedirector("TPCtracks.root");
98 TTreeSRedirector &cstream = *pcstream;
eaad8c01 99 //
eaad8c01 100 TFile f("AliESDs.root");
101 TTree * tree =(TTree*)f.Get("esdTree");
102 AliESD * esd =0;
103 tree->SetBranchAddress("ESD",&esd);
104 AliESDfriend *evf=0;
105 tree->AddFriend("esdFriendTree","AliESDfriends.root");
106 tree->SetBranchAddress("ESDfriend",&evf);
107
108 Int_t nevents = tree->GetEntries();
109 TClonesArray *clusters = new TClonesArray("AliTPCclusterMI",160);
110 for (Int_t irow=0; irow<160; irow++){
111 new ((*clusters)[irow]) AliTPCclusterMI; // iitial dummy clusters
112 }
113
114 for (Int_t ievent=0; ievent<nevents; ievent++){
115 tree->GetEntry(ievent);
116 if (!esd) continue;
117 if (!evf) continue;
118 esd->SetESDfriend(evf); //Attach the friend to the ESD
119 for (Int_t itrack =0; itrack<esd->GetNumberOfTracks(); itrack++){
120 // Int_t itrack = 0;
121 if (esd->GetTrack(itrack)->GetFriendTrack()==0) continue;
122 AliESDtrack * etrack = esd->GetTrack(itrack);
123 AliESDfriendTrack * ftrack = (AliESDfriendTrack *)esd->GetTrack(itrack)->GetFriendTrack();
124 AliTPCseed * seed = (AliTPCseed*)(ftrack->GetCalibObject(0));
125 if (!seed) continue;
126 for (Int_t irow=0; irow<160; irow++){
127 if (seed->GetClusterFast(irow)){
128 AliTPCclusterMI * cl = new ((*clusters)[irow]) AliTPCclusterMI(*(seed->GetClusterFast(irow)));
129 cl->SetLabel(itrack,0);
130 }
131 else{
132 AliTPCclusterMI * cl = (AliTPCclusterMI*)clusters->At(irow);
133 cl->SetX(0); cl->SetY(0); cl->SetZ(0); cl->SetQ(0); cl->SetLabel(-1,0);
134 }
135 }
136 Float_t dEdx = seed->GetdEdx();
137 Float_t dEdxI = seed->CookdEdx(0.05,0.6,0,77);
138 Float_t dEdxO = seed->CookdEdx(0.05,0.6,78,155);
139 Int_t ncl = seed->GetNumberOfClusters();
140 cstream<<"Tracks"<<
141 "Run="<<run<<
a82f6396 142 "Ncl="<<ncl<<
eaad8c01 143 "Event="<<ievent<<
144 "dEdx="<<dEdx<<
145 "dEdxI="<<dEdxI<<
146 "dEdxO="<<dEdxO<<
147 "Track.="<<seed<<
148 "Etrack.="<<etrack<<
149 "Cl.="<<clusters<<
150 "\n";
151 }
152 }
a2411279 153 delete pcstream;
a82f6396 154 //
155 // Fit signal part
156 //
157 TFile fs("TPCsignal.root");
158 TTree *treeB =(TTree*)fs.Get("SignalB");
663296cb 159 //
160 // Fit central electrode part
161 //
162 TTreeSRedirector * pcestream = new TTreeSRedirector("TimeRoot.root");
163 TTree * treece = (TTree*)fs.Get("Signalce");
164 if (tree) {
b78c817f 165 LaserCalib(*pcestream, treece, 800,1000, 0.7);
663296cb 166 delete pcestream;
167 }
ba6ac751 168 FitSignals(treeB,"Max-Median>150&&RMS06<1.0&&RMS09<1.5&&abs(Median-Mean09)<0.2&&abs(Mean06-Mean09)<0.2",1000);
663296cb 169 //
eaad8c01 170}
171
172
663296cb 173void FitSignals(TTree * treeB, TCut cut, Int_t max){
eaad8c01 174 AliSignalProcesor proc;
175 TF1 * f1 = proc.GetAsymGauss();
176 TTreeSRedirector cstream("FitSignal.root");
177 TFile *f = cstream.GetFile();
178
179 char lname[100];
180 sprintf(lname,"Fit%s", cut.GetTitle());
181 TEventList *list = new TEventList(lname,lname);
182 sprintf(lname,">>Fit%s", cut.GetTitle());
183 treeB->Draw(lname,cut);
184 treeB->SetEventList(list);
663296cb 185 Int_t nFits=0;
eaad8c01 186 for (Int_t ievent=0; ievent<list->GetN(); ievent++){
663296cb 187 if (nFits>max) break;
188 if (nFits%50==0) printf("%d\n",nFits);
eaad8c01 189 char ename[100];
190 sprintf(ename,"Fit%d", ievent);
191 Double_t nsample = treeB->Draw("Graph.fY-Mean09:Graph.fX","","",1,ievent);
192 Double_t * signal = treeB->GetV1();
193 Double_t * time = treeB->GetV2();
194 Double_t maxpos =0;
195 Double_t max = 0;
196 for (Int_t ipos = 0; ipos<nsample; ipos++){
197 if (signal[ipos]>max){
198 max = signal[ipos];
199 maxpos = ipos;
200 }
201 }
202
203 Int_t first = TMath::Max(maxpos-10,0.);
204 Int_t last = TMath::Min(maxpos+60, nsample);
205 //
206 f->cd();
207 TH1F his(ename,ename,last-first,first,last);
208 for (Int_t ipos=0; ipos<last-first; ipos++){
209 his.SetBinContent(ipos+1,signal[ipos+first]);
210 }
211 treeB->Draw("Sector:Row:Pad","","",1,ievent);
212 Double_t sector = treeB->GetV1()[0];
213 Double_t row = treeB->GetV2()[0];
214 Double_t pad = treeB->GetV3()[0];
215 // TGraph graph(last-first,&time[first],&signal[first]);
216 f1->SetParameters(0.75*max,maxpos,1.1,0.8,0.25,0.2);
217 // TH1F * his = (TH1F*)graph.GetHistogram();
218 his.Fit(f1,"q");
219 his.Write(ename);
220 gPad->Clear();
221 his.Draw();
222 gPad->Update();
223 Double_t params[6];
224 for (Int_t ipar=0; ipar<6; ipar++) params[ipar] = f1->GetParameters()[ipar];
225 Double_t chi2 = TFitter::GetFitter()->Chisquare(6,params);
226 TMatrixD cov(6,6);
227 cov.SetMatrixArray(TFitter::GetFitter()->GetCovarianceMatrix());
228 //
229 // tail cancellation
230 //
231 Double_t x0[1000];
232 Double_t x1[1000];
233 Double_t x2[1000];
234 for (Int_t ipos=0; ipos<last-first; ipos++){
235 x0[ipos] = signal[ipos+first];
236 }
237 proc.TailCancelationALTRO1(x0,x1,0.85*0.339,0.09,last-first);
238 proc.TailCancelationALTRO1(x1,x2,0.85,0.789,last-first);
239 //
240 sprintf(ename,"Cancel1_%d", ievent);
241 TH1F his1(ename,ename,last-first,first,last);
242 for (Int_t ipos=0; ipos<last-first; ipos++){
243 his1.SetBinContent(ipos+1,x1[ipos]);
244 }
245 his1.Write(ename);
246 sprintf(ename,"Cancel2_%d", ievent);
247 TH1F his2(ename,ename,last-first,first,last);
248 for (Int_t ipos=0; ipos<last-first; ipos++){
249 his2.SetBinContent(ipos+1,x1[ipos]);
250 }
251 f1->SetParameters(0.75*max,maxpos,1.1,0.8,0.25,0.2);
252 his2.Fit(f1,"q");
253 his2.Write(ename);
254 Double_t params2[6];
255 for (Int_t ipar=0; ipar<6; ipar++) params2[ipar] = f1->GetParameters()[ipar];
256 Double_t chi22 = TFitter::GetFitter()->Chisquare(6,params2);
257 TMatrixD cov2(6,6);
258 cov2.SetMatrixArray(TFitter::GetFitter()->GetCovarianceMatrix());
259
260 TGraph gr0(last-first, &time[first],x0);
261 TGraph gr1(last-first, &time[first],x1);
262 TGraph gr2(last-first, &time[first],x2);
263 //
264 cstream<<"Fit"<<
265 "Sector="<<sector<<
266 "Row="<<row<<
267 "Pad="<<pad<<
268 "First="<<first<<
269 "Max="<<max<<
270 "MaxPos="<<maxpos<<
271 "chi2="<<chi2<<
272 "chi22="<<chi22<<
273 "Cov="<<&cov<<
274 "Cov2="<<&cov2<<
275 "gr0.="<<&gr0<<
276 "gr1.="<<&gr1<<
277 "gr2.="<<&gr2<<
278 "p0="<<params[0]<<
279 "p1="<<params[1]<<
280 "p2="<<params[2]<<
281 "p3="<<params[3]<<
282 "p4="<<params[4]<<
283 "p5="<<params[5]<<
284 "p02="<<params2[0]<<
285 "p12="<<params2[1]<<
286 "p22="<<params2[2]<<
287 "p32="<<params2[3]<<
288 "p42="<<params2[4]<<
289 "p52="<<params2[5]<<
290 "\n";
291 // delete his;
663296cb 292 nFits++;
eaad8c01 293 }
294
295}
a82f6396 296
297
663296cb 298void LaserCalib(TTreeSRedirector & cstream, TTree * chain, Float_t tmin, Float_t tmax, Float_t fraction){
299 //
300 //
301 //
ba6ac751 302 const Double_t kMaxDelta=10;
663296cb 303 AliTPCParamSR param;
304 param.Update();
305 TFile fce("TPCsignal.root");
306 TTree * treece =(TTree*)fce.Get("Signalce");
307 if (chain) treece=chain;
308 //
309 TBranch * brsector = treece->GetBranch("Sector");
310 TBranch * brpad = treece->GetBranch("Pad");
311 TBranch * brrow = treece->GetBranch("Row");
b78c817f 312 TBranch * brTimeStamp = treece->GetBranch("TimeStamp");
663296cb 313 //
314 TBranch * brtime = treece->GetBranch("Time");
315 TBranch * brrms = treece->GetBranch("RMS06");
316 TBranch * brmax = treece->GetBranch("Max");
317 TBranch * brsum = treece->GetBranch("Qsum");
318
319 Int_t sector, pad, row=0;
320 Double_t time=0, rms=0, qMax=0, qSum=0;
b78c817f 321 UInt_t timeStamp=0;
663296cb 322 brsector->SetAddress(&sector);
323 brrow->SetAddress(&row);
324 brpad->SetAddress(&pad);
b78c817f 325 brTimeStamp->SetAddress(&timeStamp);
663296cb 326
327 brtime->SetAddress(&time);
328 brrms->SetAddress(&rms);
329 brmax->SetAddress(&qMax);
330 brsum->SetAddress(&qSum);
331
332
333 brsector->GetEntry(0);
334 //
335 Int_t firstSector = sector;
336 Int_t lastSector = sector;
337 Int_t fentry = 0;
338 Int_t lentry = 0;
339 //
340 // find time offset for differnt events
341 //
342 Int_t count = 0;
343 Double_t padTimes[500000];
344 TRobustEstimator restim;
345 Double_t meanS[72], sigmaS[72];
346 Int_t firstS[72], lastS[72];
347 Double_t sectorsID[72];
348 for (Int_t isector=0;isector<72; isector++){
349 firstS[isector]=-1;
350 lastS[isector] =-1;
351 };
ba6ac751 352 TH1F hisT("hisT","hisT",100,tmin,tmax);
353 treece->Draw("Time>>hisT","");
354 Float_t cbin = hisT.GetBinCenter(hisT.GetMaximumBin());
663296cb 355
356 for (Int_t ientry=0; ientry<treece->GetEntriesFast(); ientry++){
357 treece->GetEvent(ientry);
358 //
359 if (sector!=lastSector && sector==firstSector){
360 //if (sector!=lastSector){
361 lentry = ientry;
b78c817f 362 TTimeStamp stamp(timeStamp);
363 stamp.Print();
663296cb 364 printf("\nEvent\t%d\tFirst\t%d\tLast\t%d\t%d\n",count, fentry, lentry, lentry-fentry);
365 //
366 //
367 Int_t ngood=0;
368 for (Int_t ientry2=fentry; ientry2<lentry; ientry2++){
369 // brtime->GetEvent(ientry2);
370 // brsector->GetEvent(ientry2);
371 treece->GetEvent(ientry2);
ba6ac751 372 if (time>tmin&&time<tmax && TMath::Abs(time-cbin)<kMaxDelta){
663296cb 373 padTimes[ngood]=time;
374 ngood++;
375 if (firstS[sector]<0) firstS[sector]= ngood;
376 if (firstS[sector]>=0) lastS[sector] = ngood;
377 }
378 }
379 //
380 //
381 Double_t mean,sigma;
ba6ac751 382 restim.EvaluateUni(ngood,padTimes,mean, sigma,int(float(ngood)*fraction));
663296cb 383 printf("Event\t%d\t%f\t%f\n",count, mean, sigma);
384 for (Int_t isector=0; isector<72; isector++){
385 sectorsID[isector]=sector;
386 if (firstS[isector]>=0 &&lastS[isector]>=0 && lastS[isector]>firstS[isector] ){
387 Int_t ngoodS = lastS[isector]-firstS[isector];
388 restim.EvaluateUni(ngoodS, &padTimes[firstS[isector]],meanS[isector],
ba6ac751 389 sigmaS[isector],int(float(ngoodS)*fraction));
663296cb 390 }
391 }
392 TGraph graphM(72,sectorsID,meanS);
393 TGraph graphS(72,sectorsID,sigmaS);
394 cstream<<"TimeS"<<
ba6ac751 395 "CBin="<<cbin<<
663296cb 396 "Event="<<count<<
ba6ac751 397 "GM="<<&graphM<<
398 "GS="<<&graphS<<
663296cb 399 "\n";
400
401
402 for (Int_t ientry2=fentry; ientry2<lentry-1; ientry2++){
403 treece->GetEvent(ientry2);
404 Double_t x = param.GetPadRowRadii(sector,row);
405 Int_t maxpad = AliTPCROC::Instance()->GetNPads(sector,row);
406 Double_t y = (pad - 2.5 - 0.5*maxpad)*param.GetPadPitchWidth(sector);
407 Double_t alpha = TMath::DegToRad()*(10.+20.*(sector%18));
408 Double_t gx = x*TMath::Cos(alpha)-y*TMath::Sin(alpha);
409 Double_t gy = y*TMath::Cos(alpha)+x*TMath::Sin(alpha);
410
411 Int_t npadS = lastS[sector]-firstS[sector];
412 cstream<<"Time"<<
413 "Event="<<count<<
b78c817f 414 "TimeStamp="<<timeStamp<<
ba6ac751 415 "CBin="<<cbin<<
663296cb 416 "x="<<x<<
417 "y="<<y<<
418 "gx="<<gx<<
419 "gy="<<gy<<
420 "Sector="<<sector<<
421 "Row="<<row<<
422 "Pad="<<pad<<
423 "Time="<<time<<
424 "RMS="<<rms<<
425 "Time0="<<mean<<
426 "Sigma0="<<sigma<<
427 "TimeS0="<<meanS[sector]<<
428 "SigmaS0="<<sigmaS[sector]<<
429 "npad0="<<ngood<<
430 "npadS="<<npadS<<
431 "Max="<<qMax<<
432 "Sum="<<qSum<<
433 "\n";
434 }
435 treece->GetEvent(ientry);
436 fentry = ientry;
437 count++;
438 for (Int_t isector=0;isector<72; isector++){
439 firstS[isector]=-1;
440 lastS[isector] =-1;
441 }
442 }
443 lastSector=sector;
444 }
445}
446
447
448
a82f6396 449
450TChain *MakeChainCL(Int_t first, Int_t last){
451 TChain *chaincl = new TChain("TreeR","TreeR");
452 //
453 char fname[100];
454 for (Int_t i=first;i<last; i++){
455 if (i>0) sprintf(fname,"TPC.RecPoints%d.root/Event%d/TreeR",i,i);
456 if (i==0) sprintf(fname,"TPC.RecPoints.root/Event%d/TreeR",i);
457 chaincl->Add(fname);
458 }
459 return chaincl;
460}
461
462TTree* GetTree(Int_t ievent){
463 char fname[100];
464 char tname[100];
465 if (ievent>0) sprintf(fname,"TPC.RecPoints%d.root",ievent);
466 if (ievent==0) sprintf(fname,"TPC.RecPoints.root");
467 sprintf(tname,"Event%d/TreeR",ievent);
468 TFile * f = new TFile(fname);
469 TTree * tree = (TTree*)f->Get(tname);
470 return tree;
471
472}