16#ifndef TRACKINGITSU_INCLUDE_TRACKER_H_
17#define TRACKINGITSU_INCLUDE_TRACKER_H_
55 static constexpr int NLayers{7};
58 using LogFunc = std::function<
void(
const std::string& s)>;
66 LogFunc = [](
const std::string& s) { std::cout << s <<
'\n'; },
67 LogFunc = [](
const std::string& s) { std::cerr << s <<
'\n'; });
69 void setParameters(
const std::vector<TrackingParameters>& p) { mTrkParams = p; }
70 void setMemoryPool(std::shared_ptr<BoundedMemoryResource>& pool) { mMemoryPool = pool; }
82 void computeTracklets(
int iteration,
int iROFslice,
int iVertex) { mTraits->
computeLayerTracklets(iteration, iROFslice, iVertex); }
85 void findRoads(
int iteration) { mTraits->
findRoads(iteration); }
87 void extendTracks(
int iteration) { mTraits->
extendTracks(iteration); }
90 void computeRoadsMClabels();
91 void computeTracksMClabels();
92 void rectifyClusterIndices();
94 template <
typename...
T,
typename... F>
95 float evaluateTask(
void (Tracker::*task)(
T...),
std::string_view taskName,
int iteration, LogFunc logger, F&&... args);
97 TrackerTraits7* mTraits =
nullptr;
98 TimeFrame7* mTimeFrame =
nullptr;
100 std::vector<TrackingParameters> mTrkParams;
103 unsigned int mNumberOfDroppedTFs{0};
104 unsigned int mTimeFrameCounter{0};
105 double mTotalTime{0};
106 std::shared_ptr<BoundedMemoryResource> mMemoryPool;
117 static constexpr std::array<const char*, NStates> StateNames{
"TimeFrame initialisation",
"Tracklet finding",
"Cell finding",
"Neighbour finding",
"Road finding"};
120template <
typename...
T,
typename... F>
121float Tracker::evaluateTask(
void (Tracker::*task)(
T...),
std::string_view taskName,
int iteration, LogFunc logger, F&&... args)
126 auto start = std::chrono::high_resolution_clock::now();
127 (this->*task)(std::forward<F>(args)...);
128 auto end = std::chrono::high_resolution_clock::now();
130 std::chrono::duration<double, std::milli> diff_t{
end -
start};
131 diff = diff_t.count();
133 std::stringstream sstream;
134 if (taskName.empty()) {
135 sstream << diff <<
"\t";
137 sstream << std::setw(2) <<
" - " << taskName <<
" completed in: " << diff <<
" ms";
139 logger(sstream.str());
141 if (mTrkParams[0].SaveTimeBenchmarks) {
142 std::string taskNameStr(taskName);
143 std::transform(taskNameStr.begin(), taskNameStr.end(), taskNameStr.begin(),
144 [](
unsigned char c) { return std::tolower(c); });
145 std::replace(taskNameStr.begin(), taskNameStr.end(),
' ',
'_');
146 if (std::ofstream
file{
"its_time_benchmarks.txt", std::ios::app}) {
147 file <<
"trk:" << iteration <<
'\t' << taskNameStr <<
'\t' << diff <<
'\n';
152 (this->*task)(std::forward<F>(args)...);
Definition of the ITS track.
virtual void initialiseTimeFrame(const int iteration)
virtual void findRoads(const int iteration)
virtual void computeLayerTracklets(const int iteration, int iROFslice, int iVertex)
void setCorrType(const o2::base::PropagatorImpl< float >::MatCorrType type)
virtual void computeLayerCells(const int iteration)
virtual void findShortPrimaries()
virtual void setBz(float bz)
virtual void findCellsNeighbours(const int iteration)
virtual void extendTracks(const int iteration)
void setParameters(const std::vector< TrackingParameters > &p)
void setCorrType(const o2::base::PropagatorImpl< float >::MatCorrType type)
void printSummary() const
std::vector< TrackingParameters > & getParameters()
void getGlobalConfiguration()
void setMemoryPool(std::shared_ptr< BoundedMemoryResource > &pool)
void adoptTimeFrame(TimeFrame< NLayers > &tf)
void clustersToTracks(LogFunc=[](const std::string &s) { std::cout<< s<< '\n';}, LogFunc=[](const std::string &s) { std::cerr<< s<< '\n';})
GLint GLint GLsizei GLint GLenum GLenum type
typedef void(APIENTRYP PFNGLCULLFACEPROC)(GLenum mode)
constexpr bool DoTimeBenchmarks
a couple of static helper functions to create timestamp values for CCDB queries or override obsolete ...
Defining DataPointCompositeObject explicitly as copiable.
std::unique_ptr< GPUReconstructionTimeframe > tf