N 维数组赋值的 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;