[英]How do I iterate arguments of a macros in Rust?
我正在尝试创建一个宏,用固定大小填充向量的第一个元素。 其余元素应为零。 我写:
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] }
}
$(...)*
的代码将针对每个匹配项进行扩展。
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.