Julia 创建一个函数的多个稍微修改过的版本

Question

我有一个看起来像的功能

function eom!(du, u, p)
    @views a, b = u[:,1], u[:,2];
    @views da, db = du[:,1], du[:,2];

    y = # some stuff involving p and a;
    da .= f(a, b, y);
    db .= g(b, a);
end

我现在想创建第二个完全相同的函数，除了最后一行读取

db .= g(b, y);

我怎样才能最干净地做到这一点？当然，我可以只复制和粘贴并给函数命名略有不同的名称，但这似乎不理想，特别是如果似乎合理，我以后想要更多函数，其中的第二个参数g可能是别的东西。也许有一种方法可以将eom!一个表达式（通过参数p）传递给函数，该表达式将指定 的第二个参数g？或者有没有办法可以制作一些eom_generator可以输出我想要的所有功能的功能？也许宏是这样做的核心工具，但我不确定。

Answer 1

你可以产生一个闭包：

function eom_generator(g)
    return function eom!(du, u, p)
        @views a, b = u[:,1], u[:,2]
        @views da, db = du[:,1], du[:,2]

        y = nothing # some stuff involving p and a;
        da .= f(a, b, y)
        db .= g(a, b, y)
    end
end

const eom1! = eom_generator((a, b, y) -> g(b, a))
const eom2! = eom_generator((a, b, y) -> g(b, y))

但由于这是微分方程的核心，因此请务必以这种方式测试您是否没有任何性能问题。

如果您确定确实需要元编程，则可以@eval在循环中使用：

for (i, expr) in enumerate((:(g(b, a)), :(g(b, y))))
    @eval function $(Symbol("eom", i, "!"))(du, u, p)
            @views a, b = u[:,1], u[:,2]
            @views da, db = du[:,1], du[:,2]

            y = nothing # some stuff involving p and a;
            da .= f(a, b, y)
            db .= $expr
        end
    end
end