polars_arrow/array/utf8/
iterator.rs

1use super::{MutableUtf8Array, MutableUtf8ValuesArray, Utf8Array};
2use crate::array::{ArrayAccessor, ArrayValuesIter};
3use crate::bitmap::utils::{BitmapIter, ZipValidity};
4use crate::offset::Offset;
5
6unsafe impl<'a, O: Offset> ArrayAccessor<'a> for Utf8Array<O> {
7    type Item = &'a str;
8
9    #[inline]
10    unsafe fn value_unchecked(&'a self, index: usize) -> Self::Item {
11        self.value_unchecked(index)
12    }
13
14    #[inline]
15    fn len(&self) -> usize {
16        self.len()
17    }
18}
19
20/// Iterator of values of an [`Utf8Array`].
21pub type Utf8ValuesIter<'a, O> = ArrayValuesIter<'a, Utf8Array<O>>;
22
23impl<'a, O: Offset> IntoIterator for &'a Utf8Array<O> {
24    type Item = Option<&'a str>;
25    type IntoIter = ZipValidity<&'a str, Utf8ValuesIter<'a, O>, BitmapIter<'a>>;
26
27    fn into_iter(self) -> Self::IntoIter {
28        self.iter()
29    }
30}
31
32unsafe impl<'a, O: Offset> ArrayAccessor<'a> for MutableUtf8Array<O> {
33    type Item = &'a str;
34
35    #[inline]
36    unsafe fn value_unchecked(&'a self, index: usize) -> Self::Item {
37        self.value_unchecked(index)
38    }
39
40    #[inline]
41    fn len(&self) -> usize {
42        self.len()
43    }
44}
45
46/// Iterator of values of an [`MutableUtf8ValuesArray`].
47pub type MutableUtf8ValuesIter<'a, O> = ArrayValuesIter<'a, MutableUtf8ValuesArray<O>>;
48
49impl<'a, O: Offset> IntoIterator for &'a MutableUtf8Array<O> {
50    type Item = Option<&'a str>;
51    type IntoIter = ZipValidity<&'a str, MutableUtf8ValuesIter<'a, O>, BitmapIter<'a>>;
52
53    fn into_iter(self) -> Self::IntoIter {
54        self.iter()
55    }
56}
57
58unsafe impl<'a, O: Offset> ArrayAccessor<'a> for MutableUtf8ValuesArray<O> {
59    type Item = &'a str;
60
61    #[inline]
62    unsafe fn value_unchecked(&'a self, index: usize) -> Self::Item {
63        self.value_unchecked(index)
64    }
65
66    #[inline]
67    fn len(&self) -> usize {
68        self.len()
69    }
70}
71
72impl<'a, O: Offset> IntoIterator for &'a MutableUtf8ValuesArray<O> {
73    type Item = &'a str;
74    type IntoIter = ArrayValuesIter<'a, MutableUtf8ValuesArray<O>>;
75
76    fn into_iter(self) -> Self::IntoIter {
77        self.iter()
78    }
79}