arrow_rayon/parallel_array/
parallel_uint_array.rs1use 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}