是否可以创建一个函数而不是两个具有相同目的但参数类型不同的函数？ （我可以删除重复的代码吗？）[重复]

Question

是否可以创建一个函数而不是两个具有相同目的但参数类型不同的函数？

我有两个用 C 编写的函数，它们将图像从 RGB 转换为 HSV：

void png_rgb2hsv(pPNG_DATA data);
void jpg_rgb2hsv(pJPEG_DATA data);

他们做同样的事情：他们接受 data->row_pointers 并在循环 data->height 次中循环它。然后它转换 data->row_pointers 引用的值。这就是它的工作原理。但唯一的区别是数据结构使用不同的类型。在我看来，为同一件事使用两个功能是没有意义的。尤其是当我要为更多色彩空间添加更多功能时。

这个程序设计问题在 C 中是如何在实践中解决的？

更新： 大多数读者不明白这个问题。我没有问过超载。这是关于设计的问题。我问是否可以删除冗余功能，“冗余”代码。因为两个函数使用相同的代码，所以做同样的事情，但是函数参数中的类型不同，因为一种类型来自 libjpeg，第二种来自 libpng。我发现这是不可能的，因为这意味着将一个变量用于两种不同的类型。

Answer 1

使用通用宏：

void png_rgb2hsv(pPNG_DATA data);
void jpg_rgb2hsv(pJPEG_DATA data);

#define rgb2hsv(X) _Generic((X), pPNG_DATA: png_rgb2hsv, pJPEG_DATA: jpg_rgb2hsv)(X)

（如果你的编译器太旧并且不支持这个技巧（或者你因为某种原因不想使用它），那么看起来没有办法让两个函数具有相同的名称和不同的参数类型。然后，您需要选择其他解决方案。）

更新：

我误解了 OP 的问题。 如果你想为这两种类型创建一个 single 函数，你应该这样做：

void rgb2hsv_(Something *row_pointers, int height) // any member that you need goes here
{
    /* your code */
}

#define rgb2hsv(obj) \
do { \
_Generic((obj), pPNG_DATA: pPNG_DATA, pJPEG_DATA: pJPEG_DATA) tmp = (obj); \
    rgb2hsv_(tmp->row_pointers, tmp->height /*again, every member that you need should be stated here*/) \
} while (0)

另外，如果pJPEG_DATA 和pPNG_DATA 具有完全相同的内部布局（这意味着它们的成员具有相同的类型并且它们以相同的顺序列出），你可以试试这个：（它不安全和上一个一样，但至少它看起来不像是一个糟糕的黑客）

void rgb2hsv(void *ptr)
{
    pPNG_DATA data = (pPNG_DATA *) ptr; // it does not matter which of 2 types you use here
    /* put your code that uses `data` here */
}

但请记住，如果您在这些结构中的任何一个中交换 2 个成员或以任何方式更改它们的内部结构，那么这件事可能不再起作用。

（另外，你应该知道：这两种技术只是获得预期结果的棘手变通方法。最好的方法是将每个需要的成员作为单独的参数传递，不要做这个宏功夫）强>

Answer 2

主要有两种不同的方法来处理这个问题。您创建一个包含图像的结构以及数据的标签，如下所示：

struct img{
    char *filetype;
    void *data;
};

然后你创建一个函数来检查文件类型并调用你想要的任何函数。

或者你创建一个带有函数指针的结构体，指向你想要使用的任何函数，如下所示：

struct imgfuns{
    void (*rgb2hsv)(void *);
};

struct imgfuns *init_struct(void)
{
    struct imgfuns *funs = malloc(sizeof(*funs));
    if(funs == NULL)
        return NULL;
    if(png...)
        funs->rgb2hsv = png_rgb2hsv;
    else if(jpg...)
        funs->rgb2hsv = jpg_rgb2hsv;
    return funs;
}

int main(int argc, char *argv[])
{
    struct imgfuns *funs;
    funs = init_struct();
    if(funs == NULL)
      exit(1);
    /* get data from somewhere */
    funs->rgb2hsv(data);
    free(funs);
    return 0;
}

但是，如果您需要使用许多不同的函数，因为您需要为每个函数映射它们，这可能会很痛苦。但是您将获得更简洁的代码，并且您可以在一个地方处理所有格式，而不是为每种文件格式创建包装函数。

更多信息可以在这里找到：How do function pointers in C work?

Answer 3

Void 指针与指向任何对象的指针兼容。您可以使用void* 并添加“类型”参数。

void data2hsv(void *data, int datatype) {
    if (datatype == 0) {
        /* use png */
    } else {
        /* use jpg */
    }
}

编辑：对编译器撒谎

void data2hsv(void *data) {
    pPNG_DATA source = data; // if data is of pJPG_DATA type 
                             // compiler will not catch the error
    /* use source as if it was pPNG_DATA */
}

Answer 4

这在 C 中是不可能的，但在许多其他编程语言中，例如 C++，这是可能的。

但是，在 C 语言中存在一些解决该限制的技巧，请参阅this question 了解更多信息。

C 中的一种常见做法是将类型放在函数名称中，如您的示例所示。

Answer 5

以下使用一个附加参数来告知函数传递的是什么类型的图像数据：

void rgb2hsv(void *data, int data_type)
{
    char *row, *pix;
    switch (data_type) {
        case JPG: row= (pJPEG_DATA data)->row; break;
        case PNG: row= (pPNG_DATA  data)->row; break;
    }
    ....

Answer 6

您不能在 C 中重载函数，但您可以组织代码以使其达到最佳状态。我将为rgb2hsv 创建一个函数，它将获得void * 类型的参数，确定类型并使用它，处理所有可能的情况。你应该只使用这个功能。在较低级别上，您仍然可以复制您的函数，在较高级别上，您不必调用两个单独的函数。这将简化使用。