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
19
20namespace o2
21{
22using namespace framework;
23namespace its
24{
26
27TrackerDPL::TrackerDPL(std::shared_ptr<o2::base::GRPGeomRequest> gr,
28 bool isMC,
29 int trgType,
30 const TrackingMode::Type trMode,
31 const bool overrBeamEst,
32 o2::gpu::GPUDataTypes::DeviceType dType) : mGGCCDBRequest(gr),
33 mRecChain{o2::gpu::GPUReconstruction::CreateInstance(dType, true)},
34 mITSTrackingInterface{isMC, trgType, overrBeamEst}
35{
36 mITSTrackingInterface.setTrackingMode(trMode);
37}
38
40{
41 mTimer.Stop();
42 mTimer.Reset();
44 mChainITS.reset(mRecChain->AddChain<o2::gpu::GPUChainITS>());
45 mITSTrackingInterface.setTraitsFromProvider(mChainITS->GetITSVertexerTraits(),
46 mChainITS->GetITSTrackerTraits(),
47 mChainITS->GetITSTimeframe());
48}
49
51{
52 end();
53}
54
56{
57 auto cput = mTimer.CpuTime();
58 auto realt = mTimer.RealTime();
59 mTimer.Start(false);
60 mITSTrackingInterface.updateTimeDependentParams(pc);
61 mITSTrackingInterface.run(pc);
62 mTimer.Stop();
63 LOGP(info, "CPU Reconstruction time for this TF {} s (cpu), {} s (wall)", mTimer.CpuTime() - cput, mTimer.RealTime() - realt);
64}
65
67{
68 mITSTrackingInterface.finaliseCCDB(matcher, obj);
69}
70
75
76void TrackerDPL::end()
77{
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, TrackingMode::Type trMode, 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);
127 outputs.emplace_back("ITS", "VERTICESMCTRCONT", 0, Lifetime::Timeframe);
128 }
129 }
130
131 return DataProcessorSpec{
132 "its-tracker",
133 inputs,
134 outputs,
135 AlgorithmSpec{adaptFromTask<TrackerDPL>(ggRequest,
136 useMC,
137 trgType,
138 trMode,
139 overrBeamEst,
140 dType)},
141 Options{}};
142}
143
144} // namespace its
145} // namespace o2
static GRPGeomHelper & instance()
void setRequest(std::shared_ptr< GRPGeomRequest > req)
void run(framework::ProcessingContext &pc)
void setTrackingMode(TrackingMode::Type 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.
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
TrackerDPL(std::shared_ptr< o2::base::GRPGeomRequest > gr, bool isMC, int trgType, const TrackingMode::Type trMode=TrackingMode::Unset, const bool overrBeamEst=false, o2::gpu::GPUDataTypes::DeviceType dType=o2::gpu::GPUDataTypes::DeviceType::CPU)
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 TimeFrame.h:65
framework::DataProcessorSpec getTrackerSpec(bool useMC, bool useGeom, int useTrig, TrackingMode::Type 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 ...