CbmLitTrackFinderNN.h

Go to the documentation of this file.

/* Copyright (C) 2008-2013 GSI/JINR-LIT, Darmstadt/Dubna
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Andrey Lebedev [committer] */
 
#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_ */