Implement DoubleEndedIterator for PyTupleIterator
This commit is contained in:
parent
ae982b8ad0
commit
4ce3e9649f
|
@ -0,0 +1 @@
|
||||||
|
Add implementation `DoubleEndedIterator` for `PyTupleIterator`.
|
|
@ -0,0 +1 @@
|
||||||
|
The `PyTupleIterator` type returned by `PyTuple::iter` is now public and hence can be named by downstream crates.
|
|
@ -495,7 +495,7 @@ impl<T: Element> PyBuffer<T> {
|
||||||
|
|
||||||
err::error_on_minusone(py, unsafe {
|
err::error_on_minusone(py, unsafe {
|
||||||
ffi::PyBuffer_ToContiguous(
|
ffi::PyBuffer_ToContiguous(
|
||||||
target.as_ptr() as *mut raw::c_void,
|
target.as_mut_ptr().cast(),
|
||||||
#[cfg(Py_3_11)]
|
#[cfg(Py_3_11)]
|
||||||
&*self.0,
|
&*self.0,
|
||||||
#[cfg(not(Py_3_11))]
|
#[cfg(not(Py_3_11))]
|
||||||
|
|
|
@ -75,6 +75,7 @@ pub mod iter {
|
||||||
pub use super::dict::PyDictIterator;
|
pub use super::dict::PyDictIterator;
|
||||||
pub use super::frozenset::PyFrozenSetIterator;
|
pub use super::frozenset::PyFrozenSetIterator;
|
||||||
pub use super::set::PySetIterator;
|
pub use super::set::PySetIterator;
|
||||||
|
pub use super::tuple::PyTupleIterator;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Implementations core to all native types
|
// Implementations core to all native types
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
use std::convert::TryInto;
|
use std::convert::TryInto;
|
||||||
|
use std::iter::FusedIterator;
|
||||||
|
|
||||||
use crate::ffi::{self, Py_ssize_t};
|
use crate::ffi::{self, Py_ssize_t};
|
||||||
#[cfg(feature = "experimental-inspect")]
|
#[cfg(feature = "experimental-inspect")]
|
||||||
|
@ -232,16 +233,23 @@ pub struct PyTupleIterator<'a> {
|
||||||
length: usize,
|
length: usize,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
impl<'a> PyTupleIterator<'a> {
|
||||||
|
unsafe fn get_item(&self, index: usize) -> &'a PyAny {
|
||||||
|
#[cfg(any(Py_LIMITED_API, PyPy))]
|
||||||
|
let item = self.tuple.get_item(index).expect("tuple.get failed");
|
||||||
|
#[cfg(not(any(Py_LIMITED_API, PyPy)))]
|
||||||
|
let item = self.tuple.get_item_unchecked(index);
|
||||||
|
item
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
impl<'a> Iterator for PyTupleIterator<'a> {
|
impl<'a> Iterator for PyTupleIterator<'a> {
|
||||||
type Item = &'a PyAny;
|
type Item = &'a PyAny;
|
||||||
|
|
||||||
#[inline]
|
#[inline]
|
||||||
fn next(&mut self) -> Option<&'a PyAny> {
|
fn next(&mut self) -> Option<Self::Item> {
|
||||||
if self.index < self.length {
|
if self.index < self.length {
|
||||||
#[cfg(any(Py_LIMITED_API, PyPy))]
|
let item = unsafe { self.get_item(self.index) };
|
||||||
let item = self.tuple.get_item(self.index).expect("tuple.get failed");
|
|
||||||
#[cfg(not(any(Py_LIMITED_API, PyPy)))]
|
|
||||||
let item = unsafe { self.tuple.get_item_unchecked(self.index) };
|
|
||||||
self.index += 1;
|
self.index += 1;
|
||||||
Some(item)
|
Some(item)
|
||||||
} else {
|
} else {
|
||||||
|
@ -256,12 +264,27 @@ impl<'a> Iterator for PyTupleIterator<'a> {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
impl<'a> DoubleEndedIterator for PyTupleIterator<'a> {
|
||||||
|
#[inline]
|
||||||
|
fn next_back(&mut self) -> Option<Self::Item> {
|
||||||
|
if self.index < self.length {
|
||||||
|
let item = unsafe { self.get_item(self.length - 1) };
|
||||||
|
self.length -= 1;
|
||||||
|
Some(item)
|
||||||
|
} else {
|
||||||
|
None
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
impl<'a> ExactSizeIterator for PyTupleIterator<'a> {
|
impl<'a> ExactSizeIterator for PyTupleIterator<'a> {
|
||||||
fn len(&self) -> usize {
|
fn len(&self) -> usize {
|
||||||
self.length.saturating_sub(self.index)
|
self.length.saturating_sub(self.index)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
impl FusedIterator for PyTupleIterator<'_> {}
|
||||||
|
|
||||||
impl<'a> IntoIterator for &'a PyTuple {
|
impl<'a> IntoIterator for &'a PyTuple {
|
||||||
type Item = &'a PyAny;
|
type Item = &'a PyAny;
|
||||||
type IntoIter = PyTupleIterator<'a>;
|
type IntoIter = PyTupleIterator<'a>;
|
||||||
|
@ -510,6 +533,33 @@ mod tests {
|
||||||
|
|
||||||
assert_eq!(3_i32, iter.next().unwrap().extract::<'_, i32>().unwrap());
|
assert_eq!(3_i32, iter.next().unwrap().extract::<'_, i32>().unwrap());
|
||||||
assert_eq!(iter.size_hint(), (0, Some(0)));
|
assert_eq!(iter.size_hint(), (0, Some(0)));
|
||||||
|
|
||||||
|
assert!(iter.next().is_none());
|
||||||
|
assert!(iter.next().is_none());
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn test_iter_rev() {
|
||||||
|
Python::with_gil(|py| {
|
||||||
|
let ob = (1, 2, 3).to_object(py);
|
||||||
|
let tuple: &PyTuple = ob.downcast(py).unwrap();
|
||||||
|
assert_eq!(3, tuple.len());
|
||||||
|
let mut iter = tuple.iter().rev();
|
||||||
|
|
||||||
|
assert_eq!(iter.size_hint(), (3, Some(3)));
|
||||||
|
|
||||||
|
assert_eq!(3_i32, iter.next().unwrap().extract::<'_, i32>().unwrap());
|
||||||
|
assert_eq!(iter.size_hint(), (2, Some(2)));
|
||||||
|
|
||||||
|
assert_eq!(2_i32, iter.next().unwrap().extract::<'_, i32>().unwrap());
|
||||||
|
assert_eq!(iter.size_hint(), (1, Some(1)));
|
||||||
|
|
||||||
|
assert_eq!(1_i32, iter.next().unwrap().extract::<'_, i32>().unwrap());
|
||||||
|
assert_eq!(iter.size_hint(), (0, Some(0)));
|
||||||
|
|
||||||
|
assert!(iter.next().is_none());
|
||||||
|
assert!(iter.next().is_none());
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue