gpt4 book ai didi

rust - 如何为我的 rust 结构实现 trait FromPyObject

转载 作者:行者123 更新时间:2023-11-29 08:07:07 45 4
gpt4 key购买 nike

考虑一个通过 pyo3 暴露给 python 的简单 rust 类

use pyo3::prelude::*;

#[pyclass(name=MyClass)]
pub struct PyMyClass {
// Some fields
}

#[pymethods]
impl PyMyStruct {
#[new]
fn py_new(obj: PyRawObject) {
obj.init({
PyMyStruct {
// ...
}
});
}
}

现在有一个函数应该以这种方式用 python 中的两个结构调用:

a = MyStruct()
b = MyStruct()

c = foo(a,b)

因此定义

#[pyfunction]
fn foo(a: PyMyStruct, b: PyMyStruct) -> PyResult<PyMyStruct> {
// some high performance logic implemented in rust ...
}

现在编译器声称 PyMyStruct 应该实现特征 FromPyObject:

impl FromPyObject<'_> for PyMyStruct {
fn extract(ob: &'_ PyAny) ->PyResult<Self> {
// I dont know what to do here :(
}
}

但我不知道如何从 PyAny 中检索实例、指针或任何 PyMyStruct...有人可以帮助我吗?

最佳答案

我对这个 crate 了解不多,但我认为您应该引用您的元素。请注意,它们与 Python 解释器的其余部分共享,因此您不能真正拥有它们。

也就是只写:

#[pyfunction]
fn foo(a: &PyMyStruct, b: &PyMyStruct) -> PyResult<PyMyStruct> {
// some high performance logic implemented in rust ...
}

或者如果你想要可变对象:

#[pyfunction]
fn foo(a: &mut PyMyStruct, b: &mut PyMyStruct) -> PyResult<PyMyStruct> {
// some high performance logic implemented in rust ...
}

之所以可行,是因为有 these impls :

impl<'a, T> FromPyObject<'a> for &'a T where
T: PyTryFrom<'a>
impl<'a, T> FromPyObject<'a> for &'a mut T where
T: PyTryFrom<'a>,

然后这个other one :

impl<'v, T> PyTryFrom<'v> for T where
T: PyTypeInfo

如果您使用 &mut 变体并将同一个对象传递给此函数两次,会发生什么情况?好吧,我不确定,但浏览代码我猜你会得到对同一个对象的两个 &mut 引用,因此未定义的行为。如果我是你,我会使用非 mut 变体,如果你真的想更改对象,我会使用 RefCell

关于rust - 如何为我的 rust 结构实现 trait FromPyObject,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/57132033/

45 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com