CbmLitTrackFinderNN.h

Go to the documentation of this file.

 
#ifndef CBMLITTRACKFINDERNN_H_
#define CBMLITTRACKFINDERNN_H_
 
#include "base/CbmLitHitData.h"
#include "base/CbmLitPtrTypes.h"
#include "base/CbmLitTypes.h"
#include "interface/CbmLitTrackFinder.h"
#include <vector>
 
using std::vector;
 
class CbmLitTrackFinderNN : public CbmLitTrackFinder {
public:
  CbmLitTrackFinderNN();
 
  virtual ~CbmLitTrackFinderNN();
 
  virtual LitStatus DoFind(HitPtrVector& hits,
                           TrackPtrVector& trackSeeds,
                           TrackPtrVector& tracks);
 
  /* Setters */
  void SetSeedSelection(TrackSelectionPtr seedSelection) {
    fSeedSelection = seedSelection;
  }
  void SetFinalSelection(TrackSelectionPtr finalSelection) {
    fFinalSelection = finalSelection;
  }
  void SetPropagator(TrackPropagatorPtr propagator) {
    fPropagator = propagator;
  }
  void SetFilter(TrackUpdatePtr filter) { fFilter = filter; }
  void SetNofStations(Int_t nofStations) { fNofStations = nofStations; }
  void SetNofIterations(Int_t nofIterations) { fNofIterations = nofIterations; }
  void SetMaxNofMissingHits(const vector<Int_t>& maxNofMissingHits) {
    fMaxNofMissingHits = maxNofMissingHits;
  }
  void SetPDG(const vector<Int_t>& pdg) { fPDG = pdg; }
  void SetChiSqStripHitCut(const vector<litfloat>& chiSqStripHitCut) {
    fChiSqStripHitCut = chiSqStripHitCut;
  }
  void SetChiSqPixelHitCut(const vector<litfloat>& chiSqPixelHitCut) {
    fChiSqPixelHitCut = chiSqPixelHitCut;
  }
  void SetSigmaCoef(const vector<litfloat>& sigmaCoef) {
    fSigmaCoef = sigmaCoef;
  }
 
protected:
  void ArrangeHits(HitPtrIterator itBegin, HitPtrIterator itEnd);
 
  void InitTrackSeeds(TrackPtrIterator itBegin, TrackPtrIterator itEnd);
 
  void FollowTracks(TrackPtrIterator itBegin, TrackPtrIterator itEnd);
 
  void RemoveHits(TrackPtrIterator itBegin, TrackPtrIterator itEnd);
 
  void CopyToOutput(TrackPtrIterator itBegin,
                    TrackPtrIterator itEnd,
                    TrackPtrVector& tracks);
 
private:
  TrackPtrVector fTracks;         // Local copy of tracks.
  CbmLitHitData fHitData;         // Hit storage.
  std::set<Int_t> fUsedHitsSet;   // Sets with hits that have been used.
  std::set<Int_t> fUsedSeedsSet;  // Set with track seeds that have been used.
 
  TrackSelectionPtr fSeedSelection;   // Seed selection tool.
  TrackSelectionPtr fFinalSelection;  // Final track selection tool.
  TrackPropagatorPtr fPropagator;     // Track propagation tool.
  TrackUpdatePtr fFilter;             // KF update tool.
 
  Int_t fNofStations;    // Number of tracking stations.
  Int_t fNofIterations;  // Number of tracking iterations
  Int_t fIteration;      // Current tracking iteration
  // Tracking parameters for each iteration
  vector<Int_t>
    fMaxNofMissingHits;  // Maximum number of acceptable missing hits.
  vector<Int_t> fPDG;    // Particle hypothesis for tracking.
  vector<litfloat> fChiSqStripHitCut;  // Chi-square cut for strip hits.
  vector<litfloat> fChiSqPixelHitCut;  // Chi-square cut for pixel hits.
  vector<litfloat>
    fSigmaCoef;  // Sigma coefficient for preliminary hit selection
};
 
#endif /* CBMLITTRACKFINDERNN_H_ */