Add some new FFIs

Author: kngwyu

src/npyffi/array.rs

@@ -313,6 +313,8 @@ impl PyArrayAPI {
     impl_api![299; PyDataMem_NEW_ZEROED(size: usize, elsize: usize) -> *mut c_void];
     impl_api![300; PyArray_CheckAnyScalarExact(obj: *mut PyObject) -> c_int];
     impl_api![301; PyArray_MapIterArrayCopyIfOverlap(a: *mut PyArrayObject, index: *mut PyObject, copy_if_overlap: c_int, extra_op: *mut PyArrayObject) -> *mut PyObject];
+    impl_api![302; PyArray_ResolveWritebackIfCopy(self_: *mut PyArrayObject) -> c_int];
+    impl_api![303; PyArray_SetWritebackIfCopyBase(arr: *mut PyArrayObject, base: *mut PyArrayObject) -> c_int];
 }
 
 /// Define PyTypeObject related to Array API

src/npyffi/mod.rs

@@ -28,13 +28,17 @@ Please make sure that you get gil, by `let gil = Python::acquire_gil();`"
 
 // Define Array&UFunc APIs
 macro_rules! impl_api {
-    [ $offset:expr; $fname:ident ( $($arg:ident : $t:ty),* ) $( -> $ret:ty )* ] => {
+    [$offset: expr; $fname: ident ( $($arg: ident : $t: ty),* ) $( -> $ret: ty )* ] => {
         #[allow(non_snake_case)]
         pub unsafe fn $fname(&self, $($arg : $t), *) $( -> $ret )* {
             let fptr = self.get($offset)
                            as *const extern fn ($($arg : $t), *) $( -> $ret )*;
             (*fptr)($($arg), *)
         }
+    };
+    // To allow fn a(b: type,) -> ret
+    [$offset: expr; $fname: ident ( $($arg: ident : $t:ty,)* ) $( -> $ret: ty )* ] => {
+        impl_api![$offset; $fname( $($arg: $t),*) $( -> $ret )*];
     }
 }
 

src/npyffi/ufunc.rs

@@ -81,4 +81,18 @@ impl PyUFuncAPI {
     impl_api![39; PyUFunc_DefaultTypeResolver(ufunc: *mut PyUFuncObject, casting: NPY_CASTING, operands: *mut *mut PyArrayObject, type_tup: *mut PyObject, out_dtypes: *mut *mut PyArray_Descr) -> c_int];
     impl_api![40; PyUFunc_ValidateCasting(ufunc: *mut PyUFuncObject, casting: NPY_CASTING, operands: *mut *mut PyArrayObject, dtypes: *mut *mut PyArray_Descr) -> c_int];
     impl_api![41; PyUFunc_RegisterLoopForDescr(ufunc: *mut PyUFuncObject, user_dtype: *mut PyArray_Descr, function: PyUFuncGenericFunction, arg_dtypes: *mut *mut PyArray_Descr, data: *mut c_void) -> c_int];
+    impl_api![42; PyUFunc_FromFuncAndDataAndSignatureAndIdentity(
+        ufunc: *mut PyUFuncObject,
+        data: *mut *mut c_void,
+        types: *mut c_char,
+        ntypes: c_int,
+        nin: c_int,
+        nout: c_int,
+        identity: c_int,
+        name: *const c_char,
+        doc: *const c_char,
+        unused: c_int,
+        signature: *const c_char,
+        identity_value: *const c_char,
+    ) -> c_int];
 }