#include <PndSttCellTrackFinder.h>

Public Member Functions
	PndSttCellTrackFinder (TClonesArray *tubeArray)

virtual	~PndSttCellTrackFinder ()

void	FindTracks ()

void	AddHits (TClonesArray *hits, TString branchName)

void	SetUseGPU (Bool_t val)

void	SetRunTimeBased (Bool_t val)

void	SetClusterTime (double val)

void	SetDevTubeNeighboringsPointer (int *dev_pointer)

void	SetCalcWithCorrectedIsochrones (Bool_t val)

PndSttCellTrackFinderData *	GetTrackFinderDataObject ()

std::map< Int_t, FairHit * >	GetCorrectedIsochrones ()

int	GetNumPrimaryTracklets ()

PndTrackCand	GetFirstTrackCand (int i)

PndRiemannTrack	GetFirstRiemannTrack (int i)

PndTrackCand	GetCombiTrackCand (int i)

PndTrack	GetCombiTrack (int i)

PndRiemannTrack	GetCombiRiemannTrack (int i)

std::vector< std::vector < Double_t > >	GetTimeStampsTrackletGen ()

std::vector< std::vector < Double_t > >	GetTimeStampsGenerateNeighborhoodData ()

int	NumFirstTrackCands ()

int	NumFirstRiemannTracks ()

int	NumCombinedTracks ()

int	NumCombinedRiemannTracks ()

int	NumHits ()

int	NumHitsWithoutDouble ()

int	NumUnambiguousNeighbors ()

void	SetCalcFirstTrackletInf (Bool_t val)

void	SetVerbose (Int_t val)

void	SetBz (Double_t val)

void	StoreTrackData ()

void	Reset ()

Private Member Functions
	ClassDef (PndSttCellTrackFinder, 1)

Private Attributes
std::vector< std::vector < Double_t > >	fTimeStampsTrackletGen

std::vector< std::vector < Double_t > >	fTimeStampsGenerateNeighborhoodData

Int_t	fVerbose

Double_t	fBz

double	fClusterTime

Bool_t	fUseGPU

Bool_t	fRunTimeBased

int *	fDev_tubeNeighborings

Bool_t	fCalcWithCorrectedIsochrones

Bool_t	fCalcFirstTrackletInf

std::vector< FairHit * >	fHits

std::multimap< int, PndSttSkewedHit * >	fCombinedSkewedHits

std::map< int, FairLink >	fMapHitToFairLink

PndSttCellTrackFinderData *	fTrackFinderData

PndSttCellTrackletGenerator *	fTrackletGenerator

PndSttHitCorrector *	fHitCorrector

std::vector< PndTrackCand >	fFirstTrackCand

std::vector< PndRiemannTrack >	fFirstRiemannTrack

std::vector< PndTrackCand >	fCombiTrackCand

std::vector< PndTrack >	fCombiTrack

std::vector< PndRiemannTrack >	fCombiRiemannTrack

Detailed Description

Definition at line 24 of file PndSttCellTrackFinder.h.

Constructor & Destructor Documentation

PndSttCellTrackFinder::PndSttCellTrackFinder ( TClonesArray * tubeArray )

inline

Definition at line 26 of file PndSttCellTrackFinder.h.

References fTrackFinderData.

                                                        :
                          fVerbose(0), fBz(2.), fClusterTime(250.0), fUseGPU(kFALSE), fRunTimeBased(kFALSE), fDev_tubeNeighborings(0), fCalcWithCorrectedIsochrones(kFALSE), fCalcFirstTrackletInf(kFALSE),  fTrackFinderData(0), fTrackletGenerator(0), fHitCorrector(0) {
 
                 //Generate TrackFinderData-Object
                 fTrackFinderData= new PndSttCellTrackFinderData(tubeArray);
 
         };

virtual PndSttCellTrackFinder::~PndSttCellTrackFinder ( )

inlinevirtual

Definition at line 34 of file PndSttCellTrackFinder.h.

References fCombinedSkewedHits, fHitCorrector, fHits, fTrackFinderData, fTrackletGenerator, and i.

                                          {
                 delete fTrackFinderData;
                 delete fHitCorrector;
                 delete fTrackletGenerator;
 
                 for (size_t i = 0; i < fHits.size(); ++i) {
                         delete fHits.at(i);
                 }
                 for (std::multimap<int, PndSttSkewedHit*>::iterator it =
                                 fCombinedSkewedHits.begin(); it != fCombinedSkewedHits.end();
                                 ++it) {
                         delete (*it).second;
                 }
         }

Member Function Documentation

void PndSttCellTrackFinder::AddHits	(	TClonesArray *	hits,
		TString	branchName
	)

Definition at line 20 of file PndSttCellTrackFinder.cxx.

Referenced by PndSttCellTrackFinderTask::Exec().

                                                                           {
 
         //std::cout << "hits added!  branch name: " << branchName << "#################################################################"<<std::endl;
         //std::cout << "size of tconesarray: " << hits->GetEntriesFast() << "#################################################################"<<std::endl;
         fTrackFinderData->AddHits(hits, branchName);
 
 }

PndSttCellTrackFinder::ClassDef	(	PndSttCellTrackFinder	,
		1
	)

private

void PndSttCellTrackFinder::FindTracks ( )

Definition at line 28 of file PndSttCellTrackFinder.cxx.

References fVerbose, and MAX_THREADS_PER_BLOCK.

Referenced by PndSttCellTrackFinderTask::Exec().

                                        {
 
         // init TrackFinderData
         vector<Double_t> timesGenerateNeighorhoodData;
 
         //if(fRunTimeBased){ // Bug?????????????? J.R. 01/06-2018
         fTrackFinderData->SetRunTimeBased(fRunTimeBased);
         //}
         fTrackFinderData->SetClusterTime(fClusterTime);
         //store timestamps for runtime analysis
 
         timesGenerateNeighorhoodData.push_back(TTimeStamp());
         fTrackFinderData->GenerateNeighborhoodData();
         timesGenerateNeighorhoodData.push_back(TTimeStamp());
 
         //fTrackFinderData->PrintInfo();
 
 
         fTrackletGenerator = new PndSttCellTrackletGenerator(fTrackFinderData);
 
         fTrackletGenerator->SetCalcFirstTrackletInf(fCalcFirstTrackletInf);
         fTrackletGenerator->SetVerbose(fVerbose);
         fTrackletGenerator->SetBz(fBz);
 
         //calculation on GPU is only possible without multiple STTHits and a maximum number of MAX_THREADS_PER_BLOCK
         if (fUseGPU && (fTrackFinderData->GetAllowDoubleHits() == kFALSE)
                         && (fTrackFinderData->GetNumHitsWithoutDouble()
                                         < MAX_THREADS_PER_BLOCK)) {
                 fTrackletGenerator->SetUseGPU(fUseGPU);
                 fTrackletGenerator->SetDevTubeNeighboringsPointer(
                                 fDev_tubeNeighborings);
         }
 
         fTrackFinderData->SetClusterTime(fClusterTime);
         fTrackFinderData->SetRunTimeBased(fRunTimeBased);
 
         if (fCalcWithCorrectedIsochrones) {
                 fHitCorrector = new PndSttHitCorrector(fTrackFinderData);
                 fHitCorrector->CorrectHits();
                 map<int, FairHit*> correctedIsochrones =
                                 fHitCorrector->GetCorrectedHits();
 
                 //Get tracks with corrected hits
                 fTrackletGenerator->SetCorrectedHits(correctedIsochrones);
         }
 
         fTrackletGenerator->FindTracks();
         //fTrackletGenerator->PrintInfo();
 
         StoreTrackData();
 
         // cpu time measurement
         fTimeStampsTrackletGen.push_back(fTrackletGenerator->GetTimeStamps());
         fTimeStampsGenerateNeighborhoodData.push_back(timesGenerateNeighorhoodData);
 }

PndRiemannTrack PndSttCellTrackFinder::GetCombiRiemannTrack ( int i )

inline

Definition at line 106 of file PndSttCellTrackFinder.h.

References fCombiRiemannTrack, and i.

Referenced by PndSttCellTrackFinderTask::Exec().

                                                     {
                 return fCombiRiemannTrack[i];
         }

PndTrack PndSttCellTrackFinder::GetCombiTrack ( int i )

inline

Definition at line 101 of file PndSttCellTrackFinder.h.

References fCombiTrack, and i.

Referenced by PndSttCellTrackFinderTask::Exec().

                                       {
                 return fCombiTrack[i];
         }

PndTrackCand PndSttCellTrackFinder::GetCombiTrackCand ( int i )

inline

Definition at line 95 of file PndSttCellTrackFinder.h.

References fCombiTrackCand, and i.

Referenced by PndSttCellTrackFinderTask::Exec().

                                               {
                 return fCombiTrackCand[i];
         }

std::map<Int_t, FairHit*> PndSttCellTrackFinder::GetCorrectedIsochrones ( )

inline

Definition at line 74 of file PndSttCellTrackFinder.h.

References fHitCorrector, and PndSttHitCorrector::GetCorrectedHits().

Referenced by PndSttCellTrackFinderTask::Exec().

                                                          {
                 return fHitCorrector->GetCorrectedHits();
         }

PndRiemannTrack PndSttCellTrackFinder::GetFirstRiemannTrack ( int i )

inline

Definition at line 89 of file PndSttCellTrackFinder.h.

References fFirstRiemannTrack, and i.

Referenced by PndSttCellTrackFinderTask::Exec().

                                                     {
                 return fFirstRiemannTrack[i];
         }

PndTrackCand PndSttCellTrackFinder::GetFirstTrackCand ( int i )

inline

Definition at line 83 of file PndSttCellTrackFinder.h.

References fFirstTrackCand, and i.

Referenced by PndSttCellTrackFinderTask::Exec().

                                               {
                 return fFirstTrackCand[i];
         }

int PndSttCellTrackFinder::GetNumPrimaryTracklets ( )

inline

Definition at line 78 of file PndSttCellTrackFinder.h.

References fTrackletGenerator, and PndSttCellTrackletGenerator::GetNumPrimaryTracklets().

Referenced by PndSttCellTrackFinderTask::Exec().

                                     {
                 return fTrackletGenerator->GetNumPrimaryTracklets();
         }

std::vector<std::vector<Double_t> > PndSttCellTrackFinder::GetTimeStampsGenerateNeighborhoodData ( )

inline

Definition at line 116 of file PndSttCellTrackFinder.h.

References fTimeStampsGenerateNeighborhoodData.

Referenced by PndSttCellTrackFinderTask::FinishTask().

                                                                                 {
                 return fTimeStampsGenerateNeighborhoodData;
         }

std::vector<std::vector<Double_t> > PndSttCellTrackFinder::GetTimeStampsTrackletGen ( )

inline

Definition at line 111 of file PndSttCellTrackFinder.h.

References fTimeStampsTrackletGen.

Referenced by PndSttCellTrackFinderTask::FinishTask().

                                                                    {
                 return fTimeStampsTrackletGen;
         }

PndSttCellTrackFinderData* PndSttCellTrackFinder::GetTrackFinderDataObject ( )

inline

Definition at line 70 of file PndSttCellTrackFinder.h.

References fTrackFinderData.

Referenced by PndSttCellTrackFinderTask::Init().

                                                               {
                 return fTrackFinderData;
         }

int PndSttCellTrackFinder::NumCombinedRiemannTracks ( )

inline

Definition at line 136 of file PndSttCellTrackFinder.h.

References fCombiRiemannTrack.

Referenced by PndSttCellTrackFinderTask::Exec().

                                        {
                 return fCombiRiemannTrack.size();
         }

int PndSttCellTrackFinder::NumCombinedTracks ( )

inline

Definition at line 131 of file PndSttCellTrackFinder.h.

References fCombiTrackCand.

Referenced by PndSttCellTrackFinderTask::Exec().

                                 {
                 return fCombiTrackCand.size();
         }

int PndSttCellTrackFinder::NumFirstRiemannTracks ( )

inline

Definition at line 126 of file PndSttCellTrackFinder.h.

References fFirstRiemannTrack.

Referenced by PndSttCellTrackFinderTask::Exec().

                                     {
                 return fFirstRiemannTrack.size();
         }

int PndSttCellTrackFinder::NumFirstTrackCands ( )

inline

Definition at line 121 of file PndSttCellTrackFinder.h.

References fFirstTrackCand.

Referenced by PndSttCellTrackFinderTask::Exec().

                                  {
                 return fFirstTrackCand.size();
         }

int PndSttCellTrackFinder::NumHits ( )

inline

Definition at line 141 of file PndSttCellTrackFinder.h.

References fTrackFinderData, and PndSttCellTrackFinderData::GetNumHits().

Referenced by PndSttCellTrackFinderTask::FinishEvent().

                       {
                 return fTrackFinderData->GetNumHits();
         }

int PndSttCellTrackFinder::NumHitsWithoutDouble ( )

inline

Definition at line 145 of file PndSttCellTrackFinder.h.

References fTrackFinderData, and PndSttCellTrackFinderData::GetNumHitsWithoutDouble().

Referenced by PndSttCellTrackFinderTask::FinishEvent().

                                    {
 
                 return fTrackFinderData->GetNumHitsWithoutDouble();
         }

int PndSttCellTrackFinder::NumUnambiguousNeighbors ( )

inline

Definition at line 150 of file PndSttCellTrackFinder.h.

References fTrackFinderData, and PndSttCellTrackFinderData::GetSeparations().

Referenced by PndSttCellTrackFinderTask::FinishEvent().

                                       {
                 return fTrackFinderData->GetSeparations()[1].size()
                                 + fTrackFinderData->GetSeparations()[2].size();
         }

void PndSttCellTrackFinder::Reset ( )

inline

Definition at line 183 of file PndSttCellTrackFinder.h.

References PndSttCellTrackFinderData::clear(), fCombiRiemannTrack, fCombiTrack, fCombiTrackCand, fFirstRiemannTrack, fFirstTrackCand, fHitCorrector, fHits, fTrackFinderData, and fTrackletGenerator.

Referenced by PndSttCellTrackFinderTask::Exec().

                      {
                 fHits.clear();
                 fTrackFinderData->clear();
 
                 fFirstTrackCand.clear();
                 fFirstRiemannTrack.clear();
 
                 fCombiTrackCand.clear();
                 fCombiTrack.clear();
                 fCombiRiemannTrack.clear();
 
                 delete fHitCorrector;
                 delete fTrackletGenerator;
 
         }

void PndSttCellTrackFinder::SetBz ( Double_t val )

inline

Definition at line 165 of file PndSttCellTrackFinder.h.

References fBz, and val.

Referenced by PndSttCellTrackFinderTask::Init().

                                  {
                 fBz = val;
         }

void PndSttCellTrackFinder::SetCalcFirstTrackletInf ( Bool_t val )

inline

Definition at line 155 of file PndSttCellTrackFinder.h.

References fCalcFirstTrackletInf, and val.

Referenced by PndSttCellTrackFinderTask::Init().

                                                  {
                 fCalcFirstTrackletInf = val;
         }

void PndSttCellTrackFinder::SetCalcWithCorrectedIsochrones ( Bool_t val )

inline

Definition at line 66 of file PndSttCellTrackFinder.h.

References fCalcWithCorrectedIsochrones, and val.

Referenced by PndSttCellTrackFinderTask::Init().

                                                         {
                 fCalcWithCorrectedIsochrones = val;
         }

void PndSttCellTrackFinder::SetClusterTime ( double val )

inline

Definition at line 59 of file PndSttCellTrackFinder.h.

References fClusterTime, and val.

Referenced by PndSttCellTrackFinderTask::Init().

                                        {
                 fClusterTime = val;
         }; // J.R. 20/04-2018

void PndSttCellTrackFinder::SetDevTubeNeighboringsPointer ( int * dev_pointer )

inline

Definition at line 62 of file PndSttCellTrackFinder.h.

References fDev_tubeNeighborings.

Referenced by PndSttCellTrackFinderTask::Init().

                                                              {
                 fDev_tubeNeighborings = dev_pointer;
         }

void PndSttCellTrackFinder::SetRunTimeBased ( Bool_t val )

inline

Definition at line 56 of file PndSttCellTrackFinder.h.

References fRunTimeBased, and val.

Referenced by PndSttCellTrackFinderTask::Init().

                                          {
                 fRunTimeBased = val;
         }

void PndSttCellTrackFinder::SetUseGPU ( Bool_t val )

inline

Definition at line 53 of file PndSttCellTrackFinder.h.

References fUseGPU, and val.

Referenced by PndSttCellTrackFinderTask::Init().

                                    {
                 fUseGPU = val;
         }

void PndSttCellTrackFinder::SetVerbose ( Int_t val )

inline

Definition at line 160 of file PndSttCellTrackFinder.h.

References fVerbose, and val.

Referenced by PndSttCellTrackFinderTask::Init().

                                    {
                 fVerbose = val;
         }

void PndSttCellTrackFinder::StoreTrackData ( )

inline

Definition at line 170 of file PndSttCellTrackFinder.h.

References fCalcFirstTrackletInf, fCombiRiemannTrack, fCombiTrack, fCombiTrackCand, fFirstRiemannTrack, fFirstTrackCand, fTrackletGenerator, PndSttCellTrackletGenerator::GetCombiRiemannTracks(), PndSttCellTrackletGenerator::GetCombiTrackCands(), PndSttCellTrackletGenerator::GetCombiTracks(), PndSttCellTrackletGenerator::GetFirstRiemannTracks(), and PndSttCellTrackletGenerator::GetFirstTrackCands().

                               {
                 fFirstTrackCand = fTrackletGenerator->GetFirstTrackCands();
 
                 if (fCalcFirstTrackletInf)
                         fFirstRiemannTrack = fTrackletGenerator->GetFirstRiemannTracks();
 
                 fCombiTrackCand = fTrackletGenerator->GetCombiTrackCands();
                 fCombiRiemannTrack = fTrackletGenerator->GetCombiRiemannTracks();
                 fCombiTrack = fTrackletGenerator->GetCombiTracks();
 
         }