]> git.uio.no Git - u/mrichter/AliRoot.git/blob - ITS/AliITSTrackV1.cxx
Fixes added to get into account the new magnetic field conversion factor automatically
[u/mrichter/AliRoot.git] / ITS / AliITSTrackV1.cxx
1 //   ITS Track Class
2 //Origin  A. Badala' and G.S. Pappalardo:  e-mail Angela.Badala@ct.infn.it, Giuseppe.S.Pappalardo@ct.infn.it
3 // It contain all the usefull information for the track and the method to calculate, modify or extract them
4 // The track is mainly caracterized by the state vector of elements (fX0, fX1, fX2, fX3, fX4) and the
5 // corresponding covariance matrix of elements (C00, C10, ..... C44) that is triangular
6 //
7 #include <Riostream.h>
8 #include <TMath.h>
9 #include <TVector.h>
10 #include <TObjArray.h>
11 #include "AliRun.h"
12 #include "AliITSRad.h"
13 #include "AliITSTrackV1.h"
14 #include "AliGenerator.h"
15 //#include "AliMagF.h"
16
17
18 ClassImp(AliITSTrackV1)
19
20 AliITSTrackV1::AliITSTrackV1() {
21 //Origin  A. Badala' and G.S. Pappalardo:  e-mail Angela.Badala@ct.infn.it, Giuseppe.S.Pappalardo@ct.infn.it 
22 // default constructor   
23  
24   fTPCtrack = 0;
25   fC00=fC10=fC11=fC20=fC21=fC22=fC30=fC31=fC32=fC33=fC40=fC41=fC42=fC43=fC44=0.;
26   flistCluster = 0;   
27   fNumClustInTrack =0;
28   fChi2=-1;
29   flabel =0;
30   fLayer = -1; 
31   fClusterInTrack = 0; 
32   frtrack=0.;
33   fnoclust=0;
34   fMass=0.13956995; //a pion by default
35   fFieldFactor = 0.0;
36    
37 }
38 AliITSTrackV1::AliITSTrackV1(const char *opt, Double_t fieldfactor) {
39 //Origin  A. Badala' and G.S. Pappalardo:  e-mail Angela.Badala@ct.infn.it, Giuseppe.S.Pappalardo@ct.infn.it 
40 // default constructor   
41  
42   fTPCtrack = 0;
43   fC00=fC10=fC11=fC20=fC21=fC22=fC30=fC31=fC32=fC33=fC40=fC41=fC42=fC43=fC44=0.;
44   flistCluster = 0;   
45   fNumClustInTrack =0;
46   fChi2=-1;
47   flabel =0; 
48   fVertex.ResizeTo(3); 
49   fErrorVertex.ResizeTo(3);
50   fLayer = -1; 
51   fClusterInTrack = 0; 
52   frtrack=0.;
53   fnoclust=0;     
54   fd2.ResizeTo(6);
55   ftgl2.ResizeTo(6); 
56   fdtgl.ResizeTo(6);
57   fMass=0.13956995; //a pion by default 
58   
59 //////////////////////////////////////// gets magnetic field factor ////////////////////////////////
60
61  // AliMagF * fieldPointer = gAlice->Field();
62   // fFieldFactor =(Double_t)fieldPointer-> SolenoidField()/10/.2;
63     fFieldFactor = fieldfactor;
64   //cout<< " field factor = "<<fFieldFactor<<"\n"; getchar();
65
66 /////////////////////////////////////////////////////////////////////////////////////////////////////////
67    
68 }
69
70
71  
72 AliITSTrackV1::AliITSTrackV1(const AliITSTrackV1 &cobj) {
73 //Origin  A. Badala' and G.S. Pappalardo:  e-mail Angela.Badala@ct.infn.it, Giuseppe.S.Pappalardo@ct.infn.it
74 // copy constructor    
75
76   fClusterInTrack = new TMatrix(6,9);
77   Int_t i,j;
78   for(i=0; i<6; i++){
79   for(j=0; j<9; j++) (*fClusterInTrack)(i,j)=-1.;   
80   }  
81   flistCluster = new TObjArray; 
82   fVertex.ResizeTo(3); 
83   fErrorVertex.ResizeTo(3);
84   fVertex = cobj.fVertex;
85   fErrorVertex = cobj.fErrorVertex;
86   flabel = cobj.flabel;
87   fLayer=cobj.fLayer;
88   fTPCtrack = cobj.fTPCtrack;
89   fNumClustInTrack = cobj.fNumClustInTrack; 
90   fChi2= cobj.fChi2;
91   fX0=cobj.fX0; fX1=cobj.fX1; fX2=cobj.fX2; fX3=cobj.fX3; fX4=cobj.fX4; 
92   frtrack=cobj.frtrack;
93   fDv=cobj.fDv;
94   fZv=cobj.fZv;
95   fsigmaDv=cobj.fsigmaDv;
96   fsigmaZv=cobj.fsigmaZv;
97   fd2.ResizeTo(6);
98   ftgl2.ResizeTo(6); 
99   fdtgl.ResizeTo(6); 
100   fd2=cobj.fd2;
101   ftgl2=cobj.ftgl2;
102   fdtgl=cobj.fdtgl;
103   fnoclust=cobj.fnoclust;   
104     
105   fC00=cobj.fC00; fC10=cobj.fC10; fC11=cobj.fC11; fC20=cobj.fC20; fC21=cobj.fC21;
106   fC22=cobj.fC22; fC30=cobj.fC30; fC31=cobj.fC31; fC32=cobj.fC32; fC33=cobj.fC33; 
107   fC40=cobj.fC40; fC41=cobj.fC41; fC42=cobj.fC42; fC43=cobj.fC43; fC44=cobj.fC44; 
108  
109   *fClusterInTrack = *cobj.fClusterInTrack;
110   
111   fFieldFactor=cobj.fFieldFactor;
112   fMass=cobj.fMass; 
113    
114   for(i=0; i<cobj.flistCluster->GetSize(); i++) 
115     flistCluster->AddLast(cobj.flistCluster->At(i));
116  
117 }
118
119 AliITSTrackV1::AliITSTrackV1(AliTPCtrack &obj, Double_t fieldfactor)
120
121 //Origin  A. Badala' and G.S. Pappalardo:  e-mail Angela.Badala@ct.infn.it, Giuseppe.S.Pappalardo@ct.infn.it 
122 // special constructor to convert a TPC track into an ITS track
123
124 //////////////////////////////////////// gets magnetic field factor ////////////////////////////////
125
126    // AliMagF * fieldPointer = gAlice->Field();
127   // fFieldFactor =(Double_t)fieldPointer-> SolenoidField()/10/.2;
128     fFieldFactor = fieldfactor;
129  // cout<< " field factor dentro alitrack = "<<fFieldFactor<<"\n";/* getchar();*/
130
131 /////////////////////////////////////////////////////////////////////////////////////////////////////////
132
133
134   fTPCtrack = &obj;
135   fVertex.ResizeTo(3); 
136   fErrorVertex.ResizeTo(3);
137
138   fd2.ResizeTo(6);
139   ftgl2.ResizeTo(6); 
140   fdtgl.ResizeTo(6); 
141   AliGenerator *gener = gAlice->Generator();
142   Float_t vxg,vyg,vzg;
143   gener->GetOrigin(vxg,vyg,vzg);
144   
145  
146   fVertex(0)=(Double_t)vxg;
147   fVertex(1)=(Double_t)vyg; 
148   fVertex(2)=(Double_t)vzg;    
149   
150   fLayer = 7;
151   fClusterInTrack = new TMatrix(6,9);
152   
153   Int_t i,j;
154   for(i=0; i<6; i++){
155   for(j=0; j<9; j++) (*fClusterInTrack)(i,j)=-1.;   
156   }  
157   flistCluster = new TObjArray; 
158   fNumClustInTrack = 0;
159   fnoclust=0;        
160   LmTPC(); 
161
162 }
163
164
165 AliITSTrackV1::~AliITSTrackV1() {
166 //Origin  A. Badala' and G.S. Pappalardo:  e-mail Angela.Badala@ct.infn.it, Giuseppe.S.Pappalardo@ct.infn.it 
167 //destructor
168  
169   if(flistCluster) {
170     flistCluster->Delete();
171     delete flistCluster;
172   }
173   if(fClusterInTrack) delete fClusterInTrack;
174
175 }     
176
177 void AliITSTrackV1::PutCElements(Double_t C00, Double_t C10, Double_t C11, Double_t C20, Double_t C21, 
178 Double_t C22, Double_t C30, Double_t C31, Double_t C32, Double_t C33, Double_t C40, 
179 Double_t C41, Double_t C42, Double_t C43, Double_t C44){
180 //Origin  A. Badala' and G.S. Pappalardo:  e-mail Angela.Badala@ct.infn.it, Giuseppe.S.Pappalardo@ct.infn.it
181 // puts elements of covariance matrix
182
183   fC00=C00; fC10=C10; fC11=C11; fC20=C20; fC21=C21; fC22=C22; fC30=C30; fC31=C31; fC32=C32; fC33=C33;
184   fC40=C40; fC41=C41; fC42=C42; fC43=C43; fC44=C44; 
185 }
186   
187 void AliITSTrackV1::GetCElements(Double_t &C00, Double_t &C10, Double_t &C11, Double_t &C20, Double_t &C21, 
188 Double_t &C22, Double_t &C30, Double_t &C31, Double_t &C32, Double_t &C33, Double_t &C40, 
189 Double_t &C41, Double_t &C42, Double_t &C43, Double_t &C44) const {
190 //Origin  A. Badala' and G.S. Pappalardo:  e-mail Angela.Badala@ct.infn.it, Giuseppe.S.Pappalardo@ct.infn.it
191 // gets elements of covariance matrix
192
193
194   C00=fC00; C10=fC10; C11=fC11; C20=fC20; C21=fC21; C22=fC22; C30=fC30; C31=fC31; C32=fC32; C33=fC33;
195   C40=fC40; C41=fC41; C42=fC42; C43=fC43; C44=fC44; 
196
197 }
198
199 void AliITSTrackV1::GetXElements(Double_t &X0, Double_t &X1, Double_t &X2, Double_t &X3, Double_t &X4) const {
200 //Origin  A. Badala' and G.S. Pappalardo:  e-mail Angela.Badala@ct.infn.it, Giuseppe.S.Pappalardo@ct.infn.it
201 // gets the elements of the state vector
202   X0=fX0; X1=fX1; X2=fX2; X3=fX3; X4=fX4; 
203 }
204
205 void AliITSTrackV1::PutXElements(Double_t X0, Double_t X1, Double_t X2, Double_t X3, Double_t X4){
206 //Origin  A. Badala' and G.S. Pappalardo:  e-mail Angela.Badala@ct.infn.it, Giuseppe.S.Pappalardo@ct.infn.it
207 // puts the elements of the state vector
208   fX0=X0; fX1=X1; fX2=X2; fX3=X3; fX4=X4; 
209 }   
210
211 void AliITSTrackV1::LmTPC() {
212 //Origin  A. Badala' and G.S. Pappalardo:  e-mail Angela.Badala@ct.infn.it, Giuseppe.S.Pappalardo@ct.infn.it
213 // Transform the TPC state vector from TPC-local to master and build a new state vector ITS-type
214 // The covariance matrix is also modified accordingly      
215
216    
217   Double_t alpha = fTPCtrack->GetAlpha(); 
218  
219   //printf("LmTPC: alpha %f\n",alpha); 
220
221   Double_t yTPC   = fTPCtrack->GetY();
222   Double_t zTPC   = fTPCtrack->GetZ();
223   Double_t cTPC   = fTPCtrack->GetC();
224   Double_t etaTPC = fTPCtrack->GetEta();
225   Double_t tglTPC = fTPCtrack->GetTgl();
226   
227
228   Double_t xm, ym, zm;
229   Double_t sina = TMath::Sin(alpha);
230   Double_t cosa = TMath::Cos(alpha);
231   Double_t xl= fTPCtrack->GetX();
232   xm = xl * cosa - yTPC*sina;
233   ym = xl * sina + yTPC*cosa;
234   zm = zTPC;  
235   //cout<<" xl e alpha = "<<xl<<" "<<alpha<<"\n"; getchar(); 
236
237   Double_t x0m,y0m;
238
239   ///////////////////////////////////// determine yo //////////////////////////////////////////////////
240   
241   Double_t vxl=fVertex(0)*cosa+fVertex(1)*sina;
242   Double_t vyl= -fVertex(0)*sina+fVertex(1)*cosa;
243   Double_t xo,yo, signy;
244   Double_t r = 1./cTPC;
245   xo =  etaTPC / cTPC;
246   Double_t yo1, yo2, diffsq1, diffsq2;  
247   yo1 = yTPC +  TMath::Sqrt(r*r - (xl-xo)*(xl-xo)); 
248   yo2 = yTPC -  TMath::Sqrt(r*r - (xl-xo)*(xl-xo));   
249   diffsq1=TMath::Abs((yo1-vyl)*(yo1-vyl)+(xo-vxl)*(xo-vxl)-r*r);
250   diffsq2=TMath::Abs((yo2-vyl)*(yo2- vyl)+(xo-vxl)*(xo-vxl)-r*r);
251   if(diffsq1<diffsq2) {yo=yo1; signy=1.;} else {yo=yo2; signy=-1.;};
252   
253   ////////////////////////////////////////////////////////////////////////////////////////////////////
254
255   x0m = xo * cosa - yo * sina;
256   y0m = xo * sina + yo * cosa;  
257
258   frtrack=TMath::Sqrt(xm*xm+ym*ym);
259         
260   Double_t phi=TMath::ATan2(ym,xm);  if(phi<0) phi=2.*TMath::Pi()+phi;     
261
262   fX0=phi;
263   fX1=zm;
264   fX3=tglTPC;
265   fX4=cTPC;
266
267   
268   Double_t dd=TMath::Sqrt((x0m-fVertex(0))*(x0m-fVertex(0))+(y0m-fVertex(1))*(y0m-fVertex(1)));
269   Double_t signdd;
270   if (r>0) signdd=1.; else signdd=-1.;
271   fX2=signdd*dd-r;
272   //cout<<" fvertex = "<<fVertex(0)<<" "<<fVertex(1)<<" "<<fVertex(2)<<"\n";
273       
274   Double_t cov[15];
275   fTPCtrack->GetCovariance(cov);
276
277   Double_t  dfidy, dDdy, dDdC, dDdeta;
278
279   dfidy=(xm*cosa+ym*sina)/(frtrack*frtrack);
280   dDdy=signdd*((y0m-fVertex(1))*cosa-(x0m-fVertex(0))*sina)/dd;
281   Double_t dyodr=signy*(r+(xl-xo)*etaTPC)/TMath::Sqrt(r*r-(xl-xo)*(xl-xo));
282   Double_t dyomdr=sina*etaTPC+cosa*dyodr;
283   Double_t dxomdr=cosa*etaTPC-sina*dyodr;
284   Double_t ddddR=((x0m-fVertex(0))*dxomdr+(y0m-fVertex(1))*dyomdr)/dd;
285   dDdC=-r*r*(signdd*ddddR-1.);
286   Double_t dyoldxol=signy*(xl-xo)/TMath::Sqrt(r*r-(xl-xo)*(xl-xo));
287   Double_t dxomdeta=r*(cosa-sina*dyoldxol);
288   Double_t dyomdeta=r*(sina+cosa*dyoldxol);
289   dDdeta=signdd*((x0m-fVertex(0))*dxomdeta+(y0m-fVertex(1))*dyomdeta)/dd;
290   
291   Double_t f00=dfidy;
292   Double_t f20=dDdy;    
293   Double_t f22=dDdC;    
294   Double_t f23=dDdeta;
295   
296   Double_t t00=cov[0]*f00;
297   Double_t t02=cov[0]*f20+cov[6]*f22+cov[3]*f23;
298   Double_t t20=cov[6]*f00;
299   Double_t t22=cov[6]*f20+cov[9]*f22+cov[8]*f23;
300   
301   fC00=f00*t00;
302   fC10=cov[1]*f00;
303   fC11=cov[2];
304   fC20=f20*t00+f22*t20+f23*cov[3]*f00;
305   fC21=f20*cov[1]+f22*cov[7]+f23*cov[4];
306   fC22=f20*t02+f22*t22+f23*(cov[3]*f20+cov[8]*f22+cov[5]*f23);
307   fC30=cov[10]*f00;
308   fC31=cov[11];
309   fC32=cov[10]*f20+cov[13]*f22+cov[12]*f23;
310   fC33=cov[14];
311   fC40=t20;
312   fC41=cov[7];
313   fC42=t22;
314   fC43=cov[13];
315   fC44=cov[9];
316   
317    
318 }
319
320
321 AliITSTrackV1 &AliITSTrackV1::operator=(AliITSTrackV1 obj) {
322 //Origin  A. Badala' and G.S. Pappalardo:  e-mail Angela.Badala@ct.infn.it, Giuseppe.S.Pappalardo@ct.infn.it  
323 // assignement operator
324
325   if (flistCluster) {
326     flistCluster->Delete();
327     delete flistCluster;
328   }
329   delete fClusterInTrack;
330   fClusterInTrack = new TMatrix(6,9);
331   flistCluster = new TObjArray; 
332   flabel = obj.flabel;
333   fTPCtrack = obj.fTPCtrack; 
334   fNumClustInTrack = obj.fNumClustInTrack; 
335   fChi2= obj.fChi2;
336   fVertex=obj.fVertex;
337   fErrorVertex=obj.fErrorVertex;
338   fX0=obj.fX0; fX1=obj.fX1; fX2=obj.fX2; fX3=obj.fX3; fX4=obj.fX4;  
339   fLayer=obj.fLayer;
340   frtrack=obj.frtrack;
341   fDv=obj.fDv;
342   fZv=obj.fZv;
343   fsigmaDv=obj.fsigmaDv;
344   fsigmaZv=obj.fsigmaZv;
345   fd2=obj.fd2;
346   ftgl2=obj.ftgl2; 
347   fdtgl=obj.fdtgl;
348   fnoclust=obj.fnoclust; 
349   
350   fC00=obj.fC00; fC10=obj.fC10; fC11=obj.fC11; fC20=obj.fC20; fC21=obj.fC21;
351   fC22=obj.fC22; fC30=obj.fC30; fC31=obj.fC31; fC32=obj.fC32; fC33=obj.fC33; 
352   fC40=obj.fC40; fC41=obj.fC41; fC42=obj.fC42; fC43=obj.fC43; fC44=obj.fC44;
353    
354   fMass=obj.fMass;   
355   
356   *fClusterInTrack = *obj.fClusterInTrack;
357   Int_t i;
358   for(i=0; i<obj.flistCluster->GetSize(); i++) flistCluster->AddLast(obj.flistCluster->At(i));
359
360   return *this;
361   
362 }
363
364 void AliITSTrackV1::PutCluster(Int_t layerc, TVector vecclust) {
365 //Origin  A. Badala' and G.S. Pappalardo:  e-mail Angela.Badala@ct.infn.it, Giuseppe.S.Pappalardo@ct.infn.it
366 // put information for clusters
367   
368   (*fClusterInTrack)(layerc,0) = vecclust(0);
369   (*fClusterInTrack)(layerc,1) = vecclust(1);
370   (*fClusterInTrack)(layerc,2) = vecclust(2);
371   (*fClusterInTrack)(layerc,3) = vecclust(3);
372   (*fClusterInTrack)(layerc,4) = vecclust(4);  
373   (*fClusterInTrack)(layerc,5) = vecclust(5);
374   (*fClusterInTrack)(layerc,6) = vecclust(6);
375   (*fClusterInTrack)(layerc,7) = vecclust(7);
376   (*fClusterInTrack)(layerc,8) = vecclust(8);
377
378 }
379
380
381 void AliITSTrackV1::GetClusters() {
382 //Origin  A. Badala' and G.S. Pappalardo:  e-mail Angela.Badala@ct.infn.it, Giuseppe.S.Pappalardo@ct.infn.it
383 // prints the clusters belonging to the current track
384
385   TMatrix a(*fClusterInTrack);
386   TMatrix b(6,3);
387   Int_t i;
388   for(i=0;i<6; i++){
389    b(i,0)=a(i,6); b(i,1)=a(i,7); b(i,2)=a(i,8); 
390   }
391   a.Print();
392  // b.Print(); 
393  
394 }
395
396
397 TVector AliITSTrackV1::GetLabTrack(Int_t lay) {
398 //Origin  A. Badala' and G.S. Pappalardo:  e-mail Angela.Badala@ct.infn.it, Giuseppe.S.Pappalardo@ct.infn.it
399 // gets the label of the track
400
401   TVector vecLabel(3);
402   vecLabel(0)=( (Float_t) (*fClusterInTrack)(lay,6) );
403   vecLabel(1)=( (Float_t) (*fClusterInTrack)(lay,7) );
404   vecLabel(2)=( (Float_t) (*fClusterInTrack)(lay,8) );
405   return vecLabel;
406 }
407
408 void AliITSTrackV1::Search(TVector VecTotLabref, Long_t &labref, Int_t &freq){
409 //Origin  A. Badala' and G.S. Pappalardo:  e-mail Angela.Badala@ct.infn.it, Giuseppe.S.Pappalardo@ct.infn.it  
410 // define label
411
412   Int_t vecfreq[18];
413   
414   Int_t i,j;
415   for(i=0; i<18; i++) vecfreq[i]=0;
416   
417   for(i=0; i<18; i++) {
418     for(j=0; j<18; j++) {
419      // if(VecTotLabref(i) == 0.) VecTotLabref(i)=-3.;  //commentato il 5-3-2001
420       if( (VecTotLabref(i)>=0.) && (VecTotLabref(i)==VecTotLabref(j)) ) vecfreq[i]++;    
421     }  
422   }
423   Int_t imax=-1000;
424   Long_t  labdefault= (Long_t)1000000.;
425   freq=0;
426   for(i=0; i<18; i++) {
427     if(vecfreq[i]>freq) {freq=vecfreq[i]; imax=i;}  
428   }
429   if(imax<0) labref=labdefault; else labref=(Long_t) VecTotLabref(imax);
430
431
432
433 void AliITSTrackV1::Propagation(Double_t rk) {
434 //Origin  A. Badala' and G.S. Pappalardo:  e-mail Angela.Badala@ct.infn.it, Giuseppe.S.Pappalardo@ct.infn.it  
435 //Propagation of track 
436   Double_t duepi=2.*TMath::Pi();
437   Double_t rkm1=frtrack;
438 //cout<<" rk e rkm1 dentro Propagation   "<<rk<<" "<<rkm1<<"\n";
439   
440   //
441   Double_t aAk=ArgA(rk), aAkm1=ArgA(rkm1);
442   Double_t ak=Arga(rk), akm1=Arga(rkm1);
443   fX0+=TMath::ASin(aAk)-TMath::ASin(aAkm1);
444
445   if(fX0>duepi) fX0-=duepi;
446   if(fX0<0.) fX0+=duepi;
447         
448   Double_t tgl=fX3;
449   Double_t c=fX4;
450   Double_t d=fX2;
451   Double_t cy=c/2;
452   fX1+=tgl/cy*(TMath::ASin(ak)-TMath::ASin(akm1));  
453   frtrack=rk;
454
455         
456   Double_t bk=ArgB(rk), bkm1=ArgB(rkm1);        
457   Double_t ck=ArgC(rk), ckm1=ArgC(rkm1);  
458
459   Double_t f02=ck/TMath::Sqrt(1.-aAk*aAk) - ckm1/TMath::Sqrt(1.-aAkm1*aAkm1);
460   Double_t f04=bk/TMath::Sqrt(1.-aAk*aAk) - bkm1/TMath::Sqrt(1.-aAkm1*aAkm1);   
461   Double_t f12=tgl*d*(1./rk - 1./rkm1);
462   Double_t f13=rk - rkm1;
463   
464
465   Double_t c00=fC00;
466   Double_t c10=fC10;
467   Double_t c11=fC11;
468   Double_t c20=fC20;
469   Double_t c21=fC21;
470   Double_t c22=fC22;
471   Double_t c30=fC30;
472   Double_t c31=fC31;
473   Double_t c32=fC32;
474   Double_t c33=fC33;
475   Double_t c40=fC40;
476   Double_t c41=fC41;
477   Double_t c42=fC42;
478   Double_t c43=fC43;
479   Double_t c44=fC44;
480   
481   Double_t r10=c10+c21*f02+c41*f04;
482   Double_t r20=c20+c22*f02+c42*f04;
483   Double_t r30=c30+c32*f02+c43*f04;
484   Double_t r40=c40+c42*f02+c44*f04;
485   Double_t r21=c21+c22*f12+c32*f13;
486   Double_t r31=c31+c32*f12+c33*f13;                                                                     
487   Double_t r41=c41+c42*f12+c43*f13;
488
489   fC00=c00+c20*f02+c40*f04+f02*r20+f04*r40;
490   fC10=r10+f12*r20+f13*r30;
491   fC11=c11+c21*f12+c31*f13+f12*r21+f13*r31;
492   fC20=r20;
493   fC21=r21;
494   fC30=r30;
495   fC31=r31;
496   fC40=r40;
497   fC41=r41;
498   
499 }
500
501 void AliITSTrackV1::AddEL(AliITSRad *rl, Double_t signdE, Bool_t flagtot, Double_t mass) {
502 //Origin  A. Badala' and G.S. Pappalardo:  e-mail Angela.Badala@ct.infn.it, Giuseppe.S.Pappalardo@ct.infn.it  
503 //  add energy loss
504
505   mass=fMass;  
506   
507   TVector s(6);  
508   s(0)=0.0026+0.00283; s(1)=0.018; s(2)=0.0094; s(3)=0.0095; s(4)=0.0091; s(5)=0.0087;
509   //s(0)=0.0026+0.00283*2.; s(1)=0.018*2.; s(2)=0.0094; s(3)=0.0095; s(4)=0.0091; s(5)=0.0087;
510   //0.00277 is added in the first layer to take into account the energy loss in the beam pipe
511
512   //for(int k=0; k<6; k++) cout<<s(k)<<" "; cout<<"\n";
513   Int_t k;
514   //for(k=0; k<6; k++) s(k)=s(k)*1.6;    //forint
515   for(k=0; k<6; k++) s(k)=s(k)*1.7;    //forint
516   
517   Double_t phi=fX0;
518   
519   if(phi<0.174 ) s(5)=s(5)+0.012; 
520   if(phi>6.1 ) s(5)=s(5)+0.012; // to take into account rail 
521   if(phi>2.96 && phi<3.31 ) s(5)=s(5)+0.012;   
522   
523   /* 
524   Double_t tgl=fX3;
525   Double_t theta=((TMath::Pi())/2.)-TMath::ATan(tgl);
526   //phi*=180./TMath::Pi();
527   //theta*=180./TMath::Pi();
528   //Double_t rad90=(TMath::Pi())/2.;
529   Double_t rad40=(TMath::Pi())*40./180.;
530   Double_t rad100=(TMath::Pi())*100/180;
531   Double_t rad360=(TMath::Pi())*2.;
532   Int_t imax=rl->Getimax();
533   Int_t jmax=rl->Getjmax();
534   Int_t i=(Int_t) ( (theta-rad40)/rad100*imax);
535   Int_t j=(Int_t) ( phi/rad360*jmax );
536   //Int_t i=(Int_t)( ((theta-((TMath::Pi())/4.))/((TMath::Pi())/2.))*imax );
537   //Int_t j=(Int_t)( (phi/((TMath::Pi())*2.))*jmax );
538   if(i<0) i=0;
539   if(i>=imax) i=imax-1;
540   if(j<0) j=0;
541   if(j>=jmax) j=jmax-1;
542   
543   s(0) = 0.0028/TMath::Sin(theta)+( rl->GetRadMatrix1() )(i,j);   // 0.0028 takes into account the beam pipe
544   s(1) = ( rl->GetRadMatrix2() )(i,j);
545   s(2) = ( rl->GetRadMatrix3() )(i,j);
546   s(3) = ( rl->GetRadMatrix4() )(i,j);
547   s(4) = ( rl->GetRadMatrix5() )(i,j);
548   s(5) = ( rl->GetRadMatrix6() )(i,j);
549   
550
551   */  
552   
553   //for(int k=0; k<6; k++) cout<<s(k)<<" "; getchar();
554     
555   //if(phi>60) {cout<<" phi = "<<phi<<"\n"; getchar();}
556   //if(theta<45 || theta>135) {cout<<" theta = "<<theta<<"\n"; getchar();}
557   //cout<<" dentro AddEl: phi, theta = "<<phi<<" "<<theta<<"\n"; getchar(); 
558     
559   Double_t cl=1.+fX3*fX3;  // cl=1/(cosl)**2 = 1 + (tgl)**2
560   Double_t sqcl=TMath::Sqrt(cl);
561   Double_t pt=GetPt();
562      
563   Double_t p2=pt*pt*cl;
564   Double_t e=TMath::Sqrt(p2+mass*mass);
565   Double_t beta2=p2/(p2+mass*mass);
566   
567   Double_t dE;
568   if(flagtot) {
569     Double_t stot=s(0)+s(1)+s(2)+s(3)+s(4)+s(5);
570     dE=0.153/beta2*(log(5940*beta2/(1-beta2)) - beta2)*stot*21.82*sqcl;
571   } else {
572     dE=0.153/beta2*(log(5940*beta2/(1-beta2)) - beta2)*s(fLayer-1)*21.82*sqcl;
573   }   
574   dE=signdE*dE/1000.; 
575           
576   e+=dE;
577   Double_t p=TMath::Sqrt(e*e-mass*mass);   
578   Double_t sign=1.;
579   if(fX4 < 0.) sign=-1.; 
580   pt=sign*p/sqcl; 
581   Double_t cc=(0.299792458*0.2*fFieldFactor)/(pt*100.);
582   fX4=cc;
583   
584 }
585
586 void  AliITSTrackV1::Correct(Double_t rk) {
587 //Origin  A. Badala' and G.S. Pappalardo:  e-mail Angela.Badala@ct.infn.it, Giuseppe.S.Pappalardo@ct.infn.it  
588 // correct track to take into account real geometry detector
589
590   Double_t duepi=2.*TMath::Pi();
591   Double_t rkm1=frtrack;
592   Double_t aAk=ArgA(rk), aAkm1=ArgA(rkm1);
593   Double_t ak=Arga(rk), akm1=Arga(rkm1);
594
595   fX0+=TMath::ASin(aAk)-TMath::ASin(aAkm1);
596   if(fX0>duepi) fX0-=duepi;
597   if(fX0<0.) fX0+=duepi;
598         
599   Double_t tgl=fX3;
600   Double_t c=fX4;
601   Double_t cy=c/2;
602   fX1+=tgl/cy*(TMath::ASin(ak)-TMath::ASin(akm1));
603   frtrack=rk;
604                 
605 }
606
607 void AliITSTrackV1::AddMS(AliITSRad *rl, Double_t mass) {
608 //Origin  A. Badala' and G.S. Pappalardo:  e-mail Angela.Badala@ct.infn.it, Giuseppe.S.Pappalardo@ct.infn.it       
609 //////////   Modification of the covariance matrix to take into account multiple scattering  ///////////     
610    
611   mass=fMass;
612  
613   TVector s(6);
614
615   s(0)=0.0026+0.00283; s(1)=0.018; s(2)=0.0094; s(3)=0.0095; s(4)=0.0091; s(5)=0.0087;
616  //  s(0)=0.0026+0.00283*2.; s(1)=0.018*2.; s(2)=0.0094; s(3)=0.0095; s(4)=0.0091; s(5)=0.0087;
617 //0.00277 is added in the first layer to take into account the energy loss in the beam pipe 
618
619   Int_t k;
620   //for(k=0; k<6; k++) s(k)=s(k)*1.6;   // forint
621   for(k=0; k<6; k++) s(k)=s(k)*1.7;   // forint
622   
623   Double_t phi=fX0;
624  if(phi<0.174 ) s(5)=s(5)+0.012; //aggiunta provvisoria
625   if(phi>6.1 ) s(5)=s(5)+0.012; //aggiunta provvisoria  
626    if(phi>2.96 && phi< 3.31) s(5)=s(5)+0.012; //aggiunta provvisoria 
627   
628       
629   Double_t tgl=fX3;
630   /*
631   Double_t theta=((TMath::Pi())/2.)-TMath::ATan(tgl);
632   Double_t rad40=(TMath::Pi())*40./180.;      // rivedere
633   Double_t rad100=(TMath::Pi())*100/180;
634   Double_t rad360=(TMath::Pi())*2.;
635   Int_t imax=rl->Getimax();
636   Int_t jmax=rl->Getjmax();
637   Int_t i=(Int_t) ( (theta-rad40)/rad100*imax);
638   Int_t j=(Int_t) ( phi/rad360*jmax);
639
640   if(i<0) i=0;
641   if(i>=imax) i=imax-1;
642   if(j<0) j=0;
643   if(j>=jmax) j=jmax-1;
644   
645   s(0) = 0.0028/TMath::Sin(theta)+( rl->GetRadMatrix1() )(i,j);   // 0.0028 takes into account the beam pipe
646   s(1) = ( rl->GetRadMatrix2() )(i,j);
647   s(2) = ( rl->GetRadMatrix3() )(i,j);
648   s(3) = ( rl->GetRadMatrix4() )(i,j);
649   s(4) = ( rl->GetRadMatrix5() )(i,j);
650   s(5) = ( rl->GetRadMatrix6() )(i,j);
651    */   
652   //Double_t mass=0.1396;
653   Int_t layer=(Int_t)GetLayer();
654   
655   Double_t cosl=TMath::Cos(TMath::ATan(tgl));   
656   Double_t d=fX2;
657   Double_t c=fX4;
658   Double_t cy=c/2.;
659   Double_t q20=1./(cosl*cosl);   
660   Double_t q30=c*tgl;
661    
662   Double_t q40=cy*(frtrack*frtrack-d*d)/(1.+ 2.*cy*d);
663   Double_t dd=d+cy*d*d-cy*frtrack*frtrack;
664   Double_t dprova=frtrack*frtrack - dd*dd;
665   Double_t q41=0.;
666   if(dprova>0.) q41=-1./cosl*TMath::Sqrt(dprova)/(1.+ 2.*cy*d);
667                 
668   Double_t p2=(GetPt()*GetPt())/(cosl*cosl);
669   Double_t beta2=p2/(p2+mass*mass);
670  // Double_t theta2=14.1*14.1/(beta2*p2*1.e6)*(s(layer-1)/cosl);
671    Double_t theta2=14.1*14.1/(beta2*p2*1.e6)*(s(layer-1)/TMath::Abs(cosl));
672
673   fC22+=theta2*(q40*q40+q41*q41);
674   fC32+=theta2*q20*q40;
675   fC33+=theta2*q20*q20;
676   fC42+=theta2*q30*q40;
677   fC43+=theta2*q30*q20;
678   fC44+=theta2*q30*q30;
679     
680 }
681 void AliITSTrackV1::PrimaryTrack(AliITSRad *rl) {
682 //Origin  A. Badala' and G.S. Pappalardo:  e-mail Angela.Badala@ct.infn.it, Giuseppe.S.Pappalardo@ct.infn.it  
683 // calculation of part of covariance matrix for vertex constraint
684
685   Double_t rlayer[6];
686         
687   rlayer[0]=4.; rlayer[1]=7.;  rlayer[2]=14.9;  rlayer[3]=23.8;  
688   rlayer[4]=39.1;  rlayer[5]=43.6;
689         
690   Double_t cy=fX4/2.;   
691   Double_t tgl=(fX1-fZv)*cy/TMath::ASin(cy*frtrack);
692   Double_t frtrack=1.;  
693   fX0=0.;
694   fX1=frtrack*tgl; 
695   fX2=fDv;
696   fX3=tgl; 
697
698   fC00=fC10=fC11=fC20=fC21=fC22=fC30=fC31=fC32=fC33=fC40=fC41=fC42=fC43=0.;
699
700   AddEL(rl,1.,1);
701   fLayer=0;
702   Int_t i;
703   for (i=0; i<6; i++) {
704     Propagation(rlayer[i]);
705     fLayer++;
706     fd2(i)=fC22;
707     ftgl2(i)=fC33;
708     fdtgl(i)=fC32; 
709     AddMS(rl);    
710     AddEL(rl,-1,0);        
711   }             
712 }       
713  
714   
715 Int_t AliITSTrackV1::DoNotCross(Double_t rk) const{
716 //Origin  A. Badala' and G.S. Pappalardo:  e-mail Angela.Badala@ct.infn.it, Giuseppe.S.Pappalardo@ct.infn.it  
717 // determine if the track cross a layer
718
719   Double_t c=fX4;
720   Double_t d=fX2;
721   Double_t cy=c/2.;
722   return (TMath::Abs((cy*rk+(1.+cy*d)*d/rk)/(1.+2.*cy*d))>=1.)?1:0;
723 }
724   
725
726 Double_t AliITSTrackV1::ArgA(Double_t rk) const {
727 //Origin  A. Badala' and G.S. Pappalardo:  e-mail Angela.Badala@ct.infn.it, Giuseppe.S.Pappalardo@ct.infn.it
728 // quantity usefull in Propagation  
729   Double_t c=fX4;
730   Double_t d=fX2;
731   Double_t cy=c/2.;
732   Double_t arg=(cy*rk + (1 + cy*d)*d/rk)/(1.+ 2.*cy*d);         
733   if (TMath::Abs(arg) < 1.) return arg;
734   //cout<<"class AliITSTrack: ArgA out of range !\n";/* getchar();*/
735   return (arg>0) ? 0.99999999999 : -0.9999999999;
736 }
737    
738 Double_t AliITSTrackV1::Arga(Double_t rk) const {
739 //Origin  A. Badala' and G.S. Pappalardo:  e-mail Angela.Badala@ct.infn.it, Giuseppe.S.Pappalardo@ct.infn.it
740 // quantity usefull in Propagation 
741   Double_t c=fX4;
742   Double_t d=fX2;
743   Double_t cy=c/2.;
744   Double_t arg=(rk*rk - d*d)/(1.+ 2.*cy*d);             
745   if (arg<0.) {/*cout<<"class AliITSTrack: Arga out of range !\n";*/ arg=0.;} 
746   return cy*TMath::Sqrt(arg);
747 }
748         
749 Double_t AliITSTrackV1::ArgB(Double_t rk) const {
750 //Origin  A. Badala' and G.S. Pappalardo:  e-mail Angela.Badala@ct.infn.it, Giuseppe.S.Pappalardo@ct.infn.it
751 // quantity usefull in Propagation 
752   Double_t c=fX4;
753   Double_t d=fX2;
754   Double_t cy=c/2.;        
755   return (rk*rk - d*d)/(rk*(1.+ 2.*cy*d)*(1.+ 2.*cy*d));
756 }
757    
758 Double_t AliITSTrackV1::ArgC(Double_t rk) const {
759 //Origin  A. Badala' and G.S. Pappalardo:  e-mail Angela.Badala@ct.infn.it, Giuseppe.S.Pappalardo@ct.infn.it
760 // quantity usefull in Propagation 
761   Double_t c=fX4;
762   Double_t d=fX2;
763   Double_t cy=c/2.;             
764   return  (1./rk - 2.*cy*ArgA(rk)/(1.+ 2.*cy*d));
765 }
766
767
768 Double_t AliITSTrackV1::GetPredChi2(Double_t m[2], Double_t sigma[2] ) const {
769 //Origin  A. Badala' and G.S. Pappalardo:  e-mail Angela.Badala@ct.infn.it, Giuseppe.S.Pappalardo@ct.infn.it
770 // This function calculates a predicted chi2 increment.
771  
772   Double_t r00=sigma[0], r01=0., r11=sigma[1];
773   r00+=fC00; r01+=fC10; r11+=fC11;
774   Double_t det=r00*r11-r01*r01;
775   if(TMath::Abs(det) < 1.e-15) {cout<<" Problems on matrix in GetPredChi2 "<<det<<"\n";
776   return 1e10;}
777   Double_t tmp=r00; r00=r11; r11=tmp; r01=-r01;
778   Double_t dphi=m[0]-fX0; 
779   Double_t dz=m[1]-fX1;
780   Double_t chi2 = (dphi*r00*dphi +2.*r01*dphi*dz + dz*r11*dz)/det;
781   return chi2;
782   
783 }