Общие указатели функций в C

У меня есть функция, которая берет блок данных и размер блока и указатель функции в качестве аргумента. Затем он выполняет итерацию по данным и выполняет вычисления для каждого элемента блока данных. Ниже приведена основная информация о том, что я делаю:

int myfunction(int* data, int size, int (*functionAsPointer)(int)){
    //walking through the data and calculating something
    for (int n = 0; n < size; n++){
        data[n] = (*function)(data[n]);
    }
}

Функции, которые я передаю как аргументы, выглядят примерно так:

int mycalculation(int input){
    //doing some math with input
    //...
    return input;
}

Это хорошо работает, но теперь мне нужно передать дополнительную переменную моему элементу. Что-то вдоль линий

int mynewcalculation(int input, int someVariable){
    //e.g.
    input = input * someVariable;
    //...
    return input;
}

Есть ли элегантный способ добиться этого и в то же время сохранить мою общую дизайнерскую идею?

Ответы

Ответ 1

Обычный способ сделать его полностью общим - с void *, что, конечно, вызывает всевозможные проблемы с безопасностью:

int map_function(int* data, int size, int (*f_ptr)(int, void*), void *userdata){
    //walking through the data and calculating something
    for (int n = 0; n < size; n++){
        data[n] = (*f_ptr)(data[n], userdata);
    }
}

int simple_calculation(int input, void *userdata) {
    // ignore userdata and do some math with input
    return input;
}

int calculation_with_single_extra_arg(int input, void *userdata) {
    int input2 = * (int*) userdata;
    // do some math with input and input2
    return input;
}

int calculation_with_many_extra_args(int input, void *userdata) {
    my_data_t data = (my_data_t *) userdata;
    return input * (input * data->a + data->b) + data->c;
}

int main(int argc, char **argv) {
    int array[100];
    my_data_t userdata = { 1, 2, 3 };

    populate(array, 100);

    map_function(data, calculation_with_many_extra_args, &userdata);
}

Любая передаваемая функция должна иметь этот прототип, но вы можете записать любые данные, которые вы хотите, через userdata. Там просто нет никакой проверки типов.

Вы также можете использовать va_args, как предлагает dbingham; это не очень сильно отличается, поскольку прототип вашей функции для отображения должен быть int(int, va_list).

Изменить: Я одобряю подход void*. va_list в любом случае не добавляет каких-либо типов безопасности и добавляет больше возможностей для пользовательской ошибки (в частности, вызов va_end дважды или не реализация петли va_arg вправо). Void * также не добавляет лишних строк кода; он прошел чисто, и пользователь просто разыгрывает его (надеюсь) правильный тип (который, вероятно, является структурой, в общем случае).