Abseil Common Libraries (C++) (grcp 依赖)
https://abseil.io/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
434 lines
16 KiB
434 lines
16 KiB
#ifndef ABSL_STRINGS_INTERNAL_STR_FORMAT_ARG_H_ |
|
#define ABSL_STRINGS_INTERNAL_STR_FORMAT_ARG_H_ |
|
|
|
#include <string.h> |
|
#include <wchar.h> |
|
|
|
#include <cstdio> |
|
#include <iomanip> |
|
#include <limits> |
|
#include <sstream> |
|
#include <string> |
|
#include <type_traits> |
|
|
|
#include "absl/base/port.h" |
|
#include "absl/meta/type_traits.h" |
|
#include "absl/numeric/int128.h" |
|
#include "absl/strings/internal/str_format/extension.h" |
|
#include "absl/strings/string_view.h" |
|
|
|
class Cord; |
|
class CordReader; |
|
|
|
namespace absl { |
|
|
|
class FormatCountCapture; |
|
class FormatSink; |
|
|
|
namespace str_format_internal { |
|
|
|
template <typename T, typename = void> |
|
struct HasUserDefinedConvert : std::false_type {}; |
|
|
|
template <typename T> |
|
struct HasUserDefinedConvert< |
|
T, void_t<decltype(AbslFormatConvert( |
|
std::declval<const T&>(), std::declval<const ConversionSpec&>(), |
|
std::declval<FormatSink*>()))>> : std::true_type {}; |
|
template <typename T> |
|
class StreamedWrapper; |
|
|
|
// If 'v' can be converted (in the printf sense) according to 'conv', |
|
// then convert it, appending to `sink` and return `true`. |
|
// Otherwise fail and return `false`. |
|
// Raw pointers. |
|
struct VoidPtr { |
|
VoidPtr() = default; |
|
template <typename T, |
|
decltype(reinterpret_cast<uintptr_t>(std::declval<T*>())) = 0> |
|
VoidPtr(T* ptr) // NOLINT |
|
: value(ptr ? reinterpret_cast<uintptr_t>(ptr) : 0) {} |
|
uintptr_t value; |
|
}; |
|
ConvertResult<Conv::p> FormatConvertImpl(VoidPtr v, const ConversionSpec& conv, |
|
FormatSinkImpl* sink); |
|
|
|
// Strings. |
|
ConvertResult<Conv::s> FormatConvertImpl(const std::string& v, |
|
const ConversionSpec& conv, |
|
FormatSinkImpl* sink); |
|
ConvertResult<Conv::s> FormatConvertImpl(string_view v, |
|
const ConversionSpec& conv, |
|
FormatSinkImpl* sink); |
|
ConvertResult<Conv::s | Conv::p> FormatConvertImpl(const char* v, |
|
const ConversionSpec& conv, |
|
FormatSinkImpl* sink); |
|
template <class AbslCord, |
|
typename std::enable_if< |
|
std::is_same<AbslCord, ::Cord>::value>::type* = nullptr, |
|
class AbslCordReader = ::CordReader> |
|
ConvertResult<Conv::s> FormatConvertImpl(const AbslCord& value, |
|
const ConversionSpec& conv, |
|
FormatSinkImpl* sink) { |
|
if (conv.conv().id() != ConversionChar::s) return {false}; |
|
|
|
bool is_left = conv.flags().left; |
|
size_t space_remaining = 0; |
|
|
|
int width = conv.width(); |
|
if (width >= 0) space_remaining = width; |
|
|
|
size_t to_write = value.size(); |
|
|
|
int precision = conv.precision(); |
|
if (precision >= 0) |
|
to_write = std::min(to_write, static_cast<size_t>(precision)); |
|
|
|
space_remaining = Excess(to_write, space_remaining); |
|
|
|
if (space_remaining > 0 && !is_left) sink->Append(space_remaining, ' '); |
|
|
|
string_view piece; |
|
for (AbslCordReader reader(value); |
|
to_write > 0 && reader.ReadFragment(&piece); to_write -= piece.size()) { |
|
if (piece.size() > to_write) piece.remove_suffix(piece.size() - to_write); |
|
sink->Append(piece); |
|
} |
|
|
|
if (space_remaining > 0 && is_left) sink->Append(space_remaining, ' '); |
|
return {true}; |
|
} |
|
|
|
using IntegralConvertResult = |
|
ConvertResult<Conv::c | Conv::numeric | Conv::star>; |
|
using FloatingConvertResult = ConvertResult<Conv::floating>; |
|
|
|
// Floats. |
|
FloatingConvertResult FormatConvertImpl(float v, const ConversionSpec& conv, |
|
FormatSinkImpl* sink); |
|
FloatingConvertResult FormatConvertImpl(double v, const ConversionSpec& conv, |
|
FormatSinkImpl* sink); |
|
FloatingConvertResult FormatConvertImpl(long double v, |
|
const ConversionSpec& conv, |
|
FormatSinkImpl* sink); |
|
|
|
// Chars. |
|
IntegralConvertResult FormatConvertImpl(char v, const ConversionSpec& conv, |
|
FormatSinkImpl* sink); |
|
IntegralConvertResult FormatConvertImpl(signed char v, |
|
const ConversionSpec& conv, |
|
FormatSinkImpl* sink); |
|
IntegralConvertResult FormatConvertImpl(unsigned char v, |
|
const ConversionSpec& conv, |
|
FormatSinkImpl* sink); |
|
|
|
// Ints. |
|
IntegralConvertResult FormatConvertImpl(short v, // NOLINT |
|
const ConversionSpec& conv, |
|
FormatSinkImpl* sink); |
|
IntegralConvertResult FormatConvertImpl(unsigned short v, // NOLINT |
|
const ConversionSpec& conv, |
|
FormatSinkImpl* sink); |
|
IntegralConvertResult FormatConvertImpl(int v, const ConversionSpec& conv, |
|
FormatSinkImpl* sink); |
|
IntegralConvertResult FormatConvertImpl(unsigned v, const ConversionSpec& conv, |
|
FormatSinkImpl* sink); |
|
IntegralConvertResult FormatConvertImpl(long v, // NOLINT |
|
const ConversionSpec& conv, |
|
FormatSinkImpl* sink); |
|
IntegralConvertResult FormatConvertImpl(unsigned long v, // NOLINT |
|
const ConversionSpec& conv, |
|
FormatSinkImpl* sink); |
|
IntegralConvertResult FormatConvertImpl(long long v, // NOLINT |
|
const ConversionSpec& conv, |
|
FormatSinkImpl* sink); |
|
IntegralConvertResult FormatConvertImpl(unsigned long long v, // NOLINT |
|
const ConversionSpec& conv, |
|
FormatSinkImpl* sink); |
|
IntegralConvertResult FormatConvertImpl(uint128 v, const ConversionSpec& conv, |
|
FormatSinkImpl* sink); |
|
template <typename T, enable_if_t<std::is_same<T, bool>::value, int> = 0> |
|
IntegralConvertResult FormatConvertImpl(T v, const ConversionSpec& conv, |
|
FormatSinkImpl* sink) { |
|
return FormatConvertImpl(static_cast<int>(v), conv, sink); |
|
} |
|
|
|
// We provide this function to help the checker, but it is never defined. |
|
// FormatArgImpl will use the underlying Convert functions instead. |
|
template <typename T> |
|
typename std::enable_if<std::is_enum<T>::value && |
|
!HasUserDefinedConvert<T>::value, |
|
IntegralConvertResult>::type |
|
FormatConvertImpl(T v, const ConversionSpec& conv, FormatSinkImpl* sink); |
|
|
|
template <typename T> |
|
ConvertResult<Conv::s> FormatConvertImpl(const StreamedWrapper<T>& v, |
|
const ConversionSpec& conv, |
|
FormatSinkImpl* out) { |
|
std::ostringstream oss; |
|
oss << v.v_; |
|
if (!oss) return {false}; |
|
return str_format_internal::FormatConvertImpl(oss.str(), conv, out); |
|
} |
|
|
|
// Use templates and dependent types to delay evaluation of the function |
|
// until after FormatCountCapture is fully defined. |
|
struct FormatCountCaptureHelper { |
|
template <class T = int> |
|
static ConvertResult<Conv::n> ConvertHelper(const FormatCountCapture& v, |
|
const ConversionSpec& conv, |
|
FormatSinkImpl* sink) { |
|
const absl::enable_if_t<sizeof(T) != 0, FormatCountCapture>& v2 = v; |
|
|
|
if (conv.conv().id() != str_format_internal::ConversionChar::n) |
|
return {false}; |
|
*v2.p_ = static_cast<int>(sink->size()); |
|
return {true}; |
|
} |
|
}; |
|
|
|
template <class T = int> |
|
ConvertResult<Conv::n> FormatConvertImpl(const FormatCountCapture& v, |
|
const ConversionSpec& conv, |
|
FormatSinkImpl* sink) { |
|
return FormatCountCaptureHelper::ConvertHelper(v, conv, sink); |
|
} |
|
|
|
// Helper friend struct to hide implementation details from the public API of |
|
// FormatArgImpl. |
|
struct FormatArgImplFriend { |
|
template <typename Arg> |
|
static bool ToInt(Arg arg, int* out) { |
|
if (!arg.vtbl_->to_int) return false; |
|
*out = arg.vtbl_->to_int(arg.data_); |
|
return true; |
|
} |
|
|
|
template <typename Arg> |
|
static bool Convert(Arg arg, const str_format_internal::ConversionSpec& conv, |
|
FormatSinkImpl* out) { |
|
return arg.vtbl_->convert(arg.data_, conv, out); |
|
} |
|
|
|
template <typename Arg> |
|
static const void* GetVTablePtrForTest(Arg arg) { |
|
return arg.vtbl_; |
|
} |
|
}; |
|
|
|
// A type-erased handle to a format argument. |
|
class FormatArgImpl { |
|
private: |
|
enum { kInlinedSpace = 8 }; |
|
|
|
using VoidPtr = str_format_internal::VoidPtr; |
|
|
|
union Data { |
|
const void* ptr; |
|
const volatile void* volatile_ptr; |
|
char buf[kInlinedSpace]; |
|
}; |
|
|
|
struct VTable { |
|
bool (*convert)(Data, const str_format_internal::ConversionSpec& conv, |
|
FormatSinkImpl* out); |
|
int (*to_int)(Data); |
|
}; |
|
|
|
template <typename T> |
|
struct store_by_value |
|
: std::integral_constant<bool, (sizeof(T) <= kInlinedSpace) && |
|
(std::is_integral<T>::value || |
|
std::is_floating_point<T>::value || |
|
std::is_pointer<T>::value || |
|
std::is_same<VoidPtr, T>::value)> {}; |
|
|
|
enum StoragePolicy { ByPointer, ByVolatilePointer, ByValue }; |
|
template <typename T> |
|
struct storage_policy |
|
: std::integral_constant<StoragePolicy, |
|
(std::is_volatile<T>::value |
|
? ByVolatilePointer |
|
: (store_by_value<T>::value ? ByValue |
|
: ByPointer))> { |
|
}; |
|
|
|
// An instance of an FormatArgImpl::VTable suitable for 'T'. |
|
template <typename T> |
|
struct TypedVTable; |
|
|
|
// To reduce the number of vtables we will decay values before hand. |
|
// Anything with a user-defined Convert will get its own vtable. |
|
// For everything else: |
|
// - Decay char* and char arrays into `const char*` |
|
// - Decay any other pointer to `const void*` |
|
// - Decay all enums to their underlying type. |
|
// - Decay function pointers to void*. |
|
template <typename T, typename = void> |
|
struct DecayType { |
|
static constexpr bool kHasUserDefined = |
|
str_format_internal::HasUserDefinedConvert<T>::value; |
|
using type = typename std::conditional< |
|
!kHasUserDefined && std::is_convertible<T, const char*>::value, |
|
const char*, |
|
typename std::conditional<!kHasUserDefined && |
|
std::is_convertible<T, VoidPtr>::value, |
|
VoidPtr, const T&>::type>::type; |
|
}; |
|
template <typename T> |
|
struct DecayType<T, |
|
typename std::enable_if< |
|
!str_format_internal::HasUserDefinedConvert<T>::value && |
|
std::is_enum<T>::value>::type> { |
|
using type = typename std::underlying_type<T>::type; |
|
}; |
|
|
|
public: |
|
template <typename T> |
|
explicit FormatArgImpl(const T& value) { |
|
using D = typename DecayType<T>::type; |
|
static_assert( |
|
std::is_same<D, const T&>::value || storage_policy<D>::value == ByValue, |
|
"Decayed types must be stored by value"); |
|
Init(static_cast<D>(value)); |
|
} |
|
|
|
private: |
|
friend struct str_format_internal::FormatArgImplFriend; |
|
template <typename T, StoragePolicy = storage_policy<T>::value> |
|
struct Manager; |
|
|
|
template <typename T> |
|
struct Manager<T, ByPointer> { |
|
static Data SetValue(const T& value) { |
|
Data data; |
|
data.ptr = &value; |
|
return data; |
|
} |
|
|
|
static const T& Value(Data arg) { return *static_cast<const T*>(arg.ptr); } |
|
}; |
|
|
|
template <typename T> |
|
struct Manager<T, ByVolatilePointer> { |
|
static Data SetValue(const T& value) { |
|
Data data; |
|
data.volatile_ptr = &value; |
|
return data; |
|
} |
|
|
|
static const T& Value(Data arg) { |
|
return *static_cast<const T*>(arg.volatile_ptr); |
|
} |
|
}; |
|
|
|
template <typename T> |
|
struct Manager<T, ByValue> { |
|
static Data SetValue(const T& value) { |
|
Data data; |
|
memcpy(data.buf, &value, sizeof(value)); |
|
return data; |
|
} |
|
|
|
static T Value(Data arg) { |
|
T value; |
|
memcpy(&value, arg.buf, sizeof(T)); |
|
return value; |
|
} |
|
}; |
|
|
|
template <typename T> |
|
void Init(const T& value); |
|
|
|
template <typename T> |
|
static int ToIntVal(const T& val) { |
|
using CommonType = typename std::conditional<std::is_signed<T>::value, |
|
int64_t, uint64_t>::type; |
|
if (static_cast<CommonType>(val) > |
|
static_cast<CommonType>(std::numeric_limits<int>::max())) { |
|
return std::numeric_limits<int>::max(); |
|
} else if (std::is_signed<T>::value && |
|
static_cast<CommonType>(val) < |
|
static_cast<CommonType>(std::numeric_limits<int>::min())) { |
|
return std::numeric_limits<int>::min(); |
|
} |
|
return static_cast<int>(val); |
|
} |
|
|
|
Data data_; |
|
const VTable* vtbl_; |
|
}; |
|
|
|
template <typename T> |
|
struct FormatArgImpl::TypedVTable { |
|
private: |
|
static bool ConvertImpl(Data arg, |
|
const str_format_internal::ConversionSpec& conv, |
|
FormatSinkImpl* out) { |
|
return str_format_internal::FormatConvertImpl(Manager<T>::Value(arg), conv, |
|
out) |
|
.value; |
|
} |
|
|
|
template <typename U = T, typename = void> |
|
struct ToIntImpl { |
|
static constexpr int (*value)(Data) = nullptr; |
|
}; |
|
|
|
template <typename U> |
|
struct ToIntImpl<U, |
|
typename std::enable_if<std::is_integral<U>::value>::type> { |
|
static int Invoke(Data arg) { return ToIntVal(Manager<T>::Value(arg)); } |
|
static constexpr int (*value)(Data) = &Invoke; |
|
}; |
|
|
|
template <typename U> |
|
struct ToIntImpl<U, typename std::enable_if<std::is_enum<U>::value>::type> { |
|
static int Invoke(Data arg) { |
|
return ToIntVal(static_cast<typename std::underlying_type<T>::type>( |
|
Manager<T>::Value(arg))); |
|
} |
|
static constexpr int (*value)(Data) = &Invoke; |
|
}; |
|
|
|
public: |
|
static constexpr VTable value{&ConvertImpl, ToIntImpl<>::value}; |
|
}; |
|
|
|
template <typename T> |
|
constexpr FormatArgImpl::VTable FormatArgImpl::TypedVTable<T>::value; |
|
|
|
template <typename T> |
|
void FormatArgImpl::Init(const T& value) { |
|
data_ = Manager<T>::SetValue(value); |
|
vtbl_ = &TypedVTable<T>::value; |
|
} |
|
|
|
extern template struct FormatArgImpl::TypedVTable<str_format_internal::VoidPtr>; |
|
|
|
extern template struct FormatArgImpl::TypedVTable<bool>; |
|
extern template struct FormatArgImpl::TypedVTable<char>; |
|
extern template struct FormatArgImpl::TypedVTable<signed char>; |
|
extern template struct FormatArgImpl::TypedVTable<unsigned char>; |
|
extern template struct FormatArgImpl::TypedVTable<short>; // NOLINT |
|
extern template struct FormatArgImpl::TypedVTable<unsigned short>; // NOLINT |
|
extern template struct FormatArgImpl::TypedVTable<int>; |
|
extern template struct FormatArgImpl::TypedVTable<unsigned>; |
|
extern template struct FormatArgImpl::TypedVTable<long>; // NOLINT |
|
extern template struct FormatArgImpl::TypedVTable<unsigned long>; // NOLINT |
|
extern template struct FormatArgImpl::TypedVTable<long long>; // NOLINT |
|
extern template struct FormatArgImpl::TypedVTable< |
|
unsigned long long>; // NOLINT |
|
extern template struct FormatArgImpl::TypedVTable<uint128>; |
|
|
|
extern template struct FormatArgImpl::TypedVTable<float>; |
|
extern template struct FormatArgImpl::TypedVTable<double>; |
|
extern template struct FormatArgImpl::TypedVTable<long double>; |
|
|
|
extern template struct FormatArgImpl::TypedVTable<const char*>; |
|
extern template struct FormatArgImpl::TypedVTable<std::string>; |
|
extern template struct FormatArgImpl::TypedVTable<string_view>; |
|
} // namespace str_format_internal |
|
} // namespace absl |
|
|
|
#endif // ABSL_STRINGS_INTERNAL_STR_FORMAT_ARG_H_
|
|
|