CbmRoot
Loading...
Searching...
No Matches
Public Member Functions | Protected Member Functions | Private Attributes | List of all members
lit::parallel::LitTrackFinderNN Class Reference

#include <LitTrackFinderNN.h>

Collaboration diagram for lit::parallel::LitTrackFinderNN:
[legend]

Public Member Functions

 LitTrackFinderNN ()
 Constructor.
 
virtual ~LitTrackFinderNN ()
 Destructor.
 
void DoFind (const vector< lit::parallel::LitScalPixelHit * > &hits, const vector< lit::parallel::LitScalTrack * > &trackSeeds, vector< lit::parallel::LitScalTrack * > &tracks)
 Main function for track reconstruction.
 
void SetDetectorLayout (const LitDetectorLayoutScal &layout)
 
void SetNofIterations (int nofIterations)
 
void SetMaxNofMissingHits (const vector< int > &maxNofMissingHits)
 
void SetPDG (const vector< int > &pdg)
 
void SetChiSqStripHitCut (const vector< fscal > &chiSqStripHitCut)
 
void SetChiSqPixelHitCut (const vector< fscal > &chiSqPixelHitCut)
 
void SetSigmaCoef (const vector< fscal > &sigmaCoef)
 

Protected Member Functions

void ArrangeHits (const vector< lit::parallel::LitScalPixelHit * > &hits)
 
void InitTrackSeeds (const vector< lit::parallel::LitScalTrack * > &trackSeeds)
 Initialize track seeds and copy to local array.
 
void PropagateVirtualStations (LitTrackParamScal &par)
 
void PropagateToStation (unsigned char stationId, LitTrackParamScal &par)
 
void FollowTracks ()
 Follow tracks through detector.
 
void SelectTracks ()
 
void RemoveHits ()
 Write already used hits to a used hits set.
 
void CopyToOutput (vector< lit::parallel::LitScalTrack * > &tracks)
 Copy tracks to output array.
 

Private Attributes

vector< lit::parallel::LitScalTrack * > fTracks
 
LitHitData fHitData
 
set< int > fUsedHitsSet
 
set< int > fUsedSeedsSet
 
LitDetectorLayoutScal fLayout
 
int fNofIterations
 
int fIteration
 
vector< int > fMaxNofMissingHits
 
vector< int > fPDG
 
vector< fscal > fChiSqStripHitCut
 
vector< fscal > fChiSqPixelHitCut
 
vector< fscal > fSigmaCoef
 

Detailed Description

Definition at line 32 of file LitTrackFinderNN.h.

Constructor & Destructor Documentation

◆ LitTrackFinderNN()

lit::parallel::LitTrackFinderNN::LitTrackFinderNN ( )

Constructor.

Definition at line 31 of file LitTrackFinderNN.cxx.

References fChiSqPixelHitCut, fChiSqStripHitCut, fHitData, fIteration, fLayout, fMaxNofMissingHits, fNofIterations, fPDG, fSigmaCoef, fTracks, fUsedHitsSet, and fUsedSeedsSet.

◆ ~LitTrackFinderNN()

lit::parallel::LitTrackFinderNN::~LitTrackFinderNN ( )
virtual

Destructor.

Definition at line 47 of file LitTrackFinderNN.cxx.

Member Function Documentation

◆ ArrangeHits()

void lit::parallel::LitTrackFinderNN::ArrangeHits ( const vector< lit::parallel::LitScalPixelHit * > & hits)
protected

Definition at line 78 of file LitTrackFinderNN.cxx.

References fHitData, fUsedHitsSet, hits, and lit::parallel::LitScalPixelHit::refId.

Referenced by DoFind().

◆ CopyToOutput()

void lit::parallel::LitTrackFinderNN::CopyToOutput ( vector< lit::parallel::LitScalTrack * > & tracks)
protected

Copy tracks to output array.

Definition at line 301 of file LitTrackFinderNN.cxx.

References fTracks, fUsedSeedsSet, lit::parallel::LitScalTrack::GetPreviousTrackId(), lit::parallel::LitScalTrack::IsGood(), and tracks.

Referenced by DoFind().

◆ DoFind()

void lit::parallel::LitTrackFinderNN::DoFind ( const vector< lit::parallel::LitScalPixelHit * > & hits,
const vector< lit::parallel::LitScalTrack * > & trackSeeds,
vector< lit::parallel::LitScalTrack * > & tracks )

Main function for track reconstruction.

Parameters
[in]Arrayof hits.
[in]Arrayof track seeds.
[out]Outputarray of reconstructed tracks.

Definition at line 49 of file LitTrackFinderNN.cxx.

References ArrangeHits(), CopyToOutput(), fHitData, fIteration, fLayout, fNofIterations, FollowTracks(), fTracks, fUsedHitsSet, fUsedSeedsSet, hits, InitTrackSeeds(), RemoveHits(), SelectTracks(), and tracks.

Referenced by CbmLitFindGlobalTracksParallel::DoTracking().

◆ FollowTracks()

void lit::parallel::LitTrackFinderNN::FollowTracks ( )
protected

Follow tracks through detector.

Definition at line 187 of file LitTrackFinderNN.cxx.

References lit::parallel::LitScalTrack::AddHit(), lit::parallel::LitTrackParam< T >::C0, lit::parallel::LitTrackParam< T >::C5, fChiSqStripHitCut, fHitData, fIteration, fLayout, fMaxNofMissingHits, fSigmaCoef, fTracks, lit::parallel::LitScalTrack::GetNofHits(), lit::parallel::LitScalTrack::GetNofMissingHits(), lit::parallel::LitScalTrack::GetParamLast(), hits, lit::parallel::LitScalTrack::IncChiSq(), lit::parallel::LitFiltration(), lit::parallel::LitLineExtrapolation(), lit::parallel::NDF(), PropagateToStation(), lit::parallel::LitScalTrack::SetLastStationId(), lit::parallel::LitScalTrack::SetNDF(), lit::parallel::LitScalTrack::SetNofMissingHits(), lit::parallel::LitScalTrack::SetParamFirst(), lit::parallel::LitScalTrack::SetParamLast(), sqrt(), lit::parallel::LitScalPixelHit::X, lit::parallel::LitTrackParam< T >::X, lit::parallel::LitScalPixelHit::Y, lit::parallel::LitTrackParam< T >::Y, and lit::parallel::LitScalPixelHit::Z.

Referenced by DoFind().

◆ InitTrackSeeds()

void lit::parallel::LitTrackFinderNN::InitTrackSeeds ( const vector< lit::parallel::LitScalTrack * > & trackSeeds)
protected

Initialize track seeds and copy to local array.

Definition at line 91 of file LitTrackFinderNN.cxx.

References fTracks, lit::parallel::LitScalTrack::GetParamFirst(), lit::parallel::LitScalTrack::GetPreviousTrackId(), PropagateVirtualStations(), lit::parallel::LitTrackParam< T >::Qp, lit::parallel::LitScalTrack::SetParamFirst(), lit::parallel::LitScalTrack::SetParamLast(), and lit::parallel::LitScalTrack::SetPreviousTrackId().

Referenced by DoFind().

◆ PropagateToStation()

void lit::parallel::LitTrackFinderNN::PropagateToStation ( unsigned char stationId,
LitTrackParamScal & par )
protected

Definition at line 143 of file LitTrackFinderNN.cxx.

References fLayout, lit::parallel::LitVirtualStation< T >::GetField(), lit::parallel::LitMaterialGrid::GetMaterial(), lit::parallel::LitVirtualStation< T >::GetMaterial(), lit::parallel::LitStation< T >::GetNofVirtualStations(), lit::parallel::LitStation< T >::GetVirtualStation(), lit::parallel::LitVirtualStation< T >::GetZ(), lit::parallel::LitFieldGrid::IsEmpty(), lit::parallel::LitMaterialGrid::IsEmpty(), lit::parallel::LitAddMaterial(), lit::parallel::LitLineExtrapolation(), lit::parallel::LitRK4Extrapolation(), lit::parallel::LitTrackParam< T >::X, and lit::parallel::LitTrackParam< T >::Y.

Referenced by FollowTracks().

◆ PropagateVirtualStations()

void lit::parallel::LitTrackFinderNN::PropagateVirtualStations ( LitTrackParamScal & par)
protected

Definition at line 111 of file LitTrackFinderNN.cxx.

References fLayout, lit::parallel::LitVirtualStation< T >::GetField(), lit::parallel::LitMaterialGrid::GetMaterial(), lit::parallel::LitVirtualStation< T >::GetMaterial(), lit::parallel::LitVirtualStation< T >::GetZ(), lit::parallel::LitFieldGrid::IsEmpty(), lit::parallel::LitMaterialGrid::IsEmpty(), lit::parallel::LitAddMaterial(), lit::parallel::LitLineExtrapolation(), lit::parallel::LitRK4Extrapolation(), lit::parallel::LitTrackParam< T >::X, and lit::parallel::LitTrackParam< T >::Y.

Referenced by InitTrackSeeds().

◆ RemoveHits()

void lit::parallel::LitTrackFinderNN::RemoveHits ( )
protected

Write already used hits to a used hits set.

Definition at line 287 of file LitTrackFinderNN.cxx.

References fTracks, fUsedHitsSet, lit::parallel::LitScalTrack::GetHit(), lit::parallel::LitScalTrack::GetNofHits(), lit::parallel::LitScalTrack::IsGood(), and lit::parallel::LitScalPixelHit::refId.

Referenced by DoFind().

◆ SelectTracks()

void lit::parallel::LitTrackFinderNN::SelectTracks ( )
protected

Definition at line 277 of file LitTrackFinderNN.cxx.

References lit::parallel::DoSelectSharedHits(), fTracks, lit::parallel::LitScalTrack::GetNofHits(), and lit::parallel::LitScalTrack::IsGood().

Referenced by DoFind().

◆ SetChiSqPixelHitCut()

void lit::parallel::LitTrackFinderNN::SetChiSqPixelHitCut ( const vector< fscal > & chiSqPixelHitCut)
inline

Definition at line 60 of file LitTrackFinderNN.h.

References fChiSqPixelHitCut.

Referenced by CbmLitFindGlobalTracksParallel::DoTracking().

◆ SetChiSqStripHitCut()

void lit::parallel::LitTrackFinderNN::SetChiSqStripHitCut ( const vector< fscal > & chiSqStripHitCut)
inline

Definition at line 59 of file LitTrackFinderNN.h.

References fChiSqStripHitCut.

Referenced by CbmLitFindGlobalTracksParallel::DoTracking().

◆ SetDetectorLayout()

void lit::parallel::LitTrackFinderNN::SetDetectorLayout ( const LitDetectorLayoutScal & layout)
inline

Definition at line 55 of file LitTrackFinderNN.h.

References fLayout.

Referenced by CbmLitFindGlobalTracksParallel::DoTracking().

◆ SetMaxNofMissingHits()

void lit::parallel::LitTrackFinderNN::SetMaxNofMissingHits ( const vector< int > & maxNofMissingHits)
inline

Definition at line 57 of file LitTrackFinderNN.h.

References fMaxNofMissingHits.

Referenced by CbmLitFindGlobalTracksParallel::DoTracking().

◆ SetNofIterations()

void lit::parallel::LitTrackFinderNN::SetNofIterations ( int nofIterations)
inline

Definition at line 56 of file LitTrackFinderNN.h.

References fNofIterations.

Referenced by CbmLitFindGlobalTracksParallel::DoTracking().

◆ SetPDG()

void lit::parallel::LitTrackFinderNN::SetPDG ( const vector< int > & pdg)
inline

Definition at line 58 of file LitTrackFinderNN.h.

References fPDG.

Referenced by CbmLitFindGlobalTracksParallel::DoTracking().

◆ SetSigmaCoef()

void lit::parallel::LitTrackFinderNN::SetSigmaCoef ( const vector< fscal > & sigmaCoef)
inline

Definition at line 61 of file LitTrackFinderNN.h.

References fSigmaCoef.

Referenced by CbmLitFindGlobalTracksParallel::DoTracking().

Member Data Documentation

◆ fChiSqPixelHitCut

vector<fscal> lit::parallel::LitTrackFinderNN::fChiSqPixelHitCut
private

Definition at line 105 of file LitTrackFinderNN.h.

Referenced by LitTrackFinderNN(), and SetChiSqPixelHitCut().

◆ fChiSqStripHitCut

vector<fscal> lit::parallel::LitTrackFinderNN::fChiSqStripHitCut
private

Definition at line 104 of file LitTrackFinderNN.h.

Referenced by FollowTracks(), LitTrackFinderNN(), and SetChiSqStripHitCut().

◆ fHitData

LitHitData lit::parallel::LitTrackFinderNN::fHitData
private

Definition at line 94 of file LitTrackFinderNN.h.

Referenced by ArrangeHits(), DoFind(), FollowTracks(), and LitTrackFinderNN().

◆ fIteration

int lit::parallel::LitTrackFinderNN::fIteration
private

Definition at line 100 of file LitTrackFinderNN.h.

Referenced by DoFind(), FollowTracks(), and LitTrackFinderNN().

◆ fLayout

LitDetectorLayoutScal lit::parallel::LitTrackFinderNN::fLayout
private

Definition at line 98 of file LitTrackFinderNN.h.

Referenced by DoFind(), FollowTracks(), LitTrackFinderNN(), PropagateToStation(), PropagateVirtualStations(), and SetDetectorLayout().

◆ fMaxNofMissingHits

vector<int> lit::parallel::LitTrackFinderNN::fMaxNofMissingHits
private

Definition at line 102 of file LitTrackFinderNN.h.

Referenced by FollowTracks(), LitTrackFinderNN(), and SetMaxNofMissingHits().

◆ fNofIterations

int lit::parallel::LitTrackFinderNN::fNofIterations
private

Definition at line 99 of file LitTrackFinderNN.h.

Referenced by DoFind(), LitTrackFinderNN(), and SetNofIterations().

◆ fPDG

vector<int> lit::parallel::LitTrackFinderNN::fPDG
private

Definition at line 103 of file LitTrackFinderNN.h.

Referenced by LitTrackFinderNN(), and SetPDG().

◆ fSigmaCoef

vector<fscal> lit::parallel::LitTrackFinderNN::fSigmaCoef
private

Definition at line 106 of file LitTrackFinderNN.h.

Referenced by FollowTracks(), LitTrackFinderNN(), and SetSigmaCoef().

◆ fTracks

vector<lit::parallel::LitScalTrack*> lit::parallel::LitTrackFinderNN::fTracks
private

Definition at line 93 of file LitTrackFinderNN.h.

Referenced by CopyToOutput(), DoFind(), FollowTracks(), InitTrackSeeds(), LitTrackFinderNN(), RemoveHits(), and SelectTracks().

◆ fUsedHitsSet

set<int> lit::parallel::LitTrackFinderNN::fUsedHitsSet
private

Definition at line 95 of file LitTrackFinderNN.h.

Referenced by ArrangeHits(), DoFind(), LitTrackFinderNN(), and RemoveHits().

◆ fUsedSeedsSet

set<int> lit::parallel::LitTrackFinderNN::fUsedSeedsSet
private

Definition at line 96 of file LitTrackFinderNN.h.

Referenced by CopyToOutput(), DoFind(), and LitTrackFinderNN().

The documentation for this class was generated from the following files: