0

我正在开发一个库，其函数通常采用向量类型（__v4si或 4 个带符号整数的向量）作为参数。（请注意，到目前为止，这与C++ STL 模板类无关vector；这是一个更原始的构造，用于让编译器生成矢量化SIMD 代码。）

在我的C代码中，我通常调用一个包装宏，它接受一个 int 参数列表并__v4si像这样初始化：

#define MakeIndex(dims...) ((__v4si){ dims })

这当然在C++中也可以正常工作，但我想利用C++更具表现力的类型系统来清理对我的库 API 的调用。例如，我现在写如下内容：

long idx = IndexDotProduct(MakeIndex(1, 2, 3), MakeIndex(4, 5, 6, 7));

其中宏扩展为：

long idx = IndexDotProduct(((__v4si){1, 2, 3}), ((__v4si){4, 5, 6, 7}));

相反，我希望能够写一些类似的东西：

long idx = IndexDotProduct({1, 2, 3}, {4, 5, 6, 7});

所以，本质上（我认为）我想定义一个类，它只是原始__v4si类型周围的语法糖，但它有一个用于列表初始值设定项的隐式强制转换运算符。

我如何在C++ 11中做到这一点？

解决方案

这是一个适用于C和C++代码的公式（现在使用从我的库头文件复制和粘贴的更详细的名称）：

typedef struct vMAT_Index {
    __v4si v;
#ifdef __cplusplus
    vMAT_Index(__v4si v) : v(v) { }
    vMAT_Index(int v0 = 0, int v1 = 0, int v2 = 0, int v3 = 0) : v((__v4si){ v0, v1, v2, v3 }) { }
#endif
} vMAT_Index;

#define vMAT_MakeIndex(dims...) ((vMAT_Index){ .v = { dims } })

static inline long
vMAT_Index_dot(vMAT_Index a,
               vMAT_Index b)
{
    __v4si c = a.v * b.v;
    return (long)c[0] + c[1] + c[2] + c[3];
}

在C代码中，您仍然可以像这样使用辅助宏：

long idx = vMAT_Index_dot(vMAT_MakeIndex(1, 2, 3), vMAT_MakeIndex(4, 5, 6, 7));

但是现在在C++中你可以写：

long idx = vMAT_Index_dot({ 1, 2, 3 }, { 4, 5, 6, 7 });

感谢 nosid 提供了重要的答案！

4

1 回答 1

1

使用隐式构造函数从大括号初始化器列表中自动创建向量对象：

struct vector
{
    vector(__v4si v);
    vector(int i0, int i1, int i2, int i3);
};

long IndexDotProduct(vector lhs, vector rhs);

long idx = IndexDotProduct(((__v4si){ 1, 2, 3 }), { 4, 5, 6, 7 });
于 2013-04-07T13:09:16.810 回答