C++：如何使用模板函数中的类型进行分支？

Question

我对模板不太熟悉。如何编写名为 get 的模板函数，该函数根据模板类型选择从中获取的数组？请参阅下面的示例：

struct Foo
{
    int iArr[10];
    char cArr[10];

    // How to pick array here based on template type?
    template < typename T >
    T get( int idx )
    {
        // This does NOT work!
        switch ( T )
        {
        case int:
            return iArr[ idx ];
        case char:
            return cArr[ idx ];
        }
    }
};

// Expected behaviour of get()
Foo foo;
int i  = foo.get< int >( 2 );
char c = foo.get< char >( 4 );

Answer 1

虽然 Jason 提出的解决方案有效，但它远非惯用，并且更难以维护，因为 switch 语句 1) 中的 case 值没有明显的含义（“幻数”） ") 和 2) 很容易与 switch_value 特化中的值不同步。我建议这样做：

struct Foo {
    Foo() : iArr(), cArr() { }

    template<typename T>
    T get(std::size_t const idx) const     {
        return Foo::get_dispatcher<T>::impl(*this, idx);
    }

private:
    int iArr[10];
    char cArr[10];

    template<typename T>
    struct get_dispatcher;
};

template<>
struct Foo::get_dispatcher<int> {
    static int impl(Foo const& foo, std::size_t const idx) {
        return foo.iArr[idx];
    }
};

template<>
struct Foo::get_dispatcher<char> {
    static char impl(Foo const& foo, std::size_t const idx) {
        return foo.cArr[idx];
    }
};

使用 int 或 char 以外的任何类型调用 Foo::get<> 将产生编译器错误。

Answer 2

您需要添加某种类型的值结构，您可以使用它来获取 switch 语句的值。例如：

template<typename T>
struct switch_value {};

template<>
struct switch_value<int>
{
    enum { value = 1 };
};

template<>
struct switch_value<char>
{
    enum { value = 2 };
};

//then inside you structure Foo
template <typename T>
T get( int idx )
{
    switch ( switch_value<T>::value )
    {
    case 1:
        return iArr[ idx ];
    case 2:
        return cArr[ idx ];
    }
}

这里的好处是，如果您使用没有有效值的类型，则应该抛出编译器错误，因为 switch_value 的默认版本没有定义成员 value，因此如果您没有专门针对特定类型的结构，那么这样的模板实例化将会失败。

Answer 3

所有这些答案都太过分了。

template <typename T>
T get( int idx )
{
   if ( boost::is_same<T, int>::value)
      return *(T*)&iArr[ idx ];
   else
      return *(T*)&cArr[ idx ];
}

Answer 4

您可以专门化成员函数：

struct foo
{
    int  iArr[10];
    char cArr[10];

    template<typename T>
    T &get(int ipos) { assert( false && "Foo.get: Invalid type!" ); return T(); }

    template<>
    int &get<int>(int ipos) { return iArr[ipos]; }

    template<> 
    char &get<char>(int ipos) {return cArr[ipos]; }

    // add more specialisations for any other types...
};

Works with msvc++ 2010。希望这有帮助。

Jason 建议的 switch 专业化也应该可以正常工作。

Answer 5

对于您的示例来说，这可能有点过分了，但如果您确实一次只需要存储一个数组，那么您可以使用 boost::variant 类和访问者，例如，

#include <boost/variant.hpp>
#include <iostream>

template< typename T >
class GetVisitor : public boost::static_visitor<T>
{
  public:
    GetVisitor(int index) : index_(index) {};

    template <typename U >
    T operator() (U const& vOperand) const
    {
        return vOperand[index_];
    }

  private:
    int index_;
};


int main ()
{
    int  iArr[10] = { 1, 2, 3, 4, 5, 6, 7, 8, 9, 10 };
    char cArr[10] = { 'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j' };

    boost::variant<int*, char*>  intVariant(iArr);   //- assign integer array to variant
    boost::variant<int*, char*>  charVariant(cArr);  //- assign character array to another variant

    int  testInt  = boost::apply_visitor(GetVisitor<int>(2),  intVariant);  
    char testChar = boost::apply_visitor(GetVisitor<char>(9), charVariant);

    std::cout << "returned integer is "   << testInt  << std::endl;
    std::cout << "returned character is " << testChar << std::endl;

    return 0;
}

output is:   
returned integer is 3   
returned character is j   

GetVisitor 类隐含的对变体的限制是变体的所有成员都必须实现：

T operator[](int)

因此您还可以添加 std::vector 和 std::deque 作为变体的潜在成员。

Answer 6

我认为除了只关注模板函数之外，这就是您想要的：

在 .h 文件中，

template < typename T >
struct Foo
{
    T arr[10];

    T get( int idx )
    {
      return arr[ idx ];
    }
};

您可以像这样使用它：

Foo<int> foo1;
Foo<char> foo2;
int i  = foo1.get( 2 );
char c = foo2.get( 4 );