计算宏中重复的长度

Question

我正在尝试实现一个宏以允许创建 MATLAB 风格的矩阵。我有一个基本的工作宏，但我还有很长的路要走。

我希望能够强制执行正确的结构（每行中的元素数量相同），但我不确定如何在宏中执行此操作。我想我想强制每个内部重复具有相同的长度 - 这是我可以做的吗？

到目前为止，这是我的代码：

pub struct Matrix<T> {
    pub cols: usize,
    pub rows: usize,
    pub data: Vec<T>
}

macro_rules! mat {
    ( $($( $x:expr ),*);* ) => {
        {
            let mut vec = Vec::new();
            let mut rows = 0;

            $(
                $(
                    vec.push($x);
                )*
                rows += 1;
            )*
            Matrix { cols : vec.len()/rows, rows: rows, data: vec}
        }
    };
}

它有效，但正如您所见，它不是很安全。它对结构没有限制。

我想用这个宏做更多的事情，但我认为这是一个好的开始！

更新：

这是我制定的蹩脚实现的一些操场代码。如果有人有更好的建议，请告诉我！否则我会自己关闭它。

Answer 1

首先，快速解决您的问题的标题：请参阅The Little Book of Rust Macros 中的计数章节。总结一下：没有直接的方法，您需要编写一个宏来扩展为您可以在常规代码中计数的内容。

现在，解决您的实际问题：嘿男孩。

这并不是您想要的那么多计数，如果子序列具有不同的长度，则会在编译时失败。

首先，没有干净的方法可以从宏触发编译失败。您可以触发一些其他预先存在的错误，但您无法控制实际的错误消息。

其次，根本没有简单的方法可以在宏中进行“变量”比较。有时您可以与固定标记序列进行比较，但这里不这样做。

所以这是双重不可行的。

最简单的做法是在运行时检查构造过程中的长度，如果不匹配则返回错误或恐慌。

难道真的不可能吗？我不相信是这样。如果您愿意接受难以理解的错误消息和复杂性的巨大跳跃，您可以检查两个标记序列之间的长度是否相等，如下所示：

macro_rules! tts_equal_len {
    (($_lhs:tt $($lhs_tail:tt)*), ($_rhs:tt $($rhs_tail:tt)*)) => {
        tts_equal_len!(($($lhs_tail)*), ($($rhs_tail)*))
    };
    (($($_lhs_tail:tt)+), ()) => { do_something_bad!() };
    ((), ($($_rhs_tail:tt)+)) => { do_something_bad!() };
    ((), ()) => { do_something_good!() };
}

macro_rules! do_something_bad { () => { { println!("kaboom!") } } }
macro_rules! do_something_good { () => { { println!("no kaboom!") } } }

fn main() {
    tts_equal_len!((,,,), (,,,));
    tts_equal_len!((,,,), (,,));
    tts_equal_len!((,), (,,));
}

同样，真正的问题是找到某种在编译时失败的方法，以便用户了解编译失败的原因。

Answer 2

macro_rules! count {
    () => (0usize);
    ( $x:tt $($xs:tt)* ) => (1usize + count!($($xs)*));
}

macro_rules! mat {
    ( $( $x:expr ),* ) => { {
        let vec = vec![$($x),*];
        Matrix { cols : vec.len(), rows: 1, data: vec }
    } };
    ( $( $x0:expr ),* ; $($( $x:expr ),*);* ) => { {
        let mut _assert_width0 = [(); count!($($x0)*)];
        let mut vec = Vec::new();
        let rows = 1usize;
        let cols = count!($($x0)*);

        $( vec.push($x0); )*

        $(
            let rows = rows + 1usize;
            let _assert_width = [(); count!($($x)*)];
            _assert_width0 = _assert_width;
            $( vec.push($x); )*
        )*

        Matrix { cols : cols, rows: rows, data: vec }
    } }
}

操场

的count!宏扩展为代表的参数它得到作为输入数的常量表达式。它只是mat!宏的帮手。如果需要对很多项进行计数而编译器无法应付，请参阅The Little Book of Rust Macros 中的 Counting 章节，其中有更复杂的宏进行计数。

我的宏版本使用虚拟变量和赋值来验证所有行的宽度是否相同。首先，我更改了宏的模式以将第一行与后续行分开处理。第一个变量 ,_assert_width0用一个单位数组 ( (), 这使得数组不占用内存)初始化, 数组的大小是第一行中的项目数。然后，_assert_width也用单元数组初始化，数组的大小是每个后续行中的项目数。然后，_assert_width分配给_assert_width0。这里的神奇之处在于，如果一行的宽度与第一行的宽度不匹配，这一行将引发编译器错误，因为数组的类型不匹配（您可能有例如[(); 3]和[(); 4]）。但是，如果您不知道宏中发生了什么，则错误不是很清楚：

<anon>:38:24: 38:37 error: mismatched types:
 expected `[(); 3]`,
    found `[(); 4]`
(expected an array with a fixed size of 3 elements,
    found one with 4 elements) [E0308]
<anon>:38           _assert_width0 = _assert_width;
                                     ^~~~~~~~~~~~~
<anon>:47:13: 47:44 note: in this expansion of mat! (defined in <anon>)
<anon>:38:24: 38:37 help: see the detailed explanation for E0308