CbmLitTrackFinderBranch.h

Go to the documentation of this file.

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