detector_writer.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
 
// Project include(s)
#include "detray/io/frontend/detail/detector_components_writer.hpp"
#include "detray/io/frontend/detector_writer_config.hpp"
#include "detray/io/frontend/impl/json_writers.hpp"
#include "detray/io/utils/create_path.hpp"
#include "detray/utils/logging.hpp"
 
// System include(s)
#include <filesystem>
#include <ios>
 
namespace detray::io {

template <class detector_t>
void write_detector(detector_t& det, const typename detector_t::name_map& names,
                    detector_writer_config& cfg) {
  // How to open the file
  const std::ios_base::openmode out_mode{std::ios_base::out |
                                         std::ios_base::binary};
  auto mode =
      cfg.replace_files() ? (out_mode | std::ios_base::trunc) : out_mode;
 
  const auto file_path = detray::io::create_path(cfg.path());
 
  if (det.material_store().all_empty()) {
    cfg.write_material(false);
  }
 
  // Get the writer
  io::detail::detector_components_writer<detector_t> writer{};
  if (cfg.format() == io::format::json) {
    detail::add_json_writers(writer, cfg);
  }
 
  if (cfg.compactify_json()) {
    DETRAY_ERROR_HOST("Compactifying json files is not yet implemented");
  }
 
  writer.write(det, names, mode, file_path);
}
 
}  // namespace detray::io