26#include "TStopwatch.h"
66namespace globaltracking
73 bool useFT0,
bool calib,
bool skipTPCOnly,
bool useMC)
74 : mDataRequest(dr), mGGCCDBRequest(gr), mUseFT0(useFT0), mCalibMode(calib), mSkipTPCOnly(skipTPCOnly), mUseMC(useMC)
88 std::shared_ptr<DataRequest> mDataRequest;
89 std::shared_ptr<o2::base::GRPGeomRequest> mGGCCDBRequest;
94 bool mCalibMode =
false;
95 bool mSkipTPCOnly =
false;
106 mMatching.
setDebugTreeFileName(maxLanes == 1 ?
"dbg_TPCITSmatch.root" : fmt::format(
"dbg_TPCITSmatch_{}.root", lane));
111 mTPCCorrMapsLoader.
init(ic);
119 updateTimeDependentParams(pc);
124 auto& matchedTracks = pc.
outputs().
make<std::vector<o2::dataformats::TrackTPCITS>>(
Output{
"GLO",
"TPCITS", 0});
125 auto& ABTrackletRefs = pc.
outputs().
make<std::vector<o2::itsmft::TrkClusRef>>(
Output{
"GLO",
"TPCITSAB_REFS", 0});
126 auto& ABTrackletClusterIDs = pc.
outputs().
make<std::vector<int>>(
Output{
"GLO",
"TPCITSAB_CLID", 0});
127 auto& matchLabels = mUseMC ? pc.
outputs().
make<std::vector<o2::MCCompLabel>>(
Output{
"GLO",
"TPCITS_MC", 0}) : dummyMCLab;
128 auto& ABTrackletLabels = mUseMC ? pc.
outputs().
make<std::vector<o2::MCCompLabel>>(
Output{
"GLO",
"TPCITSAB_MC", 0}) : dummyMCLabAB;
129 auto& calib = mCalibMode ? pc.
outputs().
make<std::vector<o2::dataformats::Triplet<float, float, float>>>(
Output{
"GLO",
"TPCITS_VDTGL", 0}) : dummyCalib;
131 mMatching.
run(recoData, matchedTracks, ABTrackletRefs, ABTrackletClusterIDs, matchLabels, ABTrackletLabels, calib);
139 LOGF(info,
"TPC-ITS matching total timing: Cpu: %.3e Real: %.3e s in %d slots",
140 mTimer.CpuTime(), mTimer.RealTime(), mTimer.Counter() - 1);
156 LOG(info) <<
"ITS-TPC Matching params updated from ccdb";
160 LOG(info) <<
"its cluster dictionary updated";
165 LOG(info) <<
"ITS Alpide param updated";
167 par.printKeyValues();
171 LOG(info) <<
"ITS GeometryTGeo loaded from ccdb";
175#ifdef ENABLE_UPGRADES
177 LOG(info) <<
"it3 cluster dictionary updated";
189 static bool initOnceDone =
false;
219 LOGP(fatal,
"USEMatCorrTGeo cannot work w/o full geometry request in the GRPGeomHelper");
223 bool updateMaps =
false;
231 LOGP(info,
"Updating TPC fast transform map with new VDrift factor of {} wrt reference {} and DriftTimeOffset correction {} wrt {} from source {}",
246 std::vector<OutputSpec> outputs;
247 auto dataRequest = std::make_shared<DataRequest>();
249 dataRequest->setMatchingInputStrict();
251 dataRequest->inputs.emplace_back(
"MatchParam",
"GLO",
"ITSTPCPARAM", 0, Lifetime::Condition,
ccdbParamSpec(
"GLO/Config/ITSTPCParam"));
253 dataRequest->requestTracks(
src, useMC);
254 dataRequest->requestTPCClusters(
false);
256#ifdef ENABLE_UPGRADES
258 dataRequest->requestIT3Clusters(useMC);
260 dataRequest->requestITSClusters(useMC);
263 dataRequest->requestITSClusters(useMC);
266 dataRequest->requestFT0RecPoints(
false);
268 outputs.emplace_back(
"GLO",
"TPCITS", 0, Lifetime::Timeframe);
269 outputs.emplace_back(
"GLO",
"TPCITSAB_REFS", 0, Lifetime::Timeframe);
270 outputs.emplace_back(
"GLO",
"TPCITSAB_CLID", 0, Lifetime::Timeframe);
273 outputs.emplace_back(
"GLO",
"TPCITS_VDTGL", 0, Lifetime::Timeframe);
277 outputs.emplace_back(
"GLO",
"TPCITS_MC", 0, Lifetime::Timeframe);
278 outputs.emplace_back(
"GLO",
"TPCITSAB_MC", 0, Lifetime::Timeframe);
282 auto ggRequest = std::make_shared<o2::base::GRPGeomRequest>(
true,
291 ggRequest->addInput({
"itsTGeo",
"ITS",
"GEOMTGEO", 0, Lifetime::Condition,
ccdbParamSpec(
"ITS/Config/Geometry")}, dataRequest->inputs);
295 {
"nthreads", VariantType::Int, 1, {
"Number of afterburner threads"}},
296 {
"ignore-bc-check", VariantType::Bool,
false, {
"Do not check match candidate against BC filling"}},
297 {
"debug-tree-flags", VariantType::Int, 0, {
"DebugFlagTypes bit-pattern for debug tree"}}};
303 "itstpc-track-matcher",
306 AlgorithmSpec{adaptFromTask<TPCITSMatchingDPL>(dataRequest, ggRequest, sclOpts, useFT0, calib, skipTPCOnly, useMC)},
Class of a TPC cluster in TPC-native coordinates (row, time)
Definition of the ITS/MFT clusterer settings.
Definition of the ITSMFT compact cluster.
Helper class to access load maps from CCDB.
gsl::span< const o2::MCCompLabel > MCLabelsTr
Definition of the Names Generator class.
Definition of the GeometryManager class.
Definition of the BuildTopologyDictionary class for ITS3.
Header of the AggregatedRunInfo struct.
Helper for geometry and GRP related CCDB requests.
Definition of the GeometryTGeo class.
Definition of a container to keep Monte Carlo truth external to simulation objects.
Configurable params for TPC ITS matching.
Class to perform TPC ITS matching.
Definition of the ITS track.
Result of refitting TPC-ITS matched track.
Helper class to extract VDrift from different sources.
Helper class to obtain TPC clusters / digits / labels from DPL.
void checkUpdates(o2::framework::ProcessingContext &pc)
static GRPGeomHelper & instance()
void setRequest(std::shared_ptr< GRPGeomRequest > req)
static bool isGeometryLoaded()
static const DPLAlpideParam< N > & Instance()
T get(const char *key) const
decltype(auto) make(const Output &spec, Args... args)
ServiceRegistryRef services()
ConfigParamRegistry const & options()
DataAllocator & outputs()
The data allocator is used to allocate memory for the output data.
InputRecord & inputs()
The inputs associated with this processing context.
bool isITSTriggered() const
void setDebugTreeFileName(std::string name)
get the name of output debug file
void setUseBCFilling(bool v)
void reportTiming()
clear results of previous event reco
void setITSROFrameLengthMUS(float fums)
set ITS ROFrame duration in BC (continuous mode only)
void setTPCVDrift(const o2::tpc::VDriftCorrFact &v)
void setNHBPerTF(int n)
ITS readout mode.
void setITSTriggered(bool v)
void setTPCCorrMaps(o2::gpu::CorrectionMapsHelper *maph)
print settings
void setSkipTPCOnly(bool v)
void setMCTruthOn(bool v)
request VDrift calibration
void run(const o2::globaltracking::RecoContainer &inp, pmr::vector< o2::dataformats::TrackTPCITS > &matchedTracks, pmr::vector< o2::itsmft::TrkClusRef > &ABTrackletRefs, pmr::vector< int > &ABTrackletClusterIDs, pmr::vector< o2::MCCompLabel > &matchLabels, pmr::vector< o2::MCCompLabel > &ABTrackletLabels, pmr::vector< o2::dataformats::Triplet< float, float, float > > &calib)
void setVDriftCalib(bool v)
void setBunchFilling(const o2::BunchFilling &bf)
void setDebugFlag(UInt_t flag, bool on=true)
set the name of output debug file
void setITSROFrameLengthInBC(int nbc)
void setITSTimeBiasInBC(int n)
void setITSDictionary(const o2::itsmft::TopologyDictionary *d)
set flag to use MC truth
auto getUseMatCorrFlag() const
void setNThreads(int n)
perform all initializations
void endOfStream(framework::EndOfStreamContext &ec) final
This is invoked whenever we have an EndOfStream event.
void finaliseCCDB(framework::ConcreteDataMatcher &matcher, void *obj) final
~TPCITSMatchingDPL() override=default
TPCITSMatchingDPL(std::shared_ptr< DataRequest > dr, std::shared_ptr< o2::base::GRPGeomRequest > gr, const o2::tpc::CorrectionMapsLoaderGloOpts &sclOpts, bool useFT0, bool calib, bool skipTPCOnly, bool useMC)
void init(InitContext &ic) final
void run(ProcessingContext &pc) final
void setLumiScaleType(int32_t v)
void setLumiScaleMode(int32_t v)
void setCheckCTPIDCConsistency(bool v)
static void adopt(GeometryTGeo *raw, bool canDelete=false)
void extractCCDBInputs(o2::framework::ProcessingContext &pc)
void updateVDrift(float vdriftCorr, float vdrifRef, float driftTimeOffset=0)
bool accountCCDBInputs(const o2::framework::ConcreteDataMatcher &matcher, void *obj)
static void requestCCDBInputs(std::vector< o2::framework::InputSpec > &inputs, std::vector< o2::framework::ConfigParamSpec > &options, const CorrectionMapsLoaderGloOpts &gloOpts)
void init(o2::framework::InitContext &ic)
static void requestCCDBInputs(std::vector< o2::framework::InputSpec > &inputs, bool laser=true, bool itstpcTgl=true)
void extractCCDBInputs(o2::framework::ProcessingContext &pc, bool laser=true, bool itstpcTgl=true)
const VDriftCorrFact & getVDriftObject() const
bool accountCCDBInputs(const o2::framework::ConcreteDataMatcher &matcher, void *obj)
static std::string_view getSourceName(Source s)
Defining PrimaryVertex explicitly as messageable.
std::vector< ConfigParamSpec > ccdbParamSpec(std::string const &path, int runDependent, std::vector< CCDBMetadata > metadata={}, int qrate=0)
std::vector< ConfigParamSpec > Options
framework::DataProcessorSpec getTPCITSMatchingSpec(o2::dataformats::GlobalTrackID::mask_t src, bool useFT0, bool calib, bool skipTPCOnly, bool useGeom, bool useMC, const o2::tpc::CorrectionMapsLoaderGloOpts &sclOpts)
create a processor spec
std::vector< T, fair::mq::pmr::polymorphic_allocator< T > > vector
a couple of static helper functions to create timestamp values for CCDB queries or override obsolete ...
void collectData(o2::framework::ProcessingContext &pc, const DataRequest &request)
int lumiType
what estimator to used for corrections scaling: 0: no scaling, 1: CTP, 2: IDC
bool checkCTPIDCconsistency
int lumiMode
what corrections method to use: 0: classical scaling, 1: Using of the derivative map,...
float refTimeOffset
additive time offset reference (\mus)
float refVDrift
reference vdrift for which factor was extracted
float getTimeOffset() const
float timeOffsetCorr
additive time offset correction (\mus)
float corrFact
drift velocity correction factor (multiplicative)
LOG(info)<< "Compressed in "<< sw.CpuTime()<< " s"