1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
use crate::prelude::*;
macro_rules! get_any_value_unchecked {
($self:ident, $index:expr) => {{
let (chunk_idx, idx) = $self.index_to_chunked_index($index);
debug_assert!(chunk_idx < $self.chunks.len());
let arr = &**$self.chunks.get_unchecked(chunk_idx);
debug_assert!(idx < arr.len());
$self.arr_to_any_value(arr, idx)
}};
}
macro_rules! get_any_value {
($self:ident, $index:expr) => {{
let (chunk_idx, idx) = $self.index_to_chunked_index($index);
let arr = &*$self.chunks[chunk_idx];
assert!(idx < arr.len());
unsafe { $self.arr_to_any_value(arr, idx) }
}};
}
impl<T> ChunkAnyValue for ChunkedArray<T>
where
T: PolarsNumericType,
{
#[inline]
unsafe fn get_any_value_unchecked(&self, index: usize) -> AnyValue {
get_any_value_unchecked!(self, index)
}
fn get_any_value(&self, index: usize) -> AnyValue {
get_any_value!(self, index)
}
}
impl ChunkAnyValue for BooleanChunked {
#[inline]
unsafe fn get_any_value_unchecked(&self, index: usize) -> AnyValue {
get_any_value_unchecked!(self, index)
}
fn get_any_value(&self, index: usize) -> AnyValue {
get_any_value!(self, index)
}
}
impl ChunkAnyValue for Utf8Chunked {
#[inline]
unsafe fn get_any_value_unchecked(&self, index: usize) -> AnyValue {
get_any_value_unchecked!(self, index)
}
fn get_any_value(&self, index: usize) -> AnyValue {
get_any_value!(self, index)
}
}
impl ChunkAnyValue for ListChunked {
#[inline]
unsafe fn get_any_value_unchecked(&self, index: usize) -> AnyValue {
get_any_value_unchecked!(self, index)
}
fn get_any_value(&self, index: usize) -> AnyValue {
get_any_value!(self, index)
}
}
impl ChunkAnyValue for CategoricalChunked {
#[inline]
unsafe fn get_any_value_unchecked(&self, index: usize) -> AnyValue {
get_any_value_unchecked!(self, index)
}
fn get_any_value(&self, index: usize) -> AnyValue {
get_any_value!(self, index)
}
}
#[cfg(feature = "object")]
impl<T: PolarsObject> ChunkAnyValue for ObjectChunked<T> {
#[inline]
unsafe fn get_any_value_unchecked(&self, index: usize) -> AnyValue {
match self.get_object_unchecked(index) {
None => AnyValue::Null,
Some(v) => AnyValue::Object(v),
}
}
fn get_any_value(&self, index: usize) -> AnyValue {
match self.get_object(index) {
None => AnyValue::Null,
Some(v) => AnyValue::Object(v),
}
}
}