HDF5SourceIDHandler.hpp

Go to the documentation of this file.

 
#ifndef HDF5LIBS_INCLUDE_HDF5LIBS_HDF5SOURCEIDHANDLER_HPP_
#define HDF5LIBS_INCLUDE_HDF5LIBS_HDF5SOURCEIDHANDLER_HPP_
 
#include "daqdataformats/Fragment.hpp"
#include "daqdataformats/SourceID.hpp"
#include "detdataformats/DetID.hpp"
#include "confmodel/Session.hpp"
 
//#include "hdf5libs/hdf5filelayout/Structs.hpp"
//#include "daqdataformats/TimeSliceHeader.hpp"
//#include "daqdataformats/TriggerRecordHeader.hpp"
//#include "logging/Logging.hpp"
 
#include <highfive/H5File.hpp>
#include <highfive/H5Group.hpp>
 
#include <cstdint>
#include <map>
#include <set>
#include <string>
#include <vector>
 
namespace dunedaq {
namespace hdf5libs {
 
class HDF5SourceIDHandler
{
public:
  typedef std::map<daqdataformats::SourceID, std::string> source_id_path_map_t;
  typedef std::map<daqdataformats::SourceID, std::vector<uint64_t>> source_id_geo_id_map_t; // NOLINT(build/unsigned)
  typedef std::map<daqdataformats::SourceID::Subsystem, std::set<daqdataformats::SourceID>> subsystem_source_id_map_t;
  typedef std::map<daqdataformats::FragmentType, std::set<daqdataformats::SourceID>> fragment_type_source_id_map_t;
  typedef std::map<detdataformats::DetID::Subdetector, std::set<daqdataformats::SourceID>> subdetector_source_id_map_t;
 
  static source_id_geo_id_map_t make_source_id_geo_id_map(const confmodel::Session* session);
 
  static void store_file_level_geo_id_info(HighFive::File& h5_file, const source_id_geo_id_map_t& the_map);
 
  static void store_record_header_source_id(HighFive::Group& record_group, const daqdataformats::SourceID& source_id);
 
  static void store_record_level_path_info(HighFive::Group& record_group, const source_id_path_map_t& the_map);
 
  static void store_record_level_fragment_type_map(HighFive::Group& record_group,
                                                   const fragment_type_source_id_map_t& the_map);
 
  static void store_record_level_subdetector_map(HighFive::Group& record_group,
                                                 const subdetector_source_id_map_t& the_map);
 
  explicit HDF5SourceIDHandler(const uint32_t version); // NOLINT(build/unsigned)
 
  void fetch_file_level_geo_id_info(const HighFive::File& h5_file, source_id_geo_id_map_t& the_map);
 
  void fetch_record_level_geo_id_info(const HighFive::Group& record_group, source_id_geo_id_map_t& the_map);
 
  daqdataformats::SourceID fetch_record_header_source_id(const HighFive::Group& record_group);
 
  void fetch_source_id_path_info(const HighFive::Group& record_group, source_id_path_map_t& the_map);
 
  void fetch_fragment_type_source_id_info(const HighFive::Group& record_group, fragment_type_source_id_map_t& the_map);
 
  void fetch_subdetector_source_id_info(const HighFive::Group& record_group, subdetector_source_id_map_t& the_map);
 
  static void add_source_id_path_to_map(source_id_path_map_t& source_id_path_map,
                                        const daqdataformats::SourceID& source_id,
                                        const std::string& hdf5_path);
 
  static void add_source_id_geo_id_to_map(source_id_geo_id_map_t& source_id_geo_id_map,
                                          const daqdataformats::SourceID& source_id,
                                          uint64_t geo_id); // NOLINT(build/unsigned)
 
  static void add_fragment_type_source_id_to_map(fragment_type_source_id_map_t& fragment_type_source_id_map,
                                                 const daqdataformats::FragmentType fragment_type,
                                                 const daqdataformats::SourceID& source_id);
 
  static void add_subdetector_source_id_to_map(subdetector_source_id_map_t& subdetector_source_id_map,
                                               const detdataformats::DetID::Subdetector subdetector,
                                               const daqdataformats::SourceID& source_id);
 
  static void add_subsystem_source_id_to_map(subsystem_source_id_map_t& subsystem_source_id_map,
                                             const daqdataformats::SourceID::Subsystem subsystem,
                                             const daqdataformats::SourceID& source_id);
 
private:
  uint32_t m_version; // NOLINT(build/unsigned)
 
  static std::string get_json_string(const daqdataformats::SourceID& source_id);
 
  static std::string get_json_string(const source_id_path_map_t& source_id_path_map);
 
  static std::string get_json_string(const source_id_geo_id_map_t& source_id_geo_id_map);
 
  static std::string get_json_string(const fragment_type_source_id_map_t& fragment_type_source_id_map);
 
  static std::string get_json_string(const subdetector_source_id_map_t& subdetector_source_id_map);
 
  static void parse_json_string(const std::string& json_string, daqdataformats::SourceID& source_id);
 
  static void parse_json_string(const std::string& json_string, source_id_path_map_t& source_id_path_map);
 
  static void parse_json_string(const std::string& json_string, source_id_geo_id_map_t& source_id_geo_id_map);
 
  static void parse_json_string(const std::string& json_string, fragment_type_source_id_map_t& fragment_type_source_id_map);
 
  static void parse_json_string(const std::string& json_string, subdetector_source_id_map_t& subdetector_source_id_map);
 
  template<typename C, typename T>
  static void write_attribute(HighFive::AnnotateTraits<C>& h5annt, const std::string& name, T value);
 
  template<typename C, typename T>
  static T get_attribute(const HighFive::AnnotateTraits<C>& h5annt, const std::string& name);
};
 
template<typename C, typename T>
void
HDF5SourceIDHandler::write_attribute(HighFive::AnnotateTraits<C>& h5annt, const std::string& name, T value)
{
  if (!(h5annt.hasAttribute(name)))
    h5annt.createAttribute(name, value);
  // else
  //   ers::warning(HDF5AttributeExists(ERS_HERE, name));
}
 
template<typename C, typename T>
T
HDF5SourceIDHandler::get_attribute(const HighFive::AnnotateTraits<C>& h5annt, const std::string& name)
{
  // if (!h5annt.hasAttribute(name)) {
  //   throw InvalidHDF5Attribute(ERS_HERE, name);
  // }
  auto attr = h5annt.getAttribute(name);
  T value;
  attr.read(value);
  return value;
}
 
} // namespace hdf5libs
} // namespace dunedaq
 
#endif // HDF5LIBS_INCLUDE_HDF5LIBS_HDF5SOURCEIDHANDLER_HPP_