detector_io_options.hpp

// This file is part of the ACTS project.
//
// Copyright (C) 2016 CERN for the benefit of the ACTS project
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at https://mozilla.org/MPL/2.0/.
 
#pragma once
 
// detray IO include(s)
#include "detray/io/frontend/detector_reader_config.hpp"
#include "detray/io/frontend/detector_writer_config.hpp"
 
// Detray test include(s)
#include "detray/options/options_handling.hpp"
 
// Boost
#include "detray/options/boost_program_options.hpp"
 
// System include(s)
#include <stdexcept>
#include <string>
 
namespace detray::options {

template <>
void add_options<detray::io::detector_reader_config>(
    boost::program_options::options_description &desc,
    const detray::io::detector_reader_config & /*unused*/) {
  desc.add_options()("geometry_file",
                     boost::program_options::value<std::string>(),
                     "Detector geometry input file")(
      "grid_file", boost::program_options::value<std::string>(),
      "Detector surface grid input file")(
      "material_file", boost::program_options::value<std::string>(),
      "Detector material input file");
}

template <>
void configure_options<detray::io::detector_reader_config>(
    const boost::program_options::variables_map &vm,
    detray::io::detector_reader_config &cfg) {
  // Input files
  if (vm.count("geometry_file") != 0u) {
    cfg.add_file(vm["geometry_file"].as<std::string>());
  } else {
    throw std::invalid_argument("Please specify a geometry input file!\n\n");
  }
  if (vm.count("material_file") != 0u) {
    cfg.add_file(vm["material_file"].as<std::string>());
  }
  if (vm.count("grid_file") != 0u) {
    cfg.add_file(vm["grid_file"].as<std::string>());
  }
}

template <>
void add_options<detray::io::detector_writer_config>(
    boost::program_options::options_description &desc,
    const detray::io::detector_writer_config &cfg) {
  desc.add_options()(
      "outdir",
      boost::program_options::value<std::string>()->default_value(cfg.path()),
      "Output directory for detector files")(
      "compactify_json", "not implemented")("write_material",
                                            "toggle material output")(
      "write_grids", "toggle grid output")("replace_files",
                                           "whether to replace existing files");
}

template <>
void configure_options<detray::io::detector_writer_config>(
    const boost::program_options::variables_map &vm,
    detray::io::detector_writer_config &cfg) {
  if (!vm["outdir"].defaulted()) {
    cfg.path(vm["outdir"].as<std::string>());
  }
  cfg.compactify_json(vm.count("compactify_json") != 0u);
  cfg.write_material(vm.count("write_material") != 0u);
  cfg.write_grids(vm.count("write_grids") != 0u);
  cfg.replace_files(vm.count("replace_files") != 0u);
}
 
}  // namespace detray::options