detector_reader_config.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/builders/volume_builder_interface.hpp"
 
// System include(s)
#include <ostream>
#include <string>
#include <vector>
 
namespace detray::io {

struct detector_reader_config {
  std::vector<std::string> m_files;
  detray::volume_builder_options m_builder_opts{};
  bool m_do_check{true};
  bool m_verbose{false};

  const std::vector<std::string>& files() const { return m_files; }
  const detray::volume_builder_options& builder_options() const {
    return m_builder_opts;
  }

  bool deduplicate() const { return m_builder_opts.deduplicate(); }
  bool do_check() const { return m_do_check; }
  bool verbose_check() const { return m_verbose; }

  detector_reader_config& add_file(const std::string& file_name) {
    m_files.push_back(file_name);
    return *this;
  }

  detector_reader_config& deduplicate(bool toggle) {
    m_builder_opts.deduplicate(toggle);
    return *this;
  }

  detector_reader_config& do_check(const bool check) {
    m_do_check = check;
    return *this;
  }

  detector_reader_config& verbose_check(const bool verbose) {
    if (verbose && !m_do_check) {
      m_do_check = true;
    }
    m_verbose = verbose;
    return *this;
  }


  friend std::ostream& operator<<(std::ostream& out,
                                  const detector_reader_config& cfg) {
    out << "\nDetector reader\n"
        << "----------------------------\n"
        << "  Detector files        : \n";
    for (const auto& file_name : cfg.files()) {
      out << "    -> " << file_name << "\n";
    }
    out << "  Deduplicate data      : " << std::boolalpha << cfg.deduplicate()
        << std::noboolalpha << "\n";
 
    return out;
  }
};
 
}  // namespace detray::io