在宏输出中转义逗号

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));产生一个额外的括号,因此会有嵌套的括号.如果我不在外面放一个支架,我会收到错误错误:

意外的标记: ,

那么可以,在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)来解决它.