如何使用RTTI来确定代码的行为

时间:2013-08-24 01:17:15

标签: c++ stl rtti

我有一段代码将RTTI用作案例选择器

#include <iostream>
#include <vector>
#include <typeinfo>
#include <complex>
#include <algorithm>

using namespace std;

typedef complex<int> data;
typedef std::vector< data > List;

template <typename T> void zeros(T &v)
{
  if (typeid(typename T::value_type)==typeid(complex<int>))
    std::fill(v.begin(), v.end(), complex<int>(0, 0));
  else std::fill(v.begin(), v.end(), static_cast<typename T::value_type>(0));
}

int main(int argc, char *argv[])
{
  List A;

  zeros(A);
  return 0;
}

在上面的代码中,我使用typename T :: value_type来确定列表的数据类型,如果有复杂的话,我将用复数零填充列表。否则,将其设为相应类型的零。在上面的代码中,如果我将数据设置为int,它会编译并运行没有任何问题。但是,如果我将其设置为另一种类型,则无法编译。我想知道是否有任何方法可以使上面的代码适用于不同类型的数据(对于float,double,long,int,complex,complex和complex)。

P.S。我想在两种情况下填充向量的原因是因为我要出于某种原因创建自己的复合体,我必须告诉代码如何创建复零或零。但是对于测试,我甚至无法使用内置数据类型

1 个答案:

答案 0 :(得分:4)

似乎你想要一个默认的初始化复合体(0,0)和一个零初始化的int。您可以使用值初始化来实现此目的:

template <typename T> void zeros(T &v)
{
    std::fill(v.begin(), v.end(), typename T::value_type{});
}

如果你真的想根据什么是T::value_type来执行一些分支,那么最好使用一些类型特征,如下所示:

template<typename T>
struct default_value {
    static const T value;
};

template<>
struct default_value<std::complex<int>> { // assuming you'll use other complex type
    static const std::complex<int> value;
};

template<typename T>
const T default_value<T>::value = T{};
const std::complex<int> default_value<std::complex<int>>::value = std::complex<int>(0, 0);

// now use it
template <typename T> void zeros(T &v)
{
    std::fill(v.begin(), v.end(), default_value<typename T::value_type>::value);
}

但同样,内置类型的默认构造和值初始化应该足够了。