如何在Rust中迭代宏的参数?

Fomalhaut

我正在尝试创建一个宏,该宏以固定的大小填充向量的第一个元素。其余元素应为零。我写:

const SIZE: usize = 3;

#[derive(Debug)]
struct MyVec {
    data: [f32; SIZE]
}

macro_rules! my_vec {
    ($($x:expr),*) => [{
        let mut data: [f32; SIZE] = [0.0; SIZE];
        for (i, e) in x.enumerate() {
            data[i] = e;
        }
        MyVec { data }
    }]
}

fn main() {
    let v = my_vec![1.0, 2.0];
    println!("{:?}", v);  // must be MyVec { data: [1.0, 2.0, 0.0] }
}

看来我尝试以错误的方式迭代参数。我该如何解决?

阿卜杜勒·尼亚斯(Abdul Niyas)总理

你可以这样

const SIZE: usize = 3;

#[derive(Debug)]
struct MyVec {
    data: [f32; SIZE],
}

macro_rules! my_vec {
    ( $( $x:expr ),* ) => {
        {
            let mut data: [f32; SIZE] = [0.0; SIZE];
            let mut index = 0;
            $(
                #[allow(unused_assignments)]
                {
                    data[index] = $x;
                    index = index + 1;
                }
            )*
            MyVec { data }
        }
    };
}

fn main() {
    let v = my_vec![1.0, 2.0];
    println!("{:?}", v); // print MyVec { data: [1.0, 2.0, 0.0] }
}

内的代码$(...)*将针对每个匹配项进行扩展。

本文收集自互联网,转载请注明来源。

如有侵权,请联系 [email protected] 删除。

编辑于
0

我来说两句

0 条评论
登录 后参与评论

相关文章