[Fix] support latest spdlog (#423)

* support formatting `PixelFormat` & `DataType`

* format enum for legacy spdlog

* fix format
pull/438/head
lzhangzz 2022-05-03 14:49:19 +08:00 committed by GitHub
parent 8aba06d2f3
commit 16ee9c7843
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 50 additions and 8 deletions

View File

@ -3,7 +3,11 @@
#ifndef MMDEPLOY_SRC_UTILS_FORMATTER_H_
#define MMDEPLOY_SRC_UTILS_FORMATTER_H_
#include <ostream>
#include "core/logger.h"
#include "core/types.h"
#include "spdlog/fmt/ostr.h"
#if FMT_VERSION >= 50000
#include "spdlog/fmt/bundled/ranges.h"
@ -17,12 +21,55 @@ class Value;
MMDEPLOY_API std::string format_value(const Value& value);
inline std::string to_string(PixelFormat format) {
switch (format) {
case PixelFormat::kBGR:
return "BGR";
case PixelFormat::kRGB:
return "RGB";
case PixelFormat::kGRAYSCALE:
return "GRAYSCALE";
case PixelFormat::kNV12:
return "NV12";
case PixelFormat::kNV21:
return "NV21";
case PixelFormat::kBGRA:
return "BGRA";
default:
return "invalid_format_enum";
}
}
inline std::string to_string(DataType type) {
switch (type) {
case DataType::kFLOAT:
return "FLOAT";
case DataType::kHALF:
return "HALF";
case DataType::kINT8:
return "INT8";
case DataType::kINT32:
return "INT32";
case DataType::kINT64:
return "INT64";
default:
return "invalid_data_type_enum";
}
}
inline std::ostream& operator<<(std::ostream& os, PixelFormat format) {
return os << to_string(format);
}
inline std::ostream& operator<<(std::ostream& os, DataType type) { return os << to_string(type); }
} // namespace mmdeploy
namespace fmt {
#if FMT_VERSION >= 50000
// `Value` maybe an incomplete type at this point, making `operator<<` not usable
template <>
struct formatter<mmdeploy::Value> {
constexpr auto parse(format_parse_context& ctx) { return ctx.begin(); }
@ -38,11 +85,6 @@ inline void format_arg(BasicFormatter<char> &f, const char *, const mmdeploy::Va
f.writer() << mmdeploy::format_value(d);
}
template <typename T, std::enable_if_t<std::is_enum<std::decay_t<T> >::value, bool> = true>
void format_arg(BasicFormatter<char> &f, const char *, const T &v) {
f.writer() << (int)v;
}
template <typename T>
auto format_arg(BasicFormatter<char>& f, const char*, const T& v)
-> std::void_t<decltype(begin(v), end(v))> {