polars_core/datatypes/
static_array_collect.rs

1use arrow::array::ArrayFromIter;
2use arrow::bitmap::BitmapBuilder;
3
4use crate::chunked_array::object::{ObjectArray, PolarsObject};
5
6impl<'a, T: PolarsObject> ArrayFromIter<&'a T> for ObjectArray<T> {
7    fn arr_from_iter<I: IntoIterator<Item = &'a T>>(iter: I) -> Self {
8        ObjectArray::from(Vec::from_iter(iter.into_iter().cloned()))
9    }
10
11    fn try_arr_from_iter<E, I: IntoIterator<Item = Result<&'a T, E>>>(iter: I) -> Result<Self, E> {
12        let values: Vec<T> = iter
13            .into_iter()
14            .map(|r| r.cloned())
15            .collect::<Result<_, E>>()?;
16        Ok(ObjectArray::from(values))
17    }
18}
19
20impl<'a, T: PolarsObject> ArrayFromIter<Option<&'a T>> for ObjectArray<T> {
21    fn arr_from_iter<I: IntoIterator<Item = Option<&'a T>>>(iter: I) -> Self {
22        let iter = iter.into_iter();
23        let size = iter.size_hint().0;
24
25        let mut null_mask_builder = BitmapBuilder::with_capacity(size);
26        let mut values: Vec<T> = Vec::with_capacity(size);
27
28        for val in iter {
29            null_mask_builder.push(val.is_some());
30            values.push(val.cloned().unwrap_or_default());
31        }
32
33        ObjectArray::from(values).with_validity(null_mask_builder.into_opt_validity())
34    }
35
36    fn try_arr_from_iter<E, I: IntoIterator<Item = Result<Option<&'a T>, E>>>(
37        iter: I,
38    ) -> Result<Self, E> {
39        let iter = iter.into_iter();
40        let size = iter.size_hint().0;
41
42        let mut null_mask_builder = BitmapBuilder::with_capacity(size);
43        let mut values: Vec<T> = Vec::with_capacity(size);
44
45        for val in iter {
46            let val = val?;
47            null_mask_builder.push(val.is_some());
48            values.push(val.cloned().unwrap_or_default());
49        }
50
51        Ok(ObjectArray::from(values).with_validity(null_mask_builder.into_opt_validity()))
52    }
53}
polars_core/datatypes/static_array_collect.rs

polars_core/datatypes/
static_array_collect.rs
