]>
Commit | Line | Data |
---|---|---|
4c039060 | 1 | /************************************************************************** |
2 | * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. * | |
3 | * * | |
4 | * Author: The ALICE Off-line Project. * | |
5 | * Contributors are mentioned in the code where appropriate. * | |
6 | * * | |
7 | * Permission to use, copy, modify and distribute this software and its * | |
8 | * documentation strictly for non-commercial purposes is hereby granted * | |
9 | * without fee, provided that the above copyright notice appears in all * | |
10 | * copies and that both the copyright notice and this permission notice * | |
11 | * appear in the supporting documentation. The authors make no claims * | |
12 | * about the suitability of this software for any purpose. It is * | |
13 | * provided "as is" without express or implied warranty. * | |
14 | **************************************************************************/ | |
15 | ||
16 | /* | |
17 | $Log$ | |
18 | */ | |
19 | ||
8c555625 | 20 | /////////////////////////////////////////////////////////////////////////////// |
21 | // AliTPCPRF2D - // | |
22 | // Pad response function object in two dimesions // | |
23 | // This class contains the basic functions for the // | |
24 | // calculation of PRF according generic charge distribution // | |
25 | // In Update function object calculate table of response function // | |
26 | // in discrete x and y position // | |
27 | // This table is used for interpolation od response function in any position // | |
28 | // (function GetPRF) // | |
29 | // // | |
30 | // Origin: Marian Ivanov, Uni. of Bratislava, ivanov@fmph.uniba.sk // | |
31 | // // | |
32 | /////////////////////////////////////////////////////////////////////////////// | |
33 | #include "TMath.h" | |
34 | #include "AliTPCPRF2D.h" | |
35 | #include "TF2.h" | |
36 | #include <iostream.h> | |
37 | #include <string.h> | |
38 | #include "TCanvas.h" | |
39 | #include "TPad.h" | |
40 | #include "TStyle.h" | |
41 | #include "TH1.h" | |
42 | #include "TH2.h" | |
43 | #include "TPaveText.h" | |
44 | #include "TText.h" | |
45 | ||
46 | extern TStyle * gStyle; | |
47 | ||
48 | static const Float_t sqrt12=3.46; | |
49 | static const Int_t NPRF = 100; | |
50 | ||
51 | ||
52 | static Double_t funGauss2D(Double_t *x, Double_t * par) | |
53 | { | |
54 | return ( TMath::Exp(-(x[0]*x[0])/(2*par[0]*par[0]))* | |
55 | TMath::Exp(-(x[1]*x[1])/(2*par[1]*par[1]))); | |
56 | ||
57 | } | |
58 | ||
59 | static Double_t funCosh2D(Double_t *x, Double_t * par) | |
60 | { | |
61 | return ( 1/(TMath::CosH(3.14159*x[0]/(2*par[0]))* | |
62 | TMath::CosH(3.14159*x[1]/(2*par[1])))); | |
63 | } | |
64 | ||
65 | static Double_t funGati2D(Double_t *x, Double_t * par) | |
66 | { | |
67 | //par[1] = is equal to k3X | |
68 | //par[0] is equal to pad wire distance | |
69 | Float_t K3=par[1]; | |
70 | Float_t K3R=TMath::Sqrt(K3); | |
71 | Float_t K2=(TMath::Pi()/2)*(1-K3R/2.); | |
72 | Float_t K1=K2*K3R/(4*TMath::ATan(K3R)); | |
73 | Float_t l=x[0]/par[0]; | |
74 | Float_t tan2=TMath::TanH(K2*l); | |
75 | tan2*=tan2; | |
76 | Float_t res = K1*(1-tan2)/(1+K3*tan2); | |
77 | //par[4] = is equal to k3Y | |
78 | K3=par[4]; | |
79 | K3R=TMath::Sqrt(K3); | |
80 | K2=(TMath::Pi()/2)*(1-K3R/2.); | |
81 | K1=K2*K3R/(4*TMath::ATan(K3R)); | |
82 | l=x[1]/par[0]; | |
83 | tan2=TMath::TanH(K2*l); | |
84 | tan2*=tan2; | |
85 | res = res*K1*(1-tan2)/(1+K3*tan2); | |
86 | return res; | |
87 | } | |
88 | ||
89 | ||
90 | /////////////////////////////////////////////////////////////////////////// | |
91 | /////////////////////////////////////////////////////////////////////////// | |
92 | /////////////////////////////////////////////////////////////////////////// | |
93 | /////////////////////////////////////////////////////////////////////////// | |
94 | ||
95 | ClassImp(AliTPCPRF2D) | |
96 | ||
97 | AliTPCPRF2D::AliTPCPRF2D() | |
98 | { | |
99 | ffcharge = 0; | |
100 | fNPRF =NPRF ; | |
101 | fSigmaX = 0; | |
102 | ||
103 | fGRF = 0; | |
104 | fkNorm = 1; | |
105 | forigsigmaY=0; | |
106 | forigsigmaX=0; | |
107 | fNdiv = 5; | |
108 | //chewron default values | |
109 | SetPad(0.8,0.8); | |
110 | SetChevron(0.2,0.0,1.0); | |
111 | SetY(-0.2,0.2,2); | |
112 | // SetGauss(0.22,0.22,1); | |
113 | } | |
114 | ||
115 | AliTPCPRF2D::~AliTPCPRF2D() | |
116 | { | |
117 | if (ffcharge!=0) delete [] ffcharge; | |
118 | if (fGRF !=0 ) fGRF->Delete(); | |
119 | } | |
120 | ||
121 | void AliTPCPRF2D::SetY(Float_t y1, Float_t y2, Int_t nYdiv) | |
122 | { | |
123 | // | |
124 | //set virtual line position | |
125 | //first and last line and number of lines | |
126 | fNYdiv = nYdiv; | |
127 | if (ffcharge!=0) delete [] ffcharge; | |
128 | ffcharge = new Float_t[fNPRF*fNYdiv]; | |
129 | fY1=y1; | |
130 | fY2=y2; | |
131 | } | |
132 | ||
133 | void AliTPCPRF2D::SetPad(Float_t width, Float_t height) | |
134 | { | |
135 | //set base chevron parameters | |
136 | fHeightFull=height; | |
137 | fWidth=width; | |
138 | } | |
139 | void AliTPCPRF2D::SetChevron(Float_t hstep, | |
140 | Float_t shifty, | |
141 | Float_t fac) | |
142 | { | |
143 | //set shaping of chewron parameters | |
144 | fHeightS=hstep; | |
145 | fShiftY=shifty; | |
146 | fK=fWidth*fac/hstep; | |
147 | } | |
148 | ||
149 | void AliTPCPRF2D::SetChParam(Float_t width, Float_t height, | |
150 | Float_t hstep, Float_t shifty, Float_t fac) | |
151 | { | |
152 | SetPad(width,height); | |
153 | SetChevron(hstep,shifty,fac); | |
154 | } | |
155 | ||
156 | ||
157 | Float_t AliTPCPRF2D::GetPRF(Float_t xin, Float_t yin, Bool_t inter) | |
158 | { | |
159 | if (ffcharge==0) return 0; | |
160 | // Float_t y=Float_t(fNYdiv-1)*(yin-fY1)/(fY2-fY1); | |
161 | //transform position to "wire position" | |
162 | Float_t y=fDYtoWire*(yin-fY1); | |
163 | if (fNYdiv == 1) y=fY1; | |
164 | //normaly it find nearest line charge | |
165 | if (inter ==kFALSE){ | |
166 | Int_t i=Int_t(0.5+y); | |
167 | if (y<0) i=Int_t(-0.5+y); | |
168 | if ((i<0) || (i>=fNYdiv) ) return 0; | |
169 | fcharge = &(ffcharge[i*fNPRF]); | |
170 | return GetPRFActiv(xin); | |
171 | } | |
172 | else{ | |
173 | //make interpolation from more fore lines | |
174 | Int_t i= Int_t(y); | |
175 | if ((i<0) || (i>=fNYdiv) ) return 0; | |
176 | Float_t z0=0; | |
177 | Float_t z1=0; | |
178 | Float_t z2=0; | |
179 | Float_t z3=0; | |
180 | if (i>0) { | |
181 | fcharge =&(ffcharge[(i-1)*fNPRF]); | |
182 | z0 = GetPRFActiv(xin); | |
183 | } | |
184 | fcharge =&(ffcharge[i*fNPRF]); | |
185 | z1=GetPRFActiv(xin); | |
186 | if ((i+1)<fNYdiv){ | |
187 | fcharge =&(ffcharge[(i+1)*fNPRF]); | |
188 | z2 = GetPRFActiv(xin); | |
189 | } | |
190 | if ((i+2)<fNYdiv){ | |
191 | fcharge =&(ffcharge[(i+2)*fNPRF]); | |
192 | z3 = GetPRFActiv(xin); | |
193 | } | |
194 | Float_t a,b,c,d,K,L; | |
195 | a=z1; | |
196 | b=(z2-z0)/2.; | |
197 | K=z2-a-b; | |
198 | L=(z3-z1)/2.-b; | |
199 | d=L-2*K; | |
200 | c=K-d; | |
201 | Float_t dy=y-Float_t(i); | |
202 | Float_t res = a+b*dy+c*dy*dy+d*dy*dy*dy; | |
203 | //Float_t res = z1*(1-dy)+z2*dy; | |
204 | return res; | |
205 | } | |
206 | return 0.; | |
207 | } | |
208 | ||
209 | ||
210 | Float_t AliTPCPRF2D::GetPRFActiv(Float_t xin) | |
211 | { | |
212 | //x xin DStep unit | |
213 | //return splaine aproximaton | |
214 | Float_t x = (xin*fDStepM1)+fNPRF/2; | |
215 | Int_t i = Int_t(x); | |
216 | ||
217 | if ( (i>0) && ((i+2)<fNPRF)) { | |
218 | Float_t a,b,c,d,K,L; | |
219 | a = fcharge[i]; | |
220 | b = (fcharge[i+1]-fcharge[i-1])*0.5; | |
221 | K = fcharge[i+1]-a-b; | |
222 | L = (fcharge[i+2]-fcharge[i])*0.5-b; | |
223 | d=L-2.*K; | |
224 | c=K-d; | |
225 | Float_t dx=x-Float_t(i); | |
226 | Float_t res = a+b*dx+c*dx*dx+d*dx*dx*dx; | |
227 | return res; | |
228 | } | |
229 | else return 0; | |
230 | } | |
231 | ||
232 | ||
233 | Float_t AliTPCPRF2D::GetGRF(Float_t xin, Float_t yin) | |
234 | { | |
235 | if (fGRF != 0 ) | |
236 | return fkNorm*fGRF->Eval(xin,yin)/fInteg; | |
237 | else | |
238 | return 0.; | |
239 | } | |
240 | ||
241 | ||
242 | void AliTPCPRF2D::SetParam( TF2 * GRF, Float_t kNorm, | |
243 | Float_t sigmaX, Float_t sigmaY) | |
244 | { | |
245 | if (fGRF !=0 ) fGRF->Delete(); | |
246 | fGRF = GRF; | |
247 | fkNorm = kNorm; | |
248 | if (sigmaX ==0) sigmaX=(fWidth+fK*fHeightS)/sqrt12; | |
249 | if (sigmaY ==0) sigmaY=(fWidth+fK*fHeightS)/sqrt12; | |
250 | forigsigmaX=sigmaX; | |
251 | forigsigmaY=sigmaY; | |
252 | fDStep = TMath::Sqrt(sigmaX*sigmaX+fWidth*fWidth/6.)/10.; | |
253 | // Update(); | |
254 | sprintf(fType,"User"); | |
255 | } | |
256 | ||
257 | ||
258 | void AliTPCPRF2D::SetGauss(Float_t sigmaX, Float_t sigmaY, | |
259 | Float_t kNorm) | |
260 | { | |
261 | fkNorm = kNorm; | |
262 | if (fGRF !=0 ) fGRF->Delete(); | |
263 | fGRF = new TF2("fun",funGauss2D,-5.,5.,-5.,5.,4); | |
264 | funParam[0]=sigmaX; | |
265 | funParam[1]=sigmaY; | |
266 | funParam[2]=fK; | |
267 | funParam[3]=fHeightS; | |
268 | forigsigmaX=sigmaX; | |
269 | forigsigmaY=sigmaY; | |
270 | fGRF->SetParameters(funParam); | |
271 | fDStep = TMath::Sqrt(sigmaX*sigmaX+fWidth*fWidth/6.)/10.; | |
272 | //by default I set the step as one tenth of sigma | |
273 | //Update(); | |
274 | sprintf(fType,"Gauss"); | |
275 | } | |
276 | ||
277 | void AliTPCPRF2D::SetCosh(Float_t sigmaX, Float_t sigmaY, | |
278 | Float_t kNorm) | |
279 | { | |
280 | fkNorm = kNorm; | |
281 | if (fGRF !=0 ) fGRF->Delete(); | |
282 | fGRF = new TF2("fun", funCosh2D,-5.,5.,-5.,5.,4); | |
283 | funParam[0]=sigmaX; | |
284 | funParam[1]=sigmaY; | |
285 | funParam[2]=fK; | |
286 | funParam[3]=fHeightS; | |
287 | fGRF->SetParameters(funParam); | |
288 | forigsigmaX=sigmaX; | |
289 | forigsigmaY=sigmaY; | |
290 | fDStep = TMath::Sqrt(sigmaX*sigmaX+fWidth*fWidth/6.)/10.; | |
291 | //by default I set the step as one tenth of sigma | |
292 | //Update(); | |
293 | sprintf(fType,"Cosh"); | |
294 | } | |
295 | ||
296 | void AliTPCPRF2D::SetGati(Float_t K3X, Float_t K3Y, | |
297 | Float_t padDistance, | |
298 | Float_t kNorm) | |
299 | { | |
300 | fkNorm = kNorm; | |
301 | if (fGRF !=0 ) fGRF->Delete(); | |
302 | fGRF = new TF2("fun", funGati2D,-5.,5.,-5.,5.,5); | |
303 | fK3X=K3X; | |
304 | fK3Y=K3Y; | |
305 | fPadDistance=padDistance; | |
306 | funParam[0]=padDistance; | |
307 | funParam[1]=K3X; | |
308 | funParam[2]=fK; | |
309 | funParam[3]=fHeightS; | |
310 | funParam[4]=K3Y; | |
311 | fGRF->SetParameters(funParam); | |
312 | forigsigmaX=padDistance; | |
313 | forigsigmaY=padDistance; | |
314 | fDStep = TMath::Sqrt(padDistance*padDistance+fWidth*fWidth/6.)/10.; | |
315 | //by default I set the step as one tenth of sigma | |
316 | //Update(); | |
317 | sprintf(fType,"Gati"); | |
318 | } | |
319 | ||
320 | ||
321 | ||
322 | void AliTPCPRF2D::Update() | |
323 | { | |
324 | for (Int_t i=0; i<fNYdiv; i++){ | |
325 | if (fNYdiv == 1) fActualY = fY1; | |
326 | else | |
327 | fActualY = fY1+Float_t(i)*(fY2-fY1)/Float_t(fNYdiv-1); | |
328 | fcharge = &(ffcharge[i*fNPRF]); | |
329 | Update1(); | |
330 | } | |
331 | } | |
332 | ||
333 | ||
334 | ||
335 | void AliTPCPRF2D::Update1() | |
336 | { | |
337 | //initialize to 0 | |
338 | ||
339 | ||
340 | Int_t i; | |
341 | Float_t x; | |
342 | for (i =0; i<fNPRF;i++) fcharge[i] = 0; | |
343 | if ( fGRF == 0 ) return; | |
344 | //////////////////////////////////////////////////////// | |
345 | //I'm waiting for normal integral | |
346 | //in this moment only sum | |
347 | Float_t x2= 4*forigsigmaX; | |
348 | Float_t y2= 4*forigsigmaY; | |
349 | Float_t dx = forigsigmaX/Float_t(fNdiv*6); | |
350 | Float_t dy = forigsigmaY/Float_t(fNdiv*6); | |
351 | fInteg = 0; | |
352 | for (x=0.;x<x2;x+=dx) | |
353 | for (Float_t y=0;y<y2;y+=dy) fInteg+=fGRF->Eval(x,y)*dx*dy; | |
354 | fInteg*=4; | |
355 | ///////////////////////////////////////////////////// | |
356 | ||
357 | ||
358 | if ( fInteg == 0 ) fInteg = 1; | |
359 | ||
360 | //integrate charge over pad for different distance of pad | |
361 | for (i =0; i<fNPRF;i++) | |
362 | { //x in cm fWidth in cm | |
363 | //calculate integral | |
364 | Float_t xch = fDStep * (Float_t)(i-fNPRF/2); | |
365 | Float_t k=1; | |
366 | fcharge[i]=0; | |
367 | for (Float_t y=-fHeightFull/2.-fShiftY; | |
368 | y<fHeightFull/2.;y+=fHeightS){ | |
369 | Float_t y2=TMath::Min((y+fHeightS),Float_t(fHeightFull/2.)); | |
370 | Float_t y1=TMath::Max((y),Float_t(-fHeightFull/2.)); | |
371 | Float_t x1; | |
372 | ||
373 | if (k>0) | |
374 | x1 = (y2-y1)*fK-(fWidth+fK*fHeightS)/2.; | |
375 | else | |
376 | x1 =-(fWidth+fK*fHeightS)/2. ; | |
377 | Float_t x2=x1+fWidth; | |
378 | ||
379 | if (y2>y1) { | |
380 | ||
381 | if ((x2-x1)*fNdiv<forigsigmaX) dx=(x2-x1); | |
382 | else{ | |
383 | dx= forigsigmaX/Float_t(fNdiv); | |
384 | dx = (x2-x1)/Float_t(Int_t(3+(x2-x1)/dx)); | |
385 | } | |
386 | Float_t dy; | |
387 | if ((y2-y1)*fNdiv<forigsigmaY) dy=(y2-y1); | |
388 | else{ | |
389 | dy= forigsigmaY/Float_t(fNdiv); | |
390 | dy = (y2-y1)/Float_t(Int_t(3+(y2-y1)/dy)); | |
391 | } | |
392 | ||
393 | for (x=x1;x<x2;x+=dx) | |
394 | for (Float_t y=y1;y<y2;y+=dy){ | |
395 | if ( (y>(fActualY-(4.0*forigsigmaY))) && | |
396 | (y<(fActualY+(4.0*forigsigmaY)))){ | |
397 | Float_t xt=x-k*fK*(y-y1); | |
398 | if ((TMath::Abs(xch-xt)<4*forigsigmaX)){ | |
399 | ||
400 | Float_t z0=fGRF->Eval(xch-(xt+dx/2.),fActualY-(y+dy/2.)); | |
401 | ||
402 | Float_t z1=fGRF->Eval(xch-(xt+dx/2.),fActualY-y); | |
403 | Float_t z2=fGRF->Eval(xch-xt,fActualY-(y+dy/2.)); | |
404 | Float_t z3=fGRF->Eval(xch-(xt-dx/2.),fActualY-y); | |
405 | Float_t z4=fGRF->Eval(xch-xt,fActualY-(y-dy/2.)); | |
406 | if (z0<0) z0=0; | |
407 | if (z1<0) z1=0; | |
408 | if (z2<0) z2=0; | |
409 | if (z3<0) z3=0; | |
410 | if (z4<0) z4=0; | |
411 | ||
412 | // Float_t a=(z1-z3)/2; | |
413 | // Float_t b=(z2-z4)/2; | |
414 | Float_t c= (z3+z1-2*z0)/2.; | |
415 | Float_t d= (z2+z4-2*z0)/2.; | |
416 | Float_t z= (z0+c/12.+d/12.); | |
417 | ||
418 | //Float_t z= fGRF->Eval(xch-xt,fActualY-y); | |
419 | if (z>0.) fcharge[i]+=z*dx*dy/fInteg; | |
420 | } | |
421 | } | |
422 | } | |
423 | } | |
424 | k*=-1; | |
425 | } | |
426 | }; | |
427 | ||
428 | fSigmaX = 0; | |
429 | Float_t sum =0; | |
430 | Float_t mean=0; | |
431 | for (x =-fNPRF*fDStep; x<fNPRF*fDStep;x+=fDStep) | |
432 | { //x in cm fWidth in cm | |
433 | Float_t weight = GetPRFActiv(x); | |
434 | fSigmaX+=x*x*weight; | |
435 | mean+=x*weight; | |
436 | sum+=weight; | |
437 | }; | |
438 | if (sum>0){ | |
439 | mean/=sum; | |
440 | fSigmaX = TMath::Sqrt(fSigmaX/sum-mean*mean); | |
441 | } | |
442 | else fSigmaX=0; | |
443 | //calculate conversion coefitient to convert position to virtual wire | |
444 | fDYtoWire=Float_t(fNYdiv-1)/(fY2-fY1); | |
445 | fDStepM1=1/fDStep; | |
446 | } | |
447 | ||
448 | void AliTPCPRF2D::Streamer(TBuffer &R__b) | |
449 | { | |
450 | // Stream an object of class AliTPCPRF2D | |
451 | ||
452 | if (R__b.IsReading()) { | |
453 | Version_t R__v = R__b.ReadVersion(); if (R__v) { } | |
454 | TObject::Streamer(R__b); | |
455 | //read chewron parameters | |
456 | R__b >> fSigmaX; | |
457 | R__b >> fHeightFull; | |
458 | R__b >> fHeightS; | |
459 | R__b >> fShiftY; | |
460 | R__b >> fWidth; | |
461 | R__b >> fK; | |
462 | R__b >> fActualY; | |
463 | //read charge parameters | |
464 | R__b >> fType[0]; | |
465 | R__b >> fType[1]; | |
466 | R__b >> fType[2]; | |
467 | R__b >> fType[3]; | |
468 | R__b >> fType[4]; | |
469 | R__b >> forigsigmaX; | |
470 | R__b >> forigsigmaY; | |
471 | R__b >> fkNorm; | |
472 | R__b >> fK3X; | |
473 | R__b >> fK3Y; | |
474 | R__b >> fPadDistance; | |
475 | R__b >> fInteg; | |
476 | ||
477 | //read functions | |
478 | if (fGRF!=0) { | |
479 | delete [] fGRF; | |
480 | fGRF=0; | |
481 | } | |
482 | if (strncmp(fType,"User",3)==0){ | |
483 | fGRF= new TF2; | |
484 | R__b>>fGRF; | |
485 | } | |
486 | if (strncmp(fType,"Gauss",3)==0) | |
487 | fGRF = new TF2("fun",funGauss2D,-5.,5.,-5.,5.,4); | |
488 | if (strncmp(fType,"Cosh",3)==0) | |
489 | fGRF = new TF2("fun",funCosh2D,-5.,5.,-5.,5.,4); | |
490 | if (strncmp(fType,"Gati",3)==0) | |
491 | fGRF = new TF2("fun",funGati2D,-5.,5.,-5.,5.,5); | |
492 | ||
493 | //read interpolation parameters | |
494 | R__b >>fY1; | |
495 | R__b >>fY2; | |
496 | R__b >>fNYdiv; | |
497 | R__b >>fDStep; | |
498 | R__b >>fNPRF; | |
499 | if (ffcharge!=0) delete [] ffcharge; | |
500 | ffcharge = new Float_t[fNPRF*fNYdiv]; | |
501 | R__b.ReadFastArray(ffcharge,fNPRF*fNYdiv); | |
502 | R__b.ReadFastArray(funParam,5); | |
503 | if (fGRF!=0) fGRF->SetParameters(funParam); | |
504 | //calculate conversion coefitient to convert position to virtual wire | |
505 | fDYtoWire=Float_t(fNYdiv-1)/(fY2-fY1); | |
506 | fDStepM1=1/fDStep; | |
507 | } else { | |
508 | R__b.WriteVersion(AliTPCPRF2D::IsA()); | |
509 | TObject::Streamer(R__b); | |
510 | //write chewron parameters | |
511 | R__b << fSigmaX; | |
512 | R__b << fHeightFull; | |
513 | R__b << fHeightS; | |
514 | R__b << fShiftY; | |
515 | R__b << fWidth; | |
516 | R__b << fK; | |
517 | R__b << fActualY; | |
518 | //write charge parameters | |
519 | R__b << fType[0]; | |
520 | R__b << fType[1]; | |
521 | R__b << fType[2]; | |
522 | R__b << fType[3]; | |
523 | R__b << fType[4]; | |
524 | ||
525 | R__b << forigsigmaX; | |
526 | R__b << forigsigmaY; | |
527 | R__b << fkNorm; | |
528 | R__b << fK3X; | |
529 | R__b << fK3Y; | |
530 | R__b << fPadDistance; | |
531 | R__b << fInteg; | |
532 | ||
533 | if (strncmp(fType,"User",3)==0) R__b <<fGRF; | |
534 | //write interpolation parameters | |
535 | R__b <<fY1; | |
536 | R__b <<fY2; | |
537 | R__b <<fNYdiv; | |
538 | R__b <<fDStep; | |
539 | R__b <<fNPRF; | |
540 | R__b.WriteFastArray(ffcharge,fNPRF*fNYdiv); | |
541 | R__b.WriteFastArray(funParam,5); | |
542 | } | |
543 | } | |
544 | ||
545 | ||
546 | ||
547 | ||
548 | void AliTPCPRF2D::DrawX(Float_t x1 ,Float_t x2,Float_t y, Bool_t inter) | |
549 | { | |
550 | if (fGRF==0) return ; | |
551 | const Int_t N=100; | |
552 | char s[100]; | |
553 | TCanvas * c1 = new TCanvas("canPRF","Pad response function",700,900); | |
554 | c1->cd(); | |
555 | TPad * pad1 = new TPad("pad1PRF","",0.05,0.61,0.95,0.97,21); | |
556 | pad1->Draw(); | |
557 | TPad * pad2 = new TPad("pad2PRF","",0.05,0.22,0.95,0.60,21); | |
558 | pad2->Draw(); | |
559 | ||
560 | // pad1->cd(); | |
561 | //pad2->cd(); | |
562 | gStyle->SetOptFit(1); | |
563 | gStyle->SetOptStat(0); | |
564 | sprintf(s,"PRF response function for chevron pad"); | |
565 | TH1F * hPRFc = new TH1F("hPRFc",s,N+1,x1,x2); | |
566 | Float_t x=x1; | |
567 | Float_t y1; | |
568 | // Float_t y2; | |
569 | ||
570 | for (Float_t i = 0;i<N+1;i++) | |
571 | { | |
572 | x+=(x2-x1)/Float_t(N); | |
573 | y1 = GetPRF(x,y,inter); | |
574 | hPRFc->Fill(x,y1); | |
575 | }; | |
576 | ||
577 | pad1->cd(); | |
578 | fGRF->SetRange(x1,x1,x2,x2); | |
579 | fGRF->SetNpx(25); | |
580 | fGRF->SetNpy(25); | |
581 | fGRF->Draw("lego2"); | |
582 | // hPRFo->Fit("gaus"); | |
583 | gStyle->SetOptStat(1); | |
584 | pad2->cd(); | |
585 | hPRFc->Fit("gaus"); | |
586 | c1->cd(); | |
587 | TPaveText * comment = new TPaveText(0.05,0.02,0.95,0.20,"NDC"); | |
588 | comment->SetTextAlign(12); | |
589 | comment->SetFillColor(42); | |
590 | TText *title = comment->AddText("Chevron pad parameters:"); | |
591 | title->SetTextSize(0.03); | |
592 | sprintf(s,"Full height of pad: %2.2f",fHeightFull); | |
593 | comment->AddText(s); | |
594 | sprintf(s,"Height of one chevron unit h: %2.2f cm",2*fHeightS); | |
595 | comment->AddText(s); | |
596 | sprintf(s,"Width of one chevron unit w: %2.2f cm",fWidth); | |
597 | comment->AddText(s); | |
598 | sprintf(s,"Overlap factor: %2.2f",fK*fHeightS/fWidth); | |
599 | comment->AddText(s); | |
600 | sprintf(s,"Y position: %2.2f ",y); | |
601 | comment->AddText(s); | |
602 | sprintf(s,"Sigma x of original distribution: %2.2f ",forigsigmaX); | |
603 | comment->AddText(s); | |
604 | sprintf(s,"Sigma y of original distribution: %2.2f ",forigsigmaY); | |
605 | comment->AddText(s); | |
606 | sprintf(s,"Type of original distribution: %s ",fType); | |
607 | comment->AddText(s); | |
608 | comment->Draw(); | |
609 | } | |
610 | ||
611 | ||
612 | ||
613 | void AliTPCPRF2D::Draw(Float_t x1 ,Float_t x2,Float_t y1, Float_t y2, | |
614 | Bool_t inter, Int_t Nx, Int_t Ny) | |
615 | { | |
616 | char s[100]; | |
617 | if (fGRF==0) return ; | |
618 | TCanvas * c1 = new TCanvas("canPRF","Pad response function",700,900); | |
619 | c1->cd(); | |
620 | TPad * pad1 = new TPad("pad1PRF","",0.05,0.61,0.95,0.97,21); | |
621 | pad1->Draw(); | |
622 | TPad * pad2 = new TPad("pad2PRF","",0.05,0.22,0.95,0.60,21); | |
623 | pad2->Draw(); | |
624 | ||
625 | // pad1->cd(); | |
626 | //pad2->cd(); | |
627 | gStyle->SetOptFit(1); | |
628 | gStyle->SetOptStat(0); | |
629 | sprintf(s,"PRF response function for chevron pad"); | |
630 | TH2F * hPRFc = new TH2F("hPRFc",s,Nx+1,x1,x2,Ny+1,y1,y2); | |
631 | Float_t dx=(x2-x1)/Float_t(Nx); | |
632 | Float_t dy=(y2-y1)/Float_t(Ny) ; | |
633 | Float_t x,y,z; | |
634 | // Float_t y2; | |
635 | for ( x = x1;x<=x2;x+=dx){ | |
636 | for(y = y1;y<=y2;y+=dy) | |
637 | { | |
638 | z = GetPRF(x,y,inter); | |
639 | hPRFc->Fill(x,y,z); | |
640 | }; | |
641 | } | |
642 | pad1->cd(); | |
643 | fGRF->SetRange(x1,y1,x2,y2); | |
644 | fGRF->SetNpx(25); | |
645 | fGRF->SetNpy(25); | |
646 | fGRF->Draw("lego2"); | |
647 | // hPRFo->Fit("gaus"); | |
648 | gStyle->SetOptStat(1); | |
649 | pad2->cd(); | |
650 | hPRFc->Draw("lego2"); | |
651 | c1->cd(); | |
652 | TPaveText * comment = new TPaveText(0.05,0.02,0.95,0.20,"NDC"); | |
653 | comment->SetTextAlign(12); | |
654 | comment->SetFillColor(42); | |
655 | TText *title = comment->AddText("Chevron pad parameters:"); | |
656 | title->SetTextSize(0.03); | |
657 | sprintf(s,"Full height of pad: %2.2f",fHeightFull); | |
658 | comment->AddText(s); | |
659 | sprintf(s,"Height of one chevron unit h: %2.2f cm",2*fHeightS); | |
660 | comment->AddText(s); | |
661 | sprintf(s,"Width of one chevron unit w: %2.2f cm",fWidth); | |
662 | comment->AddText(s); | |
663 | sprintf(s,"Overlap factor: %2.2f",fK*fHeightS/fWidth); | |
664 | comment->AddText(s); | |
665 | sprintf(s,"Sigma x of original distribution: %2.2f ",forigsigmaX); | |
666 | comment->AddText(s); | |
667 | sprintf(s,"Sigma y of original distribution: %2.2f ",forigsigmaY); | |
668 | comment->AddText(s); | |
669 | sprintf(s,"Type of original distribution: %s ",fType); | |
670 | comment->AddText(s); | |
671 | comment->Draw(); | |
672 | } | |
673 | ||
674 | void AliTPCPRF2D::DrawDist(Float_t x1 ,Float_t x2,Float_t y1, Float_t y2, | |
675 | Bool_t inter, Int_t Nx, Int_t Ny, Float_t thr) | |
676 | { | |
677 | const Float_t minth=0.00001; | |
678 | if (thr<minth) thr=minth; | |
679 | char s[100]; | |
680 | if (fGRF==0) return ; | |
681 | TCanvas * c1 = new TCanvas("padDistortion","COG distortion",700,900); | |
682 | c1->cd(); | |
683 | TPad * pad1 = new TPad("CHARGE","",0.05,0.61,0.95,0.97,21); | |
684 | pad1->Draw(); | |
685 | TPad * pad2 = new TPad("dist","",0.05,0.22,0.95,0.60,21); | |
686 | pad2->Draw(); | |
687 | ||
688 | // pad1->cd(); | |
689 | //pad2->cd(); | |
690 | gStyle->SetOptFit(1); | |
691 | gStyle->SetOptStat(0); | |
692 | sprintf(s,"COG distortion (threshold=%2.2f)",thr); | |
693 | TH2F * hPRFDist = new TH2F("hDistortion",s,Nx+1,x1,x2,Ny+1,y1,y2); | |
694 | Float_t dx=(x2-x1)/Float_t(Nx); | |
695 | Float_t dy=(y2-y1)/Float_t(Ny) ; | |
696 | Float_t x,y,z,ddx; | |
697 | // Float_t y2; | |
698 | for ( x = x1;x<(x2+dx/2.);x+=dx) | |
699 | for(y = y1;y<=(y2+dx/2.);y+=dy) | |
700 | { | |
701 | Float_t sumx=0; | |
702 | Float_t sum=0; | |
703 | for (Float_t padx=-fWidth;padx<(fWidth*1.1);padx+=fWidth) | |
704 | { | |
705 | z = GetPRF(x-padx,y,inter); | |
706 | if (z>thr){ | |
707 | sum+=z; | |
708 | sumx+=z*padx; | |
709 | } | |
710 | }; | |
711 | if (sum>minth) | |
712 | { | |
713 | ddx = (x-(sumx/sum)); | |
714 | } | |
715 | else ddx=-1; | |
716 | if (TMath::Abs(ddx)<10) hPRFDist->Fill(x,y,ddx); | |
717 | } | |
718 | pad1->cd(); | |
719 | fGRF->SetRange(x1,y1,x2,y2); | |
720 | fGRF->SetNpx(25); | |
721 | fGRF->SetNpy(25); | |
722 | fGRF->Draw("lego2"); | |
723 | // hPRFo->Fit("gaus"); | |
724 | // gStyle->SetOptStat(1); | |
725 | pad2->cd(); | |
726 | hPRFDist->Draw("lego2"); | |
727 | ||
728 | c1->cd(); | |
729 | TPaveText * comment = new TPaveText(0.05,0.02,0.95,0.20,"NDC"); | |
730 | comment->SetTextAlign(12); | |
731 | comment->SetFillColor(42); | |
732 | // TText *title = comment->AddText("Distortion of COG method"); | |
733 | // title->SetTextSize(0.03); | |
734 | TText * title = comment->AddText("Chevron pad parameters:"); | |
735 | title->SetTextSize(0.03); | |
736 | sprintf(s,"Full height of pad: %2.2f",fHeightFull); | |
737 | comment->AddText(s); | |
738 | sprintf(s,"Height of one chevron unit h: %2.2f cm",2*fHeightS); | |
739 | comment->AddText(s); | |
740 | sprintf(s,"Width of one chevron unit w: %2.2f cm",fWidth); | |
741 | comment->AddText(s); | |
742 | sprintf(s,"Overlap factor: %2.2f",fK*fHeightS/fWidth); | |
743 | comment->AddText(s); | |
744 | sprintf(s,"Sigma x of original distribution: %2.2f ",forigsigmaX); | |
745 | comment->AddText(s); | |
746 | sprintf(s,"Sigma y of original distribution: %2.2f ",forigsigmaY); | |
747 | comment->AddText(s); | |
748 | sprintf(s,"Type of original distribution: %s ",fType); | |
749 | comment->AddText(s); | |
750 | comment->Draw(); | |
751 | ||
752 | } | |
753 |