枚举变量作为动态模板参数

时间:2012-04-02 22:04:25

标签: c++ templates template-meta-programming

我们假设我们有一个枚举类型:

enum DataType { INT, DOUBLE };

一个类型映射器:

template<DataType T>
struct TypeTraits {}; 

template<>
struct TypeTraits<INT> { typedef int T; };

template<>
struct TypeTraits<DOUBLE> { typedef double T; };

代表操作的一些模板(不要担心丑陋的void指针和类似C的类型转换):

struct Operation {
  DataType rettype;

  Operation(DataType rettype) : rettype(rettype);
  virtual void* compute();
};

template<DataType RetType>
class Constant : public Operation {
  typedef typename TypeTraits<RetType>::T RType;
  RType val;

  Constant(RType val) : val(val), Operation(RetType) {}; 
  virtual void* compute(){ return &val; }
};  

template<DataType T1, DataType T2, DataType RetType>
class Add : public Operation {
  typedef typename TypeTraits<RetType>::T1 T1Type;
  typedef typename TypeTraits<RetType>::T2 T2Type;
  typedef typename TypeTraits<RetType>::RetType RType;
  RType val;

  Operation *c1, *c2;

  Add(Operation *c1, Operation *c2) : c1(c1), c2(c2), Operation(RetType) {}; 

  virtual void* compute(){
    T1Type *a = (T1Type *)c1->compute();
    T2Type *b = (T2Type *)c2->compute();
    val = *a + *b; 
    return &val;
  }   
};  

抽象树表示:

class AbstractNode {
  enum Type { ADD, INT_CONSTANT, DOUBLE_CONSTANT };

  Type type;
  int intval;
  double doubleval;
  child1 *AbstractNode;
  child2 *AbstractNode;
}

我们正在从输入中读取序列化的抽象树,以便将其转换为操作树,然后 - 计算结果。

我们想写一些类似的东西:

algebrator(Operation *op){
  if(op->type == AbstractNode::INT_CONSTANT)
    return new Constant<INT>(op->intval);
  else if(op->type == AbstractNode::DOUBLE_CONSTANT)
    return new Constant<DOUBLE>(op->doubleval);
  else {
    Operation *c1 = algebrator(op->child1),
              *c2 = algebrator(op->child2);
    DataType rettype = add_types_resolver(c1->rettype, c2->rettype);
    return new Add<c1->rettype, c2->rettype, rettype>(c1, c2);
  }
}

其中add_types_resolver是指定基于操作参数类型的添加操作的返回类型的东西。

我们当然失败了,编译器会让我们陷入困境。我们不能将变量用作模板变量!这是因为实例化模板所需的所有信息必须在赞美期间可用!


现在 - 问题。

除了编写大量if-else或switch-case语句之外,还有其他解决方案吗?我们不能要求编译器以任何方式在编译期间扩展所有情况吗?模板由枚举参数化,因此我们有一个保证,即这样的过程是有限的。

请 - 不要写回答“我认为整个例子搞砸了”。我只是想知道是否有办法用变量提供模板,知道它来自有限的小集合。

整个事情可能看起来像是一种矫枉过正,但我​​真的很好奇如何在这种不寻常的情况下实例化类。

2 个答案:

答案 0 :(得分:2)

Quick'n'dirty宏解决方案:

column_type.cc:

enum ColumnType {
  INT = 1,
  DOUBLE = 2,
  BOOL = 3
};

typed_call_test.cc(用法示例):

#include <iostream>
#include "column_type.cc"
#include "typed_call.cc"

template <ColumnType T>
void PrintType() {
  ::std::cout << T <<::std::endl;
}

int main() {
  ColumnType type = INT;
  // this won't compile:
  // PrintType<type>();                  
  // and instead of writing this:
  switch (type) {                  
    case INT:                  
      PrintType<INT>();                  
      break;                  
    case DOUBLE:                  
      PrintType<DOUBLE>();                  
      break;                  
    case BOOL:                  
      PrintType<BOOL>();                  
      break;                  
  }                  
  // now you can write this:
  TYPED_CALL(PrintType, type, );

  return 0;
}

typed_call.cc(“library”):

// Requirements:
// |function| return type must be void
//
// Usage:
//
// having for instance such |type| variable:
//   ColumnType type = INT;
// and such |foo| function definition:
//   template <ColumnType T>
//   void foo(t1 arg1, t2 arg2) {
//     …
//   }
//
// instead of writing (won't compile):
//   foo<type>(arg1, arg2);                  
// write this:
//   TYPED_CALL(foo, type, arg1, arg2);
//
//
// for foo with 0 arguments write this:
//   TYPED_CALL(foo, type, );
//
#define TYPED_CALL(function, type, args...) {                        \
  switch (type) {                                                    \
    case INT:                                                        \
      function<INT>(args);                                           \
      break;                                                         \
    case DOUBLE:                                                     \
      function<DOUBLE>(args);                                        \
      break;                                                         \
    case BOOL:                                                       \
      function<BOOL>(args);                                          \
      break;                                                         \
  }                                                                  \
}

#define BASE_TYPED_CALL(function, type, args...) {                   \
  switch (type) {                                                    \
    case INT:                                                        \
      function<int>(args);                                           \
      break;                                                         \
    case DOUBLE:                                                     \
      function<double>(args);                                        \
      break;                                                         \
    case BOOL:                                                       \
      function<bool>(args);                                          \
      break;                                                         \
  }                                                                  \
}

要使此解决方案“升级”,您可以使用函数替换宏(仍然包含类似的开关构造)。但是,您可能希望将函数(带有()运算符的对象)作为此函数的参数传递,而不是像此宏中的普通函数。顺便说一下:这正是他们在谷歌中的表现。


第一个旁注:嗨,我的同学来自华沙大学的Columnar和分布式数据仓库课程!本课程产生了许多令人费解的C ++模板问题:)

第二个旁注:这是我的Typetraits的等价物:

template <ColumnType T>
struct EnumToBuiltin {
};

template <>
struct EnumToBuiltin<INT> {
  typedef int type;
};
template <>
struct EnumToBuiltin<DOUBLE> {
  typedef double type;
};
template <>
struct EnumToBuiltin<BOOL> {
  typedef bool type;
};


template <typename T>
struct BuiltinToEnum {
};

template <>
struct BuiltinToEnum<int> {
  static const ColumnType type = INT;
};
template <>
struct BuiltinToEnum<double> {
  static const ColumnType type = DOUBLE;
};
template <>
struct BuiltinToEnum<bool> {
  static const ColumnType type = BOOL;
};

答案 1 :(得分:0)

  

在编译期间,我们不能要求编译器以任何方式扩展所有情况吗?

你可以这样做。介绍条件 - &gt; TMP,它是动态的。如果你在正确的地方做,那么你将有很少的条件写。将类型和常量引入为编译时信息将有助于最大限度地减少这种情况。