有几个很好的理由可以选择
#include <cstdlib>
template<typename T, std::size_t N>
constexpr std::size_t ARRAY_COUNT_FUNC(T (&arr)[N]) { return N; }
而不是
#define ARRAY_COUNT_MACRO(arr) (sizeof(arr)/sizeof(*arr))
一个重要的区别是,当一个指针(不是数组)传递给ARRAY_COUNT_MACRO
时,它会静默地返回一个无用的答案,但是将相同的参数传递给ARRAY_COUNT_FUNC
将导致编译器错误指出错误。
但宏确实有一个优点:它的论点没有评估。
#include <utility>
struct S {
int member_array[5];
};
// OK:
std::size_t count1 = ARRAY_COUNT_MACRO(std::declval<S&>().member_array);
// ERROR: std::declval is odr-used!
std::size_t count2 = ARRAY_COUNT_FUNC(std::declval<S&>().member_array);
还有另一种方法可以同时获益吗?例如,如果参数不是数组,则会导致编译错误,并且不会使用其参数。
答案 0 :(得分:6)
如Ch here所描述的那样无耻地从Chromium项目中扯下来。
#include <utility>
#include <cstdlib>
template<typename T, std::size_t N>
constexpr std::size_t ARRAY_COUNT_FUNC(T (&arr)[N]) { return N; }
#define ARRAY_COUNT_MACRO(arr) (sizeof(arr)/sizeof(*arr))
// Template for typesafey goodness.
template <typename T, size_t N>
char (&ArraySizeHelper(T (&array)[N]))[N];
// sizeof to avoid actually calling the function.
#define arraysize(array) (sizeof(ArraySizeHelper(array)))
struct S {
int member_array[5];
};
int main()
{
// OK:
std::size_t count1 = ARRAY_COUNT_MACRO(std::declval<S&>().member_array);
// ERROR: std::declval is odr-used!
//std::size_t count2 = ARRAY_COUNT_FUNC(std::declval<S&>().member_array);
// OK:
std::size_t count2 = arraysize(std::declval<S&>().member_array);
// ERROR:
// int * p;
// std::size_t count3 = arraysize(p);
}
答案 1 :(得分:2)
...然后我记得<type_traits>
有一个std::is_array
模板。另一种解决方案:
#include <type_traits>
template<typename T>
constexpr auto ArrayCountImpl(std::nullptr_t)
-> typename std::enable_if<std::is_array<typename
std::remove_reference<T>::type>::value,
std::size_t>::type
{ return std::extent<typename std::remove_reference<T>::type>::value; }
template<typename T>
std::size_t ArrayCountImpl(...)
{ static_assert(!std::is_same<T,T>::value,
"Argument type is not an array"); }
#define ARRAY_COUNT_MACRO_2(arr) (ArrayCountImpl<decltype(arr)>(nullptr))