如何将模板参数派生的编译时信息添加到模板中?

时间:2012-11-14 14:35:16

标签: c++ templates c++11

我使用Stroustrup在GoingNative 2012(从23:00开始)提供的用户定义的文字来讨论单元实现。这是代码:

 #include <iostream>

 using std::cout;
 using std::endl;

 template<int M, int K, int S> 
 struct Unit { // a unit in the MKS system
   enum {m=M,kg=K,s=S}; 
 };

 template<typename Unit> // a magnitude with a unit
 struct Value {
   double val;
   constexpr Value(double d) : val(d) {}
 };

 using Meter = Unit<1,0,0>;
 using Second = Unit<0,0,1>;

 using Distance = Value< Meter >;
 using Time = Value< Second >;
 using Velocity = Value< Unit<1,0,-1> >;

 constexpr Value<Meter> operator "" _m(long double d)
 // a f-p literal with suffix 'm'
 {
   return Distance(d);
 }

 constexpr Value<Second> operator"" _s(long double d)
 // a f-p literal with suffix 's'
 {
   return Time(d);
 }

 constexpr Velocity operator/(Distance d, Time t)
 {
   return ( d.val / t.val );
 }

 int main(void)
 {
    Distance s = 100._m;
    Time t = 22._s;
    Velocity v = s/t;

    cout << "s " << s.val << "\nt " << t.val << endl;
    cout << "v " << v.val << endl;

   return 0;
 }

正如您所见,我可以自由地定义operator/来计算速度。输出是(需要gcc-4.7):

$ g++ -std=c++0x test_units_02.cc && ./a.out
s 100
t 22
v 4.54545

到目前为止一切顺利。现在我想将一个包含单位表示的字符串添加到struct Unit(或Value?)。无论我想写什么方式

cout << "v " << v.val << v.unit << endl;

并获得类似

的内容
v 4.54545 m^1 s^-1

v 4.54545 m^1 kg^0 s^-1

它不需要美观,因为它只是用于检查。并学习如何做到这一点;)。

当然,优雅的解决方案是在编译时对所有内容进行评估。

我有一些镜头,但我不会因为我没有结果的尝试而让你感到困惑......

2 个答案:

答案 0 :(得分:6)

首先,我们将unit成员添加到Value

 template<typename Unit> // a magnitude with a unit
 struct Value {
   double val;
   constexpr static Unit unit = {};
   constexpr Value(double d) : val(d) {}
 };

然后我们写一个stream out运算符:

 template<int M, int K, int S>
 std::ostream &operator<<(std::ostream &os, Unit<M, K, S>) {
   return os << "m^" << M << " kg^" << K << " s^" << S;
 }

在编译时生成一个字符串可能,但需要constexpr编译时字符串类(例如boost::mpl::string)和十进制格式 - 所有这些都是可行的但是在这种情况下并不特别值得。

答案 1 :(得分:1)

Unit已经拥有了我们需要的信息,因此我们可以通过向Value添加函数来实现此目的,或者我们可以重载operator<<

template<typename U>
struct Value
{
    double val;
    constexpr Value(double d) : val(d) {}

    std::string Units() const
    {
        return "m^" + to_string(U::m) + 
               " kg^" + to_string(U::kg) +
               " s^" + to_string(U::s);
    }
};

template <typename U>
std::ostream& operator<<(std::ostream& out, Value<U> val)
{
    out << val.val
        << " m^" << U::m << " kg^" << U::kg << " s^" << U::s;
    return out;
}

我们还可以提供通用operator/

template <typename U1, typename U2>
Value<Unit<U1::m - U2::m, U1::kg - U2::kg, U1::s - U2::s>>
operator/(Value<U1> v1, Value<U2> v2)
{
    return (v1.val / v2.val);
}

这为我们提供了更大的灵活性:

void demo()
{
    auto accel = Distance(100) / Time(22) / Time(1);
    cout << accel << endl; // Print with units.
    cout << accel.val << endl; // Print without units.
    cout << accel.Units() << endl; // Print only units.
}