Project
Loading...
Searching...
No Matches
Tracker.h
Go to the documentation of this file.
1// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
2// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
3// All rights not expressly granted are reserved.
4//
5// This software is distributed under the terms of the GNU General Public
6// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
7//
8// In applying this license CERN does not waive the privileges and immunities
9// granted to it by virtue of its status as an Intergovernmental Organization
10// or submit itself to any jurisdiction.
15
16#ifndef TRACKINGITSU_INCLUDE_TRACKER_H_
17#define TRACKINGITSU_INCLUDE_TRACKER_H_
18
19#include <array>
20#include <chrono>
21#include <cmath>
22#include <fstream>
23#include <iomanip>
24#include <iosfwd>
25#include <memory>
26#include <string_view>
27#include <utility>
28#include <sstream>
29
33#include "ITStracking/ROframe.h"
37#include "ITStracking/Road.h"
39
42
43namespace o2
44{
45
46namespace gpu
47{
48class GPUChainITS;
49}
50namespace its
51{
52
54{
55 static constexpr int NLayers{7};
58 using LogFunc = std::function<void(const std::string& s)>;
59
60 public:
62
64
66 LogFunc = [](const std::string& s) { std::cout << s << '\n'; },
67 LogFunc = [](const std::string& s) { std::cerr << s << '\n'; });
68
69 void setParameters(const std::vector<TrackingParameters>& p) { mTrkParams = p; }
70 void setMemoryPool(std::shared_ptr<BoundedMemoryResource>& pool) { mMemoryPool = pool; }
71 std::vector<TrackingParameters>& getParameters() { return mTrkParams; }
73 void setBz(float bz) { mTraits->setBz(bz); }
75 bool isMatLUT() const { return mTraits->isMatLUT(); }
76 void setNThreads(int n) { mTraits->setNThreads(n); }
77 int getNThreads() const { return mTraits->getNThreads(); }
78 void printSummary() const;
79
80 private:
81 void initialiseTimeFrame(int iteration) { mTraits->initialiseTimeFrame(iteration); }
82 void computeTracklets(int iteration, int iROFslice, int iVertex) { mTraits->computeLayerTracklets(iteration, iROFslice, iVertex); }
83 void computeCells(int iteration) { mTraits->computeLayerCells(iteration); }
84 void findCellsNeighbours(int iteration) { mTraits->findCellsNeighbours(iteration); }
85 void findRoads(int iteration) { mTraits->findRoads(iteration); }
86 void findShortPrimaries() { mTraits->findShortPrimaries(); }
87 void extendTracks(int iteration) { mTraits->extendTracks(iteration); }
88
89 // MC interaction
90 void computeRoadsMClabels();
91 void computeTracksMClabels();
92 void rectifyClusterIndices();
93
94 template <typename... T, typename... F>
95 float evaluateTask(void (Tracker::*task)(T...), std::string_view taskName, int iteration, LogFunc logger, F&&... args);
96
97 TrackerTraits7* mTraits = nullptr;
98 TimeFrame7* mTimeFrame = nullptr;
99
100 std::vector<TrackingParameters> mTrkParams;
101 o2::gpu::GPUChainITS* mRecoChain = nullptr;
102
103 unsigned int mNumberOfDroppedTFs{0};
104 unsigned int mTimeFrameCounter{0};
105 double mTotalTime{0};
106 std::shared_ptr<BoundedMemoryResource> mMemoryPool;
107
108 enum State {
109 TFInit = 0,
110 Trackleting,
111 Celling,
112 Neighbouring,
113 Roading,
114 NStates,
115 };
116 State mCurState;
117 static constexpr std::array<const char*, NStates> StateNames{"TimeFrame initialisation", "Tracklet finding", "Cell finding", "Neighbour finding", "Road finding"};
118};
119
120template <typename... T, typename... F>
121float Tracker::evaluateTask(void (Tracker::*task)(T...), std::string_view taskName, int iteration, LogFunc logger, F&&... args)
122{
123 float diff{0.f};
124
125 if constexpr (constants::DoTimeBenchmarks) {
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();
129
130 std::chrono::duration<double, std::milli> diff_t{end - start};
131 diff = diff_t.count();
132
133 std::stringstream sstream;
134 if (taskName.empty()) {
135 sstream << diff << "\t";
136 } else {
137 sstream << std::setw(2) << " - " << taskName << " completed in: " << diff << " ms";
138 }
139 logger(sstream.str());
140
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';
148 }
149 }
150
151 } else {
152 (this->*task)(std::forward<F>(args)...);
153 }
154
155 return diff;
156}
157
158} // namespace its
159} // namespace o2
160
161#endif /* TRACKINGITSU_INCLUDE_TRACKER_H_ */
useful math constants
uint32_t c
Definition RawData.h:2
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)
Definition Tracker.h:69
void setCorrType(const o2::base::PropagatorImpl< float >::MatCorrType type)
Definition Tracker.h:74
void printSummary() const
Definition Tracker.cxx:409
bool isMatLUT() const
Definition Tracker.h:75
std::vector< TrackingParameters > & getParameters()
Definition Tracker.h:71
void getGlobalConfiguration()
Definition Tracker.cxx:336
void setBz(float bz)
Definition Tracker.h:73
void setMemoryPool(std::shared_ptr< BoundedMemoryResource > &pool)
Definition Tracker.h:70
int getNThreads() const
Definition Tracker.h:77
void adoptTimeFrame(TimeFrame< NLayers > &tf)
Definition Tracker.cxx:403
void clustersToTracks(LogFunc=[](const std::string &s) { std::cout<< s<< '\n';}, LogFunc=[](const std::string &s) { std::cerr<< s<< '\n';})
Definition Tracker.cxx:44
void setNThreads(int n)
Definition Tracker.h:76
GLdouble n
Definition glcorearb.h:1982
GLuint GLuint end
Definition glcorearb.h:469
GLint GLint GLsizei GLint GLenum GLenum type
Definition glcorearb.h:275
typedef void(APIENTRYP PFNGLCULLFACEPROC)(GLenum mode)
GLuint start
Definition glcorearb.h:469
constexpr bool DoTimeBenchmarks
Definition Constants.h:40
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