在宏输出中转义逗号

Escaping commas in macro output

我正在尝试编写一个宏,使我能够转换 (a, b, c, d)(a, a + b, a + b + c, a + b + c + d),等等。这是我目前得到的:

macro_rules! pascal_next {
    ($x: expr) => ($x);
    ($x: expr, $y: expr) => (
        ($x, $x + $y)
    );
    ($x: expr, $y: expr, $($rest: expr),+) => (
        ($x, pascal_next!(
                $x + $y, $($rest),+
            )
        )
    );
}

但是有一个问题,它实际上会输出(a, (a + b, (a + b + c, a + b + c +d)))。由来是第二个匹配规则($x: expr, $y: expr) => (($x, $x + $y));,多了一个括号,这样就会嵌套括号。如果我不在外面放一个括号,我会得到错误 error:

unexpected token: ,

那么可以在 Rust 宏中输出一个逗号 , 吗?

没有;宏 的结果必须 是一个完整的语法结构,如表达式或项目。你绝对不能有随机的语法位,比如逗号或右大括号。

你可以通过简单地不输出任何东西来解决这个问题,直到你有一个完整的、最终的表达式。看哪!

#![feature(trace_macros)]

macro_rules! pascal_impl {
    /*
    The input to this macro takes the following form:

    ```ignore
    (
        // The current output accumulator.
        ($($out:tt)*);

        // The current additive prefix.
        $prefix:expr;

        // The remaining, comma-terminated elements.
        ...
    )
    ```
    */

    /*
    Termination condition: there is no input left.  As
    such, dump the output.
    */
    (
        $out:expr;
        $_prefix:expr;
    ) => {
        $out
    };

    /*
    Otherwise, we have more to scrape!
    */
    (
        ($($out:tt)*);
        $prefix:expr;
        $e:expr, $($rest:tt)*
    ) => {
        pascal_impl!(
            ($($out)* $prefix+$e,);
            $prefix+$e;
            $($rest)*
        )
    };
}

macro_rules! pascal {
    ($($es:expr),+) => { pascal_impl!((); 0; $($es),+,) };
}

trace_macros!(true);

fn main() {
    println!("{:?}", pascal!(1, 2, 3, 4));
}

注意:要在稳定的编译器上使用它,您需要删除 #![feature(trace_macros)]trace_macros!(true); 行。其他都应该没问题。

它的作用是递归地处理输入,将部分(可能语义无效)输出作为输入传递给递归的下一层。这让我们可以建立一个 "open list",否则我们无法做到。

然后,一旦我们没有输入,我们只需将我们的部分输出重新解释为一个完整的表达式,然后...完成。

我包含跟踪内容的原因是我可以向您展示它运行时的样子:

pascal! { 1 , 2 , 3 , 4 }
pascal_impl! { (  ) ; 0 ; 1 , 2 , 3 , 4 , }
pascal_impl! { ( 0 + 1 , ) ; 0 + 1 ; 2 , 3 , 4 , }
pascal_impl! { ( 0 + 1 , 0 + 1 + 2 , ) ; 0 + 1 + 2 ; 3 , 4 , }
pascal_impl! { ( 0 + 1 , 0 + 1 + 2 , 0 + 1 + 2 + 3 , ) ; 0 + 1 + 2 + 3 ; 4 , }
pascal_impl! { ( 0 + 1 , 0 + 1 + 2 , 0 + 1 + 2 + 3 , 0 + 1 + 2 + 3 + 4 , ) ; 0 + 1 + 2 + 3 + 4 ; }

输出为:

(1, 3, 6, 10)

需要注意的一件事:大量未注释的整数文字会导致编译时间急剧增加。如果发生这种情况,您可以通过简单地注释 all 整数文字(如 1i32)来解决它。