R3BPDGConverter.h

Go to the documentation of this file.

#pragma once
 
#include <cstdint>
#include <cstdlib>
#include <string>
#include <unordered_map>
 
namespace R3B
{
    class PDGConverter
    {
      public:
        enum class Type : uint8_t
        {
            basic,
            nuclear,
            invalid,
        };
 
        PDGConverter() = default;
 
        auto get_pid(const std::string& name) -> int32_t;
 
        auto get_name(int32_t pid) -> std::string;

        static auto get_type(const std::string& name) -> Type;

        static auto get_type(int32_t pid) -> Type;
 
        auto reset();
        static constexpr auto nuclear_id_min = int32_t{ 1000000000 };
        static constexpr auto nuclear_id_max = int32_t{ 1009999990 };
 
      private:
        std::unordered_map<int32_t, std::string> pid_to_name_;
        std::unordered_map<std::string, int32_t> name_to_pid_;
 
        void register_entry(int32_t pid);
        void register_entry(const std::string& name);
 
        auto convert_basic_pid_to_name(int32_t pid) -> std::string;
    };
 
} // namespace R3B