polars_core/chunked_array/object/
iterator.rsuse arrow::array::Array;
use arrow::trusted_len::TrustedLen;
use crate::chunked_array::object::{ObjectArray, PolarsObject};
pub struct ObjectIter<'a, T: PolarsObject> {
array: &'a ObjectArray<T>,
current: usize,
current_end: usize,
}
impl<'a, T: PolarsObject> ObjectIter<'a, T> {
pub fn new(array: &'a ObjectArray<T>) -> Self {
ObjectIter::<T> {
array,
current: 0,
current_end: array.len(),
}
}
}
impl<'a, T: PolarsObject> std::iter::Iterator for ObjectIter<'a, T> {
type Item = Option<&'a T>;
#[inline]
fn next(&mut self) -> Option<Self::Item> {
if self.current == self.current_end {
None
} else if unsafe { self.array.is_null_unchecked(self.current) } {
self.current += 1;
Some(None)
} else {
let old = self.current;
self.current += 1;
unsafe { Some(Some(self.array.value_unchecked(old))) }
}
}
fn size_hint(&self) -> (usize, Option<usize>) {
(
self.array.len() - self.current,
Some(self.array.len() - self.current),
)
}
}
impl<T: PolarsObject> std::iter::DoubleEndedIterator for ObjectIter<'_, T> {
fn next_back(&mut self) -> Option<Self::Item> {
if self.current_end == self.current {
None
} else {
self.current_end -= 1;
Some(if self.array.is_null(self.current_end) {
None
} else {
unsafe { Some(self.array.value_unchecked(self.current_end)) }
})
}
}
}
impl<T: PolarsObject> std::iter::ExactSizeIterator for ObjectIter<'_, T> {}
impl<'a, T: PolarsObject> IntoIterator for &'a ObjectArray<T> {
type Item = Option<&'a T>;
type IntoIter = ObjectIter<'a, T>;
fn into_iter(self) -> Self::IntoIter {
ObjectIter::<'a, T>::new(self)
}
}
pub struct OwnedObjectIter<T: PolarsObject> {
array: ObjectArray<T>,
current: usize,
current_end: usize,
}
impl<T: PolarsObject> OwnedObjectIter<T> {
pub fn new(array: ObjectArray<T>) -> Self {
let current_end = array.len();
OwnedObjectIter::<T> {
array,
current: 0,
current_end,
}
}
}
unsafe impl<T: PolarsObject> TrustedLen for OwnedObjectIter<T> {}
impl<T: PolarsObject> ObjectArray<T> {
pub(crate) fn into_iter_cloned(self) -> OwnedObjectIter<T> {
OwnedObjectIter::<T>::new(self)
}
}
impl<T: PolarsObject> std::iter::Iterator for OwnedObjectIter<T> {
type Item = Option<T>;
#[inline]
fn next(&mut self) -> Option<Self::Item> {
if self.current == self.current_end {
None
} else if unsafe { self.array.is_null_unchecked(self.current) } {
self.current += 1;
Some(None)
} else {
let old = self.current;
self.current += 1;
unsafe { Some(Some(self.array.value_unchecked(old).clone())) }
}
}
fn size_hint(&self) -> (usize, Option<usize>) {
(
self.array.len() - self.current,
Some(self.array.len() - self.current),
)
}
}