Program Listing for File get_root_ids.hpp

↰ Return to documentation for file (include/dish2/load/get_root_ids.hpp)

#pragma once
#ifndef DISH2_LOAD_GET_ROOT_IDS_HPP_INCLUDE
#define DISH2_LOAD_GET_ROOT_IDS_HPP_INCLUDE

#include <set>
#include <string>

#include "../../../third-party/conduit/include/uitsl/math/math_utils.hpp"
#include "../../../third-party/conduit/include/uitsl/utility/keyname_directory_transform.hpp"
#include "../../../third-party/Empirical/include/emp/base/always_assert.hpp"
#include "../../../third-party/Empirical/include/emp/base/vector.hpp"

namespace dish2 {

emp::vector<size_t> get_root_ids() {

  const auto res = uitsl::keyname_directory_transform(
    "root_id",
    {
      {"a", "genome|population"},
      {"ext", R"(\.bin|\.bin\.gz|\.bin\.xz|\.json|\.json\.gz|\.json\.xz)"}
    },
    ".",
    uitsl::stoszt,
    true
  );

  // ensure each candidate has a unique root id
  emp_always_assert(
    std::set<size_t>( std::begin(res), std::end(res) ).size() == res.size()
  );

  return res;

}

} // namespace dish2

#endif // #ifndef DISH2_LOAD_GET_ROOT_IDS_HPP_INCLUDE