动态分配指针联合 - C.

Question

我正在用C编写一个用于低级图像处理的基本库.我知道其他(非常好的)库存在; 这对我来说是一次学习经历,而不是达到目的的手段.

我已经定义了以下(简化,针对此问题)构造:

union img_rawdata{
    uint8_t*  u8;
    uint16_t* u16;
    float*    flt;
};

enum img_type{
    IMG_U8,
    IMG_U16,
    IMG_FLT
};

struct image{
    union img_rawdata  rawdata;
    enum img_type      type;
    unsigned int       num_values;
};

我的问题是:在联合中动态分配正确的指针有什么好处？

现在,我看到的唯一方法是使用switch语句,如:

void alloc_img(struct image* img, enum img_type type, unsigned int num_vals){
    switch (type){
        case IMG_U8:
            img->rawdata.u8 = (uint8_t*)malloc(num_vals*sizeof(uint8_t));
        break;            
        case IMG_U16:
            img->rawdata.u16 = (uint16_t*)malloc(num_vals*sizeof(uint16_t));
        break;
        case IMG_FLT:
            img->rawdata.flt = (float*)malloc(num_vals*sizeof(float));
        break;
    }
}

这似乎并不那么糟糕; 但是,在我的实现中,实际的内存分配大约是50行(因为rawdata不是一维的,错误检查等).

是否有任何可以减少代码冗余的预处理器魔法,或者这是编写此代码的最佳方式？

或者,或者,是否有一种不同的方式来解决完全避免这个问题的问题？

Answer 1

[假设所有类型的指针void *都包含相同的大小]

修改你喜欢的内容

union img_rawdata {
  void * untyped;
  uint8_t * u8;
  uint16_t * u16;
  float * flt;
};

enum img_type {
  IMG_UNDEF = -1
  IMG_U8 = 0,
  IMG_U16,
  IMG_FLT,
  IMG_MAX
};

并添加

const size_t img_type_size[IMG_MAX] = {
  sizeof(*((union img_rawdata *) NULL)->u8),
  sizeof(*((union img_rawdata *) NULL)->u16),
  sizeof(*((union img_rawdata *) NULL)->flt)
};

然后将开关更换为:

assert(IMG_UNDEF < type && IMG_MAX > type);
img->rawdata.untyped = malloc(num_vals * img_type_size[type]);