如何用宏实现Rust自定义结构体中Array2的切片操作？

推荐方案：实现Index和IndexMut trait #

在Rust中，最符合惯用风格的方式是为Foo实现Index和IndexMut trait，这样可以直接使用原生索引语法bar[[0, 1]]，完全不需要暴露内部的data字段，代码简洁且直观。

实现代码 #

use ndarray::{Array2};
use std::ops::{Index, IndexMut};

struct Foo {
    data: Array2<u32>,
    v1: i32,
    v2: i32,
}

// 实现不可变索引（支持读取）
impl Index<[usize; 2]> for Foo {
    type Output = u32;

    fn index(&self, idx: [usize; 2]) -> &Self::Output {
        &self.data[idx]
    }
}

// 实现可变索引（支持写入）
impl IndexMut<[usize; 2]> for Foo {
    fn index_mut(&mut self, idx: [usize; 2]) -> &mut Self::Output {
        &mut self.data[idx]
    }
}

fn main() {
    let mut bar = Foo {
        data: Array2::zeros((4, 3)),
        v1: 10,
        v2: 20,
    };

    // 直接用索引语法读取值
    println!("{}", bar[[0, 1]]);
    // 直接用索引语法写入值
    bar[[0, 1]] = 42;
    println!("{}", bar[[0, 1]]); // 输出42
}

这个方案的核心优势：

• 遵循Rust原生语法，使用者无需学习额外的宏规则
• 同时支持不可变读取和可变写入，覆盖所有访问场景
• 自带类型安全检查，编译器会自动校验索引合法性

备选方案：实现slice!宏 #

如果一定要用宏的方式，也可以封装一个简单的宏来隐藏data字段的访问：

宏实现代码 #

use ndarray::Array2;

struct Foo {
    data: Array2<u32>,
    v1: i32,
    v2: i32,
}

// 不可变访问宏
macro_rules! slice {
    ($obj:expr, [$x:expr, $y:expr]) => {
        $obj.data[[$x, $y]]
    };
}

// 可变访问宏（用于写入）
macro_rules! slice_mut {
    ($obj:expr, [$x:expr, $y:expr]) => {
        &mut $obj.data[[$x, $y]]
    };
}

fn main() {
    let mut bar = Foo {
        data: Array2::zeros((4, 3)),
        v1: 10,
        v2: 20,
    };

    // 使用宏读取值
    println!("{}", slice!(bar, [0, 1]));
    // 使用可变宏写入值
    *slice_mut!(bar, [0, 1]) = 42;
    println!("{}", slice!(bar, [0, 1])); // 输出42
}

不过这种方案需要额外定义两个宏区分可变/不可变访问，使用者需要记忆宏的用法，不如原生索引语法直观，因此并非最优解。

内容的提问来源于stack exchange，提问作者javirk