将函数作为参数传递以避免重复代码

时间:2018-10-02 19:26:22

标签: c++ c++11 c++14 dry overloading

到目前为止,我一直都使用C ++ 11进行编码,并且我试图了解auto在较新版本中的工作方式。特别是,我有两个函数(在示例中为f1和f2)在给定的结构上工作。这两个函数几乎相同,但是它们在操作的成员上有所变化,有时操作是一个值的倒数,等等...(此示例是对实际程序的过度简化)。

我想避免使用条件和重载函数来实现此行为。您知道更清洁或更惯用的方法吗?这段代码中有我遗漏的任何问题吗?

typedef struct thing_t {
    double A;
    double B;
} THING;    

double get_A(const THING &t) {
    return t.A;
}

double get_B(const THING &t) {
    return t.B;
}

double convert(const THING &t, auto first, auto then) {
    return first(t) / then(t);
}

double f1(const THING &t) {
    return convert(t, get_A, get_B); 
}

double f2(const THING &t) {
    return convert(t, get_B, get_A);
}

int main() {
    THING t = {1.0, 2.0};
    std::cout << f1(t) << std::endl;
    std::cout << f2(t) << std::endl;
    return 0;
}

非常感谢您抽出宝贵的时间来审核我的问题。

1 个答案:

答案 0 :(得分:6)

首先,您尚不能使用auto函数参数。那是非标准的C ++。 typedef struct这件事也是C-ism。在C ++中,只需:

struct thing_t {
    double A;
    double B;
};

现在让我们谈谈概括。 convert是否需要了解其论点?也许它本身就是一个高阶函数:

template <typename F, typename G>
auto convert(F f, G g) {
    return [=](auto const& x) { return f(x) / g(x); }
}

然后get_Aget_B返回成员。我们已经有了相应的语法:指向成员数据的指针(不幸的是,它们不能直接调用,因此您需要std::mem_fn):

double f1(const thing_t& t) {
    return convert(std::mem_fn(&thing_t::A), std::mem_fn(&thing_t::B))(t);
}

C ++ 17引入了std::invoke,因此您可以使此处的实用程序功能更加用户友好。它可以在C ++ 14中实现,但可以让您编写:

template <typename F, typename G>
auto convert(F f, G g) {
    return [=](auto const& x) { return std::invoke(f, x) / std::invoke(g, x); };
}

double f1(const thing_t& t) {
    return convert(&thing_t::A, &thing_t::B)(t);
}

double f2(const thing_t& t) {
    return convert(&thing_t::B, &thing_t::A)(t);
}

您怎么看?