如何用Julia元编程优化罗马数字转换的if冗余代码？

问题背景 #

我先用常规Julia代码完成了罗马数字转换练习，但代码存在明显冗余：

function to_roman(number::Int64)
    0 < number < 4000 || error("not in range")

    parts = Char[]
    while number > 0
        if number >= 1000
            push!(parts, 'M')
            number -= 1000
        elseif number >= 500
            push!(parts, 'D')
            number -= 500
        elseif number >= 100
            push!(parts, 'C')
            number -= 100
        elseif number >= 50
            push!(parts, 'L')
            number -= 50
        elseif number >= 10
            push!(parts, 'X')
            number -= 10
        elseif number >= 5
            push!(parts, 'V')
            number -= 5
        elseif number >= 1
            push!(parts, 'I')
            number -= 1
        end
    end
    join(parts)
end

之后我用循环遍历的方式优化了冗余，但想尝试用元编程自动生成上述的if-elseif链作为练手项目，不过尝试失败：

macro gen(number)
    code = quote
        parts = Char[]
        while $(esc(number)) > 0
        end
    end
    for (i,c) in enumerate("MDCLXVI")
        exp = :(
            if $(esc(number)) >= $(values[c])
                push!(parts, $c)
                $(esc(number)) -= $(values[c])
            end
        )
        if i > 1
            exp.head = :elseif
        end
        push!(code.args, exp)
    end
    push!(code.args, :(return parts))
    dump(code)
    return code
end

问题在于无法正确将if-elseif链嵌套进while循环的AST结构中，还受LineNumberNode干扰。

解决方案 #

要正确生成嵌套在while内的if-elseif链，需要直接操作while表达式的主体部分，而不是往顶层代码块里塞表达式。以下是可行的实现：

步骤1：定义有序罗马数字映射 #

先定义从大到小排序的罗马数字与数值的对应关系，保证生成的if链逻辑正确：

const roman_pairs = [
    'M' => 1000,
    'D' => 500,
    'C' => 100,
    'L' => 50,
    'X' => 10,
    'V' => 5,
    'I' => 1,
]

步骤2：编写元编程宏 #

核心是构建完整的if-elseif链，再将其作为while循环的主体：

macro gen_roman(number)
    # 构建if-elseif链的根节点
    if_chain = nothing
    for (c, val) in roman_pairs
        cond = :($(esc(number)) >= $val)
        block = quote
            push!(parts, $c)
            $(esc(number)) -= $val
        end
        current = Expr(:if, cond, block)
        if if_chain === nothing
            if_chain = current
        else
            # 将当前if拼接到上一个if的else分支
            push!(if_chain.args, current)
            current.head = :elseif
        end
    end

    # 构建完整代码块：初始化数组 + while循环 + 返回结果
    full_code = quote
        parts = Char[]
        while $(esc(number)) > 0
            $if_chain
        end
        join(parts)
    end

    return full_code
end

步骤3：测试宏实现转换函数 #

用宏生成最终的转换函数并测试：

function to_roman(number::Int64)
    0 < number < 4000 || error("must be between 1 and 3999 inclusive")
    @gen_roman number
end

# 测试示例
println(to_roman(1994)) # 输出 MCMXCIV
println(to_roman(1111)) # 输出 MCXI

关键要点 #

• AST结构操作：while表达式的格式是Expr(:while, 条件, 主体块)，需把if-elseif链作为第三个参数传入while的Expr中，而非往顶层代码块追加。
• if-elseif链构建：循环逐个拼接节点，第一个节点为:if，后续节点改为:elseif并挂载到前一个if的args中。
• 变量作用域：用esc()处理输入的number变量，避免宏展开时的作用域冲突。

内容的提问来源于stack exchange，提问作者primfaktor