PndCAGBTracker.h

Go to the documentation of this file.

  //-*- Mode: C++ -*-
  // *****************************************************************************
  //                                                                             *
  // @Autors: I.Kulakov; M.Zyzak; I.Kisel                                        *
  // @e-mail: I.Kulakov@gsi.de; M.Zyzak@gsi.de; I.Kisel@compeng.uni-frankfurt.de *
  //                                                                             *
  // *****************************************************************************

#ifndef PNDCAGBTRACKER_H
#define PNDCAGBTRACKER_H

#include "PndCATimer.h"

#include "PndCADef.h"
#include "PndCATrackParam.h"
#include "PndCAParam.h"

#include "PndCAGBHit.h"
#include "PndCAGBTrack.h"


#include "PndCAStationArray.h"
#include "PndCATarget.h"
#include "PndCANPlets.h"
#include "PndCANPletsV.h"

#include "PndCAStationSTT.h"

class PndCATarget;
class PndCAHit;
class PndCATrack;
class PndCAHits;
class PndCATracks;
class PndCAHitV;
class PndCAHitsV;

#include <cstdio>
#include <iostream>
#include <vector>
#include <string>
using std::string;
using std::vector;

class PndCAMerger;

class PndCAGBTracker
{
 public:
  PndCAGBTracker();
  ~PndCAGBTracker();
  void Init();

  void StartEvent();
  void SetNSlices( int N );

  const PndCAGBHit *Hits() const { return &fHits[0]; }
  const PndCAGBHit &Hit( int index ) const { 
    if( index<0 || index >= (int) fHits.size() ){
      cout<<"error hit index ind "<<index<<" size "<<fHits.size()<<" nHits "<<fNHits<<endl;
      exit(0);
    }
    return fHits[index]; 
  }

  int NHits() const { return fNHits; }
  double Time() const { return fTime; }
  double StatTime( int iTimer ) const { return fStatTime[iTimer]; }
  int NTimers() const { return fNTimers; }
  int StatNEvents() const { return fStatNEvents; }
  int NTracks() const { return fNTracks; }
  PndCAGBTrack *Tracks() const { return fTracks; }
  PndCAGBTrack *Tracks() { return fTracks; }
  const PndCAGBTrack &Track( int i ) const { return fTracks[i]; }
  int *TrackHits() const { return fTrackHits; }
  int *TrackHits() { return fTrackHits; }
  int TrackHit( int i ) const { return fTrackHits[i]; }

  const PndCAParam&  GetParameters() const { return fParameters; }
  PndCAParam&  GetParametersNonConst() { return fParameters; }
  
  int NStations() const { return fParameters.NStations(); }
  
  void WriteSettings( std::ostream &out ) const;
  void ReadSettings( std::istringstream &in );
  void WriteEvent( FILE *out ) const;
  
//   void ReadTracks( std::istream &in );

  void SaveHitsInFile( string prefix ) const; // Save Hits in txt file. @prefix - prefix for file name. Ex: "./data/ev1"
//   void SaveSettingsInFile( string prefix ) const; // Save geometry in txt file. @prefix - prefix for file name. Ex: "./data/"
  bool SaveTracksInFile( string prefix ) const;
  bool ReadHitsFromFile( string prefix );
//   bool ReadSettingsFromFile( string prefix );

  double SliceTrackerTime() const { return fSliceTrackerTime; }
  double SliceTrackerCpuTime() const { return fSliceTrackerCpuTime; }

  void StoreToFile( const char *filename ) const;
  void RestoreFromFile( FILE *f );
    
  void SetHits( std::vector<PndCAGBHit> &hits );     // need for StRoot
  int  GetHitsSize() const {return fHits.size();}

  void FindTracks();  

   void CATrackFinder();

  // fuctions used by TrackFinder


  void Create1Plets( const PndCATarget& target, const PndCAHits &hits, PndCAElementsOnStation<PndCANPletV>& singlets, int iStation );
  void PickUpHits( PndCAElementsOnStation<PndCANPletV>& a, PndCAElementsOnStation<PndCANPletV>& r, int iS );

  void CreateNPlets( const PndCATarget& target, const PndCAHits& hits, PndCAElementsOnStation<PndCANPletV>& triplets, int iStation, int cellLength );

  void FindNeighbours( PndCANPlets& triplets );
  void CreateTracks( const PndCANPlets& triplets, PndCATracks& tracks );

  void InvertCholetsky(float a[15]) const;
  void MultiplySS(float const C[15], float const V[15], float K[5][5]) const;
  void MultiplyMS(float const C[5][5], float const V[15], float K[15]) const;
  void MultiplySR(float const C[15], float const r_in[5], float r_out[5]) const;
  void FilterTracks(float const r[5], float const C[15], float const m[5], float const V[15], float R[5], float W[15], float &chi2) const;
  void Merge( PndCATracks& tracks );    
  
  void FindBestCandidate(int ista,
                         PndCATrack &best_tr,
                         int currITrip,
                         PndCATrack &curr_tr,
                         unsigned char min_best_l,
                         const PndCANPlets& triplets,
                         unsigned int& nCalls);
      
  // private

  enum { kFastPrimIter,     // primary fast tracks
         kAllPrimIter,      // primary all tracks
         kAllPrimJumpIter,  // primary tracks with jumped triplets
         kAllSecIter       // secondary all tracks
  };
  int fFindIter; // current iteration number

  PndCATarget fTarget;
  float fMaxInvMom;     // max considered q/p for tracks
  float fPick_m, // coefficient for size of region on middle station for add middle hits in triplets: Dx = Pick*sigma_x Dy = Pick*sigma_y
    fPick_r,  // same for right hits
    fPick; // pick for current procedure
  float fPickNeighbour; // (PickNeighbour < dp/dp_error)  =>  triplets are neighbours
  float TRACK_PROB_CUT; // = 0.01;
  float TRACK_CHI2_CUT; // = 10.0;  // cut for tracks candidates. per one DoF
  float_v TRIPLET_CHI2_CUT; // = 5.0; // cut for selecting triplets before collecting tracks.per one DoF
  float fMaxDX0; // Set correction in order to take into account overlaping
  
  friend class PndCAPerformance; //dbg
  protected:

    vector<PndCAGBHit> fHits;     //* hit array
    int fNHits;                //* N hits in event

    int *fTrackHits;           //* track->hits reference array
    PndCAGBTrack *fTracks; //* array of tracks
    int fNTracks;              //* N tracks

    double fTime; //* total time
    static const int fNTimers = 25;
    double fStatTime[fNTimers]; //* timers
    int fStatNEvents;    //* n events proceed

    double fSliceTrackerTime; // reco time of the slice tracker;
    double fSliceTrackerCpuTime; // reco time of the slice tracker;
    PndCATFIterTimerInfo fGTi; // global
    PndCATFTimerInfo fTi; // for iterations
    PndCATFIterTimerInfo fStatGTi; // global
    PndCATFTimerInfo fStatTi; // for iterations

  
    PndCAParam fParameters;
    PndCAStationSTT fStations[50];
  
  private:
    PndCAGBTracker( const PndCAGBTracker& );
    PndCAGBTracker &operator=( const PndCAGBTracker& );
};

#endif