BasicPostProcessor.cc

Go to the documentation of this file.

#include "art/Framework/Art/BasicPostProcessor.h"

#include "art/Framework/Art/detail/exists_outside_prolog.h"
#include "art/Framework/Art/detail/fhicl_key.h"
#include "canvas/Utilities/Exception.h"
#include "cetlib/filepath_maker.h"
#include "fhiclcpp/extended_value.h"
#include "fhiclcpp/intermediate_table.h"

#include <iostream>
#include <string>

namespace {

  using table_t = fhicl::extended_value::table_t;
  using art::detail::exists_outside_prolog;
  using art::detail::fhicl_key;

  void
  verifyProcessName(fhicl::intermediate_table& raw_config)
  {
    if (exists_outside_prolog(raw_config, "process_name")) {
      auto const& process_name = raw_config.get<std::string>("process_name");
      if (process_name.empty()) {
        throw cet::exception("BAD_PROCESS_NAME")
          << "Empty process_name not permitted.";
      } else if (process_name.find('_') != std::string::npos) {
        throw cet::exception("BAD_PROCESS_NAME")
          << "Underscores not permitted in process_name: illegal value \""
          << process_name << "\"";
      }
    } else {
      std::cerr << "INFO: using default process_name of \"DUMMY\".\n";
      raw_config.put("process_name", "DUMMY");
    }
  }

  void
  verifyInterfaces(fhicl::intermediate_table& raw_config)
  {
    std::string const services{"services"};
    std::string const service_provider{"service_provider"};
    auto const ciProvider =
      fhicl_key(services, "CatalogInterface", service_provider);
    auto const ftProvider =
      fhicl_key(services, "FileTransfer", service_provider);
    if (!exists_outside_prolog(raw_config, ciProvider)) {
      raw_config.put(ciProvider, "TrivialFileDelivery");
    }
    if (!exists_outside_prolog(raw_config, ftProvider)) {
      raw_config.put(ftProvider, "TrivialFileTransfer");
    }
  }

  void
  verifySourceConfig(fhicl::intermediate_table& raw_config)
  {
    if (exists_outside_prolog(raw_config, "source.fileNames")) {
      if (exists_outside_prolog(raw_config, "source.module_type")) {
        if (raw_config.get<std::string>("source.module_type") == "EmptyEvent") {
          throw art::Exception(art::errors::Configuration)
            << "Error: source files specified for EmptyEvent source.";
        }
      } else {
        raw_config.put("source.module_type", "RootInput");
      }
    } else if (!exists_outside_prolog(raw_config, "source.module_type")) {
      raw_config.put("source.module_type", "EmptyEvent");
    }
    if (raw_config.get<std::string>("source.module_type") == "EmptyEvent" &&
        !exists_outside_prolog(raw_config, "source.maxEvents") &&
        !exists_outside_prolog(raw_config, "source.maxTime")) {
      // Default 1 event.
      raw_config.put("source.maxEvents", 1);
    }
  }

  void
  injectModuleLabels(fhicl::intermediate_table& int_table,
                     std::string const& table_spec)
  {
    if (!int_table.exists(table_spec)) {
      return;
    }
    auto& top_table_val = int_table.update(table_spec);
    if (!top_table_val.is_a(fhicl::TABLE)) {
      throw art::Exception(art::errors::Configuration)
        << "Unexpected non-table " << table_spec << ".\n";
    }
    auto& table = int_table.get<table_t&>(table_spec);
    for (auto const& tval : table) {
      if (tval.first.find('_') != std::string::npos) {
        throw art::Exception(art::errors::Configuration)
          << "Module parameter set label \"" << tval.first << "\" is illegal: "
          << "underscores are not permitted in module names.";
      }
      auto& table_val = int_table.update(table_spec + '.' + tval.first);
      if (!table_val.is_a(fhicl::TABLE)) {
        throw art::Exception(art::errors::Configuration)
          << "Unexpected non-table " << tval.first << " found in " << table_spec
          << ".\n";
      };
      int_table.put(table_spec + '.' + tval.first + ".module_label",
                    tval.first);
    }
  }

  void
  addModuleLabels(fhicl::intermediate_table& raw_config)
  {
    if (exists_outside_prolog(raw_config, "source")) {
      raw_config.put("source.module_label", "source");
    }
    injectModuleLabels(raw_config, "outputs");
    injectModuleLabels(raw_config, "physics.producers");
    injectModuleLabels(raw_config, "physics.filters");
    injectModuleLabels(raw_config, "physics.analyzers");
  }

  void
  injectServiceType(fhicl::intermediate_table& raw_config,
                    std::string const& table_spec,
                    std::vector<std::string> const& excluded = {})
  {
    if (!exists_outside_prolog(raw_config, table_spec)) {
      return;
    }
    auto& top_table_val = raw_config.update(table_spec);
    if (!top_table_val.is_a(fhicl::TABLE)) {
      throw art::Exception(art::errors::Configuration)
        << "Unexpected non-table " << table_spec << ".\n";
    }
    auto& table = raw_config.get<table_t&>(table_spec);
    for (auto const& tval : table) {
      auto& table_val = raw_config.update(table_spec + '.' + tval.first);
      if (!table_val.is_a(fhicl::TABLE)) {
        throw art::Exception(art::errors::Configuration)
          << "Unexpected non-table " << tval.first << " found in " << table_spec
          << ".\n";
      };
      if (std::find(excluded.cbegin(), excluded.cend(), tval.first) ==
          excluded.end()) {
        raw_config.put(table_spec + '.' + tval.first + ".service_type",
                       tval.first);
      }
    }
  }

  void
  addServiceType(fhicl::intermediate_table& raw_config)
  {
    std::vector<std::string> const excluded{"message", "scheduler"};
    injectServiceType(raw_config, "services", excluded);
  }
} // namespace

int
art::BasicPostProcessor::doCheckOptions(bpo::variables_map const&)
{
  return 0; // NOP
}

int
art::BasicPostProcessor::doProcessOptions(bpo::variables_map const&,
                                          fhicl::intermediate_table& raw_config)
{
  verifyProcessName(raw_config);
  verifyInterfaces(raw_config);
  verifySourceConfig(raw_config);
  // module_labels and service_type.
  addModuleLabels(raw_config);
  addServiceType(raw_config);
  return 0; // If anything had gone wrong, we would have thrown.
}