Project
Loading...
Searching...
No Matches
TrackerSpec.cxx
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.
11
12#include <vector>
13
18
19namespace o2
20{
21using namespace framework;
22namespace its
23{
25
26TrackerDPL::TrackerDPL(std::shared_ptr<o2::base::GRPGeomRequest> gr,
27 bool isMC,
28 int trgType,
29 const TrackingMode& trMode,
30 const bool overrBeamEst,
31 o2::gpu::GPUDataTypes::DeviceType dType) : mGGCCDBRequest(gr),
32 mRecChain{o2::gpu::GPUReconstruction::CreateInstance(dType, true)},
33 mITSTrackingInterface{isMC, trgType, overrBeamEst}
34{
35 mITSTrackingInterface.setTrackingMode(trMode);
36}
37
39{
40 mTimer.Stop();
41 mTimer.Reset();
43 mChainITS.reset(mRecChain->AddChain<o2::gpu::GPUChainITS>());
44 mITSTrackingInterface.setTraitsFromProvider(mChainITS->GetITSVertexerTraits(),
45 mChainITS->GetITSTrackerTraits(),
46 mChainITS->GetITSTimeframe());
47}
48
50{
51 end();
52}
53
55{
56 auto cput = mTimer.CpuTime();
57 auto realt = mTimer.RealTime();
58 mTimer.Start(false);
59 mITSTrackingInterface.updateTimeDependentParams(pc);
60 mITSTrackingInterface.run(pc);
61 mTimer.Stop();
62 LOGP(info, "CPU Reconstruction time for this TF {} s (cpu), {} s (wall)", mTimer.CpuTime() - cput, mTimer.RealTime() - realt);
63}
64
66{
67 mITSTrackingInterface.finaliseCCDB(matcher, obj);
68}
69
74
75void TrackerDPL::end()
76{
77 mITSTrackingInterface.end();
78 mITSTrackingInterface.printSummary();
79 LOGF(info, "ITS CA-Tracker total timing: Cpu: %.3e Real: %.3e s in %d slots", mTimer.CpuTime(), mTimer.RealTime(), mTimer.Counter() - 1);
80}
81
82DataProcessorSpec getTrackerSpec(bool useMC, bool useGeom, int trgType, const std::string& trModeS, const bool overrBeamEst, o2::gpu::GPUDataTypes::DeviceType dType)
83{
84 std::vector<InputSpec> inputs;
85
86 inputs.emplace_back("compClusters", "ITS", "COMPCLUSTERS", 0, Lifetime::Timeframe);
87 inputs.emplace_back("patterns", "ITS", "PATTERNS", 0, Lifetime::Timeframe);
88 inputs.emplace_back("ROframes", "ITS", "CLUSTERSROF", 0, Lifetime::Timeframe);
89 if (trgType == 1) {
90 inputs.emplace_back("phystrig", "ITS", "PHYSTRIG", 0, Lifetime::Timeframe);
91 } else if (trgType == 2) {
92 inputs.emplace_back("phystrig", "TRD", "TRKTRGRD", 0, Lifetime::Timeframe);
93 }
94 inputs.emplace_back("itscldict", "ITS", "CLUSDICT", 0, Lifetime::Condition, ccdbParamSpec("ITS/Calib/ClusterDictionary"));
95 inputs.emplace_back("itsalppar", "ITS", "ALPIDEPARAM", 0, Lifetime::Condition, ccdbParamSpec("ITS/Config/AlpideParam"));
96 auto ggRequest = std::make_shared<o2::base::GRPGeomRequest>(false, // orbitResetTime
97 true, // GRPECS=true
98 false, // GRPLHCIF
99 true, // GRPMagField
100 true, // askMatLUT
102 inputs,
103 true);
104 if (!useGeom) {
105 ggRequest->addInput({"itsTGeo", "ITS", "GEOMTGEO", 0, Lifetime::Condition, framework::ccdbParamSpec("ITS/Config/Geometry")}, inputs);
106 }
107 if (overrBeamEst) {
108 inputs.emplace_back("meanvtx", "GLO", "MEANVERTEX", 0, Lifetime::Condition, ccdbParamSpec("GLO/Calib/MeanVertex", {}, 1));
109 }
110
111 std::vector<OutputSpec> outputs;
112 outputs.emplace_back("ITS", "TRACKS", 0, Lifetime::Timeframe);
113 outputs.emplace_back("ITS", "TRACKCLSID", 0, Lifetime::Timeframe);
114 outputs.emplace_back("ITS", "ITSTrackROF", 0, Lifetime::Timeframe);
115 outputs.emplace_back("ITS", "VERTICES", 0, Lifetime::Timeframe);
116 outputs.emplace_back("ITS", "VERTICESROF", 0, Lifetime::Timeframe);
117 outputs.emplace_back("ITS", "IRFRAMES", 0, Lifetime::Timeframe);
118
119 if (useMC) {
120 inputs.emplace_back("itsmclabels", "ITS", "CLUSTERSMCTR", 0, Lifetime::Timeframe);
121 inputs.emplace_back("ITSMC2ROframes", "ITS", "CLUSTERSMC2ROF", 0, Lifetime::Timeframe);
122 outputs.emplace_back("ITS", "VERTICESMCTR", 0, Lifetime::Timeframe);
123 outputs.emplace_back("ITS", "VERTICESMCPUR", 0, Lifetime::Timeframe);
124 outputs.emplace_back("ITS", "TRACKSMCTR", 0, Lifetime::Timeframe);
125 outputs.emplace_back("ITS", "ITSTrackMC2ROF", 0, Lifetime::Timeframe);
126 }
127
128 return DataProcessorSpec{
129 "its-tracker",
130 inputs,
131 outputs,
132 AlgorithmSpec{adaptFromTask<TrackerDPL>(ggRequest,
133 useMC,
134 trgType,
135 trModeS == "sync" ? o2::its::TrackingMode::Sync : trModeS == "async" ? o2::its::TrackingMode::Async
137 overrBeamEst,
138 dType)},
139 Options{}};
140}
141
142} // namespace its
143} // namespace o2
static GRPGeomHelper & instance()
void setRequest(std::shared_ptr< GRPGeomRequest > req)
void run(framework::ProcessingContext &pc)
void setTrackingMode(TrackingMode mode=TrackingMode::Unset)
virtual void updateTimeDependentParams(framework::ProcessingContext &pc)
virtual void finaliseCCDB(framework::ConcreteDataMatcher &matcher, void *obj)
void setTraitsFromProvider(VertexerTraits *, TrackerTraits7 *, TimeFrame7 *)
void stop() final
This is invoked on stop.
TrackerDPL(std::shared_ptr< o2::base::GRPGeomRequest > gr, bool isMC, int trgType, const TrackingMode &trMode=TrackingMode::Unset, const bool overrBeamEst=false, o2::gpu::GPUDataTypes::DeviceType dType=o2::gpu::GPUDataTypes::DeviceType::CPU)
void init(framework::InitContext &ic) final
void finaliseCCDB(framework::ConcreteDataMatcher &matcher, void *obj) final
void endOfStream(framework::EndOfStreamContext &ec) final
This is invoked whenever we have an EndOfStream event.
void run(framework::ProcessingContext &pc) final
GLuint GLuint end
Definition glcorearb.h:469
std::vector< ConfigParamSpec > ccdbParamSpec(std::string const &path, int runDependent, std::vector< CCDBMetadata > metadata={}, int qrate=0)
std::vector< ConfigParamSpec > Options
o2::dataformats::Vertex< o2::dataformats::TimeStamp< int > > Vertex
Definition ROframe.h:37
framework::DataProcessorSpec getTrackerSpec(bool useMC, bool useGeom, int useTrig, const std::string &trMode, const bool overrBeamEst=false, o2::gpu::GPUDataTypes::DeviceType dType=o2::gpu::GPUDataTypes::DeviceType::CPU)
a couple of static helper functions to create timestamp values for CCDB queries or override obsolete ...