/[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.2 by pam-fi, Thu Jun 1 09:03:09 2006 UTC revision 1.6 by pam-fi, Tue Jul 11 15:32:27 2006 UTC
# Line 18  extern "C" {     Line 18  extern "C" {    
18  //  //
19  //--------------------------------------  //--------------------------------------
20  TrkTrack::TrkTrack(){  TrkTrack::TrkTrack(){
21      image = 0;      seqno = -1;
22        image = -1;
23      chi2  = 0;      chi2  = 0;
24      for(int it1=0;it1<5;it1++){      for(int it1=0;it1<5;it1++){
25          al[it1] = 0;          al[it1] = 0;
# Line 47  TrkTrack::TrkTrack(){ Line 48  TrkTrack::TrkTrack(){
48  //  //
49  //--------------------------------------  //--------------------------------------
50  TrkTrack::TrkTrack(const TrkTrack& t){  TrkTrack::TrkTrack(const TrkTrack& t){
51        seqno = t.seqno;
52      image = t.image;      image = t.image;
53      chi2  = t.chi2;      chi2  = t.chi2;
54      for(int it1=0;it1<5;it1++){      for(int it1=0;it1<5;it1++){
# Line 238  TrkLevel2::TrkLevel2(){ Line 240  TrkLevel2::TrkLevel2(){
240      Track    = new TClonesArray("TrkTrack");      Track    = new TClonesArray("TrkTrack");
241      SingletX = new TClonesArray("TrkSinglet");      SingletX = new TClonesArray("TrkSinglet");
242      SingletY = new TClonesArray("TrkSinglet");      SingletY = new TClonesArray("TrkSinglet");
243    
244    //      PhysicalTrack = new TClonesArray("TrkTrack");
245            //sostituire con TRefArray... appena ho capito come si usa
246  }  }
247  //--------------------------------------  //--------------------------------------
248  //  //
# Line 266  void TrkLevel2::Dump(){ Line 271  void TrkLevel2::Dump(){
271   * 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).
272   */   */
273  void TrkLevel2::FillCommonVar(cTrkLevel2 *l2){  void TrkLevel2::FillCommonVar(cTrkLevel2 *l2){
274        //
275    //    Track    = new TClonesArray("TrkTrack");
276    //    SingletX = new TClonesArray("TrkSinglet");
277    //    SingletY = new TClonesArray("TrkSinglet");
278  //  temporary objects:  //  temporary objects:
279      TrkSinglet* t_singlet = new TrkSinglet();      TrkSinglet* t_singlet = new TrkSinglet();
280      TrkTrack*   t_track   = new TrkTrack();      TrkTrack*   t_track   = new TrkTrack();
# Line 277  void TrkLevel2::FillCommonVar(cTrkLevel2 Line 286  void TrkLevel2::FillCommonVar(cTrkLevel2
286  //  *** TRACKS ***  //  *** TRACKS ***
287      TClonesArray &t = *Track;      TClonesArray &t = *Track;
288      for(int i=0; i<l2->ntrk; i++){      for(int i=0; i<l2->ntrk; i++){
289            t_track->seqno = i;
290          t_track->image = l2->image[i]-1;          t_track->image = l2->image[i]-1;
291    //      cout << "track "<<i<<t_track->seqno << t_track->image<<endl;
292          t_track->chi2  = l2->chi2_nt[i];          t_track->chi2  = l2->chi2_nt[i];
293          for(int it1=0;it1<5;it1++){          for(int it1=0;it1<5;it1++){
294              t_track->al[it1] = l2->al_nt[i][it1];              t_track->al[it1] = l2->al_nt[i][it1];
# Line 322  void TrkLevel2::FillCommonVar(cTrkLevel2 Line 333  void TrkLevel2::FillCommonVar(cTrkLevel2
333          new(sy[i]) TrkSinglet(*t_singlet);          new(sy[i]) TrkSinglet(*t_singlet);
334          t_singlet->Clear();          t_singlet->Clear();
335          };          };
336            
337            delete t_track;
338            delete t_singlet;
339  }  }
340  //--------------------------------------  //--------------------------------------
341  //  //
# Line 332  void TrkLevel2::Clear(){ Line 346  void TrkLevel2::Clear(){
346      for(Int_t i=0; i<12 ; i++){      for(Int_t i=0; i<12 ; i++){
347          crc[i] = -1;          crc[i] = -1;
348      };      };
349      Track->RemoveAll();  /*    Track->RemoveAll();
350      SingletX->RemoveAll();      SingletX->RemoveAll();
351      SingletY->RemoveAll();      SingletY->RemoveAll();*/
352            // modify to avoid memory leakage
353            Track->Clear();
354            SingletX->Clear();
355            SingletY->Clear();
356  }  }
357  //--------------------------------------  //--------------------------------------
358  //  //
# Line 345  void TrkLevel2::Clear(){ Line 363  void TrkLevel2::Clear(){
363   * 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.
364   */   */
365  TClonesArray *TrkLevel2::GetTracks(){  TClonesArray *TrkLevel2::GetTracks(){
366        TClonesArray *sorted = GetTracks_NFitSorted();
367        return sorted;
368            
369            // fare di meglio...
370    /*      PhysicalTrack->Clear();
371            if(ntrk() > 0) GetTracks_NFitSorted();
372            return PhysicalTrack;*/
373    };
374    
375    /*TClonesArray *TrkLevel2::GetTracks_Chi2Sorted(){
376    
377      TClonesArray *sorted = new TClonesArray("TrkTrack");      TClonesArray *sorted = new TClonesArray("TrkTrack");
378      TClonesArray &t = *Track;      TClonesArray &t = *Track;
# Line 372  TClonesArray *TrkLevel2::GetTracks(){ Line 400  TClonesArray *TrkLevel2::GetTracks(){
400          indo++;          indo++;
401      }      }
402      return sorted;      return sorted;
403    }*/
404    TClonesArray *TrkLevel2::GetTracks_NFitSorted(){
405    
406            TClonesArray *sorted = new TClonesArray("TrkTrack");    
407        TClonesArray &t = *Track;
408            TClonesArray &ts = *sorted;
409    //    TClonesArray &ts = *PhysicalTrack;
410            int N=this->ntrk();
411        vector<int> m(N); for(int i=0; i<N; i++)m[i]=1;
412    
413        int indo=0;
414        int indi=0;
415        while(N != 0){
416            int nfit =0;
417            float chi2ref=1000000;
418            // first loop to search maximum num. of fit points
419            for(int i=0; i<this->ntrk(); i++){
420                if( ((TrkTrack *)t[i])->GetNtot() >= nfit && m[i]==1){
421                    nfit =    ((TrkTrack *)t[i])->GetNtot();
422    //              cout << "1** "<<i<< " " << nfit<<endl;
423                }
424            }
425            //second loop to search minimum chi2 among selected
426            for(int i=0; i<this->ntrk(); i++){
427                if(    ((TrkTrack *)t[i])->chi2 < chi2ref
428                    && ((TrkTrack *)t[i])->GetNtot()== nfit
429                    && m[i]==1){
430                    chi2ref = ((TrkTrack *)t[i])->chi2;
431                    indi = i;
432    //              cout << "2** "<<i<< " " << nfit <<" "<<chi2ref<<endl;
433                }
434            }
435            if( ((TrkTrack *)t[indi])->HasImage() ){
436                m[((TrkTrack *)t[indi])->image] = 0;
437                N--;
438    
439    //          Int_t nfiti=((TrkTrack *)t[((TrkTrack *)t[indi])->image  ])->GetNtot();
440    //          Float_t chi2i=((TrkTrack *)t[((TrkTrack *)t[indi])->image  ])->chi2;
441                    
442    //          cout << "i** "<< ((TrkTrack *)t[indi])->image << " " << nfiti <<" "<<chi2i<<endl;
443            }
444            new(ts[indo]) TrkTrack(*(TrkTrack*)t[indi]);
445            m[indi] = 0;
446            N--;    
447            indo++;
448        }
449        return sorted;
450    //    return PhysicalTrack;
451  }  }
452  //--------------------------------------  //--------------------------------------
453  //  //
# Line 398  TrkTrack *TrkLevel2::GetStoredTrack(int Line 474  TrkTrack *TrkLevel2::GetStoredTrack(int
474  //  //
475  //--------------------------------------  //--------------------------------------
476  /**  /**
477     * Retrieves the is-th stored X singlet.
478     * @param it Singlet number, ranging from 0 to nclsx().
479     */
480    TrkSinglet *TrkLevel2::GetSingletX(int is){
481    
482            if(is >= this->nclsx()){
483                    cout << "** TrkLevel2 ** Singlet "<< is << "doen not exits! " << endl;
484                    cout << "                Stored x-singlets nclsx() = "<< this->nclsx() << endl;
485                    return 0;
486            }
487            TClonesArray &t = *(SingletX);
488            TrkSinglet *singlet = (TrkSinglet*)t[is];
489            return singlet;
490    }
491    //--------------------------------------
492    //
493    //
494    //--------------------------------------
495    /**
496     * Retrieves the is-th stored Y singlet.
497     * @param it Singlet number, ranging from 0 to nclsx().
498     */
499    TrkSinglet *TrkLevel2::GetSingletY(int is){
500    
501            if(is >= this->nclsy()){
502                    cout << "** TrkLevel2 ** Singlet "<< is << "doen not exits! " << endl;
503                    cout << "                Stored y-singlets nclsy() = "<< this->nclsx() << endl;
504                    return 0;
505            }
506            TClonesArray &t = *(SingletY);
507            TrkSinglet *singlet = (TrkSinglet*)t[is];
508            return singlet;
509    }
510    //--------------------------------------
511    //
512    //
513    //--------------------------------------
514    /**
515   * Retrieves the it-th "physical" track, sorted by the method GetNTracks().   * Retrieves the it-th "physical" track, sorted by the method GetNTracks().
516   * @param it Track number, ranging from 0 to GetNTracks().   * @param it Track number, ranging from 0 to GetNTracks().
517   */   */
# Line 409  TrkTrack *TrkLevel2::GetTrack(int it){ Line 523  TrkTrack *TrkLevel2::GetTrack(int it){
523          return 0;          return 0;
524      }      }
525      TrkTrack *track = (TrkTrack*)(*(this->GetTracks()))[it];      TrkTrack *track = (TrkTrack*)(*(this->GetTracks()))[it];
526            GetTracks()->Delete();////TEMPORANEO
527      return track;      return track;
528  }  }
529    /**
530     * Give the number of "physical" tracks, sorted by the method GetTracks().
531     */
532    Int_t TrkLevel2::GetNTracks(){
533            Int_t ntot=0;
534            ntot = GetTracks()->GetEntries();
535            GetTracks()->Delete();////TEMPORANEO
536            return ntot;
537    };
538  //--------------------------------------  //--------------------------------------
539  //  //
540  //  //
# Line 432  TrkTrack *TrkLevel2::GetTrackImage(int i Line 556  TrkTrack *TrkLevel2::GetTrackImage(int i
556          return 0;          return 0;
557      }      }
558      TrkTrack *image = (TrkTrack*)(*Track)[track->image];      TrkTrack *image = (TrkTrack*)(*Track)[track->image];
559            GetTracks()->Delete(); ////TEMPORANEO
560      return image;      return image;
561            
562  }  }
# Line 451  void TrkLevel2::LoadField(TString s){ Line 576  void TrkLevel2::LoadField(TString s){
576  //  //
577  //--------------------------------------  //--------------------------------------
578  /**  /**
579     * Get tracker-plane (mechanical) z-coordinate
580     * @param plane_id plane index (1=TOP,2,3,4,5,6=BOTTOM)
581     */
582    Float_t TrkLevel2::GetZTrk(Int_t plane_id){
583            switch(plane_id){
584                    case 1: return ZTRK1;
585                    case 2: return ZTRK2;
586                    case 3: return ZTRK3;
587                    case 4: return ZTRK4;
588                    case 5: return ZTRK5;
589                    case 6: return ZTRK6;
590                    default: return 0.;
591            };
592    };
593    //--------------------------------------
594    //
595    //
596    //--------------------------------------
597    /**
598   * Trajectory default constructor.   * Trajectory default constructor.
599   * (By default is created with z-coordinates inside the tracking volume)   * (By default is created with z-coordinates inside the tracking volume)
600    */    */
# Line 462  Trajectory::Trajectory(){ Line 606  Trajectory::Trajectory(){
606      thx = new float[npoint];      thx = new float[npoint];
607      thy = new float[npoint];      thy = new float[npoint];
608      tl = new float[npoint];      tl = new float[npoint];
609      float dz = ((ZTRKUP)-(ZTRKDW))/(npoint-1);      float dz = ((ZTRK1)-(ZTRK6))/(npoint-1);
610      for(int i=0; i<npoint; i++){      for(int i=0; i<npoint; i++){
611          x[i] = 0;          x[i] = 0;
612          y[i] = 0;          y[i] = 0;
613          z[i] = (ZTRKUP) - i*dz;          z[i] = (ZTRK1) - i*dz;
614          thx[i] = 0;          thx[i] = 0;
615          thy[i] = 0;          thy[i] = 0;
616          tl[i] = 0;          tl[i] = 0;
# Line 493  Trajectory::Trajectory(int n){ Line 637  Trajectory::Trajectory(int n){
637      thx = new float[npoint];      thx = new float[npoint];
638      thy = new float[npoint];      thy = new float[npoint];
639      tl = new float[npoint];      tl = new float[npoint];
640      float dz = ((ZTRKUP)-(ZTRKDW))/(npoint-1);      float dz = ((ZTRK1)-(ZTRK6))/(npoint-1);
641      for(int i=0; i<npoint; i++){      for(int i=0; i<npoint; i++){
642          x[i] = 0;          x[i] = 0;
643          y[i] = 0;          y[i] = 0;
644          z[i] = (ZTRKUP) - i*dz;          z[i] = (ZTRK1) - i*dz;
645          thx[i] = 0;          thx[i] = 0;
646          thy[i] = 0;          thy[i] = 0;
647          tl[i] = 0;          tl[i] = 0;
# Line 571  float Trajectory::GetLength(int ifirst, Line 715  float Trajectory::GetLength(int ifirst,
715      return l;      return l;
716    
717  }  }
718    
719  ClassImp(TrkLevel2);  ClassImp(TrkLevel2);
720  ClassImp(TrkSinglet);  ClassImp(TrkSinglet);
721  ClassImp(TrkTrack);  ClassImp(TrkTrack);
Legend:
Removed from v.1.2  
changed lines
  Added in v.1.6
  ViewVC Help
Powered by ViewVC 1.1.23