在 Julia 宏中用字符串替换变量名称

Question

我正在开发一个项目，需要在 Julia 中创建一个宏，该宏接受 Julia 代码表达式、变量名和字符串作为输入，并返回相同的表达式，其中指定的变量名替换为给定的字符串。宏尊重 Julia 中的变量作用域非常重要。

例如，如果我有以下表达式：

x = 42
expr = :(2 * x + 3)

我用变量名 x 和替换字符串“y”来调用宏，如下所示：

@my_macro expr x "y"

我想要得到修改后的表达式：

:(2 * y + 3)

我一直在尝试实现这个宏，但我在处理变量范围和确保仅对指定变量进行替换时遇到了麻烦。

对于正确实施此宏的任何指导或帮助，我将不胜感激。如果有人对如何在 Julia 中实现这一目标有任何见解、代码片段或建议，请与我分享。

Answer 1

您可以定义以下函数：

using AbstractTrees
function replace!(expr::Expr, old_val::Symbol, new_val::Symbol) 
    for e in collect(PreOrderDFS(expr))
        if typeof(e) <: Expr
            ix = findfirst(==(old_val), e.args)
            if !isnothing(ix) 
                e.args[ix] = new_val
            end
        end
    end
    expr
end

现在可以使用此功能，例如：

julia> replace!(quote; f(x,z) = 2x+4z; end, :x, :y)
quote
    #= REPL[61]:1 =#
    f(y, z) = begin
            #= REPL[61]:1 =#
            2y + 4z
        end
end

此驾驶室可在宏中使用，例如：

macro replace(mycode, oldval, newval)
    replace!(mycode, oldval.value, newval.value)
end

可用作：

julia> x=1; y=100;

julia> @replace(3x+y, :x, :y)
400