typeinfo.h

Go to the documentation of this file.

 
#ifndef TREELITE_ENUM_TYPEINFO_H_
#define TREELITE_ENUM_TYPEINFO_H_
 
#include <treelite/logging.h>
 
#include <cstdint>
#include <string>
#include <type_traits>
#include <typeinfo>
 
namespace treelite {
 
enum class TypeInfo : std::uint8_t { kInvalid = 0, kUInt32 = 1, kFloat32 = 2, kFloat64 = 3 };
 
std::string TypeInfoToString(treelite::TypeInfo info);
 
TypeInfo TypeInfoFromString(std::string const& str);
 
template <typename T>
inline TypeInfo TypeInfoFromType() {
  if (std::is_same_v<T, std::uint32_t>) {
    return TypeInfo::kUInt32;
  } else if (std::is_same_v<T, float>) {
    return TypeInfo::kFloat32;
  } else if (std::is_same_v<T, double>) {
    return TypeInfo::kFloat64;
  } else {
    return TypeInfo::kInvalid;
  }
}
 
}  // namespace treelite
 
#endif  // TREELITE_ENUM_TYPEINFO_H_