我有一个类模板,我用它来获取变量的大小:
template <class T>
class Size
{
unsigned int operator() (T) {return sizeof(T);}
};
这很好但是对于字符串我想使用strlen而不是sizeof:
template <>
class Size<char *>
{
unsigned int operator() (char *str) {return strlen(str);}
};
问题是当我使用const char *创建一个大小的实例时,它会转到非特定版本。我想知道是否有办法在模板专业化中捕获char *的const和非const版本?感谢。
答案 0 :(得分:10)
使用此技术:
#include <type_traits>
template< typename T, typename = void >
class Size
{
unsigned int operator() (T) {return sizeof(T);}
};
template< typename T >
class Size< T, typename std::enable_if<
std::is_same< T, char* >::value ||
std::is_same< T, const char* >::value
>::type >
{
unsigned int operator() ( T str ) { /* your code here */ }
};
编辑:如何在类定义之外定义方法的示例。
EDIT2:添加了帮助程序以避免重复可能很长且复杂的情况。
EDIT3:简化帮手。
#include <type_traits>
#include <iostream>
template< typename T >
struct my_condition
: std::enable_if< std::is_same< T, char* >::value ||
std::is_same< T, const char* >::value >
{};
template< typename T, typename = void >
struct Size
{
unsigned int operator() (T);
};
template< typename T >
struct Size< T, typename my_condition< T >::type >
{
unsigned int operator() (T);
};
template< typename T, typename Dummy >
unsigned int Size< T, Dummy >::operator() (T)
{
return 1;
}
template< typename T >
unsigned int Size< T, typename my_condition< T >::type >::operator() (T)
{
return 2;
}
int main()
{
std::cout << Size< int >()(0) << std::endl;
std::cout << Size< char* >()(0) << std::endl;
std::cout << Size< const char* >()(0) << std::endl;
}
打印
1
2
2
答案 1 :(得分:4)
一种方法是使用辅助函数来确定模板类型是char *
还是char const *
。您可以使用简单的结构来完成此操作。然后,您可以使用SFINAE选择Size结构的正确特化。
#include <cstring>
#include <iostream>
#include <boost/utility/enable_if.hpp>
template<typename T>
struct is_cstring {
enum { value = false };
};
template<>
struct is_cstring<char *> {
enum { value = true };
};
template<>
struct is_cstring<char const *> {
enum { value = true };
};
template<typename T, typename = void>
struct Size;
template<typename T>
struct Size<T, typename boost::disable_if<is_cstring<T> >::type> {
unsigned int operator ()(T const &) const {
return sizeof(T);
}
};
template<typename T>
struct Size<T, typename boost::enable_if<is_cstring<T> >::type> {
unsigned int operator ()(T const &str) const {
return strlen(str);
}
};
int main() {
std::string blah = "afafasa";
char *x = "asdfsadsad";
std::cout << Size<int>()(4) << std::endl;
std::cout << Size<char const *>()("blahblah") << std::endl;
std::cout << Size<char *>()(x) << std::endl;
}
打印结果为:
4
8
10
答案 2 :(得分:1)
当然,你也应该能够写作:
template <>
class Size<const char *>
{
unsigned int operator() (const char *str) {return strlen(str);}
};
template <>
class Size<char *> : public Size<const char *>
{ };
......而且,如果您需要:
template <size_t size>
class Size<char[size]> : public Size<const char *>
{ };
template <size_t size>
class Size<const char[size]> : public Size<const char *>
{ };
答案 3 :(得分:-1)
最简单的解决方案是将char *
专精取代为const char *
。它也适用于char *
。
template <>
class Size<const char *>
{
unsigned int operator() (const char *str) {return strlen(str);}
};