/[PAMELA software]/DarthVader/TrackerLevel2/src/TrkLevel2.cpp
ViewVC logotype

Contents of /DarthVader/TrackerLevel2/src/TrkLevel2.cpp

Parent Directory Parent Directory | Revision Log Revision Log


Revision 1.26 - (show annotations) (download)
Fri Feb 16 14:56:01 2007 UTC (17 years, 9 months ago) by pam-fi
Branch: MAIN
Changes since 1.25: +125 -104 lines
Magnetic field, improoved de/dx, reprocessing tools

1 /**
2 * \file TrkLevel2.cpp
3 * \author Elena Vannuccini
4 */
5 #include <TrkLevel2.h>
6 #include <iostream>
7 #include <math.h>
8 using namespace std;
9 //......................................
10 // F77 routines
11 //......................................
12 extern "C" {
13 void dotrack_(int*, double*, double*, double*, double*, int*);
14 void dotrack2_(int*, double*, double*, double*, double*,double*, double*, double*,int*);
15 // int readb_(const char*);
16 // int readb_();
17 void mini2_(int*,int*,int*);
18 void guess_();
19 void gufld_(float*, float*);
20 }
21
22 //--------------------------------------
23 //
24 //
25 //--------------------------------------
26 TrkTrack::TrkTrack(){
27 // cout << "TrkTrack::TrkTrack()" << endl;
28 seqno = -1;
29 image = -1;
30 chi2 = 0;
31 nstep = 0;
32 for(int it1=0;it1<5;it1++){
33 al[it1] = 0;
34 for(int it2=0;it2<5;it2++)coval[it1][it2] = 0;
35 };
36 for(int ip=0;ip<6;ip++){
37 xgood[ip] = 0;
38 ygood[ip] = 0;
39 xm[ip] = 0;
40 ym[ip] = 0;
41 zm[ip] = 0;
42 resx[ip] = 0;
43 resy[ip] = 0;
44 xv[ip] = 0;
45 yv[ip] = 0;
46 zv[ip] = 0;
47 axv[ip] = 0;
48 ayv[ip] = 0;
49 dedx_x[ip] = 0;
50 dedx_y[ip] = 0;
51 };
52 clx = 0;
53 cly = 0;
54 // clx = new TRefArray(6,0);
55 // cly = new TRefArray(6,0);
56 };
57 //--------------------------------------
58 //
59 //
60 //--------------------------------------
61 TrkTrack::TrkTrack(const TrkTrack& t){
62 seqno = t.seqno;
63 image = t.image;
64 chi2 = t.chi2;
65 nstep = t.nstep;
66 for(int it1=0;it1<5;it1++){
67 al[it1] = t.al[it1];
68 for(int it2=0;it2<5;it2++)coval[it1][it2] = t.coval[it1][it2];
69 };
70 for(int ip=0;ip<6;ip++){
71 xgood[ip] = t.xgood[ip];
72 ygood[ip] = t.ygood[ip];
73 xm[ip] = t.xm[ip];
74 ym[ip] = t.ym[ip];
75 zm[ip] = t.zm[ip];
76 resx[ip] = t.resx[ip];
77 resy[ip] = t.resy[ip];
78 xv[ip] = t.xv[ip];
79 yv[ip] = t.yv[ip];
80 zv[ip] = t.zv[ip];
81 axv[ip] = t.axv[ip];
82 ayv[ip] = t.ayv[ip];
83 dedx_x[ip] = t.dedx_x[ip];
84 dedx_y[ip] = t.dedx_y[ip];
85 };
86 clx = 0;
87 cly = 0;
88 if(t.clx)clx = new TRefArray(*(t.clx));
89 if(t.cly)cly = new TRefArray(*(t.cly));
90
91 };
92 //--------------------------------------
93 //
94 //
95 //--------------------------------------
96 void TrkTrack::Copy(TrkTrack& t){
97
98 t.seqno = seqno;
99 t.image = image;
100 t.chi2 = chi2;
101 t.nstep = nstep;
102 for(int it1=0;it1<5;it1++){
103 t.al[it1] = al[it1];
104 for(int it2=0;it2<5;it2++)t.coval[it1][it2] = coval[it1][it2];
105 };
106 for(int ip=0;ip<6;ip++){
107 t.xgood[ip] = xgood[ip];
108 t.ygood[ip] = ygood[ip];
109 t.xm[ip] = xm[ip];
110 t.ym[ip] = ym[ip];
111 t.zm[ip] = zm[ip];
112 t.resx[ip] = resx[ip];
113 t.resy[ip] = resy[ip];
114 t.xv[ip] = xv[ip];
115 t.yv[ip] = yv[ip];
116 t.zv[ip] = zv[ip];
117 t.axv[ip] = axv[ip];
118 t.ayv[ip] = ayv[ip];
119 t.dedx_x[ip] = dedx_x[ip];
120 t.dedx_y[ip] = dedx_y[ip];
121
122 };
123
124 };
125 //--------------------------------------
126 //
127 //
128 //--------------------------------------
129 /**
130 * Evaluates the trajectory in the apparatus associated to the track.
131 * It integrates the equations of motion in the magnetic field. The magnetic field should be previously loaded ( by calling TrkLevel2::LoadField() ), otherwise an error message is returned.
132 * @param t pointer to an object of the class Trajectory,
133 * which z coordinates should be previously initialized by calling the proper constructor ( Trajectory::Trajectory(int n, float* zin) ).
134 * @return error flag.
135 */
136 int TrkTrack::DoTrack(Trajectory* t){
137
138 double *dxout = new double[t->npoint];
139 double *dyout = new double[t->npoint];
140 double *dzin = new double[t->npoint];
141 double dal[5];
142
143 int ifail = 0;
144
145 for (int i=0; i<5; i++) dal[i] = (double)al[i];
146 for (int i=0; i<t->npoint; i++) dzin[i] = (double)t->z[i];
147
148 TrkParams::Load(1);
149 if( !TrkParams::IsLoaded(1) ){
150 cout << "int TrkTrack::DoTrack(Trajectory* t) --- ERROR --- m.field not loaded"<<endl;
151 return 0;
152 }
153 dotrack_(&(t->npoint),dzin,dxout,dyout,dal,&ifail);
154
155 for (int i=0; i<t->npoint; i++){
156 t->x[i] = (float)*dxout++;
157 t->y[i] = (float)*dyout++;
158 }
159
160 // delete [] dxout;
161 // delete [] dyout;
162 // delete [] dzin;
163
164 return ifail;
165 };
166 //--------------------------------------
167 //
168 //
169 //--------------------------------------
170 /**
171 * Evaluates the trajectory in the apparatus associated to the track.
172 * It integrates the equations of motion in the magnetic field. The magnetic field should be previously loaded ( by calling TrkLevel2::LoadField() ), otherwise an error message is returned.
173 * @param t pointer to an object of the class Trajectory,
174 * which z coordinates should be previously initialized by calling the proper constructor ( Trajectory::Trajectory(int n, float* zin) ).
175 * @return error flag.
176 */
177 int TrkTrack::DoTrack2(Trajectory* t){
178
179 double *dxout = new double[t->npoint];
180 double *dyout = new double[t->npoint];
181 double *dthxout = new double[t->npoint];
182 double *dthyout = new double[t->npoint];
183 double *dtlout = new double[t->npoint];
184 double *dzin = new double[t->npoint];
185 double dal[5];
186
187 int ifail = 0;
188
189 for (int i=0; i<5; i++) dal[i] = (double)al[i];
190 for (int i=0; i<t->npoint; i++) dzin[i] = (double)t->z[i];
191
192 TrkParams::Load(1);
193 if( !TrkParams::IsLoaded(1) ){
194 cout << "int TrkTrack::DoTrack2(Trajectory* t) --- ERROR --- m.field not loaded"<<endl;
195 return 0;
196 }
197 dotrack2_(&(t->npoint),dzin,dxout,dyout,dthxout,dthyout,dtlout,dal,&ifail);
198
199 for (int i=0; i<t->npoint; i++){
200 t->x[i] = (float)*dxout++;
201 t->y[i] = (float)*dyout++;
202 t->thx[i] = (float)*dthxout++;
203 t->thy[i] = (float)*dthyout++;
204 t->tl[i] = (float)*dtlout++;
205 }
206
207 // delete [] dxout;
208 // delete [] dyout;
209 // delete [] dzin;
210
211 return ifail;
212 };
213 //--------------------------------------
214 //
215 //
216 //--------------------------------------
217 //float TrkTrack::BdL(){
218 //};
219 //--------------------------------------
220 //
221 //
222 //--------------------------------------
223 Float_t TrkTrack::GetRigidity(){
224 Float_t rig=0;
225 if(chi2>0)rig=1./al[4];
226 if(rig<0) rig=-rig;
227 return rig;
228 };
229 //
230 Float_t TrkTrack::GetDeflection(){
231 Float_t def=0;
232 if(chi2>0)def=al[4];
233 return def;
234 };
235 //
236 Float_t TrkTrack::GetDEDX(){
237 Float_t dedx=0;
238 for(Int_t i=0; i<6; i++)dedx+=dedx_x[i]*xgood[i]+dedx_y[i]*ygood[i];
239 dedx = dedx/(this->GetNX()+this->GetNY());
240 return dedx;
241 };
242
243 //--------------------------------------
244 //
245 //
246 //--------------------------------------
247 void TrkTrack::Dump(){
248 cout << endl << "========== Track " ;
249 cout << endl << "seq. n. : "<< seqno;
250 cout << endl << "image n. : "<< image;
251 cout << endl << "al : "; for(int i=0; i<5; i++)cout << al[i] << " ";
252 cout << endl << "chi^2 : "<< chi2;
253 cout << endl << "n.step : "<< nstep;
254 cout << endl << "xgood : "; for(int i=0; i<6; i++)cout << xgood[i] ;
255 cout << endl << "ygood : "; for(int i=0; i<6; i++)cout << ygood[i] ;
256 cout << endl << "xm : "; for(int i=0; i<6; i++)cout << xm[i] << " ";
257 cout << endl << "ym : "; for(int i=0; i<6; i++)cout << ym[i] << " ";
258 cout << endl << "zm : "; for(int i=0; i<6; i++)cout << zm[i] << " ";
259 cout << endl << "xv : "; for(int i=0; i<6; i++)cout << xv[i] << " ";
260 cout << endl << "yv : "; for(int i=0; i<6; i++)cout << yv[i] << " ";
261 cout << endl << "zv : "; for(int i=0; i<6; i++)cout << zv[i] << " ";
262 cout << endl << "resx : "; for(int i=0; i<6; i++)cout << resx[i] << " ";
263 cout << endl << "resy : "; for(int i=0; i<6; i++)cout << resy[i] << " ";
264 cout << endl << "coval : "; for(int i=0; i<5; i++)cout << coval[0][i]<<" ";
265 cout << endl << " "; for(int i=0; i<5; i++)cout << coval[1][i]<<" ";
266 cout << endl << " "; for(int i=0; i<5; i++)cout << coval[2][i]<<" ";
267 cout << endl << " "; for(int i=0; i<5; i++)cout << coval[3][i]<<" ";
268 cout << endl << " "; for(int i=0; i<5; i++)cout << coval[4][i]<<" ";
269 cout << endl << "dedx_x : "; for(int i=0; i<6; i++)cout << dedx_x[i] << " ";
270 cout << endl << "dedx_y : "; for(int i=0; i<6; i++)cout << dedx_y[i] << " ";
271 cout << endl;
272 }
273 /**
274 * Set the TrkTrack position measurements
275 */
276 void TrkTrack::SetMeasure(double *xmeas, double *ymeas, double *zmeas){
277 for(int i=0; i<6; i++) xm[i]=*xmeas++;
278 for(int i=0; i<6; i++) ym[i]=*ymeas++;
279 for(int i=0; i<6; i++) zm[i]=*zmeas++;
280 }
281 /**
282 * Set the TrkTrack position resolution
283 */
284 void TrkTrack::SetResolution(double *rx, double *ry){
285 for(int i=0; i<6; i++) resx[i]=*rx++;
286 for(int i=0; i<6; i++) resy[i]=*ry++;
287 }
288 /**
289 * Set the TrkTrack good measurement
290 */
291 void TrkTrack::SetGood(int *xg, int *yg){
292 for(int i=0; i<6; i++) xgood[i]=*xg++;
293 for(int i=0; i<6; i++) ygood[i]=*yg++;
294 }
295
296 /**
297 * Load the magnetic field
298 */
299 void TrkTrack::LoadField(TString path){
300
301 // strcpy(path_.path,path.Data());
302 // path_.pathlen = path.Length();
303 // path_.error = 0;
304 // readb_();
305
306 TrkParams::Set(path,1);
307
308 };
309
310
311 /**
312 * Method to fill minimization-routine common
313 */
314 void TrkTrack::FillMiniStruct(cMini2track& track){
315
316 for(int i=0; i<6; i++){
317
318 track.xgood[i]=xgood[i];
319 track.ygood[i]=ygood[i];
320
321 track.xm[i]=xm[i];
322 track.ym[i]=ym[i];
323 track.zm[i]=zm[i];
324
325 // --- temporaneo ----------------------------
326 // andrebbe inserita la dimensione del sensore
327 float segment = 100.;
328 track.xm_a[i]=xm[i];
329 track.xm_b[i]=xm[i];
330 track.ym_a[i]=ym[i];
331 track.ym_b[i]=ym[i];
332 if( xgood[i] && !ygood[i] ){
333 track.ym_a[i] = track.ym_a[i]+segment;
334 track.ym_b[i] = track.ym_b[i]-segment;
335 }else if( !xgood[i] && ygood[i]){
336 track.xm_a[i] = track.xm_a[i]+segment;
337 track.xm_b[i] = track.xm_b[i]-segment;
338 }
339 // --- temporaneo ----------------------------
340
341 track.resx[i]=resx[i];
342 track.resy[i]=resy[i];
343 }
344
345 for(int i=0; i<5; i++) track.al[i]=al[i];
346 track.zini = 23.5;
347 // ZINI = 23.5 !!! it should be the same parameter in all codes
348
349 }
350 /**
351 * Method to set values from minimization-routine common
352 */
353 void TrkTrack::SetFromMiniStruct(cMini2track *track){
354
355 for(int i=0; i<5; i++) {
356 al[i]=track->al[i];
357 for(int j=0; j<5; j++) coval[i][j]=track->cov[i][j];
358 }
359 chi2 = track->chi2;
360 nstep = track->nstep;
361 for(int i=0; i<6; i++){
362 xv[i] = track->xv[i];
363 yv[i] = track->yv[i];
364 zv[i] = track->zv[i];
365 xm[i] = track->xm[i];
366 ym[i] = track->ym[i];
367 zm[i] = track->zm[i];
368 axv[i] = track->axv[i];
369 ayv[i] = track->ayv[i];
370 }
371
372 }
373 /**
374 * Tracking method. It calls F77 mini routine.
375 */
376 void TrkTrack::Fit(double pfixed, int& fail, int iprint){
377
378 float al_ini[] = {0.,0.,0.,0.,0.};
379
380 extern cMini2track track_;
381 fail = 0;
382 FillMiniStruct(track_);
383
384 // if fit variables have been reset, evaluate the initial guess
385 if(al[0]==-9999.&&al[1]==-9999.&&al[2]==-9999.&&al[3]==-9999.&&al[4]==-9999.)guess_();
386
387 // --------------------- free momentum
388 if(pfixed==0.) {
389 track_.pfixed=0.;
390 }
391 // --------------------- fixed momentum
392 if(pfixed!=0.) {
393 al[4]=1./pfixed;
394 track_.pfixed=pfixed;
395 }
396
397 // store temporarily the initial guess
398 for(int i=0; i<5; i++) al_ini[i]=track_.al[i];
399
400 // ------------------------------------------
401 // call mini routine
402 TrkParams::Load(1);
403 if( !TrkParams::IsLoaded(1) ){
404 cout << "void TrkTrack::Fit(double pfixed, int& fail, int iprint) --- ERROR --- m.field not loaded"<<endl;
405 return;
406 }
407 int istep=0;
408 int ifail=0;
409 mini2_(&istep,&ifail, &iprint);
410 if(ifail!=0) {
411 if(iprint)cout << "ERROR: ifail= " << ifail << endl;
412 fail = 1;
413 }
414 // ------------------------------------------
415
416 SetFromMiniStruct(&track_);
417 // cout << endl << "eta ===> " << track_.al[4] << endl;
418
419 // for(int i=0; i<5; i++) al[i]=track_.al[i];
420 // chi2=track_.chi2;
421 // nstep=track_.nstep;
422 // for(int i=0; i<6; i++) xv[i]=track_.xv[i];
423 // for(int i=0; i<6; i++) yv[i]=track_.yv[i];
424 // for(int i=0; i<6; i++) zv[i]=track_.zv[i];
425 // for(int i=0; i<6; i++) axv[i]=track_.axv[i];
426 // for(int i=0; i<6; i++) ayv[i]=track_.ayv[i];
427 // for(int i=0; i<5; i++) {
428 // for(int j=0; j<5; j++) coval[i][j]=track_.cov[i][j];
429 // }
430
431 if(fail){
432 if(iprint)cout << " >>>> fit failed >>>> drawing initial par"<<endl;
433 for(int i=0; i<5; i++) al[i]=al_ini[i];
434 }
435
436 };
437 /*
438 * Reset the fit parameters
439 */
440 void TrkTrack::FitReset(){
441 for(int i=0; i<5; i++) al[i]=-9999.;
442 chi2=0.;
443 nstep=0;
444 for(int i=0; i<6; i++) xv[i]=0.;
445 for(int i=0; i<6; i++) yv[i]=0.;
446 for(int i=0; i<6; i++) zv[i]=0.;
447 for(int i=0; i<6; i++) axv[i]=0.;
448 for(int i=0; i<6; i++) ayv[i]=0.;
449 for(int i=0; i<5; i++) {
450 for(int j=0; j<5; j++) coval[i][j]=0.;
451 }
452 }
453
454 //--------------------------------------
455 //
456 //
457 //--------------------------------------
458 void TrkTrack::Clear(){
459 // cout << "TrkTrack::Clear()"<<endl;
460 seqno = -1;
461 image = -1;
462 chi2 = 0;
463 nstep = 0;
464 for(int it1=0;it1<5;it1++){
465 al[it1] = 0;
466 for(int it2=0;it2<5;it2++)coval[it1][it2] = 0;
467 };
468 for(int ip=0;ip<6;ip++){
469 xgood[ip] = 0;
470 ygood[ip] = 0;
471 xm[ip] = 0;
472 ym[ip] = 0;
473 zm[ip] = 0;
474 resx[ip] = 0;
475 resy[ip] = 0;
476 xv[ip] = 0;
477 yv[ip] = 0;
478 zv[ip] = 0;
479 axv[ip] = 0;
480 ayv[ip] = 0;
481 dedx_x[ip] = 0;
482 dedx_y[ip] = 0;
483
484 };
485 if(clx)clx->Clear();
486 if(cly)cly->Clear();
487 };
488 //--------------------------------------
489 //
490 //
491 //--------------------------------------
492 void TrkTrack::Delete(){
493 // cout << "TrkTrack::Delete()"<<endl;
494 // Clear();
495 if(clx)delete clx;
496 if(cly)delete cly;
497 };
498 //--------------------------------------
499 //
500 //
501 //--------------------------------------
502
503 //--------------------------------------
504 //
505 //
506 //--------------------------------------
507 TrkSinglet::TrkSinglet(){
508 // cout << "TrkSinglet::TrkSinglet() " << GetUniqueID()<<endl;
509 plane = 0;
510 coord[0] = 0;
511 coord[1] = 0;
512 sgnl = 0;
513 // cls = 0;
514 };
515 //--------------------------------------
516 //
517 //
518 //--------------------------------------
519 TrkSinglet::TrkSinglet(const TrkSinglet& s){
520 // cout << "TrkSinglet::TrkSinglet(const TrkSinglet& s) " << GetUniqueID()<<endl;
521 plane = s.plane;
522 coord[0] = s.coord[0];
523 coord[1] = s.coord[1];
524 sgnl = s.sgnl;
525 // cls = 0;//<<<<pointer
526 cls = TRef(s.cls);
527 };
528 //--------------------------------------
529 //
530 //
531 //--------------------------------------
532 void TrkSinglet::Dump(){
533 int i=0;
534 cout << endl << "========== Singlet " ;
535 cout << endl << "plane : " << plane;
536 cout << endl << "coord[2] : "; while( i<2 && cout << coord[i] << " ") i++;
537 cout << endl << "sgnl : " << sgnl;
538 }
539 //--------------------------------------
540 //
541 //
542 //--------------------------------------
543 void TrkSinglet::Clear(){
544 // cout << "TrkSinglet::Clear() " << GetUniqueID()<<endl;
545 // cls=0;
546 plane=-1;
547 coord[0]=-999;
548 coord[1]=-999;
549 sgnl=0;
550
551 }
552 //--------------------------------------
553 //
554 //
555 //--------------------------------------
556 TrkLevel2::TrkLevel2(){
557 // cout <<"TrkLevel2::TrkLevel2()"<<endl;
558 for(Int_t i=0; i<12 ; i++){
559 good[i] = -1;
560 };
561 // okkio!! memory-leak
562 // Track = new TClonesArray("TrkTrack");
563 // SingletX = new TClonesArray("TrkSinglet");
564 // SingletY = new TClonesArray("TrkSinglet");
565 Track = 0;
566 SingletX = 0;
567 SingletY = 0;
568
569 }
570 //--------------------------------------
571 //
572 //
573 //--------------------------------------
574 void TrkLevel2::Set(){
575 if(!Track)Track = new TClonesArray("TrkTrack");
576 if(!SingletX)SingletX = new TClonesArray("TrkSinglet");
577 if(!SingletY)SingletY = new TClonesArray("TrkSinglet");
578 }
579 //--------------------------------------
580 //
581 //
582 //--------------------------------------
583 void TrkLevel2::Dump(){
584
585 //
586 cout << endl << endl << "=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-";
587 cout << endl << "good : "; for(int i=0; i<12; i++) cout << good[i]<<" ";
588 cout << endl << "ntrk() : " << this->ntrk() ;
589 cout << endl << "nclsx() : " << this->nclsx();
590 cout << endl << "nclsy() : " << this->nclsy();
591 // TClonesArray &t = *Track;
592 // TClonesArray &sx = *SingletX;
593 // TClonesArray &sy = *SingletY;
594 // for(int i=0; i<ntrk(); i++) ((TrkTrack *)t[i])->Dump();
595 // for(int i=0; i<nclsx(); i++) ((TrkSinglet *)sx[i])->Dump();
596 // for(int i=0; i<nclsy(); i++) ((TrkSinglet *)sy[i])->Dump();
597 if(Track){
598 TClonesArray &t = *Track;
599 for(int i=0; i<ntrk(); i++) ((TrkTrack *)t[i])->Dump();
600 }
601 if(SingletX){
602 TClonesArray &sx = *SingletX;
603 for(int i=0; i<nclsx(); i++) ((TrkSinglet *)sx[i])->Dump();
604 }
605 if(SingletY){
606 TClonesArray &sy = *SingletY;
607 for(int i=0; i<nclsy(); i++) ((TrkSinglet *)sy[i])->Dump();
608 }
609 }
610 //--------------------------------------
611 //
612 //
613 //--------------------------------------
614 /**
615 * Fills a TrkLevel2 object with values from a struct cTrkLevel2 (to get data from F77 common).
616 */
617 // void TrkLevel2::SetFromLevel2Struct(cTrkLevel2 *l2){
618
619 // // temporary objects:
620 // TrkSinglet* t_singlet = new TrkSinglet();
621 // TrkTrack* t_track = new TrkTrack();
622
623 // // **** general variables ****
624 // // good2 = l2->good2;
625 // for(Int_t i=0; i<12 ; i++){
626 // // crc[i] = l2->crc[i];
627 // good[i] = l2->good[i];
628 // };
629 // // *** TRACKS ***
630 // if(!Track) Track = new TClonesArray("TrkTrack");
631 // TClonesArray &t = *Track;
632 // for(int i=0; i<l2->ntrk; i++){
633 // t_track->seqno = i;// NBNBNBNB deve sempre essere = i
634 // t_track->image = l2->image[i]-1;
635 // // cout << "track "<<i<<t_track->seqno << t_track->image<<endl;
636 // t_track->chi2 = l2->chi2_nt[i];
637 // t_track->nstep = l2->nstep_nt[i];
638 // for(int it1=0;it1<5;it1++){
639 // t_track->al[it1] = l2->al_nt[i][it1];
640 // for(int it2=0;it2<5;it2++)
641 // t_track->coval[it1][it2] = l2->coval[i][it2][it1];
642 // };
643 // for(int ip=0;ip<6;ip++){
644 // t_track->xgood[ip] = l2->xgood_nt[i][ip];
645 // t_track->ygood[ip] = l2->ygood_nt[i][ip];
646 // t_track->xm[ip] = l2->xm_nt[i][ip];
647 // t_track->ym[ip] = l2->ym_nt[i][ip];
648 // t_track->zm[ip] = l2->zm_nt[i][ip];
649 // t_track->resx[ip] = l2->resx_nt[i][ip];
650 // t_track->resy[ip] = l2->resy_nt[i][ip];
651 // t_track->xv[ip] = l2->xv_nt[i][ip];
652 // t_track->yv[ip] = l2->yv_nt[i][ip];
653 // t_track->zv[ip] = l2->zv_nt[i][ip];
654 // t_track->axv[ip] = l2->axv_nt[i][ip];
655 // t_track->ayv[ip] = l2->ayv_nt[i][ip];
656 // t_track->dedx_x[ip] = l2->dedx_x[i][ip];
657 // t_track->dedx_y[ip] = l2->dedx_y[i][ip];
658 // // t_track->clx[ip] = 0;
659 // // t_track->cly[ip] = 0;
660 // };
661 // new(t[i]) TrkTrack(*t_track);
662 // t_track->Clear();
663 // };
664 // // *** SINGLETS ***
665 // if(!SingletX)SingletX = new TClonesArray("TrkSinglet");
666 // TClonesArray &sx = *SingletX;
667 // for(int i=0; i<l2->nclsx; i++){
668 // t_singlet->plane = l2->planex[i];
669 // t_singlet->coord[0] = l2->xs[i][0];
670 // t_singlet->coord[1] = l2->xs[i][1];
671 // t_singlet->sgnl = l2->signlxs[i];
672 // // t_singlet->cls = 0;
673 // new(sx[i]) TrkSinglet(*t_singlet);
674 // t_singlet->Clear();
675 // }
676 // if(!SingletY)SingletY = new TClonesArray("TrkSinglet");
677 // TClonesArray &sy = *SingletY;
678 // for(int i=0; i<l2->nclsy; i++){
679 // t_singlet->plane = l2->planey[i];
680 // t_singlet->coord[0] = l2->ys[i][0];
681 // t_singlet->coord[1] = l2->ys[i][1];
682 // t_singlet->sgnl = l2->signlys[i];
683 // // t_singlet->cls = 0;
684 // new(sy[i]) TrkSinglet(*t_singlet);
685 // t_singlet->Clear();
686 // };
687
688 // delete t_track;
689 // delete t_singlet;
690 // }
691 //--------------------------------------
692 //
693 //
694 //--------------------------------------
695 /**
696 * Fills a TrkLevel2 object with values from a struct cTrkLevel2 (to get data from F77 common).
697 * Ref to Level1 data (clusters) is also set.
698 */
699 void TrkLevel2::SetFromLevel2Struct(cTrkLevel2 *l2, TrkLevel1 *l1){
700
701 // temporary objects:
702 TrkSinglet* t_singlet = new TrkSinglet();
703 TrkTrack* t_track = new TrkTrack();
704 // general variables
705 // good2 = l2->good2;
706 for(Int_t i=0; i<12 ; i++){
707 // crc[i] = l2->crc[i];
708 good[i] = l2->good[i];
709 };
710 // *** TRACKS ***
711 if(!Track) Track = new TClonesArray("TrkTrack");
712 TClonesArray &t = *Track;
713 //-----------------------------------------------------
714 if(l1 && !t_track->clx)t_track->clx = new TRefArray(6,0);
715 if(l1 && !t_track->cly)t_track->cly = new TRefArray(6,0);
716 //-----------------------------------------------------
717 for(int i=0; i<l2->ntrk; i++){
718 t_track->seqno = i;// NBNBNBNB deve sempre essere = i
719 t_track->image = l2->image[i]-1;
720 // cout << "track "<<i<<t_track->seqno << t_track->image<<endl;
721 t_track->chi2 = l2->chi2_nt[i];
722 t_track->nstep = l2->nstep_nt[i];
723 for(int it1=0;it1<5;it1++){
724 t_track->al[it1] = l2->al_nt[i][it1];
725 for(int it2=0;it2<5;it2++)
726 t_track->coval[it1][it2] = l2->coval[i][it2][it1];
727 };
728 for(int ip=0;ip<6;ip++){
729 t_track->xgood[ip] = l2->xgood_nt[i][ip];
730 t_track->ygood[ip] = l2->ygood_nt[i][ip];
731 t_track->xm[ip] = l2->xm_nt[i][ip];
732 t_track->ym[ip] = l2->ym_nt[i][ip];
733 t_track->zm[ip] = l2->zm_nt[i][ip];
734 t_track->resx[ip] = l2->resx_nt[i][ip];
735 t_track->resy[ip] = l2->resy_nt[i][ip];
736 t_track->xv[ip] = l2->xv_nt[i][ip];
737 t_track->yv[ip] = l2->yv_nt[i][ip];
738 t_track->zv[ip] = l2->zv_nt[i][ip];
739 t_track->axv[ip] = l2->axv_nt[i][ip];
740 t_track->ayv[ip] = l2->ayv_nt[i][ip];
741 t_track->dedx_x[ip] = l2->dedx_x[i][ip];
742 t_track->dedx_y[ip] = l2->dedx_y[i][ip];
743 //-----------------------------------------------------
744 //-----------------------------------------------------
745 if(l1 && t_track->xgood[ip])t_track->clx->AddAt(l1->GetCluster(l2->cltrx[i][ip]-1),ip);
746 if(l1 && t_track->ygood[ip])t_track->cly->AddAt(l1->GetCluster(l2->cltry[i][ip]-1),ip);
747 //-----------------------------------------------------
748 //-----------------------------------------------------
749 };
750 new(t[i]) TrkTrack(*t_track);
751 t_track->Clear();
752 };
753 // *** SINGLETS ***
754 if(!SingletX)SingletX = new TClonesArray("TrkSinglet");
755 TClonesArray &sx = *SingletX;
756 for(int i=0; i<l2->nclsx; i++){
757 t_singlet->plane = l2->planex[i];
758 t_singlet->coord[0] = l2->xs[i][0];
759 t_singlet->coord[1] = l2->xs[i][1];
760 t_singlet->sgnl = l2->signlxs[i];
761 //-----------------------------------------------------
762 if(l1) t_singlet->cls = l1->GetCluster(l2->clsx[i]-1);
763 //-----------------------------------------------------
764 new(sx[i]) TrkSinglet(*t_singlet);
765 t_singlet->Clear();
766 }
767 if(!SingletY)SingletY = new TClonesArray("TrkSinglet");
768 TClonesArray &sy = *SingletY;
769 for(int i=0; i<l2->nclsy; i++){
770 t_singlet->plane = l2->planey[i];
771 t_singlet->coord[0] = l2->ys[i][0];
772 t_singlet->coord[1] = l2->ys[i][1];
773 t_singlet->sgnl = l2->signlys[i];
774 //-----------------------------------------------------
775 if(l1) t_singlet->cls = l1->GetCluster(l2->clsy[i]-1);
776 //-----------------------------------------------------
777 new(sy[i]) TrkSinglet(*t_singlet);
778 t_singlet->Clear();
779 };
780
781 delete t_track;
782 delete t_singlet;
783 }
784 /**
785 * Fills a struct cTrkLevel2 with values from a TrkLevel2 object (to put data into a F77 common).
786 */
787
788 void TrkLevel2::GetLevel2Struct(cTrkLevel2 *l2) const {
789
790 // general variables
791 // l2->good2 = good2 ;
792 for(Int_t i=0; i<12 ; i++){
793 // l2->crc[i] = crc[i];
794 l2->good[i] = good[i];
795 };
796 // *** TRACKS ***
797
798 if(Track){
799 l2->ntrk = Track->GetEntries();
800 for(Int_t i=0;i<l2->ntrk;i++){
801 l2->image[i] = 1 + ((TrkTrack *)Track->At(i))->image;
802 l2->chi2_nt[i] = ((TrkTrack *)Track->At(i))->chi2;
803 l2->nstep_nt[i] = ((TrkTrack *)Track->At(i))->nstep;
804 for(int it1=0;it1<5;it1++){
805 l2->al_nt[i][it1] = ((TrkTrack *)Track->At(i))->al[it1];
806 for(int it2=0;it2<5;it2++)
807 l2->coval[i][it2][it1] = ((TrkTrack *)Track->At(i))->coval[it1][it2];
808 };
809 for(int ip=0;ip<6;ip++){
810 l2->xgood_nt[i][ip] = ((TrkTrack *)Track->At(i))->xgood[ip];
811 l2->ygood_nt[i][ip] = ((TrkTrack *)Track->At(i))->ygood[ip];
812 l2->xm_nt[i][ip] = ((TrkTrack *)Track->At(i))->xm[ip];
813 l2->ym_nt[i][ip] = ((TrkTrack *)Track->At(i))->ym[ip];
814 l2->zm_nt[i][ip] = ((TrkTrack *)Track->At(i))->zm[ip];
815 l2->resx_nt[i][ip] = ((TrkTrack *)Track->At(i))->resx[ip];
816 l2->resy_nt[i][ip] = ((TrkTrack *)Track->At(i))->resy[ip];
817 l2->xv_nt[i][ip] = ((TrkTrack *)Track->At(i))->xv[ip];
818 l2->yv_nt[i][ip] = ((TrkTrack *)Track->At(i))->yv[ip];
819 l2->zv_nt[i][ip] = ((TrkTrack *)Track->At(i))->zv[ip];
820 l2->axv_nt[i][ip] = ((TrkTrack *)Track->At(i))->axv[ip];
821 l2->ayv_nt[i][ip] = ((TrkTrack *)Track->At(i))->ayv[ip];
822 l2->dedx_x[i][ip] = ((TrkTrack *)Track->At(i))->dedx_x[ip];
823 l2->dedx_y[i][ip] = ((TrkTrack *)Track->At(i))->dedx_y[ip];
824 };
825 }
826 }
827 // *** SINGLETS ***
828 if(SingletX){
829 l2->nclsx = SingletX->GetEntries();
830 for(Int_t i=0;i<l2->nclsx;i++){
831 l2->planex[i] = ((TrkSinglet *)SingletX->At(i))->plane;
832 l2->xs[i][0] = ((TrkSinglet *)SingletX->At(i))->coord[0];
833 l2->xs[i][1] = ((TrkSinglet *)SingletX->At(i))->coord[1];
834 l2->signlxs[i] = ((TrkSinglet *)SingletX->At(i))->sgnl;
835 }
836 }
837
838 if(SingletY){
839 l2->nclsy = SingletY->GetEntries();
840 for(Int_t i=0;i<l2->nclsy;i++){
841 l2->planey[i] = ((TrkSinglet *)SingletY->At(i))->plane;
842 l2->ys[i][0] = ((TrkSinglet *)SingletY->At(i))->coord[0];
843 l2->ys[i][1] = ((TrkSinglet *)SingletY->At(i))->coord[1];
844 l2->signlys[i] = ((TrkSinglet *)SingletY->At(i))->sgnl;
845 }
846 }
847 }
848 //--------------------------------------
849 //
850 //
851 //--------------------------------------
852 void TrkLevel2::Clear(){
853 for(Int_t i=0; i<12 ; i++){
854 good[i] = -1;
855 };
856 // if(Track)Track->Clear("C");
857 // if(SingletX)SingletX->Clear("C");
858 // if(SingletY)SingletY->Clear("C");
859 if(Track)Track->Delete();
860 if(SingletX)SingletX->Delete();
861 if(SingletY)SingletY->Delete();
862 }
863 // //--------------------------------------
864 // //
865 // //
866 // //--------------------------------------
867 void TrkLevel2::Delete(){
868
869 // cout << "void TrkLevel2::Delete()"<<endl;
870 Clear();
871 if(Track)delete Track;
872 if(SingletX)delete SingletX;
873 if(SingletY)delete SingletY;
874
875 }
876 //--------------------------------------
877 //
878 //
879 //--------------------------------------
880 /**
881 * Sort physical tracks and stores them in a TObjectArray, ordering by increasing chi**2 value (in case of track image, it selects the one with lower chi**2). The total number of physical tracks is given by GetNTracks() and the it-th physical track can be retrieved by means of the method GetTrack(int it).
882 * This method is overridden by PamLevel2::GetTracks(), where calorimeter and TOF information is used.
883 */
884 TRefArray *TrkLevel2::GetTracks_NFitSorted(){
885
886 if(!Track)return 0;
887
888 TRefArray *sorted = new TRefArray();
889
890 TClonesArray &t = *Track;
891 // TClonesArray &ts = *PhysicalTrack;
892 int N = ntrk();
893 vector<int> m(N); for(int i=0; i<N; i++)m[i]=1;
894 // int m[50]; for(int i=0; i<N; i++)m[i]=1;
895
896 int indo=0;
897 int indi=0;
898 while(N > 0){
899 // while(N != 0){
900 int nfit =0;
901 float chi2ref = numeric_limits<float>::max();
902
903 // first loop to search maximum num. of fit points
904 for(int i=0; i < ntrk(); i++){
905 if( ((TrkTrack *)t[i])->GetNtot() >= nfit && m[i]==1){
906 nfit = ((TrkTrack *)t[i])->GetNtot();
907 }
908 }
909 //second loop to search minimum chi2 among selected
910 for(int i=0; i<ntrk(); i++){
911 Float_t chi2 = ((TrkTrack *)t[i])->chi2;
912 if(chi2 < 0) chi2 = -chi2*1000;
913 if( chi2 < chi2ref
914 && ((TrkTrack *)t[i])->GetNtot() == nfit
915 && m[i]==1){
916 chi2ref = ((TrkTrack *)t[i])->chi2;
917 indi = i;
918 };
919 };
920 if( ((TrkTrack *)t[indi])->HasImage() ){
921 m[((TrkTrack *)t[indi])->image] = 0;
922 N--;
923
924 // cout << "i** "<< ((TrkTrack *)t[indi])->image << " " << nfiti <<" "<<chi2i<<endl;
925 };
926 sorted->Add( (TrkTrack*)t[indi] );
927
928 m[indi] = 0;
929 // cout << "SORTED "<< indo << " "<< indi << " "<< N << " "<<((TrkTrack *)t[indi])->image<<" "<<chi2ref<<endl;
930 N--;
931 indo++;
932 }
933 m.clear();
934 // cout << "GetTracks_NFitSorted(it): Done"<< endl;
935
936 return sorted;
937 // return PhysicalTrack;
938 }
939 //--------------------------------------
940 //
941 //
942 //--------------------------------------
943 /**
944 * Retrieves the is-th stored track.
945 * @param it Track number, ranging from 0 to ntrk().
946 * Fitted tracks ( images included ) are stored in a TObjectArray ( TrkLevel2::Track ) in the same order they are returned by the F77 fitting routine.
947 */
948 TrkTrack *TrkLevel2::GetStoredTrack(int is){
949
950 if(is >= this->ntrk()){
951 cout << "** TrkLevel2 ** Track "<< is << "doen not exits! " << endl;
952 cout << " Stored tracks ntrk() = "<< this->ntrk() << endl;
953 return 0;
954 }
955 if(!Track){
956 cout << "TrkTrack *TrkLevel2::GetStoredTrack(int is) >> (TClonesArray*) Track ==0 "<<endl;
957 };
958 TClonesArray &t = *(Track);
959 TrkTrack *track = (TrkTrack*)t[is];
960 return track;
961 }
962 //--------------------------------------
963 //
964 //
965 //--------------------------------------
966 /**
967 * Retrieves the is-th stored X singlet.
968 * @param it Singlet number, ranging from 0 to nclsx().
969 */
970 TrkSinglet *TrkLevel2::GetSingletX(int is){
971
972 if(is >= this->nclsx()){
973 cout << "** TrkLevel2 ** Singlet "<< is << "doen not exits! " << endl;
974 cout << " Stored x-singlets nclsx() = "<< this->nclsx() << endl;
975 return 0;
976 }
977 if(!SingletX)return 0;
978 TClonesArray &t = *(SingletX);
979 TrkSinglet *singlet = (TrkSinglet*)t[is];
980 return singlet;
981 }
982 //--------------------------------------
983 //
984 //
985 //--------------------------------------
986 /**
987 * Retrieves the is-th stored Y singlet.
988 * @param it Singlet number, ranging from 0 to nclsx().
989 */
990 TrkSinglet *TrkLevel2::GetSingletY(int is){
991
992 if(is >= this->nclsy()){
993 cout << "** TrkLevel2 ** Singlet "<< is << "doen not exits! " << endl;
994 cout << " Stored y-singlets nclsy() = "<< this->nclsx() << endl;
995 return 0;
996 }
997 if(!SingletY)return 0;
998 TClonesArray &t = *(SingletY);
999 TrkSinglet *singlet = (TrkSinglet*)t[is];
1000 return singlet;
1001 }
1002 //--------------------------------------
1003 //
1004 //
1005 //--------------------------------------
1006 /**
1007 * Retrieves the it-th "physical" track, sorted by the method GetNTracks().
1008 * @param it Track number, ranging from 0 to GetNTracks().
1009 */
1010
1011 TrkTrack *TrkLevel2::GetTrack(int it){
1012
1013 if(it >= this->GetNTracks()){
1014 cout << "** TrkLevel2 ** Track "<< it << "does not exits! " << endl;
1015 cout << " Physical tracks GetNTracks() = "<< this->ntrk() << endl;
1016 return 0;
1017 }
1018
1019 TRefArray *sorted = GetTracks(); //TEMPORANEO
1020 if(!sorted)return 0;
1021 TrkTrack *track = (TrkTrack*)sorted->At(it);
1022 sorted->Clear();
1023 delete sorted;
1024 return track;
1025 }
1026 /**
1027 * Give the number of "physical" tracks, sorted by the method GetTracks().
1028 */
1029 Int_t TrkLevel2::GetNTracks(){
1030
1031 Float_t ntot=0;
1032 if(!Track)return 0;
1033 TClonesArray &t = *Track;
1034 for(int i=0; i<ntrk(); i++) {
1035 if( ((TrkTrack *)t[i])->GetImageSeqNo() == -1 ) ntot+=1.;
1036 else ntot+=0.5;
1037 }
1038 return (Int_t)ntot;
1039
1040 };
1041 //--------------------------------------
1042 //
1043 //
1044 //--------------------------------------
1045 /**
1046 * Retrieves (if present) the image of the it-th "physical" track, sorted by the method GetNTracks().
1047 * @param it Track number, ranging from 0 to GetNTracks().
1048 */
1049 TrkTrack *TrkLevel2::GetTrackImage(int it){
1050
1051 if(it >= this->GetNTracks()){
1052 cout << "** TrkLevel2 ** Track "<< it << "does not exits! " << endl;
1053 cout << " Physical tracks GetNTracks() = "<< this->ntrk() << endl;
1054 return 0;
1055 }
1056
1057 TRefArray* sorted = GetTracks(); //TEMPORANEO
1058 if(!sorted)return 0;
1059 TrkTrack *track = (TrkTrack*)sorted->At(it);
1060
1061 if(!track->HasImage()){
1062 cout << "** TrkLevel2 ** Track "<< it << "does not have image! " << endl;
1063 return 0;
1064 }
1065 if(!Track)return 0;
1066 TrkTrack *image = (TrkTrack*)(*Track)[track->image];
1067
1068 sorted->Delete();
1069 delete sorted;
1070
1071 return image;
1072
1073 }
1074 //--------------------------------------
1075 //
1076 //
1077 //--------------------------------------
1078 /**
1079 * Loads the magnetic field.
1080 * @param s Path of the magnetic-field files.
1081 */
1082 void TrkLevel2::LoadField(TString path){
1083 //
1084 // strcpy(path_.path,path.Data());
1085 // path_.pathlen = path.Length();
1086 // path_.error = 0;
1087 // readb_();
1088
1089 TrkParams::Set(path,1);
1090
1091 //
1092 };
1093 /**
1094 * Get BY (kGauss)
1095 * @param v (x,y,z) coordinates in cm
1096 */
1097 float TrkLevel2::GetBX(float* v){
1098 float b[3];
1099 gufld_(v,b);
1100 return b[0]/10.;
1101 }
1102 /**
1103 * Get BY (kGauss)
1104 * @param v (x,y,z) coordinates in cm
1105 */
1106 float TrkLevel2::GetBY(float* v){
1107 float b[3];
1108 gufld_(v,b);
1109 return b[1]/10.;
1110 }
1111 /**
1112 * Get BY (kGauss)
1113 * @param v (x,y,z) coordinates in cm
1114 */
1115 float TrkLevel2::GetBZ(float* v){
1116 float b[3];
1117 gufld_(v,b);
1118 return b[2]/10.;
1119 }
1120 //--------------------------------------
1121 //
1122 //
1123 //--------------------------------------
1124 /**
1125 * Get tracker-plane (mechanical) z-coordinate
1126 * @param plane_id plane index (1=TOP,2,3,4,5,6=BOTTOM)
1127 */
1128 Float_t TrkLevel2::GetZTrk(Int_t plane_id){
1129 switch(plane_id){
1130 case 1: return ZTRK1;
1131 case 2: return ZTRK2;
1132 case 3: return ZTRK3;
1133 case 4: return ZTRK4;
1134 case 5: return ZTRK5;
1135 case 6: return ZTRK6;
1136 default: return 0.;
1137 };
1138 };
1139 //--------------------------------------
1140 //
1141 //
1142 //--------------------------------------
1143 /**
1144 * Trajectory default constructor.
1145 * (By default is created with z-coordinates inside the tracking volume)
1146 */
1147 Trajectory::Trajectory(){
1148 npoint = 10;
1149 x = new float[npoint];
1150 y = new float[npoint];
1151 z = new float[npoint];
1152 thx = new float[npoint];
1153 thy = new float[npoint];
1154 tl = new float[npoint];
1155 float dz = ((ZTRK1)-(ZTRK6))/(npoint-1);
1156 for(int i=0; i<npoint; i++){
1157 x[i] = 0;
1158 y[i] = 0;
1159 z[i] = (ZTRK1) - i*dz;
1160 thx[i] = 0;
1161 thy[i] = 0;
1162 tl[i] = 0;
1163 }
1164 }
1165 //--------------------------------------
1166 //
1167 //
1168 //--------------------------------------
1169 /**
1170 * Trajectory constructor.
1171 * (By default is created with z-coordinates inside the tracking volume)
1172 * \param n Number of points
1173 */
1174 Trajectory::Trajectory(int n){
1175 if(n<=0){
1176 cout << "NB! Trajectory must have at least 1 point >>> created with 10 points" << endl;
1177 n=10;
1178 }
1179 npoint = n;
1180 x = new float[npoint];
1181 y = new float[npoint];
1182 z = new float[npoint];
1183 thx = new float[npoint];
1184 thy = new float[npoint];
1185 tl = new float[npoint];
1186 float dz = ((ZTRK1)-(ZTRK6))/(npoint-1);
1187 for(int i=0; i<npoint; i++){
1188 x[i] = 0;
1189 y[i] = 0;
1190 z[i] = (ZTRK1) - i*dz;
1191 thx[i] = 0;
1192 thy[i] = 0;
1193 tl[i] = 0;
1194 }
1195 }
1196 //--------------------------------------
1197 //
1198 //
1199 //--------------------------------------
1200 /**
1201 * Trajectory constructor.
1202 * \param n Number of points
1203 * \param pz Pointer to float array, defining z coordinates
1204 */
1205 Trajectory::Trajectory(int n, float* zin){
1206 npoint = 10;
1207 if(n>0)npoint = n;
1208 x = new float[npoint];
1209 y = new float[npoint];
1210 z = new float[npoint];
1211 thx = new float[npoint];
1212 thy = new float[npoint];
1213 tl = new float[npoint];
1214 int i=0;
1215 do{
1216 x[i] = 0;
1217 y[i] = 0;
1218 z[i] = zin[i];
1219 thx[i] = 0;
1220 thy[i] = 0;
1221 tl[i] = 0;
1222 i++;
1223 }while(zin[i-1] > zin[i] && i < npoint);
1224 npoint=i;
1225 if(npoint != n)cout << "NB! Trajectory created with "<<npoint<<" points"<<endl;
1226 }
1227 void Trajectory::Delete(){
1228
1229 if(x) delete [] x;
1230 if(y) delete [] y;
1231 if(z) delete [] z;
1232 if(thx) delete [] thx;
1233 if(thy) delete [] thy;
1234 if(tl) delete [] tl;
1235
1236 }
1237 //--------------------------------------
1238 //
1239 //
1240 //--------------------------------------
1241 /**
1242 * Dump the trajectory coordinates.
1243 */
1244 void Trajectory::Dump(){
1245 cout <<endl<< "Trajectory ========== "<<endl;
1246 for (int i=0; i<npoint; i++){
1247 cout << i <<" >> " << x[i] <<" "<< y[i] <<" "<< z[i] ;
1248 cout <<" -- " << thx[i] <<" "<< thy[i] ;
1249 cout <<" -- " << tl[i] << endl;
1250 };
1251 }
1252 //--------------------------------------
1253 //
1254 //
1255 //--------------------------------------
1256 /**
1257 * Get trajectory length between two points
1258 * @param ifirst first point (default 0)
1259 * @param ilast last point (default npoint)
1260 */
1261 float Trajectory::GetLength(int ifirst, int ilast){
1262 if( ifirst<0 ) ifirst = 0;
1263 if( ilast>=npoint) ilast = npoint-1;
1264 float l=0;
1265 for(int i=ifirst;i<=ilast;i++){
1266 l=l+tl[i];
1267 };
1268 if(z[ilast] > ZINI)l=l-tl[ilast];
1269 if(z[ifirst] < ZINI) l=l-tl[ifirst];
1270
1271 return l;
1272
1273 }
1274
1275 /**
1276 * Evaluates the trajectory in the apparatus associated to the track.
1277 * It integrates the equations of motion in the magnetic field. The magnetic field should be previously loaded ( by calling TrkLevel2::LoadField() ), otherwise an error message is returned.
1278 * @param t pointer to an object of the class Trajectory,
1279 * which z coordinates should be previously initialized by calling the proper constructor ( Trajectory::Trajectory(int n, float* zin) ).
1280 * @return error flag.
1281 */
1282 int Trajectory::DoTrack2(float* al){
1283
1284 double *dxout = new double[npoint];
1285 double *dyout = new double[npoint];
1286 double *dthxout = new double[npoint];
1287 double *dthyout = new double[npoint];
1288 double *dtlout = new double[npoint];
1289 double *dzin = new double[npoint];
1290 double dal[5];
1291
1292 int ifail = 0;
1293
1294 for (int i=0; i<5; i++) dal[i] = (double)al[i];
1295 for (int i=0; i<npoint; i++) dzin[i] = (double)z[i];
1296
1297 TrkParams::Load(1);
1298 if( !TrkParams::IsLoaded(1) ){
1299 cout << "int Trajectory::DoTrack2(float* al) --- ERROR --- m.field not loaded"<<endl;
1300 return 0;
1301 }
1302 dotrack2_(&(npoint),dzin,dxout,dyout,dthxout,dthyout,dtlout,dal,&ifail);
1303
1304 for (int i=0; i<npoint; i++){
1305 x[i] = (float)*dxout++;
1306 y[i] = (float)*dyout++;
1307 thx[i] = (float)*dthxout++;
1308 thy[i] = (float)*dthyout++;
1309 tl[i] = (float)*dtlout++;
1310 }
1311
1312 return ifail;
1313 };
1314
1315 ClassImp(TrkLevel2);
1316 ClassImp(TrkSinglet);
1317 ClassImp(TrkTrack);
1318 ClassImp(Trajectory);

  ViewVC Help
Powered by ViewVC 1.1.23