OpenCL仅提供使用C99规范访问单维数组的权限。然而我的问题是二维的,我在主机端使用二维数组
我不想通过计算索引使我的代码可读性降低,而是想使用C宏来获取元素A[i][j]
。不幸的是,我在这方面做得很差,而且对C的经验也不多。我想我对如何做到这一点有了一般的想法,但如果有人可以批评,我将不胜感激。
这将是:
#define 2d_access(u, y, x) (u[y][x])
其中u是矩阵,y是行,x是列,宏将返回
价值u[y][x]
矩阵是静态分配的,因此宏将具有WIDTH组件。
#define 2d_access(u, y, x) (u[y * WIDTH] + x])
答案 0 :(得分:7)
由于到目前为止所有答案都依赖于恒定宽度,这里是任意宽度列的(重量级)解决方案:
#define matrix_type(t) struct { size_t width; t array[]; }
#define matrix_alloc(t, w, h) malloc(offsetof(matrix_type(t), array[(w) * (h)]))
#define matrix_init(m, t, w, h) \
matrix_type(t) *m = matrix_alloc(t, w, h); \
if(!m) matrix_alloc_error(); else m->width = (w);
#define matrix_index(m, w, h) m->array[m->width * (w) + (h)]
// redefine if you want to handle malloc errors
#define matrix_alloc_error()
只需使用free
释放数组。
当然,你也可以添加一个高度字段,并进行边界检查等。您甚至可以将它们写为实际函数,或使用宏自动声明struct
类型,这样您就不必为所有内容使用匿名struct
类型。如果您在堆栈中需要它,则可以使用alloca
,但代价是可移植性。
如果您有一个恒定的矩阵大小,您可以使用一些投射黑客来实现“原生”2D索引(通过[]
运算符):
#define CAT_(x, y) x##y
#define CAT(x, y) CAT_(x, y)
#define MANGLE(x) CAT(x, _hidden_do_not_use_0xdeadbeef_)
#define matrix_init(m, t, w, h) \
t MANGLE(m)[(w) * (h)]; \
t (*m)[(w)] = (void *)MANGLE(m);
// because of the funky typing, `m[0][1]` does what you'd expect it to.
请注意,与其他解决方案不同,这会创建第二个变量,这可能不是很干净,但我认为我使用了非常清晰的修改方法,所以它不会妨碍实践。
答案 1 :(得分:6)
更干净的是为您正在使用的每个阵列定义一个宏,因此您可以使其看起来与2D阵列访问完全相同。因此,给定数组A
,您将定义:
#define A(r, c) (A[(r)*WIDTH + (c)])
请注意替换值周围的括号。这处理替换是表达式的情况,如A(i + 1, j)
。没有括号,这将扩展到A[i + 1*WIDTH + j]
,这不是你想要的:
A[i + 1*WIDTH + j] = A[WIDTH + i + j] != A[(i + 1)*WIDTH + j]
为了避免第二个参数出现同样的问题,两者都包含在替换文本的括号中。
答案 2 :(得分:2)
没有批评,你已经给出了解决方案:
#define access_2d(u, y, x) (u[(y) * WIDTH + (x)])
好吧,也许我的想法不同,但我会将其定义为
// x before y
#define access_2d(u, x, y) (u[(y) * WIDTH + (x)])
但这不是更好,只是一种偏好。