arrow_rayon/parallel_array/
parallel_uint_array.rs

1use arrow_array::types::{UInt16Type, UInt32Type, UInt64Type, UInt8Type};
2use arrow_array::{UInt16Array, UInt32Array, UInt64Array, UInt8Array};
3use rayon::iter::ParallelIterator;
4
5use crate::parallel_primitive_array::{ParallelPrimitiveArray, ParallelPrimitiveArrayRef};
6
7pub type ParallelUInt8Array = ParallelPrimitiveArray<UInt8Type>;
8pub type ParallelUInt8ArrayRef<'data> = ParallelPrimitiveArrayRef<'data, UInt8Type>;
9pub type ParallelUInt16Array = ParallelPrimitiveArray<UInt16Type>;
10pub type ParallelUInt16ArrayRef<'data> = ParallelPrimitiveArrayRef<'data, UInt16Type>;
11pub type ParallelUInt32Array = ParallelPrimitiveArray<UInt32Type>;
12pub type ParallelUInt32ArrayRef<'data> = ParallelPrimitiveArrayRef<'data, UInt32Type>;
13pub type ParallelUInt64Array = ParallelPrimitiveArray<UInt64Type>;
14pub type ParallelUInt64ArrayRef<'data> = ParallelPrimitiveArrayRef<'data, UInt64Type>;
15
16pub trait UInt8ArrayRefParallelIterator<'data> {
17    type Iter: ParallelIterator<Item = Option<u8>>;
18
19    fn par_iter(&'data self) -> Self::Iter;
20}
21
22impl<'data> UInt8ArrayRefParallelIterator<'data> for UInt8Array {
23    type Iter = ParallelUInt8ArrayRef<'data>;
24
25    fn par_iter(&'data self) -> Self::Iter {
26        ParallelUInt8ArrayRef::new(self)
27    }
28}
29
30pub trait UInt16ArrayRefParallelIterator<'data> {
31    type Iter: ParallelIterator<Item = Option<u16>>;
32
33    fn par_iter(&'data self) -> Self::Iter;
34}
35
36impl<'data> UInt16ArrayRefParallelIterator<'data> for UInt16Array {
37    type Iter = ParallelUInt16ArrayRef<'data>;
38
39    fn par_iter(&'data self) -> Self::Iter {
40        ParallelUInt16ArrayRef::new(self)
41    }
42}
43
44pub trait UInt32ArrayRefParallelIterator<'data> {
45    type Iter: ParallelIterator<Item = Option<u32>>;
46
47    fn par_iter(&'data self) -> Self::Iter;
48}
49
50impl<'data> UInt32ArrayRefParallelIterator<'data> for UInt32Array {
51    type Iter = ParallelUInt32ArrayRef<'data>;
52
53    fn par_iter(&'data self) -> Self::Iter {
54        ParallelUInt32ArrayRef::new(self)
55    }
56}
57
58pub trait UInt64ArrayRefParallelIterator<'data> {
59    type Iter: ParallelIterator<Item = Option<u64>>;
60
61    fn par_iter(&'data self) -> Self::Iter;
62}
63
64impl<'data> UInt64ArrayRefParallelIterator<'data> for UInt64Array {
65    type Iter = ParallelUInt64ArrayRef<'data>;
66
67    fn par_iter(&'data self) -> Self::Iter {
68        ParallelUInt64ArrayRef::new(self)
69    }
70}