Skip to content

Commit b2ba951

Browse files
committed
[PWGHF] Add UPC w/o ML process functions to validation checks
- taskD0: Add doprocessDataWithDCAFitterNWithUpc to both doprocess array and centrality/occupancy check - taskDplus: Add doprocessDataWithUpc to doprocess array validation This ensures both UPC process functions (with and without ML) are properly validated in the init() checks. Addresses review comments from @zhangbiao-phy on PR #13603
1 parent 210abd3 commit b2ba951

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

PWGHF/D2H/Tasks/taskD0.cxx

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -237,14 +237,14 @@ struct HfTaskD0 {
237237

238238
void init(InitContext&)
239239
{
240-
std::array<bool, 13> doprocess{doprocessDataWithDCAFitterN, doprocessDataWithDCAFitterNCent, doprocessDataWithKFParticle, doprocessMcWithDCAFitterN, doprocessMcWithDCAFitterNCent, doprocessMcWithKFParticle, doprocessDataWithDCAFitterNMl, doprocessDataWithDCAFitterNMlCent, doprocessDataWithKFParticleMl, doprocessMcWithDCAFitterNMl, doprocessMcWithDCAFitterNMlCent, doprocessMcWithKFParticleMl, doprocessDataWithDCAFitterNMlWithUpc};
240+
std::array<bool, 14> doprocess{doprocessDataWithDCAFitterN, doprocessDataWithDCAFitterNCent, doprocessDataWithKFParticle, doprocessMcWithDCAFitterN, doprocessMcWithDCAFitterNCent, doprocessMcWithKFParticle, doprocessDataWithDCAFitterNMl, doprocessDataWithDCAFitterNMlCent, doprocessDataWithKFParticleMl, doprocessMcWithDCAFitterNMl, doprocessMcWithDCAFitterNMlCent, doprocessMcWithKFParticleMl, doprocessDataWithDCAFitterNWithUpc, doprocessDataWithDCAFitterNMlWithUpc};
241241
if ((std::accumulate(doprocess.begin(), doprocess.end(), 0)) == 0) {
242242
LOGP(fatal, "At least one process function should be enabled at a time.");
243243
}
244244
if ((doprocessDataWithDCAFitterN || doprocessDataWithDCAFitterNCent || doprocessMcWithDCAFitterN || doprocessMcWithDCAFitterNCent || doprocessDataWithDCAFitterNMl || doprocessDataWithDCAFitterNMlCent || doprocessMcWithDCAFitterNMl || doprocessMcWithDCAFitterNMlCent) && (doprocessDataWithKFParticle || doprocessMcWithKFParticle || doprocessDataWithKFParticleMl || doprocessMcWithKFParticleMl)) {
245245
LOGP(fatal, "DCAFitterN and KFParticle can not be enabled at a time.");
246246
}
247-
if ((storeCentrality || storeOccupancyAndIR) && !(doprocessDataWithDCAFitterNCent || doprocessMcWithDCAFitterNCent || doprocessDataWithDCAFitterNMlCent || doprocessMcWithDCAFitterNMlCent || doprocessDataWithDCAFitterNMlWithUpc)) {
247+
if ((storeCentrality || storeOccupancyAndIR) && !(doprocessDataWithDCAFitterNCent || doprocessMcWithDCAFitterNCent || doprocessDataWithDCAFitterNMlCent || doprocessMcWithDCAFitterNMlCent || doprocessDataWithDCAFitterNWithUpc || doprocessDataWithDCAFitterNMlWithUpc)) {
248248
LOGP(fatal, "Can't enable the storeCentrality and storeOccupancu without cent process or UPC process");
249249
}
250250
auto vbins = (std::vector<double>)binsPt;

PWGHF/D2H/Tasks/taskDplus.cxx

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -149,7 +149,7 @@ struct HfTaskDplus {
149149

150150
void init(InitContext&)
151151
{
152-
std::array<bool, 5> doprocess{doprocessData, doprocessDataWithMl, doprocessMc, doprocessMcWithMl, doprocessDataWithMlWithUpc};
152+
std::array<bool, 6> doprocess{doprocessData, doprocessDataWithMl, doprocessMc, doprocessMcWithMl, doprocessDataWithUpc, doprocessDataWithMlWithUpc};
153153
if ((std::accumulate(doprocess.begin(), doprocess.end(), 0)) != 1) {
154154
LOGP(fatal, "Only one process function should be enabled! Please check your configuration!");
155155
}

0 commit comments

Comments
 (0)