macros - 在宏输出中转义逗号

Question

我正在尝试编写一个使我能够转换的宏(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));，多出一个括号，这样就有将嵌套括号。如果我不在外面放一个括号，我会得到错误错误:

unexpected token: ,

那么是否可以在 Rust 宏中输出一个逗号 , 呢？

Answer 1

没有；宏的结果必须是一个完整的语法结构，如表达式或项。您绝对不能有随机的语法位，例如逗号或右大括号。

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

#![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) ; 行。其他一切都应该没问题。

它所做的是递归地处理输入，将部分(并且可能语义无效)输出作为输入传递到下一级递归。这让我们可以建立一个“开放列表”，否则我们无法做到这一点。

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

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

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)

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