将“* mut * mut f32”转换为“＆[＆[f32]]”

Question

我想转换数组。

例：

func()-> *mut *mut f32;
...
let buffer = func();
for n in 0..48000 {
    buffer[0][n] = 1.0;
    buffer[1][n] = 3.0;
}

Answer 1

在Rust &[T] / &mut [T]中称为切片。 切片不是数组; 它是指向数组开头和此数组中项目数的指针。 因此，要创建&mut [T] out *mut T ，您需要知道指针后面的数组的长度。

*mut *mut T看起来像一个2D，可能是锯齿状的数组的C实现，即一个数组数组（这与连续的2D数组不同，你可能知道）。 没有自由的方法将它转换为&mut [&mut [T]] ，因为正如我之前所说， *mut T是一个指针大小的数字，而&mut [T]是两个指针大小的数字。 所以你不能，例如，将*mut T为&mut [T] ，这将是一个大小不匹配。 因此，由于布局不匹配，您无法简单地将*mut *mut f32为&mut [&mut [f32]] 。

为了安全地处理存储在*mut *mut f32 ，首先需要确定外部数组的长度和所有内部数组的长度。 为简单起见，让我们考虑一下静态地知道它们：

const ROWS: usize = 48000;
const COLUMNS: usize = 48000;

现在，既然知道了长度，就可以将外部指针转换为原始指针切片：

use std::slice;

let buffer: *mut *mut f32 = func();

let buf_slice: &mut [*mut f32] = unsafe {
    slice::from_raw_parts_mut(buffer, ROWS);
};

现在，您需要浏览此切片并将每个项目转换为切片，将结果收集到矢量中：

let matrix: Vec<&mut [f32]> = buf_slice.iter_mut()
    .map(|p| unsafe { slice::from_raw_parts_mut(p, COLUMNS) })
    .collect();

现在您确实可以通过索引访问缓冲区：

for n in 0..COLUMNS {
    matrix[0][n] = 1.0;
    matrix[1][n] = 3.0;
}

（为了便于阅读，我在绑定上放置了显式类型，其中大多数实际上都可以省略）

因此，将原始指针转换为切片时需要考虑两件事：

1. 你需要知道数组的确切长度才能从中创建一个切片; 如果您知道，可以使用slice::from_raw_parts()或slice::from_raw_parts_mut() ;
2. 如果要转换嵌套数组，则需要重建间接的每一层，因为指针的大小与切片不同。

当然，您必须跟踪谁是缓冲区的所有者以及何时将其释放，否则您可以轻松获得指向不再存在的缓冲区的切片。 毕竟，这是unsafe 。

Answer 2

由于您的数组似乎是一个指向48000 f32 s数组的指针数组，因此您只需使用固定大小的数组（ [T; N] ）而不是切片（ [T] ）：

fn func() -> *mut *mut f32 { unimplemented!() }

fn main() {
    let buffer = func();
    let buffer: &mut [&mut [f32; 48000]; 2] = unsafe { std::mem::transmute(buffer) };
    for n in 0..48000 {
        buffer[0][n] = 1.0;
        buffer[1][n] = 3.0;
    }
}