add aod-origin-replace option

Author: aalkin

Framework/Core/include/Framework/AnalysisHelpers.h

@@ -69,6 +69,7 @@ struct IndexBuilder {
 
 namespace o2::framework
 {
+void wrongOriginReplacement(std::string_view replacement);
 std::shared_ptr<arrow::Table> makeEmptyTableImpl(const char* name, std::shared_ptr<arrow::Schema>& schema);
 
 template <soa::is_table T>

Framework/Core/include/Framework/AnalysisTask.h

@@ -525,6 +525,18 @@ DataProcessorSpec adaptAnalysisTask(ConfigContext const& ctx, Args&&... args)
   std::vector<ExpressionInfo> expressionInfos;
   std::vector<InputInfo> inputInfos;
 
+  std::string replacementOrigin;
+  header::DataOrigin newOrigin{"AOD"};
+  if (ctx.options().hasOption("aod-origin-replace")) {
+    replacementOrigin = ctx.options().get<std::string>("aod-origin-replace");
+    if (replacementOrigin.size() > 4UL) {
+      wrongOriginReplacement(replacementOrigin);
+    }
+  }
+  if (!replacementOrigin.empty()) {
+    newOrigin.runtimeInit(replacementOrigin.c_str(), std::min(replacementOrigin.size(), 4UL));
+  }
+
   constexpr const int numElements = nested_brace_constructible_size<false, std::decay_t<T>>() / 10;
 
   /// make sure options and configurables are set before expression infos are created

Framework/Core/src/AnalysisHelpers.cxx

@@ -79,6 +79,11 @@ std::shared_ptr<arrow::Table> IndexBuilder::materialize(std::vector<framework::I
 
 namespace o2::framework
 {
+void wrongOriginReplacement(std::string_view replacement)
+{
+  throw framework::runtime_error_f("Provided origin replacement string is longer than 4 symbols: %s", replacement.data());
+}
+
 std::shared_ptr<arrow::Table> makeEmptyTableImpl(const char* name, std::shared_ptr<arrow::Schema>& schema)
 {
   schema = schema->WithMetadata(std::make_shared<arrow::KeyValueMetadata>(std::vector{std::string{"label"}}, std::vector{std::string{name}}));

Framework/Core/src/WorkflowCustomizationHelpers.cxx

@@ -67,6 +67,8 @@ std::vector<ConfigParamSpec> WorkflowCustomizationHelpers::requiredWorkflowOptio
            {"aod-writer-resmode", VariantType::String, "RECREATE", {"Creation mode of the result files: NEW, CREATE, RECREATE, UPDATE"}},
            {"aod-writer-ntfmerge", VariantType::Int, -1, {"Number of time frames to merge into one file"}},
            {"aod-writer-keep", VariantType::String, "", {"Comma separated list of ORIGIN/DESCRIPTION/SUBSPECIFICATION:treename:col1/col2/..:filename"}},
+           // options to manipulate origins
+           {"aod-origin-replace", VariantType::String, "", {"Replace AOD origin with the string provided"}},
 
            {"fairmq-rate-logging", VariantType::Int, 0, {"Rate logging for FairMQ channels"}},
            {"fairmq-recv-buffer-size", VariantType::Int, 4, {"recvBufferSize option for FairMQ channels"}},