Skip to content

Commit 715080f

Browse files
aalkinaalkin
committed
Request slice infos for grouped process arguments
For grouping process functions add the slice-info input for each argument related to the grouping table by an index with the control option of the process function and follow the origin replacement of the sliced table.
1 parent 8b0d35b commit 715080f

2 files changed

Lines changed: 68 additions & 18 deletions

File tree

‎Framework/Core/include/Framework/AnalysisTask.h‎

Lines changed: 48 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -148,6 +148,38 @@ struct AnalysisDataProcessorBuilder {
148148
}.template operator()<A::originals.size(), std::decay_t<A>::originals>(std::make_index_sequence<std::decay_t<A>::originals.size()>());
149149
}
150150

151+
/// helpers to append InputSpect for a single argument with grouping
152+
template <soa::is_iterator G, soa::is_table A>
153+
requires(soa::relatedByIndex<std::decay_t<G>, std::decay_t<A>>())
154+
static void addSlicingInput(const char* name, bool value, std::vector<InputSpec>& inputs, header::DataOrigin&& newOrigin = header::DataOrigin{"AOD"})
155+
{
156+
auto key = std::string{"fIndex"} + o2::framework::cutString(soa::getLabelFromType<std::decay_t<G>>());
157+
Entry entry{soa::getLabelFromTypeForKey<std::decay_t<A>>(key), soa::getMatcherFromTypeForKey<std::decay_t<A>>(key), key, value};
158+
// replace the origin of the sliced table first, so that both the slicer source and the slice info carry the new origin
159+
bool originReplaced = (entry.matcher.origin == header::DataOrigin{"AOD"}) && (newOrigin != header::DataOrigin{"AOD"});
160+
if (originReplaced) {
161+
entry.matcher = replaceOrigin(entry.matcher, newOrigin);
162+
}
163+
auto spec = inputForEntry(entry, !o2::soa::is_smallgroups<std::decay_t<A>>);
164+
spec.metadata.emplace_back(ConfigParamSpec{std::string{"control:"} + name, VariantType::Bool, value, {"\"\""}});
165+
if (originReplaced) {
166+
spec.metadata.emplace_back(ConfigParamSpec{"aod-origin-replaced", VariantType::Bool, true, {"\"\""}});
167+
}
168+
169+
DataSpecUtils::updateInputList(inputs, std::move(spec));
170+
}
171+
172+
template <soa::is_iterator G, soa::is_table A>
173+
requires(!soa::relatedByIndex<std::decay_t<G>, std::decay_t<A>>())
174+
static void addSlicingInput(const char*, bool, std::vector<InputSpec>&, header::DataOrigin&&)
175+
{
176+
}
177+
178+
template <soa::is_table G, soa::is_table A>
179+
static void addSlicingInput(const char*, bool, std::vector<InputSpec>&, header::DataOrigin&&)
180+
{
181+
}
182+
151183
/// helper to append the inputs and expression information for normalized arguments
152184
template <soa::is_table... As>
153185
static void addInputsAndExpressions(uint32_t hash, const char* name, bool value, std::vector<InputSpec>& inputs, std::vector<ExpressionInfo>& eInfos, std::vector<InputInfo>& iInfos, header::DataOrigin&& newOrigin = header::DataOrigin{"AOD"})
@@ -162,6 +194,17 @@ struct AnalysisDataProcessorBuilder {
162194
...);
163195
}
164196

197+
/// helper to append the slicing inputs
198+
template <soa::is_iterator G, soa::is_table... As>
199+
static void addSlicingInputs(const char* name, bool value, std::vector<InputSpec>& inputs, header::DataOrigin&& newOrigin = header::DataOrigin{"AOD"})
200+
{
201+
([&name,&value,&inputs,newOrigin]() mutable {
202+
using T = std::decay_t<As>;
203+
addSlicingInput<G, T>(name, value, inputs, std::move(newOrigin));
204+
}(),
205+
...);
206+
}
207+
165208
/// helper to parse the process arguments
166209
template <typename T>
167210
inline static bool requestInputsFromArgs(T&, std::string const&, std::vector<InputSpec>&, std::vector<ExpressionInfo>&, std::vector<InputInfo>&, header::DataOrigin)
@@ -200,7 +243,11 @@ struct AnalysisDataProcessorBuilder {
200243
requires(std::is_lvalue_reference_v<A> && (std::is_lvalue_reference_v<Args> && ...))
201244
{
202245
constexpr auto hash = o2::framework::TypeIdHelpers::uniqueId<void (C::*)(A, Args...)>();
203-
addInputsAndExpressions<typename std::decay_t<A>::parent_t, Args...>(hash, name, value, inputs, eInfos, iInfos, std::move(newOrigin));
246+
// here we also add slicing inputs with the same constrol config param spec as normal inputs, so it can be removed in adjust topology
247+
// step if the process function is disabled
248+
auto newOriginCopy = newOrigin;
249+
addSlicingInputs<A, Args...>(name, value, inputs, std::move(newOrigin));
250+
addInputsAndExpressions<typename std::decay_t<A>::parent_t, Args...>(hash, name, value, inputs, eInfos, iInfos, std::move(newOriginCopy));
204251
}
205252

206253
/// 3. generic case

‎Framework/Core/test/test_AnalysisTask.cxx‎

Lines changed: 20 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -218,23 +218,26 @@ TEST_CASE("AdaptorCompilation")
218218
REQUIRE(task1ngc.inputs[4].binding == "B3s");
219219

220220
auto task2 = adaptAnalysisTask<BTask>(*cfgc, TaskName{"test2"});
221-
REQUIRE(task2.inputs.size() == 10);
222-
REQUIRE(task2.inputs[2].binding == "TracksExtension");
223-
REQUIRE(task2.inputs[1].binding == "Tracks");
224-
REQUIRE(task2.inputs[4].binding == "TracksExtra_002Extension");
225-
REQUIRE(task2.inputs[3].binding == "TracksExtra");
226-
REQUIRE(task2.inputs[6].binding == "TracksCovExtension");
227-
REQUIRE(task2.inputs[5].binding == "TracksCov");
228-
REQUIRE(task2.inputs[7].binding == "AmbiguousTracks");
229-
REQUIRE(task2.inputs[8].binding == "Calos");
230-
REQUIRE(task2.inputs[9].binding == "CaloTriggers");
231-
REQUIRE(task2.inputs[0].binding == "Collisions_001");
221+
REQUIRE(task2.inputs.size() == 11); // 10 base + 1 slice
222+
REQUIRE(task2.inputs[0].binding == "Tracks_Slice");
223+
REQUIRE(task2.inputs[1].binding == "Collisions_001");
224+
REQUIRE(task2.inputs[2].binding == "Tracks");
225+
REQUIRE(task2.inputs[3].binding == "TracksExtension");
226+
REQUIRE(task2.inputs[4].binding == "TracksExtra");
227+
REQUIRE(task2.inputs[5].binding == "TracksExtra_002Extension");
228+
REQUIRE(task2.inputs[6].binding == "TracksCov");
229+
REQUIRE(task2.inputs[7].binding == "TracksCovExtension");
230+
REQUIRE(task2.inputs[8].binding == "AmbiguousTracks");
231+
REQUIRE(task2.inputs[9].binding == "Calos");
232+
REQUIRE(task2.inputs[10].binding == "CaloTriggers");
233+
232234

233235
auto task3 = adaptAnalysisTask<CTask>(*cfgc, TaskName{"test3"});
234-
REQUIRE(task3.inputs.size() == 3);
235-
REQUIRE(task3.inputs[0].binding == "Collisions_001");
236-
REQUIRE(task3.inputs[1].binding == "Tracks");
237-
REQUIRE(task3.inputs[2].binding == "TracksExtension");
236+
REQUIRE(task3.inputs.size() == 4); // 3 base + 1 slice
237+
REQUIRE(task3.inputs[0].binding == "Tracks_Slice");
238+
REQUIRE(task3.inputs[1].binding == "Collisions_001");
239+
REQUIRE(task3.inputs[2].binding == "Tracks");
240+
REQUIRE(task3.inputs[3].binding == "TracksExtension");
238241

239242
auto task4 = adaptAnalysisTask<DTask>(*cfgc, TaskName{"test4"});
240243
REQUIRE(task4.inputs.size() == 2);
@@ -269,10 +272,10 @@ TEST_CASE("AdaptorCompilation")
269272
REQUIRE(task11.inputs.size() == 1);
270273

271274
auto task12 = adaptAnalysisTask<LTask>(*cfgc, TaskName{"test12"});
272-
REQUIRE(task12.inputs.size() == 3);
275+
REQUIRE(task12.inputs.size() == 4); // 3 base + 1 actual slice
273276

274277
auto task13 = adaptAnalysisTask<MTask>(*cfgc, TaskName{"test13"});
275-
REQUIRE(task13.inputs.size() == 3);
278+
REQUIRE(task13.inputs.size() == 4); // 3 base + 1 actual slice
276279
}
277280

278281
TEST_CASE("TestPartitionIteration")

0 commit comments

Comments
 (0)