16

我有一个类模板，我用它来获取变量的大小：

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 * 创建一个 size 实例时，它会转到非专业版本。我想知道是否有办法在模板专业化中同时捕获 char * 的 const 和非 const 版本？谢谢。

4

3 回答 3

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
于 2013-02-17T22:00:32.737 回答
5

当然，您还应该能够编写：

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 *>
{ };
于 2014-10-10T02:35:25.513 回答
4

一种方法是使用辅助函数来确定模板类型是 achar *还是 a 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
于 2013-02-17T22:08:07.490 回答