我正在尝试创建一个宏，该宏以固定的大小填充向量的第一个元素。其余元素应为零。我写：
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] }
}
看来我尝试以错误的方式迭代参数。我该如何解决？
你可以这样
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] 删除。
我来说两句