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

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

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 1.1 by mocchiut, Fri May 19 13:15:54 2006 UTC revision 1.31 by pam-fi, Wed Mar 28 09:22:28 2007 UTC
# Line 4  Line 4 
4   */   */
5  #include <TrkLevel2.h>  #include <TrkLevel2.h>
6  #include <iostream>  #include <iostream>
7    #include <math.h>
8  using namespace std;  using namespace std;
9  //......................................  //......................................
10  // F77 routines  // F77 routines
11  //......................................  //......................................
12  extern "C" {      extern "C" {    
13      void dotrack_(int*, double*, double*, double*, double*, int*);      void dotrack_(int*, double*, double*, double*, double*, int*);
14      int  readb_(const char*);      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(){  TrkTrack::TrkTrack(){
27      image = 0;  //    cout << "TrkTrack::TrkTrack()" << endl;
28        seqno = -1;
29        image = -1;
30      chi2  = 0;      chi2  = 0;
31        nstep = 0;
32      for(int it1=0;it1<5;it1++){      for(int it1=0;it1<5;it1++){
33          al[it1] = 0;          al[it1] = 0;
34          for(int it2=0;it2<5;it2++)          for(int it2=0;it2<5;it2++)coval[it1][it2] = 0;
             coval[it1][it2] = 0;  
35      };      };
36      for(int ip=0;ip<6;ip++){      for(int ip=0;ip<6;ip++){
37          xgood[ip]  = 0;          xgood[ip]  = 0;
# Line 39  TrkTrack::TrkTrack(){ Line 48  TrkTrack::TrkTrack(){
48          ayv[ip]    = 0;          ayv[ip]    = 0;
49          dedx_x[ip] = 0;          dedx_x[ip] = 0;
50          dedx_y[ip] = 0;          dedx_y[ip] = 0;
51      };          };
52        clx = 0;
53        cly = 0;
54    //    clx = new TRefArray(6,0); //forse causa memory leak???
55    //    cly = new TRefArray(6,0); //forse causa memory leak???
56  };  };
57  //--------------------------------------  //--------------------------------------
58  //  //
59  //  //
60  //--------------------------------------  //--------------------------------------
61  TrkTrack::TrkTrack(const TrkTrack& t){  TrkTrack::TrkTrack(const TrkTrack& t){
62        seqno = t.seqno;
63      image = t.image;      image = t.image;
64      chi2  = t.chi2;      chi2  = t.chi2;
65        nstep = t.nstep;
66      for(int it1=0;it1<5;it1++){      for(int it1=0;it1<5;it1++){
67          al[it1] = t.al[it1];          al[it1] = t.al[it1];
68          for(int it2=0;it2<5;it2++)          for(int it2=0;it2<5;it2++)coval[it1][it2] = t.coval[it1][it2];
             coval[it1][it2] = t.coval[it1][it2];  
69      };      };
70      for(int ip=0;ip<6;ip++){      for(int ip=0;ip<6;ip++){
71          xgood[ip]  = t.xgood[ip];          xgood[ip]  = t.xgood[ip];
# Line 68  TrkTrack::TrkTrack(const TrkTrack& t){ Line 82  TrkTrack::TrkTrack(const TrkTrack& t){
82          ayv[ip]    = t.ayv[ip];          ayv[ip]    = t.ayv[ip];
83          dedx_x[ip] = t.dedx_x[ip];          dedx_x[ip] = t.dedx_x[ip];
84          dedx_y[ip] = t.dedx_y[ip];          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  //  //
# Line 93  int TrkTrack::DoTrack(Trajectory* t){ Line 145  int TrkTrack::DoTrack(Trajectory* t){
145      for (int i=0; i<5; i++)         dal[i]  = (double)al[i];      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];      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);      dotrack_(&(t->npoint),dzin,dxout,dyout,dal,&ifail);
154            
155      for (int i=0; i<t->npoint; i++){      for (int i=0; i<t->npoint; i++){
# Line 110  int TrkTrack::DoTrack(Trajectory* t){ Line 167  int TrkTrack::DoTrack(Trajectory* t){
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(){  //float TrkTrack::BdL(){
218  //};  //};
219  //--------------------------------------  //--------------------------------------
# Line 131  Float_t TrkTrack::GetDeflection(){ Line 235  Float_t TrkTrack::GetDeflection(){
235  //  //
236  Float_t TrkTrack::GetDEDX(){  Float_t TrkTrack::GetDEDX(){
237          Float_t dedx=0;          Float_t dedx=0;
238          for(Int_t i=0; i<6; i++)dedx+=dedx_x[i]*xgood[i]+dedx_y[i]*ygood[i];  //      for(Int_t i=0; i<6; i++)dedx+=dedx_x[i]*xgood[i]+dedx_y[i]*ygood[i];
239            for(Int_t i=0; i<6; i++)dedx+=dedx_x[i]*XGood(i)+dedx_y[i]*YGood(i);
240          dedx = dedx/(this->GetNX()+this->GetNY());          dedx = dedx/(this->GetNX()+this->GetNY());
241          return dedx;          return dedx;
242  };  };
# Line 142  Float_t TrkTrack::GetDEDX(){ Line 247  Float_t TrkTrack::GetDEDX(){
247  //--------------------------------------  //--------------------------------------
248  void TrkTrack::Dump(){  void TrkTrack::Dump(){
249      cout << endl << "========== Track " ;      cout << endl << "========== Track " ;
250        cout << endl << "seq.  n. : "<< seqno;
251        cout << endl << "image n. : "<< image;
252      cout << endl << "al       : "; for(int i=0; i<5; i++)cout << al[i] << " ";      cout << endl << "al       : "; for(int i=0; i<5; i++)cout << al[i] << " ";
253      cout << endl << "chi^2    : "<< chi2;      cout << endl << "chi^2    : "<< chi2;
254      cout << endl << "xgood    : "; for(int i=0; i<6; i++)cout << xgood[i] ;      cout << endl << "n.step   : "<< nstep;
255      cout << endl << "ygood    : "; for(int i=0; i<6; i++)cout << ygood[i] ;      cout << endl << "xgood    : "; for(int i=0; i<6; i++)cout << XGood(i) ;
256        cout << endl << "ygood    : "; for(int i=0; i<6; i++)cout << YGood(i) ;
257      cout << endl << "xm       : "; for(int i=0; i<6; i++)cout << xm[i] << " ";      cout << endl << "xm       : "; for(int i=0; i<6; i++)cout << xm[i] << " ";
258      cout << endl << "ym       : "; for(int i=0; i<6; i++)cout << ym[i] << " ";      cout << endl << "ym       : "; for(int i=0; i<6; i++)cout << ym[i] << " ";
259      cout << endl << "zm       : "; for(int i=0; i<6; i++)cout << zm[i] << " ";      cout << endl << "zm       : "; for(int i=0; i<6; i++)cout << zm[i] << " ";
260        cout << endl << "xv       : "; for(int i=0; i<6; i++)cout << xv[i] << " ";
261        cout << endl << "yv       : "; for(int i=0; i<6; i++)cout << yv[i] << " ";
262        cout << endl << "zv       : "; for(int i=0; i<6; i++)cout << zv[i] << " ";
263        cout << endl << "resx     : "; for(int i=0; i<6; i++)cout << resx[i] << " ";
264        cout << endl << "resy     : "; for(int i=0; i<6; i++)cout << resy[i] << " ";
265        cout << endl << "coval    : "; for(int i=0; i<5; i++)cout << coval[0][i]<<" ";
266        cout << endl << "           "; for(int i=0; i<5; i++)cout << coval[1][i]<<" ";
267        cout << endl << "           "; for(int i=0; i<5; i++)cout << coval[2][i]<<" ";
268        cout << endl << "           "; for(int i=0; i<5; i++)cout << coval[3][i]<<" ";
269        cout << endl << "           "; for(int i=0; i<5; i++)cout << coval[4][i]<<" ";
270      cout << endl << "dedx_x   : "; for(int i=0; i<6; i++)cout << dedx_x[i] << " ";      cout << endl << "dedx_x   : "; for(int i=0; i<6; i++)cout << dedx_x[i] << " ";
271      cout << endl << "dedx_y   : "; for(int i=0; i<6; i++)cout << dedx_y[i] << " ";      cout << endl << "dedx_y   : "; for(int i=0; i<6; i++)cout << dedx_y[i] << " ";
272        cout << endl;
273  }  }
274    /**
275     * Set the TrkTrack position measurements
276     */
277    void TrkTrack::SetMeasure(double *xmeas, double *ymeas, double *zmeas){
278        for(int i=0; i<6; i++) xm[i]=*xmeas++;
279        for(int i=0; i<6; i++) ym[i]=*ymeas++;
280        for(int i=0; i<6; i++) zm[i]=*zmeas++;
281    }
282    /**
283     * Set the TrkTrack position resolution
284     */
285    void TrkTrack::SetResolution(double *rx, double *ry){
286        for(int i=0; i<6; i++) resx[i]=*rx++;
287        for(int i=0; i<6; i++) resy[i]=*ry++;
288    }
289    /**
290     * Set the TrkTrack good measurement
291     */
292    void TrkTrack::SetGood(int *xg, int *yg){
293        // NB! si perdera` l'informazione sul numero del cluster
294        for(int i=0; i<6; i++) xgood[i]=*xg++;
295        for(int i=0; i<6; i++) ygood[i]=*yg++;
296    }
297    
298    /**
299     * Load the magnetic field
300     */
301    void TrkTrack::LoadField(TString path){
302        
303    //     strcpy(path_.path,path.Data());
304    //     path_.pathlen = path.Length();
305    //     path_.error   = 0;
306    //     readb_();
307    
308        TrkParams::Set(path,1);
309        TrkParams::Load(1);
310    
311    };
312    
313    
314    /**
315     * Method to fill minimization-routine common
316     */
317    void TrkTrack::FillMiniStruct(cMini2track& track){
318    
319        for(int i=0; i<6; i++){
320    
321    //      track.xgood[i]=xgood[i];
322    //      track.ygood[i]=ygood[i];
323            track.xgood[i]=XGood(i);
324            track.ygood[i]=YGood(i);
325            
326            track.xm[i]=xm[i];
327            track.ym[i]=ym[i];
328            track.zm[i]=zm[i];
329            
330    //      --- temporaneo ----------------------------
331    //      andrebbe inserita la dimensione del sensore
332            float segment = 100.;
333            track.xm_a[i]=xm[i];
334            track.xm_b[i]=xm[i];
335            track.ym_a[i]=ym[i];
336            track.ym_b[i]=ym[i];
337            if(       XGood(i) && !YGood(i) ){
338                track.ym_a[i] = track.ym_a[i]+segment;
339                track.ym_b[i] = track.ym_b[i]-segment;
340            }else if( !XGood(i) && YGood(i)){
341                track.xm_a[i] = track.xm_a[i]+segment;
342                track.xm_b[i] = track.xm_b[i]-segment;
343            }
344    //      --- temporaneo ----------------------------
345            
346            track.resx[i]=resx[i];
347            track.resy[i]=resy[i];
348        }
349    
350        for(int i=0; i<5; i++) track.al[i]=al[i];
351        track.zini = 23.5;
352    // ZINI = 23.5 !!! it should be the same parameter in all codes
353        
354    }
355    /**
356     * Method to set values from  minimization-routine common
357     */
358    void TrkTrack::SetFromMiniStruct(cMini2track *track){
359    
360        for(int i=0; i<5; i++) {
361            al[i]=track->al[i];
362            for(int j=0; j<5; j++) coval[i][j]=track->cov[i][j];
363        }
364        chi2  = track->chi2;
365        nstep = track->nstep;
366        for(int i=0; i<6; i++){
367            xv[i]  = track->xv[i];
368            yv[i]  = track->yv[i];
369            zv[i]  = track->zv[i];
370            xm[i]  = track->xm[i];
371            ym[i]  = track->ym[i];
372            zm[i]  = track->zm[i];
373            axv[i] = track->axv[i];
374            ayv[i] = track->ayv[i];
375        }
376        
377    }
378    /**
379     * Tracking method. It calls F77 mini routine.
380     */
381    void TrkTrack::Fit(double pfixed, int& fail, int iprint){
382    
383        float al_ini[] = {0.,0.,0.,0.,0.};
384    
385        extern cMini2track track_;
386        fail = 0;
387        FillMiniStruct(track_);
388    
389        // if fit variables have been reset, evaluate the initial guess
390        if(al[0]==-9999.&&al[1]==-9999.&&al[2]==-9999.&&al[3]==-9999.&&al[4]==-9999.)guess_();
391    
392        // --------------------- free momentum
393        if(pfixed==0.) {
394            track_.pfixed=0.;
395        }
396        // --------------------- fixed momentum
397        if(pfixed!=0.) {
398            al[4]=1./pfixed;    
399            track_.pfixed=pfixed;
400        }
401    
402        //  store temporarily the initial guess
403        for(int i=0; i<5; i++) al_ini[i]=track_.al[i];
404    
405        //  ------------------------------------------
406        //  call mini routine
407        TrkParams::Load(1);
408        if( !TrkParams::IsLoaded(1) ){
409            cout << "void TrkTrack::Fit(double pfixed, int& fail, int iprint) --- ERROR --- m.field not loaded"<<endl;
410            return;
411        }
412        int istep=0;
413        int ifail=0;
414        mini2_(&istep,&ifail, &iprint);
415        if(ifail!=0) {
416            if(iprint)cout << "ERROR: ifail= " << ifail << endl;
417            fail = 1;
418        }
419        //  ------------------------------------------
420        
421        SetFromMiniStruct(&track_);
422    //    cout << endl << "eta ===> " << track_.al[4] << endl;
423    
424    //     for(int i=0; i<5; i++) al[i]=track_.al[i];
425    //     chi2=track_.chi2;
426    //     nstep=track_.nstep;
427    //     for(int i=0; i<6; i++) xv[i]=track_.xv[i];
428    //     for(int i=0; i<6; i++) yv[i]=track_.yv[i];
429    //     for(int i=0; i<6; i++) zv[i]=track_.zv[i];
430    //     for(int i=0; i<6; i++) axv[i]=track_.axv[i];
431    //     for(int i=0; i<6; i++) ayv[i]=track_.ayv[i];
432    //     for(int i=0; i<5; i++) {
433    //      for(int j=0; j<5; j++) coval[i][j]=track_.cov[i][j];
434    //     }
435    
436        if(fail){
437            if(iprint)cout << " >>>> fit failed >>>> drawing initial par"<<endl;
438            for(int i=0; i<5; i++) al[i]=al_ini[i];
439        }
440    
441    };
442    /*
443     * Reset the fit parameters
444     */
445    void TrkTrack::FitReset(){
446        for(int i=0; i<5; i++) al[i]=-9999.;
447        chi2=0.;
448        nstep=0;
449        for(int i=0; i<6; i++) xv[i]=0.;
450        for(int i=0; i<6; i++) yv[i]=0.;
451        for(int i=0; i<6; i++) zv[i]=0.;
452        for(int i=0; i<6; i++) axv[i]=0.;
453        for(int i=0; i<6; i++) ayv[i]=0.;
454        for(int i=0; i<5; i++) {
455            for(int j=0; j<5; j++) coval[i][j]=0.;
456        }
457    }
458    /*
459     * Set the tracking mode
460     */
461    void TrkTrack::SetTrackingMode(int trackmode){
462        extern cMini2track track_;
463        track_.trackmode = trackmode;
464    }
465    /*
466     * Set the factor scale for tracking precision
467     */
468    void TrkTrack::SetPrecisionFactor(double fact){
469        extern cMini2track track_;
470        track_.fact = fact;
471    }
472    /*
473     * Set the factor scale for tracking precision
474     */
475    void TrkTrack::SetStepMin(int istepmin){
476        extern cMini2track track_;
477        track_.istepmin = istepmin;
478    }
479    
480    
481    /*
482     * Method to retrieve the X-view clusters associated to the track.
483     * @param ip Tracker plane (0-5)
484     */
485    TrkCluster *TrkTrack::GetClusterX(int ip){
486        cout << " TrkCluster *TrkTrack::GetClusterX(int ip) -- momentaneamente fuori servizio --"<< endl;
487        if(!clx)return NULL;
488        TrkCluster *pt = (TrkCluster*)(clx->At(ip));
489        return pt;
490    };
491    /*
492     * Method to retrieve the Y-view clusters associated to the track.
493     * @param ip Tracker plane (0-5)
494     */
495    TrkCluster *TrkTrack::GetClusterY(int ip){
496        cout << " TrkCluster *TrkTrack::GetClusterY(int ip) -- momentaneamente fuori servizio --"<< endl;
497        if(!cly)return NULL;
498        TrkCluster *pt = (TrkCluster*)(cly->At(ip));
499        return pt;
500    };
501    
502    
503    //--------------------------------------
504    //
505    //
506    //--------------------------------------
507    void TrkTrack::Clear(){
508    //    cout << "TrkTrack::Clear()"<<endl;
509        seqno = -1;
510        image = -1;
511        chi2  = 0;
512        nstep = 0;
513        for(int it1=0;it1<5;it1++){
514            al[it1] = 0;
515            for(int it2=0;it2<5;it2++)coval[it1][it2] = 0;
516        };
517        for(int ip=0;ip<6;ip++){
518            xgood[ip]  = 0;
519            ygood[ip]  = 0;
520            xm[ip]     = 0;
521            ym[ip]     = 0;
522            zm[ip]     = 0;
523            resx[ip]   = 0;
524            resy[ip]   = 0;
525            xv[ip]     = 0;
526            yv[ip]     = 0;
527            zv[ip]     = 0;
528            axv[ip]    = 0;
529            ayv[ip]    = 0;
530            dedx_x[ip] = 0;
531            dedx_y[ip] = 0;
532    
533        };
534        if(clx)clx->Clear();
535        if(cly)cly->Clear();
536    };
537    //--------------------------------------
538    //
539    //
540    //--------------------------------------
541    void TrkTrack::Delete(){
542    //    cout << "TrkTrack::Delete()"<<endl;
543    //    Clear();
544        if(clx)delete clx;
545        if(cly)delete cly;
546    };
547    //--------------------------------------
548    //
549    //
550    //--------------------------------------
551    
552  //--------------------------------------  //--------------------------------------
553  //  //
554  //  //
555  //--------------------------------------  //--------------------------------------
556  TrkSinglet::TrkSinglet(){  TrkSinglet::TrkSinglet(){
557    //    cout << "TrkSinglet::TrkSinglet() " << GetUniqueID()<<endl;
558      plane    = 0;      plane    = 0;
559      coord[0] = 0;      coord[0] = 0;
560      coord[1] = 0;      coord[1] = 0;
561      sgnl     = 0;      sgnl     = 0;
562    //    cls      = 0;
563  };  };
564  //--------------------------------------  //--------------------------------------
565  //  //
566  //  //
567  //--------------------------------------  //--------------------------------------
568  TrkSinglet::TrkSinglet(const TrkSinglet& s){  TrkSinglet::TrkSinglet(const TrkSinglet& s){
569    //    cout << "TrkSinglet::TrkSinglet(const TrkSinglet& s) " << GetUniqueID()<<endl;
570      plane    = s.plane;      plane    = s.plane;
571      coord[0] = s.coord[0];      coord[0] = s.coord[0];
572      coord[1] = s.coord[1];      coord[1] = s.coord[1];
573      sgnl     = s.sgnl;      sgnl     = s.sgnl;
574    //      cls      = 0;//<<<<pointer
575        cls      = TRef(s.cls);
576  };  };
577  //--------------------------------------  //--------------------------------------
578  //  //
# Line 187  void TrkSinglet::Dump(){ Line 589  void TrkSinglet::Dump(){
589  //  //
590  //  //
591  //--------------------------------------  //--------------------------------------
592    void TrkSinglet::Clear(){
593    //    cout << "TrkSinglet::Clear() " << GetUniqueID()<<endl;
594    //    cls=0;
595        plane=-1;
596        coord[0]=-999;
597        coord[1]=-999;
598        sgnl=0;
599        
600    }
601    //--------------------------------------
602    //
603    //
604    //--------------------------------------
605  TrkLevel2::TrkLevel2(){  TrkLevel2::TrkLevel2(){
606      good2    = -1;    //    cout <<"TrkLevel2::TrkLevel2()"<<endl;
607      for(Int_t i=0; i<12 ; i++){      for(Int_t i=0; i<12 ; i++){
608          crc[i] = -1;                  good[i] = -1;
609      };          };
610      Track    = new TClonesArray("TrkTrack");  // okkio!! memory-leak
611      SingletX = new TClonesArray("TrkSinglet");  //     Track    = new TClonesArray("TrkTrack");
612      SingletY = new TClonesArray("TrkSinglet");  //     SingletX = new TClonesArray("TrkSinglet");
613    //     SingletY = new TClonesArray("TrkSinglet");
614        Track    = 0;
615        SingletX = 0;
616        SingletY = 0;
617    
618    }
619    //--------------------------------------
620    //
621    //
622    //--------------------------------------
623    void TrkLevel2::Set(){
624        if(!Track)Track    = new TClonesArray("TrkTrack");
625        if(!SingletX)SingletX = new TClonesArray("TrkSinglet");
626        if(!SingletY)SingletY = new TClonesArray("TrkSinglet");
627  }  }
628  //--------------------------------------  //--------------------------------------
629  //  //
630  //  //
631  //--------------------------------------  //--------------------------------------
632  void TrkLevel2::Dump(){  void TrkLevel2::Dump(){
633      TClonesArray &t  = *Track;          
634      TClonesArray &sx = *SingletX;          //
     TClonesArray &sy = *SingletY;  
   
635      cout << endl << endl << "=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-";      cout << endl << endl << "=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-=-";
636      cout << endl << "good2    : " << good2;      cout << endl << "good     : "; for(int i=0; i<12; i++) cout << good[i]<<" ";
     cout << endl << "crc      : "; for(int i=0; i<12; i++) cout << crc[i];  
637      cout << endl << "ntrk()   : " << this->ntrk() ;      cout << endl << "ntrk()   : " << this->ntrk() ;
638      cout << endl << "nclsx()  : " << this->nclsx();      cout << endl << "nclsx()  : " << this->nclsx();
639      cout << endl << "nclsy()  : " << this->nclsy();      cout << endl << "nclsy()  : " << this->nclsy();
640      for(int i=0; i<this->ntrk(); i++)     ((TrkTrack *)t[i])->Dump();  //     TClonesArray &t  = *Track;
641      for(int i=0; i<this->nclsx(); i++) ((TrkSinglet *)sx[i])->Dump();  //     TClonesArray &sx = *SingletX;
642      for(int i=0; i<this->nclsy(); i++) ((TrkSinglet *)sy[i])->Dump();  //     TClonesArray &sy = *SingletY;
643    //     for(int i=0; i<ntrk(); i++)     ((TrkTrack *)t[i])->Dump();
644    //     for(int i=0; i<nclsx(); i++) ((TrkSinglet *)sx[i])->Dump();
645    //     for(int i=0; i<nclsy(); i++) ((TrkSinglet *)sy[i])->Dump();
646        if(Track){
647            TClonesArray &t  = *Track;
648            for(int i=0; i<ntrk(); i++)     ((TrkTrack *)t[i])->Dump();
649        }      
650        if(SingletX){
651            TClonesArray &sx = *SingletX;
652            for(int i=0; i<nclsx(); i++) ((TrkSinglet *)sx[i])->Dump();
653        }
654        if(SingletY){
655            TClonesArray &sy = *SingletY;
656            for(int i=0; i<nclsy(); i++) ((TrkSinglet *)sy[i])->Dump();
657        }
658  }  }
659  //--------------------------------------  //--------------------------------------
660  //  //
# Line 222  void TrkLevel2::Dump(){ Line 663  void TrkLevel2::Dump(){
663  /**  /**
664   * Fills a TrkLevel2 object with values from a struct cTrkLevel2 (to get data from F77 common).   * Fills a TrkLevel2 object with values from a struct cTrkLevel2 (to get data from F77 common).
665   */   */
666  void TrkLevel2::FillCommonVar(cTrkLevel2 *l2){  // void TrkLevel2::SetFromLevel2Struct(cTrkLevel2 *l2){
667    
668    //      //  temporary objects:
669    //     TrkSinglet* t_singlet = new TrkSinglet();
670    //     TrkTrack*   t_track   = new TrkTrack();
671    
672    //      //  **** general variables ****
673    // //    good2 = l2->good2;
674    //     for(Int_t i=0; i<12 ; i++){
675    // //           crc[i] = l2->crc[i];
676    //              good[i] = l2->good[i];
677    //      };
678    //      //  *** TRACKS ***
679    //     if(!Track) Track = new TClonesArray("TrkTrack");
680    //     TClonesArray &t = *Track;
681    //     for(int i=0; i<l2->ntrk; i++){
682    //      t_track->seqno = i;// NBNBNBNB deve sempre essere = i
683    //      t_track->image = l2->image[i]-1;
684    //      //      cout << "track "<<i<<t_track->seqno << t_track->image<<endl;
685    //      t_track->chi2  = l2->chi2_nt[i];
686    //      t_track->nstep = l2->nstep_nt[i];
687    //      for(int it1=0;it1<5;it1++){
688    //          t_track->al[it1] = l2->al_nt[i][it1];
689    //          for(int it2=0;it2<5;it2++)
690    //              t_track->coval[it1][it2] = l2->coval[i][it2][it1];
691    //      };
692    //      for(int ip=0;ip<6;ip++){
693    //          t_track->xgood[ip]  = l2->xgood_nt[i][ip];
694    //          t_track->ygood[ip]  = l2->ygood_nt[i][ip];
695    //          t_track->xm[ip]     = l2->xm_nt[i][ip];
696    //          t_track->ym[ip]     = l2->ym_nt[i][ip];
697    //          t_track->zm[ip]     = l2->zm_nt[i][ip];
698    //          t_track->resx[ip]   = l2->resx_nt[i][ip];
699    //          t_track->resy[ip]   = l2->resy_nt[i][ip];
700    //          t_track->xv[ip]     = l2->xv_nt[i][ip];
701    //          t_track->yv[ip]     = l2->yv_nt[i][ip];
702    //          t_track->zv[ip]     = l2->zv_nt[i][ip];
703    //          t_track->axv[ip]    = l2->axv_nt[i][ip];
704    //          t_track->ayv[ip]    = l2->ayv_nt[i][ip];
705    //          t_track->dedx_x[ip] = l2->dedx_x[i][ip];
706    //          t_track->dedx_y[ip] = l2->dedx_y[i][ip];
707    // //                   t_track->clx[ip] = 0;
708    // //                   t_track->cly[ip] = 0;
709    //      };
710    //      new(t[i]) TrkTrack(*t_track);
711    //      t_track->Clear();
712    //     };
713    // //  *** SINGLETS ***
714    //     if(!SingletX)SingletX = new TClonesArray("TrkSinglet");
715    //     TClonesArray &sx = *SingletX;
716    //     for(int i=0; i<l2->nclsx; i++){
717    //      t_singlet->plane    = l2->planex[i];
718    //      t_singlet->coord[0] = l2->xs[i][0];
719    //      t_singlet->coord[1] = l2->xs[i][1];
720    //      t_singlet->sgnl     = l2->signlxs[i];
721    // //           t_singlet->cls      = 0;
722    //      new(sx[i]) TrkSinglet(*t_singlet);
723    //      t_singlet->Clear();
724    //     }
725    //     if(!SingletY)SingletY = new TClonesArray("TrkSinglet");
726    //     TClonesArray &sy = *SingletY;
727    //     for(int i=0; i<l2->nclsy; i++){
728    //      t_singlet->plane    = l2->planey[i];
729    //      t_singlet->coord[0] = l2->ys[i][0];
730    //      t_singlet->coord[1] = l2->ys[i][1];
731    //      t_singlet->sgnl     = l2->signlys[i];
732    // //           t_singlet->cls      = 0;
733    //      new(sy[i]) TrkSinglet(*t_singlet);
734    //      t_singlet->Clear();
735    //     };
736            
737    //     delete t_track;
738    //     delete t_singlet;
739    // }
740    //--------------------------------------
741    //
742    //
743    //--------------------------------------
744    /**
745     * Fills a TrkLevel2 object with values from a struct cTrkLevel2 (to get data from F77 common).
746     * Ref to Level1 data (clusters) is also set. If l1==NULL no references are set.
747     * (NB It make sense to set references only if events are stored in a tree that contains also the Level1 branch)
748     */
749    void TrkLevel2::SetFromLevel2Struct(cTrkLevel2 *l2, TrkLevel1 *l1){
750    
751    //    cout << "void TrkLevel2::SetFromLevel2Struct(cTrkLevel2 *l2, TrkLevel1 *l1)"<<endl;
752        Clear();
753  //  temporary objects:  //  temporary objects:
754      TrkSinglet* t_singlet = new TrkSinglet();      TrkSinglet* t_singlet = new TrkSinglet();
755      TrkTrack*   t_track   = new TrkTrack();      TrkTrack*   t_track   = new TrkTrack();
756    
757    //  -----------------
758  //  general variables  //  general variables
759      good2 = l2->good2;  //  -----------------
760      for(Int_t i=0; i<12 ; i++){      for(Int_t i=0; i<12 ; i++){
761          crc[i] = l2->crc[i];          good[i] = l2->good[i];
762      };      };
763    //  --------------
764  //  *** TRACKS ***  //  *** TRACKS ***
765    //  --------------
766        if(!Track) Track = new TClonesArray("TrkTrack");
767      TClonesArray &t = *Track;      TClonesArray &t = *Track;
768        //-----------------------------------------------------
769        if( l1 && !t_track->clx )t_track->clx = new TRefArray(6,0);
770        if( l1 && !t_track->cly )t_track->cly = new TRefArray(6,0);
771        //-----------------------------------------------------
772      for(int i=0; i<l2->ntrk; i++){      for(int i=0; i<l2->ntrk; i++){
773    //      cout <<" TRACK "<<i<<" ------------------ "<<endl;
774            t_track->seqno = i;// NBNBNBNB deve sempre essere = i
775          t_track->image = l2->image[i]-1;          t_track->image = l2->image[i]-1;
776          t_track->chi2  = l2->chi2_nt[i];          t_track->chi2  = l2->chi2_nt[i];
777            t_track->nstep = l2->nstep_nt[i];
778          for(int it1=0;it1<5;it1++){          for(int it1=0;it1<5;it1++){
779              t_track->al[it1] = l2->al_nt[i][it1];              t_track->al[it1] = l2->al_nt[i][it1];
780              for(int it2=0;it2<5;it2++)              for(int it2=0;it2<5;it2++)
781                  t_track->coval[it1][it2] = l2->coval[i][it2][it1];                  t_track->coval[it1][it2] = l2->coval[i][it2][it1];
782          };          };
783          for(int ip=0;ip<6;ip++){          for(int ip=0;ip<6;ip++){
784              t_track->xgood[ip]  = l2->xgood_nt[i][ip];              t_track->xgood[ip]  = l2->cltrx[i][ip];//l2->xgood_nt[i][ip];
785              t_track->ygood[ip]  = l2->ygood_nt[i][ip];              t_track->ygood[ip]  = l2->cltry[i][ip];//l2->ygood_nt[i][ip];
786              t_track->xm[ip]     = l2->xm_nt[i][ip];              t_track->xm[ip]     = l2->xm_nt[i][ip];
787              t_track->ym[ip]     = l2->ym_nt[i][ip];              t_track->ym[ip]     = l2->ym_nt[i][ip];
788              t_track->zm[ip]     = l2->zm_nt[i][ip];              t_track->zm[ip]     = l2->zm_nt[i][ip];
# Line 256  void TrkLevel2::FillCommonVar(cTrkLevel2 Line 795  void TrkLevel2::FillCommonVar(cTrkLevel2
795              t_track->ayv[ip]    = l2->ayv_nt[i][ip];              t_track->ayv[ip]    = l2->ayv_nt[i][ip];
796              t_track->dedx_x[ip] = l2->dedx_x[i][ip];              t_track->dedx_x[ip] = l2->dedx_x[i][ip];
797              t_track->dedx_y[ip] = l2->dedx_y[i][ip];              t_track->dedx_y[ip] = l2->dedx_y[i][ip];
798    //          cout << " ip "<<ip<<" l2->cltrx[i][ip] "<< l2->cltrx[i][ip]<<endl;
799    //          cout << " ip "<<ip<<" l2->cltry[i][ip] "<< l2->cltry[i][ip]<<endl;
800                //-----------------------------------------------------
801                //-----------------------------------------------------
802    //          if(l1 && t_track->xgood[ip])t_track->clx->AddAt(l1->GetCluster(l2->cltrx[i][ip]-1),ip);
803    //          if(l1 && t_track->ygood[ip])t_track->cly->AddAt(l1->GetCluster(l2->cltry[i][ip]-1),ip);
804                if(l2->xgood_nt[i][ip]){
805    //                  cout << " ip "<<ip<<" l2->cltrx[i][ip] "<< l2->cltrx[i][ip]<<endl;
806    //                  cout << " ip "<<ip<<" l2->cltrx[i][ip] "<< l2->cltrx[i][ip]<<" ";
807    //                  if( l1->GetCluster(l2->cltrx[i][ip]-1)->TestBit(l1->GetCluster(l2->cltrx[i][ip]-1)->kIsReferenced) )cout << ">> is referenced ";
808                    if(l1)t_track->clx->AddAt(l1->GetCluster(l2->cltrx[i][ip]-1),ip);
809    //                  cout << " --- "<<l1->GetCluster(l2->cltrx[i][ip]-1)->GetUniqueID()<<endl;
810    //              t_track->xgood[ip] = l2->cltrx[i][ip]; // WORK-AROUND *****
811                }else{
812                    if(l1)t_track->clx->RemoveAt(ip);
813                }
814                if(l2->ygood_nt[i][ip]){
815    //                  cout << " ip "<<ip<<" l2->cltry[i][ip] "<< l2->cltry[i][ip]<<endl;
816    //                  cout << " ip "<<ip<<" l2->cltry[i][ip] "<< l2->cltry[i][ip]<<" ";
817    //                  if( l1->GetCluster(l2->cltry[i][ip]-1)->TestBit(l1->GetCluster(l2->cltry[i][ip]-1)->kIsReferenced) )cout << ">> is referenced ";
818                    if(l1)t_track->cly->AddAt(l1->GetCluster(l2->cltry[i][ip]-1),ip);
819    //                  cout << " --- "<<l1->GetCluster(l2->cltry[i][ip]-1)->GetUniqueID()<<endl;
820    //              t_track->ygood[ip] = l2->cltry[i][ip]; // WORK-AROUND *****
821                }else{
822                    if(l1)t_track->cly->RemoveAt(ip);
823                }
824                //-----------------------------------------------------
825                //-----------------------------------------------------
826          };          };
827          new(t[i]) TrkTrack(*t_track);          new(t[i]) TrkTrack(*t_track);
828          t_track->Clear();          t_track->Clear();
829      };      };
830    //  ----------------
831  //  *** SINGLETS ***  //  *** SINGLETS ***
832    //  ----------------
833        if(!SingletX)SingletX = new TClonesArray("TrkSinglet");
834      TClonesArray &sx = *SingletX;      TClonesArray &sx = *SingletX;
835      for(int i=0; i<l2->nclsx; i++){      for(int i=0; i<l2->nclsx; i++){
836          t_singlet->plane    = l2->planex[i];          t_singlet->plane    = l2->planex[i];
837          t_singlet->coord[0] = l2->xs[i][0];          t_singlet->coord[0] = l2->xs[i][0];
838          t_singlet->coord[1] = l2->xs[i][1];          t_singlet->coord[1] = l2->xs[i][1];
839          t_singlet->sgnl     = l2->signlxs[i];          t_singlet->sgnl     = l2->signlxs[i];
840            //-----------------------------------------------------
841            if(l1) t_singlet->cls      = l1->GetCluster(l2->clsx[i]-1);
842            //-----------------------------------------------------
843          new(sx[i]) TrkSinglet(*t_singlet);          new(sx[i]) TrkSinglet(*t_singlet);
844          t_singlet->Clear();          t_singlet->Clear();
845      }      }
846        if(!SingletY)SingletY = new TClonesArray("TrkSinglet");
847      TClonesArray &sy = *SingletY;      TClonesArray &sy = *SingletY;
848      for(int i=0; i<l2->nclsy; i++){      for(int i=0; i<l2->nclsy; i++){
849          t_singlet->plane    = l2->planey[i];          t_singlet->plane    = l2->planey[i];
850          t_singlet->coord[0] = l2->ys[i][0];          t_singlet->coord[0] = l2->ys[i][0];
851          t_singlet->coord[1] = l2->ys[i][1];          t_singlet->coord[1] = l2->ys[i][1];
852          t_singlet->sgnl     = l2->signlys[i];          t_singlet->sgnl     = l2->signlys[i];
853            //-----------------------------------------------------
854            if(l1) t_singlet->cls      = l1->GetCluster(l2->clsy[i]-1);
855            //-----------------------------------------------------
856          new(sy[i]) TrkSinglet(*t_singlet);          new(sy[i]) TrkSinglet(*t_singlet);
857          t_singlet->Clear();          t_singlet->Clear();
858          };      };
859            
860        delete t_track;
861        delete t_singlet;
862    }
863    /**
864     * Fills a struct cTrkLevel2 with values from a TrkLevel2 object (to put data into a F77 common).
865     */
866    
867    void TrkLevel2::GetLevel2Struct(cTrkLevel2 *l2) const {
868      
869    //  general variables
870    //    l2->good2 = good2 ;
871        for(Int_t i=0; i<12 ; i++){
872    //      l2->crc[i] = crc[i];
873                    l2->good[i] = good[i];
874        };
875    //  *** TRACKS ***
876    
877        if(Track){
878            l2->ntrk              =  Track->GetEntries();    
879            for(Int_t i=0;i<l2->ntrk;i++){
880                l2->image[i] = 1 + ((TrkTrack *)Track->At(i))->image;
881                l2->chi2_nt[i] =  ((TrkTrack *)Track->At(i))->chi2;
882                l2->nstep_nt[i] =  ((TrkTrack *)Track->At(i))->nstep;
883                for(int it1=0;it1<5;it1++){
884                    l2->al_nt[i][it1] = ((TrkTrack *)Track->At(i))->al[it1];
885                    for(int it2=0;it2<5;it2++)
886                        l2->coval[i][it2][it1] = ((TrkTrack *)Track->At(i))->coval[it1][it2];
887                };
888                for(int ip=0;ip<6;ip++){
889                    l2->xgood_nt[i][ip] = ((TrkTrack *)Track->At(i))->XGood(ip);
890                    l2->ygood_nt[i][ip] = ((TrkTrack *)Track->At(i))->YGood(ip);
891                    l2->xm_nt[i][ip]    = ((TrkTrack *)Track->At(i))->xm[ip];
892                    l2->ym_nt[i][ip]    = ((TrkTrack *)Track->At(i))->ym[ip];
893                    l2->zm_nt[i][ip]    = ((TrkTrack *)Track->At(i))->zm[ip];
894                    l2->resx_nt[i][ip]  = ((TrkTrack *)Track->At(i))->resx[ip];
895                    l2->resy_nt[i][ip]  = ((TrkTrack *)Track->At(i))->resy[ip];
896                    l2->xv_nt[i][ip]    = ((TrkTrack *)Track->At(i))->xv[ip];
897                    l2->yv_nt[i][ip]    = ((TrkTrack *)Track->At(i))->yv[ip];
898                    l2->zv_nt[i][ip]    = ((TrkTrack *)Track->At(i))->zv[ip];
899                    l2->axv_nt[i][ip]   = ((TrkTrack *)Track->At(i))->axv[ip];
900                    l2->ayv_nt[i][ip]   = ((TrkTrack *)Track->At(i))->ayv[ip];
901                    l2->dedx_x[i][ip]   = ((TrkTrack *)Track->At(i))->dedx_x[ip];
902                    l2->dedx_y[i][ip]   = ((TrkTrack *)Track->At(i))->dedx_y[ip];
903                };
904            }
905        }
906    //  *** SINGLETS ***    
907        if(SingletX){
908            l2->nclsx              = SingletX->GetEntries();
909            for(Int_t i=0;i<l2->nclsx;i++){
910                l2->planex[i]  = ((TrkSinglet *)SingletX->At(i))->plane;
911                l2->xs[i][0]   = ((TrkSinglet *)SingletX->At(i))->coord[0];
912                l2->xs[i][1]   = ((TrkSinglet *)SingletX->At(i))->coord[1];
913                l2->signlxs[i] = ((TrkSinglet *)SingletX->At(i))->sgnl;
914            }
915        }
916    
917        if(SingletY){
918            l2->nclsy              = SingletY->GetEntries();
919            for(Int_t i=0;i<l2->nclsy;i++){
920                l2->planey[i]  = ((TrkSinglet *)SingletY->At(i))->plane;
921                l2->ys[i][0]   = ((TrkSinglet *)SingletY->At(i))->coord[0];
922                l2->ys[i][1]   = ((TrkSinglet *)SingletY->At(i))->coord[1];
923                l2->signlys[i] = ((TrkSinglet *)SingletY->At(i))->sgnl;
924            }
925        }
926  }  }
927  //--------------------------------------  //--------------------------------------
928  //  //
929  //  //
930  //--------------------------------------  //--------------------------------------
931  void TrkLevel2::Clear(){  void TrkLevel2::Clear(){
     good2    = -1;  
932      for(Int_t i=0; i<12 ; i++){      for(Int_t i=0; i<12 ; i++){
933          crc[i] = -1;          good[i] = -1;
934      };      };
935      Track->RemoveAll();  //    if(Track)Track->Clear("C");
936      SingletX->RemoveAll();  //    if(SingletX)SingletX->Clear("C");
937      SingletY->RemoveAll();  //    if(SingletY)SingletY->Clear("C");
938        if(Track)Track->Delete();
939        if(SingletX)SingletX->Delete();
940        if(SingletY)SingletY->Delete();
941    }
942    // //--------------------------------------
943    // //
944    // //
945    // //--------------------------------------
946    void TrkLevel2::Delete(){
947            
948    //    cout << "void TrkLevel2::Delete()"<<endl;
949        Clear();
950        if(Track)delete Track;
951        if(SingletX)delete SingletX;
952        if(SingletY)delete SingletY;
953    
954  }  }
955  //--------------------------------------  //--------------------------------------
956  //  //
# Line 301  void TrkLevel2::Clear(){ Line 960  void TrkLevel2::Clear(){
960   * 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).   * 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).
961   * This method is overridden by PamLevel2::GetTracks(), where calorimeter and TOF information is used.   * This method is overridden by PamLevel2::GetTracks(), where calorimeter and TOF information is used.
962   */   */
963  TClonesArray *TrkLevel2::GetTracks(){  TRefArray *TrkLevel2::GetTracks_NFitSorted(){
964    
965      TClonesArray *sorted = new TClonesArray("TrkTrack");      if(!Track)return 0;
     TClonesArray &t = *Track;  
     TClonesArray &ts = *sorted;  
     int N=this->ntrk();  
     vector<int> m(N); for(int i=0; i<N; i++)m[i]=1;  
966    
967        TRefArray *sorted = new TRefArray();
968            
969        TClonesArray &t  = *Track;
970    //    TClonesArray &ts = *PhysicalTrack;
971        int N = ntrk();
972        vector<int> m(N); for(int i=0; i<N; i++)m[i]=1;
973    //      int m[50]; for(int i=0; i<N; i++)m[i]=1;
974            
975      int indo=0;      int indo=0;
976      int indi=0;      int indi=0;
977      while(N != 0){      while(N > 0){
978          float chi2ref=1000000;  //    while(N != 0){
979          for(int i=0; i<this->ntrk(); i++){          int nfit =0;
980              if(((TrkTrack *)t[i])->chi2 < chi2ref && m[i]==1){          float chi2ref = numeric_limits<float>::max();
981                  chi2ref = ((TrkTrack *)t[i])->chi2;                  
982                  indi = i;          // first loop to search maximum num. of fit points
983            for(int i=0; i < ntrk(); i++){
984                if( ((TrkTrack *)t[i])->GetNtot() >= nfit && m[i]==1){
985                    nfit =    ((TrkTrack *)t[i])->GetNtot();
986              }              }
987          }          }
988          if( ((TrkTrack *)t[indi])->image != -1 ){          //second loop to search minimum chi2 among selected
989            for(int i=0; i<ntrk(); i++){
990                Float_t chi2 = ((TrkTrack *)t[i])->chi2;
991                if(chi2 < 0) chi2 = -chi2*1000;
992                if(    chi2 < chi2ref
993                       && ((TrkTrack *)t[i])->GetNtot() == nfit
994                       && m[i]==1){
995                    chi2ref = ((TrkTrack *)t[i])->chi2;
996                    indi = i;
997                };
998            };
999            if( ((TrkTrack *)t[indi])->HasImage() ){
1000              m[((TrkTrack *)t[indi])->image] = 0;              m[((TrkTrack *)t[indi])->image] = 0;
1001              N--;              N--;
1002          }          
1003          new(ts[indo]) TrkTrack(*(TrkTrack*)t[indi]);  //          cout << "i** "<< ((TrkTrack *)t[indi])->image << " " << nfiti <<" "<<chi2i<<endl;
1004            };
1005            sorted->Add( (TrkTrack*)t[indi] );      
1006                    
1007          m[indi] = 0;          m[indi] = 0;
1008    //      cout << "SORTED "<< indo << " "<< indi << " "<< N << " "<<((TrkTrack *)t[indi])->image<<" "<<chi2ref<<endl;
1009          N--;              N--;    
1010          indo++;          indo++;
1011      }      }
1012        m.clear();
1013    //    cout << "GetTracks_NFitSorted(it): Done"<< endl;
1014    
1015      return sorted;      return sorted;
1016    //    return PhysicalTrack;
1017  }  }
1018  //--------------------------------------  //--------------------------------------
1019  //  //
# Line 346  TrkTrack *TrkLevel2::GetStoredTrack(int Line 1031  TrkTrack *TrkLevel2::GetStoredTrack(int
1031          cout << "                Stored tracks ntrk() = "<< this->ntrk() << endl;          cout << "                Stored tracks ntrk() = "<< this->ntrk() << endl;
1032          return 0;          return 0;
1033      }      }
1034        if(!Track){
1035            cout << "TrkTrack *TrkLevel2::GetStoredTrack(int is) >> (TClonesArray*) Track ==0 "<<endl;
1036        };
1037      TClonesArray &t = *(Track);      TClonesArray &t = *(Track);
1038      TrkTrack *track = (TrkTrack*)t[is];      TrkTrack *track = (TrkTrack*)t[is];
1039      return track;      return track;
# Line 355  TrkTrack *TrkLevel2::GetStoredTrack(int Line 1043  TrkTrack *TrkLevel2::GetStoredTrack(int
1043  //  //
1044  //--------------------------------------  //--------------------------------------
1045  /**  /**
1046     * Retrieves the is-th stored X singlet.
1047     * @param it Singlet number, ranging from 0 to nclsx().
1048     */
1049    TrkSinglet *TrkLevel2::GetSingletX(int is){
1050    
1051            if(is >= this->nclsx()){
1052                    cout << "** TrkLevel2 ** Singlet "<< is << "doen not exits! " << endl;
1053                    cout << "                Stored x-singlets nclsx() = "<< this->nclsx() << endl;
1054                    return 0;
1055            }
1056            if(!SingletX)return 0;
1057            TClonesArray &t = *(SingletX);
1058            TrkSinglet *singlet = (TrkSinglet*)t[is];
1059            return singlet;
1060    }
1061    //--------------------------------------
1062    //
1063    //
1064    //--------------------------------------
1065    /**
1066     * Retrieves the is-th stored Y singlet.
1067     * @param it Singlet number, ranging from 0 to nclsx().
1068     */
1069    TrkSinglet *TrkLevel2::GetSingletY(int is){
1070    
1071            if(is >= this->nclsy()){
1072                    cout << "** TrkLevel2 ** Singlet "<< is << "doen not exits! " << endl;
1073                    cout << "                Stored y-singlets nclsy() = "<< this->nclsx() << endl;
1074                    return 0;
1075            }
1076            if(!SingletY)return 0;
1077            TClonesArray &t = *(SingletY);
1078            TrkSinglet *singlet = (TrkSinglet*)t[is];
1079            return singlet;
1080    }
1081    //--------------------------------------
1082    //
1083    //
1084    //--------------------------------------
1085    /**
1086   * Retrieves the it-th "physical" track, sorted by the method GetNTracks().   * Retrieves the it-th "physical" track, sorted by the method GetNTracks().
1087   * @param it Track number, ranging from 0 to GetNTracks().   * @param it Track number, ranging from 0 to GetNTracks().
1088   */   */
1089    
1090  TrkTrack *TrkLevel2::GetTrack(int it){  TrkTrack *TrkLevel2::GetTrack(int it){
1091            
1092      if(it >= this->GetNTracks()){          if(it >= this->GetNTracks()){
1093          cout << "** TrkLevel2 ** Track "<< it << "does not exits! " << endl;                  cout << "** TrkLevel2 ** Track "<< it << "does not exits! " << endl;
1094          cout << "                Physical tracks GetNTracks() = "<< this->ntrk() << endl;                  cout << "                Physical tracks GetNTracks() = "<< this->ntrk() << endl;
1095          return 0;                  return 0;
1096      }          }
1097      TrkTrack *track = (TrkTrack*)(*(this->GetTracks()))[it];          
1098      return track;          TRefArray *sorted = GetTracks();  //TEMPORANEO  
1099            if(!sorted)return 0;
1100            TrkTrack *track = (TrkTrack*)sorted->At(it);
1101            sorted->Clear();
1102            delete sorted;
1103            return track;
1104  }  }
1105    /**
1106     * Give the number of "physical" tracks, sorted by the method GetTracks().
1107     */
1108    Int_t TrkLevel2::GetNTracks(){
1109                    
1110            Float_t ntot=0;
1111            if(!Track)return 0;
1112            TClonesArray &t = *Track;
1113            for(int i=0; i<ntrk(); i++) {    
1114                    if( ((TrkTrack *)t[i])->GetImageSeqNo() == -1 ) ntot+=1.;
1115                    else ntot+=0.5;
1116            }
1117            return (Int_t)ntot;
1118    
1119    };
1120  //--------------------------------------  //--------------------------------------
1121  //  //
1122  //  //
# Line 383  TrkTrack *TrkLevel2::GetTrackImage(int i Line 1132  TrkTrack *TrkLevel2::GetTrackImage(int i
1132          cout << "                Physical tracks GetNTracks() = "<< this->ntrk() << endl;          cout << "                Physical tracks GetNTracks() = "<< this->ntrk() << endl;
1133          return 0;          return 0;
1134      }      }
1135      TrkTrack *track = (TrkTrack*)(*(this->GetTracks()))[it];          
1136        TRefArray* sorted = GetTracks(); //TEMPORANEO
1137        if(!sorted)return 0;
1138        TrkTrack *track = (TrkTrack*)sorted->At(it);
1139            
1140      if(!track->HasImage()){      if(!track->HasImage()){
1141          cout << "** TrkLevel2 ** Track "<< it << "does not have image! " << endl;          cout << "** TrkLevel2 ** Track "<< it << "does not have image! " << endl;
1142          return 0;          return 0;
1143      }      }
1144        if(!Track)return 0;
1145      TrkTrack *image = (TrkTrack*)(*Track)[track->image];      TrkTrack *image = (TrkTrack*)(*Track)[track->image];
1146    
1147        sorted->Delete();
1148        delete sorted;
1149    
1150      return image;      return image;
1151            
1152  }  }
# Line 400  TrkTrack *TrkLevel2::GetTrackImage(int i Line 1158  TrkTrack *TrkLevel2::GetTrackImage(int i
1158   * Loads the magnetic field.   * Loads the magnetic field.
1159   * @param s Path of the magnetic-field files.   * @param s Path of the magnetic-field files.
1160   */   */
1161  void TrkLevel2::LoadField(TString s){  void TrkLevel2::LoadField(TString path){
1162      readb_(s.Data());  //
1163    //     strcpy(path_.path,path.Data());
1164    //     path_.pathlen = path.Length();
1165    //     path_.error   = 0;
1166    //     readb_();
1167    
1168        TrkParams::Set(path,1);
1169        TrkParams::Load(1);
1170    
1171    //
1172  };  };
1173    /**
1174     * Get BY (kGauss)
1175     * @param v (x,y,z) coordinates in cm
1176     */
1177    float TrkLevel2::GetBX(float* v){
1178        float b[3];
1179        gufld_(v,b);
1180        return b[0]/10.;
1181    }
1182    /**
1183     * Get BY (kGauss)
1184     * @param v (x,y,z) coordinates in cm
1185     */
1186    float TrkLevel2::GetBY(float* v){
1187        float b[3];
1188        gufld_(v,b);
1189        return b[1]/10.;
1190    }
1191    /**
1192     * Get BY (kGauss)
1193     * @param v (x,y,z) coordinates in cm
1194     */
1195    float TrkLevel2::GetBZ(float* v){
1196        float b[3];
1197        gufld_(v,b);
1198        return b[2]/10.;
1199    }
1200    //--------------------------------------
1201    //
1202    //
1203    //--------------------------------------
1204    /**
1205     * Get tracker-plane (mechanical) z-coordinate
1206     * @param plane_id plane index (1=TOP,2,3,4,5,6=BOTTOM)
1207     */
1208    Float_t TrkLevel2::GetZTrk(Int_t plane_id){
1209            switch(plane_id){
1210                    case 1: return ZTRK1;
1211                    case 2: return ZTRK2;
1212                    case 3: return ZTRK3;
1213                    case 4: return ZTRK4;
1214                    case 5: return ZTRK5;
1215                    case 6: return ZTRK6;
1216                    default: return 0.;
1217            };
1218    };
1219    //--------------------------------------
1220    //
1221    //
1222    //--------------------------------------
1223    /**
1224     * Trajectory default constructor.
1225     * (By default is created with z-coordinates inside the tracking volume)
1226      */
1227    Trajectory::Trajectory(){
1228        npoint = 10;
1229        x = new float[npoint];
1230        y = new float[npoint];
1231        z = new float[npoint];
1232        thx = new float[npoint];
1233        thy = new float[npoint];
1234        tl = new float[npoint];
1235        float dz = ((ZTRK1)-(ZTRK6))/(npoint-1);
1236        for(int i=0; i<npoint; i++){
1237            x[i] = 0;
1238            y[i] = 0;
1239            z[i] = (ZTRK1) - i*dz;
1240            thx[i] = 0;
1241            thy[i] = 0;
1242            tl[i] = 0;
1243        }
1244    }
1245  //--------------------------------------  //--------------------------------------
1246  //  //
1247  //  //
1248  //--------------------------------------  //--------------------------------------
1249  /**  /**
1250   * Trajectory constructor.   * Trajectory constructor.
1251     * (By default is created with z-coordinates inside the tracking volume)
1252   * \param n Number of points   * \param n Number of points
1253   */   */
1254  Trajectory::Trajectory(int n){  Trajectory::Trajectory(int n){
1255        if(n<=0){
1256            cout << "NB! Trajectory must have at least 1 point >>> created with 10 points" << endl;
1257            n=10;
1258        }
1259      npoint = n;      npoint = n;
1260      x = new float[npoint];      x = new float[npoint];
1261      y = new float[npoint];      y = new float[npoint];
1262      z = new float[npoint];      z = new float[npoint];
1263        thx = new float[npoint];
1264        thy = new float[npoint];
1265        tl = new float[npoint];
1266        float dz = ((ZTRK1)-(ZTRK6))/(npoint-1);
1267      for(int i=0; i<npoint; i++){      for(int i=0; i<npoint; i++){
1268          x[i] = 0;          x[i] = 0;
1269          y[i] = 0;          y[i] = 0;
1270          z[i] = 0;          z[i] = (ZTRK1) - i*dz;
1271            thx[i] = 0;
1272            thy[i] = 0;
1273            tl[i] = 0;
1274      }      }
1275  }  }
1276  //--------------------------------------  //--------------------------------------
# Line 432  Trajectory::Trajectory(int n){ Line 1283  Trajectory::Trajectory(int n){
1283   * \param pz Pointer to float array, defining z coordinates   * \param pz Pointer to float array, defining z coordinates
1284   */   */
1285  Trajectory::Trajectory(int n, float* zin){  Trajectory::Trajectory(int n, float* zin){
1286      npoint = n;      npoint = 10;
1287        if(n>0)npoint = n;
1288      x = new float[npoint];      x = new float[npoint];
1289      y = new float[npoint];      y = new float[npoint];
1290      z = new float[npoint];      z = new float[npoint];
1291      for(int i=0; i<npoint; i++){      thx = new float[npoint];
1292        thy = new float[npoint];
1293        tl = new float[npoint];
1294        int i=0;
1295        do{
1296          x[i] = 0;          x[i] = 0;
1297          y[i] = 0;          y[i] = 0;
1298          z[i] = zin[i];          z[i] = zin[i];
1299      }          thx[i] = 0;
1300            thy[i] = 0;
1301            tl[i] = 0;
1302            i++;            
1303        }while(zin[i-1] > zin[i] && i < npoint);
1304        npoint=i;
1305        if(npoint != n)cout << "NB! Trajectory created with "<<npoint<<" points"<<endl;
1306    }
1307    void Trajectory::Delete(){
1308        
1309        if(x) delete [] x;
1310        if(y) delete [] y;
1311        if(z) delete [] z;
1312        if(thx) delete [] thx;
1313        if(thy) delete [] thy;
1314        if(tl) delete [] tl;
1315    
1316  }  }
1317  //--------------------------------------  //--------------------------------------
1318  //  //
# Line 452  Trajectory::Trajectory(int n, float* zin Line 1324  Trajectory::Trajectory(int n, float* zin
1324  void Trajectory::Dump(){  void Trajectory::Dump(){
1325      cout <<endl<< "Trajectory ========== "<<endl;      cout <<endl<< "Trajectory ========== "<<endl;
1326      for (int i=0; i<npoint; i++){      for (int i=0; i<npoint; i++){
1327          cout << i <<" >> " << x[i] <<" "<< y[i] <<" "<< z[i] << endl;;          cout << i <<" >> " << x[i] <<" "<< y[i] <<" "<< z[i] ;
1328            cout <<" -- " << thx[i] <<" "<< thy[i] ;
1329            cout <<" -- " << tl[i] << endl;
1330        };
1331    }
1332    //--------------------------------------
1333    //
1334    //
1335    //--------------------------------------
1336    /**
1337     * Get trajectory length between two points
1338     * @param ifirst first point (default 0)
1339     * @param ilast last point (default npoint)
1340     */
1341    float Trajectory::GetLength(int ifirst, int ilast){
1342        if( ifirst<0    ) ifirst = 0;
1343        if( ilast>=npoint) ilast  = npoint-1;
1344        float l=0;
1345        for(int i=ifirst;i<=ilast;i++){
1346            l=l+tl[i];
1347      };      };
1348        if(z[ilast] > ZINI)l=l-tl[ilast];
1349        if(z[ifirst] < ZINI)   l=l-tl[ifirst];
1350        
1351        return l;
1352    
1353  }  }
1354    
1355    /**
1356     * Evaluates the trajectory in the apparatus associated to the track.
1357     * 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.  
1358     * @param t pointer to an object of the class Trajectory,
1359     * which z coordinates should be previously initialized by calling the proper constructor ( Trajectory::Trajectory(int n, float* zin) ).
1360     * @return error flag.
1361     */
1362    int Trajectory::DoTrack2(float* al){
1363    
1364        double *dxout   = new double[npoint];
1365        double *dyout   = new double[npoint];
1366        double *dthxout = new double[npoint];
1367        double *dthyout = new double[npoint];
1368        double *dtlout  = new double[npoint];
1369        double *dzin    = new double[npoint];
1370        double dal[5];
1371    
1372        int ifail = 0;
1373    
1374        for (int i=0; i<5; i++)      dal[i]  = (double)al[i];
1375        for (int i=0; i<npoint; i++) dzin[i] = (double)z[i];
1376    
1377        TrkParams::Load(1);
1378        if( !TrkParams::IsLoaded(1) ){
1379            cout << "int Trajectory::DoTrack2(float* al) --- ERROR --- m.field not loaded"<<endl;
1380            return 0;
1381        }
1382        dotrack2_(&(npoint),dzin,dxout,dyout,dthxout,dthyout,dtlout,dal,&ifail);
1383        
1384        for (int i=0; i<npoint; i++){
1385            x[i]   = (float)*dxout++;
1386            y[i]   = (float)*dyout++;
1387            thx[i] = (float)*dthxout++;
1388            thy[i] = (float)*dthyout++;
1389            tl[i]  = (float)*dtlout++;
1390        }
1391    
1392        return ifail;
1393    };
1394    
1395  ClassImp(TrkLevel2);  ClassImp(TrkLevel2);
1396  ClassImp(TrkSinglet);  ClassImp(TrkSinglet);
1397  ClassImp(TrkTrack);  ClassImp(TrkTrack);
Legend:
Removed from v.1.1  
changed lines
  Added in v.1.31
  ViewVC Help
Powered by ViewVC 1.1.23