1
1
//! Safe interface for NumPy ndarray
2
2
3
3
use ndarray:: * ;
4
- use npyffi:: { self , PyArrayAPI } ;
4
+ use npyffi:: { self , PY_ARRAY_API } ;
5
5
use pyo3:: * ;
6
6
use std:: marker:: PhantomData ;
7
7
use std:: os:: raw:: c_void;
@@ -15,7 +15,7 @@ pub struct PyArray<T>(PyObject, PhantomData<T>);
15
15
16
16
pyobject_native_type_convert ! (
17
17
PyArray <T >,
18
- * npyffi:: PyArrayAPI . get_type_object( npyffi:: ArrayType :: PyArray_Type ) ,
18
+ * npyffi:: PY_ARRAY_API . get_type_object( npyffi:: ArrayType :: PyArray_Type ) ,
19
19
npyffi:: PyArray_Check ,
20
20
T
21
21
) ;
@@ -364,8 +364,8 @@ impl<T: TypeNum> PyArray<T> {
364
364
data : * mut c_void ,
365
365
) -> & ' py Self {
366
366
let dims: Vec < _ > = dims. iter ( ) . map ( |d| * d as npy_intp ) . collect ( ) ;
367
- let ptr = PyArrayAPI . PyArray_New (
368
- PyArrayAPI . get_type_object ( npyffi:: ArrayType :: PyArray_Type ) ,
367
+ let ptr = PY_ARRAY_API . PyArray_New (
368
+ PY_ARRAY_API . get_type_object ( npyffi:: ArrayType :: PyArray_Type ) ,
369
369
dims. len ( ) as i32 ,
370
370
dims. as_ptr ( ) as * mut npy_intp ,
371
371
T :: typenum_default ( ) ,
@@ -414,8 +414,8 @@ impl<T: TypeNum> PyArray<T> {
414
414
pub fn zeros < ' py > ( py : Python < ' py > , dims : & [ usize ] , is_fortran : bool ) -> & ' py Self {
415
415
let dims: Vec < npy_intp > = dims. iter ( ) . map ( |d| * d as npy_intp ) . collect ( ) ;
416
416
unsafe {
417
- let descr = PyArrayAPI . PyArray_DescrFromType ( T :: typenum_default ( ) ) ;
418
- let ptr = PyArrayAPI . PyArray_Zeros (
417
+ let descr = PY_ARRAY_API . PyArray_DescrFromType ( T :: typenum_default ( ) ) ;
418
+ let ptr = PY_ARRAY_API . PyArray_Zeros (
419
419
dims. len ( ) as i32 ,
420
420
dims. as_ptr ( ) as * mut npy_intp ,
421
421
descr,
@@ -443,7 +443,7 @@ impl<T: TypeNum> PyArray<T> {
443
443
/// # }
444
444
pub fn arange < ' py > ( py : Python < ' py > , start : f64 , stop : f64 , step : f64 ) -> & ' py Self {
445
445
unsafe {
446
- let ptr = PyArrayAPI . PyArray_Arange ( start, stop, step, T :: typenum_default ( ) ) ;
446
+ let ptr = PY_ARRAY_API . PyArray_Arange ( start, stop, step, T :: typenum_default ( ) ) ;
447
447
Self :: from_owned_ptr ( py, ptr)
448
448
}
449
449
}
@@ -463,7 +463,7 @@ impl<T: TypeNum> PyArray<T> {
463
463
pub fn copy_to < U : TypeNum > ( & self , other : & mut PyArray < U > ) -> Result < ( ) , ArrayCastError > {
464
464
let self_ptr = self . as_array_ptr ( ) ;
465
465
let other_ptr = other. as_array_ptr ( ) ;
466
- let result = unsafe { PyArrayAPI . PyArray_CopyInto ( other_ptr, self_ptr) } ;
466
+ let result = unsafe { PY_ARRAY_API . PyArray_CopyInto ( other_ptr, self_ptr) } ;
467
467
if result == -1 {
468
468
Err ( ArrayCastError :: Numpy {
469
469
from : T :: npy_data_type ( ) ,
@@ -489,7 +489,7 @@ impl<T: TypeNum> PyArray<T> {
489
489
pub fn move_to < U : TypeNum > ( & self , other : & mut PyArray < U > ) -> Result < ( ) , ArrayCastError > {
490
490
let self_ptr = self . as_array_ptr ( ) ;
491
491
let other_ptr = other. as_array_ptr ( ) ;
492
- let result = unsafe { PyArrayAPI . PyArray_MoveInto ( other_ptr, self_ptr) } ;
492
+ let result = unsafe { PY_ARRAY_API . PyArray_MoveInto ( other_ptr, self_ptr) } ;
493
493
if result == -1 {
494
494
Err ( ArrayCastError :: Numpy {
495
495
from : T :: npy_data_type ( ) ,
@@ -517,8 +517,8 @@ impl<T: TypeNum> PyArray<T> {
517
517
is_fortran : bool ,
518
518
) -> Result < & ' py PyArray < U > , ArrayCastError > {
519
519
let ptr = unsafe {
520
- let descr = PyArrayAPI . PyArray_DescrFromType ( U :: typenum_default ( ) ) ;
521
- PyArrayAPI . PyArray_CastToType (
520
+ let descr = PY_ARRAY_API . PyArray_DescrFromType ( U :: typenum_default ( ) ) ;
521
+ PY_ARRAY_API . PyArray_CastToType (
522
522
self . as_array_ptr ( ) ,
523
523
descr,
524
524
if is_fortran { -1 } else { 0 } ,
0 commit comments