Question

我正在尝试编写一个宏，该宏扩展为impls的集合，用于我写的一些自定义结构。这就是我所拥有的：

pub trait ComponentWise : Clone + Copy {
  fn new(x: f32, y: f32, z: f32) -> Self;

  fn x(&self) -> f32;
  fn y(&self) -> f32;
  fn z(&self) -> f32;

  fn has_nans(&self) -> bool {
    f32::is_nan(self.x()) || 
    f32::is_nan(self.y()) || 
    f32::is_nan(self.z())
  }
}

macro_rules! component_wise_impls { 
  ($T:ty) => {
    impl ComponentWise for $T {
      fn new(x: f32, y: f32, z: f32) -> $T {
        let x: $T = $T { vals: [x, y, z] };
        debug_assert!(!x.has_nans());
        x
      }
   };
}

pub struct Vector {
  pub vals: [f32; 3]
}

component_wise_impls!(Vector);

可悲的是，我被告知：

错误：意外令牌：矢量

让x：$ T = $ T {vals：[x，y，z]};

我有点惊讶，因为我可以做Vector { vals: [x, y, z] }，我认为这正是$T { vals: [x, y, z] };扩展到的。有没有办法在宏中编写这个new函数？

Answer 1

这是有效的：

macro_rules! component_wise_impls {
    ($T:ident) => {
        impl ComponentWise for $T {
            fn new(x: f32, y: f32, z: f32) -> $T {
                let x: $T = $T { vals: [x, y, z] };
                debug_assert!(!x.has_nans());
                x
            }
        }
    }
}

您无法使用需要表达式的ty。

从此page：

插值$argument_name可能出现在与其片段说明符一致的任何位置（即，如果它被指定为ident，它可以在允许使用标识符的任何地方使用。）

正如llogiq指出的那样，您必须实施x，y和z方法，因此完整代码为：

pub trait ComponentWise : Clone + Copy {
  fn new(x: f32, y: f32, z: f32) -> Self;

  fn x(&self) -> f32;
  fn y(&self) -> f32;
  fn z(&self) -> f32;

  fn has_nans(&self) -> bool {
    f32::is_nan(self.x()) || 
    f32::is_nan(self.y()) || 
    f32::is_nan(self.z())
  }
}

macro_rules! component_wise_impls {
    ($T:ident) => {
        impl ComponentWise for $T {
            fn new(x: f32, y: f32, z: f32) -> $T {
                let x: $T = $T { vals: [x, y, z] };
                debug_assert!(!x.has_nans());
                x
            }

            fn x(&self) -> f32 {
                self.vals[0]
            }

            fn y(&self) -> f32 {
                self.vals[1]
            }

            fn z(&self) -> f32 {
                self.vals[2]
            }
        }
    }
}

#[derive(Copy, Clone)]
pub struct Vector {
  pub vals: [f32; 3]
}

component_wise_impls!(Vector);

Rust宏中的意外令牌

1 个答案: