N-D 数组赋值的 C 宏重载

Question

已经有很多关于宏重载的问题/答案。但是，我找不到将其应用于我的特定问题的方法。

我想在 C 中方便地为我的 3D 图像赋值。 现在，我执行以下操作：

#define IMGET(im,y,x,c) im.data[(y)+im.height*((x)+im.width*(c))]
#define IMSET(im,y,x,c,v) IMGET(im,y,x,c)=v

效果很好。

但是，当我有只有 2D 的黑白图像时，我也想使用它。 类似的东西：

#define IMGET(im,y,x,c) im.data[(y)+im.height*((x)+im.width*(c))] //3D case
#define IMSET(im,y,x,c,v) IMGET(im,y,x,c)=v //3D case
#define IMGET(im,y,x) im.data[(y)+im.height*(x)] //2D case
#define IMSET(im,y,x,v) IMGET(im,y,x)=v //2D case

有可能吗？

Answer 1

@Potatoswatter 的评论给出了一个非常好的通用解决方案。但这对于我的简单问题可能有点过头了。

看了之后，找到了如下解决方案：

#define IMGET(im,y,x,...) im.data[(y)+im.height*((x)+im.width*(__VA_ARGS__+0))]

然后，我可以将这个宏用于以下所有情况：

float val = IMGET(im,y,x,c);

或：

float val = IMGET(im,y,x,c);

如果我想分配，我会这样做：

IMGET(im,y,x,c)=val;

或

IMGET(im,y,x)=val;