Skip to content

Commit c9cb606

Browse files
committed
Add ability to retain TrackQA for all global tracks
1 parent b18b96a commit c9cb606

File tree

2 files changed

+6
-0
lines changed

2 files changed

+6
-0
lines changed

Detectors/AOD/include/AODProducerWorkflow/AODProducerWorkflowSpec.h

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -237,6 +237,7 @@ class AODProducerWorkflowDPL : public Task
237237
bool mThinTracks{false};
238238
bool mPropTracks{false};
239239
bool mPropMuons{false};
240+
float mTrackQCKeepGlobalTracks{true};
240241
float mTrackQCFraction{0.00};
241242
int64_t mTrackQCNTrCut{4};
242243
float mTrackQCDCAxy{3.};

Detectors/AOD/src/AODProducerWorkflowSpec.cxx

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -501,6 +501,9 @@ void AODProducerWorkflowDPL::fillTrackTablesPerCollision(int collisionID,
501501
static std::uniform_real_distribution<> distr(0., 1.);
502502
bool writeQAData = o2::math_utils::Tsallis::downsampleTsallisCharged(data.getTrackParam(trackIndex).getPt(), mTrackQCFraction, mSqrtS, weight, distr(mGenerator));
503503
auto extraInfoHolder = processBarrelTrack(collisionID, collisionBC, trackIndex, data, bcsMap);
504+
if(!extraInfoHolder.isTPConly && mTrackQCKeepGlobalTracks){
505+
writeQAData = true;
506+
}
504507

505508
if (writeQAData) {
506509
auto trackQAInfoHolder = processBarrelTrackQA(collisionID, collisionBC, trackIndex, data, bcsMap);
@@ -1719,6 +1722,7 @@ void AODProducerWorkflowDPL::init(InitContext& ic)
17191722
LOGP(warn, "Specified non-default empty streamer mask!");
17201723
}
17211724
}
1725+
mTrackQCKeepGlobalTracks = ic.options().get<bool>("trackqc-keepglobaltracks");
17221726
mTrackQCFraction = ic.options().get<float>("trackqc-fraction");
17231727
mTrackQCNTrCut = ic.options().get<int64_t>("trackqc-NTrCut");
17241728
mTrackQCDCAxy = ic.options().get<float>("trackqc-tpc-dca");
@@ -3348,6 +3352,7 @@ DataProcessorSpec getAODProducerWorkflowSpec(GID::mask_t src, bool enableSV, boo
33483352
ConfigParamSpec{"hepmc-update", VariantType::String, "always", {"When to update HepMC Aux tables: always - force update, never - never update, all - if all keys are present, any - when any key is present (not valid yet)"}},
33493353
ConfigParamSpec{"propagate-muons", VariantType::Bool, false, {"Propagate muons to IP"}},
33503354
ConfigParamSpec{"thin-tracks", VariantType::Bool, false, {"Produce thinned track tables"}},
3355+
ConfigParamSpec{"trackqc-keepglobaltracks", VariantType::Bool, true, {"Always keep TrackQA for global tracks"}},
33513356
ConfigParamSpec{"trackqc-fraction", VariantType::Float, float(0.1), {"Fraction of tracks to QC"}},
33523357
ConfigParamSpec{"trackqc-NTrCut", VariantType::Int64, 4L, {"Minimal length of the track - in amount of tracklets"}},
33533358
ConfigParamSpec{"trackqc-tpc-dca", VariantType::Float, 3.f, {"Keep TPC standalone track with this DCAxy to the PV"}},

0 commit comments

Comments
 (0)