LitTrackFinderNN.h

Go to the documentation of this file.

/* Copyright (C) 2013-2014 GSI/JINR-LIT, Darmstadt/Dubna
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Andrey Lebedev [committer] */
 
#ifndef LITTRACKFINDERNN_H_
#define LITTRACKFINDERNN_H_
 
#include "LitDetectorLayout.h"
#include "LitHitData.h"
 
#include <set>
#include <vector>
 
using std::set;
using std::vector;
 
#include "LitScalPixelHit.h"
#include "LitScalTrack.h"
 
namespace lit
{
  namespace parallel
  {
 
    class LitTrackFinderNN {
     public:
      LitTrackFinderNN();
 
      virtual ~LitTrackFinderNN();
 
      void DoFind(const vector<lit::parallel::LitScalPixelHit*>& hits,
                  const vector<lit::parallel::LitScalTrack*>& trackSeeds, vector<lit::parallel::LitScalTrack*>& tracks);
 
 
      /* Setters */
      void SetDetectorLayout(const LitDetectorLayoutScal& layout) { fLayout = layout; }
      void SetNofIterations(int nofIterations) { fNofIterations = nofIterations; }
      void SetMaxNofMissingHits(const vector<int>& maxNofMissingHits) { fMaxNofMissingHits = maxNofMissingHits; }
      void SetPDG(const vector<int>& pdg) { fPDG = pdg; }
      void SetChiSqStripHitCut(const vector<fscal>& chiSqStripHitCut) { fChiSqStripHitCut = chiSqStripHitCut; }
      void SetChiSqPixelHitCut(const vector<fscal>& chiSqPixelHitCut) { fChiSqPixelHitCut = chiSqPixelHitCut; }
      void SetSigmaCoef(const vector<fscal>& sigmaCoef) { fSigmaCoef = sigmaCoef; }
 
     protected:
      void ArrangeHits(const vector<lit::parallel::LitScalPixelHit*>& hits);
 
      void InitTrackSeeds(const vector<lit::parallel::LitScalTrack*>& trackSeeds);
 
      void PropagateVirtualStations(LitTrackParamScal& par);
 
      void PropagateToStation(unsigned char stationId, LitTrackParamScal& par);
 
      void FollowTracks();
 
      void SelectTracks();
 
      void RemoveHits();
 
      void CopyToOutput(vector<lit::parallel::LitScalTrack*>& tracks);
 
     private:
      vector<lit::parallel::LitScalTrack*> fTracks;  // Local copy of tracks.
      LitHitData fHitData;                           // Hit storage.
      set<int> fUsedHitsSet;                         // Sets with hits that have been used.
      set<int> fUsedSeedsSet;                        // Set with track seeds that have been used.
 
      LitDetectorLayoutScal fLayout;  // Detector layout
      int fNofIterations;             // Number of tracking iterations
      int fIteration;                 // Current tracking iteration
      // Tracking parameters for each iteration
      vector<int> fMaxNofMissingHits;   // Maximum number of acceptable missing hits.
      vector<int> fPDG;                 // Particle hypothesis for tracking.
      vector<fscal> fChiSqStripHitCut;  // Chi-square cut for strip hits.
      vector<fscal> fChiSqPixelHitCut;  // Chi-square cut for pixel hits.
      vector<fscal> fSigmaCoef;         // Sigma coefficient for preliminary hit selection
    };
 
  }  // namespace parallel
}  // namespace lit
#endif /* LITTRACKFINDERNN_H_ */