Implement PyArray::copy_to, move_to

Author: kngwyu

src/array.rs

@@ -455,4 +455,64 @@ impl<T: TypeNum> PyArray<T> {
             Self::from_owned_ptr(py, ptr)
         }
     }
+
+    /// Copies self into `other`, performing a data-type conversion if necessary.
+    /// # Example
+    /// ```
+    /// # extern crate pyo3; extern crate numpy; fn main() {
+    /// use numpy::{PyArray, PyArrayModule, IntoPyArray};
+    /// let gil = pyo3::Python::acquire_gil();
+    /// let np = PyArrayModule::import(gil.python()).unwrap();
+    /// let pyarray_f = PyArray::<f64>::arange(gil.python(), &np, 2.0, 5.0, 1.0);
+    /// let mut pyarray_i = PyArray::<i64>::new(gil.python(), &np, &[3]);
+    /// assert!(pyarray_f.copy_to(&np, &mut pyarray_i).is_ok());
+    /// assert_eq!(pyarray_i.as_slice().unwrap(), &[2, 3, 4]);
+    /// # }
+    pub fn copy_to<U: TypeNum>(
+        &self,
+        np: &PyArrayModule,
+        other: &mut PyArray<U>,
+    ) -> Result<(), ArrayCastError> {
+        let self_ptr = self.as_array_ptr();
+        let other_ptr = other.as_array_ptr();
+        let result = unsafe { np.PyArray_CopyInto(other_ptr, self_ptr) };
+        if result == -1 {
+            Err(ArrayCastError::Numpy {
+                from: T::npy_data_type(),
+                to: U::npy_data_type(),
+            })
+        } else {
+            Ok(())
+        }
+    }
+
+    /// Move the data of self into `other`, performing a data-type conversion if necessary.
+    /// # Example
+    /// ```
+    /// # extern crate pyo3; extern crate numpy; fn main() {
+    /// use numpy::{PyArray, PyArrayModule, IntoPyArray};
+    /// let gil = pyo3::Python::acquire_gil();
+    /// let np = PyArrayModule::import(gil.python()).unwrap();
+    /// let pyarray_f = PyArray::<f64>::arange(gil.python(), &np, 2.0, 5.0, 1.0);
+    /// let mut pyarray_i = PyArray::<i64>::new(gil.python(), &np, &[3]);
+    /// assert!(pyarray_f.move_to(&np, &mut pyarray_i).is_ok());
+    /// assert_eq!(pyarray_i.as_slice().unwrap(), &[2, 3, 4]);
+    /// # }
+    pub fn move_to<U: TypeNum>(
+        self,
+        np: &PyArrayModule,
+        other: &mut PyArray<U>,
+    ) -> Result<(), ArrayCastError> {
+        let self_ptr = self.as_array_ptr();
+        let other_ptr = other.as_array_ptr();
+        let result = unsafe { np.PyArray_MoveInto(other_ptr, self_ptr) };
+        if result == -1 {
+            Err(ArrayCastError::Numpy {
+                from: T::npy_data_type(),
+                to: U::npy_data_type(),
+            })
+        } else {
+            Ok(())
+        }
+    }
 }

src/error.rs

@@ -28,6 +28,8 @@ pub enum ArrayCastError {
     ToRust { from: NpyDataType, to: NpyDataType },
     /// Error for casting rust's `Vec` into numpy array.
     FromVec,
+    /// Error in numpy -> numpy data conversion
+    Numpy { from: NpyDataType, to: NpyDataType },
 }
 
 impl ArrayCastError {
@@ -46,18 +48,16 @@ impl fmt::Display for ArrayCastError {
                 write!(f, "Cast failed: from={:?}, to={:?}", from, to)
             }
             ArrayCastError::FromVec => write!(f, "Cast failed: FromVec (maybe invalid dimension)"),
+            ArrayCastError::Numpy { from, to } => write!(
+                f,
+                "Cast failed: from=ndarray(dtype={:?}), to=ndarray(dtype={:?})",
+                from, to
+            ),
         }
     }
 }
 
-impl error::Error for ArrayCastError {
-    fn description(&self) -> &str {
-        match self {
-            ArrayCastError::ToRust { .. } => "ArrayCast failed(IntoArray)",
-            ArrayCastError::FromVec => "ArrayCast failed(FromVec)",
-        }
-    }
-}
+impl error::Error for ArrayCastError {}
 
 impl IntoPyErr for ArrayCastError {
     fn into_pyerr(self, msg: &str) -> PyErr {
@@ -67,6 +67,10 @@ impl IntoPyErr for ArrayCastError {
                 from, to, msg
             ),
             ArrayCastError::FromVec => format!("ArrayCastError::FromVec: {}", msg),
+            ArrayCastError::Numpy { from, to } => format!(
+                "ArrayCastError::Numpy: from: {:?}, to: {:?}, msg: {}",
+                from, to, msg
+            ),
         };
         PyErr::new::<exc::TypeError, _>(msg)
     }