在宏输出中转义逗号

时间:2015-07-02 22:14:49

标签: macros rust

我正在尝试编写一个允许我转换的宏 public class Schedule { ... private static final ImmutableMap<String, Schedule> WEEKDAY_TO_INT_MAP = ImmutableMap.<String, Schedule>builder() .put("SUNDAY", withIntWeekday(0)) .put("MONDAY", withIntWeekday(1)) .put("TUESDAY", withIntWeekday(2)) .put("WEDNESDAY", withIntWeekday(3)) .put("THURSDAY", withIntWeekday(4)) .put("FRIDAY", withIntWeekday(5)) .put("SATURDAY", withIntWeekday(6)) .build(); public static Schedule withIntWeekday(int weekdayInt) { Schedule schedule = new Schedule(); schedule.setWeekday(weekdayInt); return schedule; } @JsonCreator public static Schedule fromString(String weekday) { return WEEKDAY_TO_INT_MAP.get(weekday); } ... } (a, b, c, d)等。这是我到目前为止所得到的:

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

  

意外令牌:($x: expr, $y: expr) => (($x, $x + $y));

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

1 个答案:

答案 0 :(得分:10)

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

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

#![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)。