CbmLitTrackFinderBranch.h

Go to the documentation of this file.

 
#ifndef CBMLITTRACKFINDERBASEBRANCH_H_
#define CBMLITTRACKFINDERBASEBRANCH_H_
 
#include "base/CbmLitHitData.h"
#include "base/CbmLitPtrTypes.h"
#include "interface/CbmLitTrackFinder.h"
 
#include <map>
#include <set>
 
class CbmLitTrackSelection;
class CbmLitTrackUpdate;
class CbmLitTrackFitter;
 
using std::map;
using std::set;
 
class CbmLitTrackFinderBranch : public CbmLitTrackFinder {
public:
  CbmLitTrackFinderBranch();
 
  virtual ~CbmLitTrackFinderBranch();
 
  LitStatus DoFind(HitPtrVector& hits,
                   TrackPtrVector& trackSeeds,
                   TrackPtrVector& tracks);
 
  /* Setters */
  void SetFinalSelection(TrackSelectionPtr finalSelection) {
    fFinalSelection = finalSelection;
  }
  void SetSeedSelection(TrackSelectionPtr seedSelection) {
    fSeedSelection = seedSelection;
  }
  void SetFilter(TrackUpdatePtr filter) { fFilter = filter; }
  void SetPropagator(TrackPropagatorPtr propagator) {
    fPropagator = propagator;
  }
  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:
  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.
 
  TrackPtrVector fTracks;  // Local storage of tracks
 
  TrackSelectionPtr fSeedSelection;   // Track seed selection tool
  TrackSelectionPtr fFinalSelection;  // Final track selection tool
  TrackPropagatorPtr fPropagator;     // Track propagation tool
  TrackUpdatePtr fFilter;             // Kalman Filter track 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
  Int_t
    fMaxNofHitsInValidationGate;  // Maximum number of hits in the validation
                                  // gate for which a separate branch is created
  Int_t fMaxNofBranches;  // Maximum number of branches for one input track seed
  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 /*CBMLITTRACKFINDERBASEBRANCH_H_*/