HDF5TestUtils.hpp

Go to the documentation of this file.

 
#ifndef HDF5LIBS_TESTUTILITIES_HPP_
#define HDF5LIBS_TESTUTILITIES_HPP_
 
#include "hdf5libs/HDF5RawDataFile.hpp"
 
#include <vector>
#include <string>
#include <filesystem>
#include <regex>
 
namespace dunedaq {
 
namespace hdf5libs {
 
std::vector<std::string>
get_files_matching_pattern(const std::string& path, const std::string& pattern)
{
  std::regex regex_search_pattern(pattern);
  std::vector<std::string> file_list;
  for (const auto& entry : std::filesystem::directory_iterator(path)) {
    if (std::regex_match(entry.path().filename().string(), regex_search_pattern)) {
      file_list.push_back(entry.path());
    }
  }
  return file_list;
}
 
std::vector<std::string>
delete_files_matching_pattern(const std::string& path, const std::string& pattern)
{
  std::regex regex_search_pattern(pattern);
  std::vector<std::string> file_list;
  for (const auto& entry : std::filesystem::directory_iterator(path)) {
    if (std::regex_match(entry.path().filename().string(), regex_search_pattern)) {
      if (std::filesystem::remove(entry.path())) {
        file_list.push_back(entry.path());
      }
    }
  }
  return file_list;
}
 
uint64_t
encode_geoid(int det_id, int crate_id, int slot_id, int stream_id)
{
  return (static_cast<uint64_t>(stream_id) << 48) | (static_cast<uint64_t>(slot_id) << 32) |
         (static_cast<uint64_t>(crate_id) << 16) | det_id;
}
 
HDF5SourceIDHandler::source_id_geo_id_map_t
create_srcid_geoid_map()
{
  HDF5SourceIDHandler::source_id_geo_id_map_t map;
 
  dunedaq::daqdataformats::SourceID sid;
  sid.subsystem = dunedaq::daqdataformats::SourceID::Subsystem::kDetectorReadout;
 
  sid.id = 0;
  map[sid].push_back(encode_geoid(3, 1, 0, 0));
 
  sid.id = 1;
  map[sid].push_back(encode_geoid(3, 1, 0, 1));
 
  sid.id = 3;
  map[sid].push_back(encode_geoid(3, 1, 1, 0));
 
  sid.id = 4;
  map[sid].push_back(encode_geoid(3, 1, 1, 1));
 
  sid.id = 4;
  map[sid].push_back(encode_geoid(2, 1, 0, 0));
 
  sid.id = 5;
  map[sid].push_back(encode_geoid(2, 1, 0, 1));
 
  sid.id = 6;
  map[sid].push_back(encode_geoid(2, 1, 1, 0));
 
  sid.id = 7;
  map[sid].push_back(encode_geoid(2, 1, 1, 1));
 
  return map;
}
 
} // namespace hdf5libs
 
} // namespace dunedaq
 
#endif // HDF5LIBS_TESTUTILITIES_HPP_