如何将 FFI 数组作为 c_void 指针传递到代数的 DMatrix2?

How can I pass a FFI array as a c_void pointer to nalgebra's DMatrix2?

本文关键字:DMatrix2 指针 void FFI 数组      更新时间:2023-10-16

我有一个数组形式的矩阵C++,并希望将其传递给用 Rust 编写的共享库函数。我有这样的东西

#![crate_type = "dylib"]
extern crate libc;
use libc::c_void;
extern crate nalgebra as na;
use na::DMatrix2;
#[no_mangle]
pub extern "C" fn rust_fn(p_data: *const c_void, sizex: usize, sizey: usize) {
    let matrix = DMatrix2::from_row_vector(sizey, sizex, p_data);
    // Do something usefull with the matrix
}

这无法编译,因为我将c_void传递给from_row_vector().

我怎样才能做到这一点?

矩阵是一个双精度数组,但我试图保持接口通用,所以我也可以从例如 Python 调用函数。

我不想在从函数返回时释放矩阵(我想借用,而不是拥有矩阵)。

您可以使用std::slice::from_raw_parts来获取切片:

let slice = std::slice::from_raw_parts(p_data, sizex*sizey);

若要确保指针类型匹配,可以将参数列表中的p_data 的类型更改为*const N其中N是矩阵中使用的类型,也可以使用像 p_data as *const N 这样的强制转换。