sparrow 0.3.0
Loading...
Searching...
No Matches
dispatch.hpp
Go to the documentation of this file.
1// Copyright 2024 Man Group Operations Limited
2//
3// Licensed under the Apache License, Version 2.0 (the "License");
4// you may not use this file except in compliance with the License.
5// You may obtain a copy of the License at
6//
7// http://www.apache.org/licenses/LICENSE-2.0
8//
9// Unless required by applicable law or agreed to in writing, software
10// distributed under the License is distributed on an "AS IS" BASIS,
11// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or mplied.
12// See the License for the specific language governing permissions and
13// limitations under the License.
14
15#pragma once
16
17#include <type_traits>
18
19#include "sparrow/layout/array_wrapper.hpp"
20#include "sparrow/layout/decimal_array.hpp"
21#include "sparrow/layout/dictionary_encoded_array.hpp"
22#include "sparrow/layout/fixed_width_binary_array.hpp"
23#include "sparrow/layout/list_layout/list_array.hpp"
24#include "sparrow/layout/nested_value_types.hpp"
25#include "sparrow/layout/null_array.hpp"
26#include "sparrow/layout/primitive_array.hpp"
27#include "sparrow/layout/run_end_encoded_layout/run_end_encoded_array.hpp"
28#include "sparrow/layout/struct_layout/struct_array.hpp"
29#include "sparrow/layout/temporal/date_array.hpp"
30#include "sparrow/layout/temporal/duration_array.hpp"
31#include "sparrow/layout/temporal/interval_array.hpp"
32#include "sparrow/layout/temporal/time_array.hpp"
33#include "sparrow/layout/temporal/timestamp_array.hpp"
34#include "sparrow/layout/union_array.hpp"
35#include "sparrow/layout/variable_size_binary_layout/variable_size_binary_array.hpp"
36#include "sparrow/types/data_traits.hpp"
37#include "sparrow/types/data_type.hpp"
38
39namespace sparrow
40{
41 template <class F>
42 using visit_result_t = std::invoke_result_t<F, null_array>;
43
44 template <class F>
45 [[nodiscard]] visit_result_t<F> visit(F&& func, const array_wrapper& ar)
46 {
47 if (ar.is_dictionary())
48 {
49 switch (ar.data_type())
50 {
51 case data_type::UINT8:
52 return func(unwrap_array<dictionary_encoded_array<std::uint8_t>>(ar));
53 case data_type::INT8:
54 return func(unwrap_array<dictionary_encoded_array<std::int8_t>>(ar));
55 case data_type::UINT16:
56 return func(unwrap_array<dictionary_encoded_array<std::uint16_t>>(ar));
57 case data_type::INT16:
58 return func(unwrap_array<dictionary_encoded_array<std::int16_t>>(ar));
59 case data_type::UINT32:
60 return func(unwrap_array<dictionary_encoded_array<std::uint32_t>>(ar));
61 case data_type::INT32:
62 return func(unwrap_array<dictionary_encoded_array<std::int32_t>>(ar));
63 case data_type::UINT64:
64 return func(unwrap_array<dictionary_encoded_array<std::uint64_t>>(ar));
65 case data_type::INT64:
66 return func(unwrap_array<dictionary_encoded_array<std::int64_t>>(ar));
67 default:
68 throw std::runtime_error("data datype of dictionary encoded array must be an integer");
69 }
70 }
71 else
72 {
73 switch (ar.data_type())
74 {
75 case data_type::NA:
76 return func(unwrap_array<null_array>(ar));
77 case data_type::BOOL:
78 return func(unwrap_array<primitive_array<bool>>(ar));
79 case data_type::UINT8:
80 return func(unwrap_array<primitive_array<std::uint8_t>>(ar));
81 case data_type::INT8:
82 return func(unwrap_array<primitive_array<std::int8_t>>(ar));
83 case data_type::UINT16:
84 return func(unwrap_array<primitive_array<std::uint16_t>>(ar));
85 case data_type::INT16:
86 return func(unwrap_array<primitive_array<std::int16_t>>(ar));
87 case data_type::UINT32:
88 return func(unwrap_array<primitive_array<std::uint32_t>>(ar));
89 case data_type::INT32:
90 return func(unwrap_array<primitive_array<std::int32_t>>(ar));
91 case data_type::UINT64:
92 return func(unwrap_array<primitive_array<std::uint64_t>>(ar));
93 case data_type::INT64:
94 return func(unwrap_array<primitive_array<std::int64_t>>(ar));
95 case data_type::HALF_FLOAT:
96 return func(unwrap_array<primitive_array<float16_t>>(ar));
97 case data_type::FLOAT:
98 return func(unwrap_array<primitive_array<float32_t>>(ar));
99 case data_type::DOUBLE:
100 return func(unwrap_array<primitive_array<float64_t>>(ar));
101 case data_type::STRING:
102 return func(unwrap_array<string_array>(ar));
103 case data_type::LARGE_STRING:
104 return func(unwrap_array<big_string_array>(ar));
105 case data_type::BINARY:
106 return func(unwrap_array<binary_array>(ar));
107 case data_type::LARGE_BINARY:
108 return func(unwrap_array<big_binary_array>(ar));
109 case data_type::RUN_ENCODED:
110 return func(unwrap_array<run_end_encoded_array>(ar));
111 case data_type::LIST:
112 return func(unwrap_array<list_array>(ar));
113 case data_type::LARGE_LIST:
114 return func(unwrap_array<big_list_array>(ar));
115 case data_type::LIST_VIEW:
116 return func(unwrap_array<list_view_array>(ar));
117 case data_type::LARGE_LIST_VIEW:
118 return func(unwrap_array<big_list_view_array>(ar));
119 case data_type::FIXED_SIZED_LIST:
120 return func(unwrap_array<fixed_sized_list_array>(ar));
121 case data_type::STRUCT:
122 return func(unwrap_array<struct_array>(ar));
123 case data_type::DENSE_UNION:
124 return func(unwrap_array<dense_union_array>(ar));
125 case data_type::SPARSE_UNION:
126 return func(unwrap_array<sparse_union_array>(ar));
127 case data_type::DECIMAL32:
128 return func(unwrap_array<decimal_32_array>(ar));
129 case data_type::DECIMAL64:
130 return func(unwrap_array<decimal_64_array>(ar));
131 case data_type::DECIMAL128:
132 return func(unwrap_array<decimal_128_array>(ar));
133 case data_type::DECIMAL256:
134 return func(unwrap_array<decimal_256_array>(ar));
135 case data_type::FIXED_WIDTH_BINARY:
136 return func(unwrap_array<fixed_width_binary_array>(ar));
137 case sparrow::data_type::DATE_DAYS:
138 return func(unwrap_array<date_days_array>(ar));
139 case data_type::DATE_MILLISECONDS:
140 return func(unwrap_array<date_milliseconds_array>(ar));
141 case data_type::TIMESTAMP_SECONDS:
142 return func(unwrap_array<timestamp_array<timestamp<std::chrono::seconds>>>(ar));
143 case data_type::TIMESTAMP_MILLISECONDS:
144 return func(unwrap_array<timestamp_array<timestamp<std::chrono::milliseconds>>>(ar));
145 case data_type::TIMESTAMP_MICROSECONDS:
146 return func(unwrap_array<timestamp_array<timestamp<std::chrono::microseconds>>>(ar));
147 case data_type::TIMESTAMP_NANOSECONDS:
148 return func(unwrap_array<timestamp_array<timestamp<std::chrono::nanoseconds>>>(ar));
149 case data_type::TIME_SECONDS:
150 return func(unwrap_array<time_seconds_array>(ar));
151 case data_type::TIME_MILLISECONDS:
152 return func(unwrap_array<time_milliseconds_array>(ar));
153 case data_type::TIME_MICROSECONDS:
154 return func(unwrap_array<time_microseconds_array>(ar));
155 case data_type::TIME_NANOSECONDS:
156 return func(unwrap_array<time_nanoseconds_array>(ar));
157 case data_type::DURATION_SECONDS:
158 return func(unwrap_array<duration_seconds_array>(ar));
159 case data_type::DURATION_MILLISECONDS:
160 return func(unwrap_array<duration_milliseconds_array>(ar));
161 case data_type::DURATION_MICROSECONDS:
162 return func(unwrap_array<duration_microseconds_array>(ar));
163 case data_type::DURATION_NANOSECONDS:
164 return func(unwrap_array<duration_nanoseconds_array>(ar));
165 case data_type::INTERVAL_MONTHS:
166 return func(unwrap_array<months_interval_array>(ar));
167 case data_type::INTERVAL_DAYS_TIME:
168 return func(unwrap_array<days_time_interval_array>(ar));
169 case data_type::INTERVAL_MONTHS_DAYS_NANOSECONDS:
170 return func(unwrap_array<month_day_nanoseconds_interval_array>(ar));
171 default:
172 throw std::invalid_argument("array type not supported");
173 }
174 }
175 }
176}
array_wrapper.hpp
sparrow::array_wrapper
Base class for array type erasure.
Definition array_wrapper.hpp:71
sparrow::array_wrapper::is_dictionary
bool is_dictionary() const
Definition array_wrapper.hpp:155
sparrow::array_wrapper::data_type
enum data_type data_type() const
Definition array_wrapper.hpp:150
sparrow::dictionary_encoded_array
Definition dictionary_encoded_array.hpp:95
sparrow::timestamp_array
Array of timestamps.
Definition timestamp_array.hpp:104
data_traits.hpp
data_type.hpp
date_array.hpp
decimal_array.hpp
dictionary_encoded_array.hpp
duration_array.hpp
fixed_width_binary_array.hpp
interval_array.hpp
list_array.hpp
sparrow::timestamp
date::zoned_time< Duration, TimeZonePtr > timestamp
Definition data_type.hpp:129
sparrow::unwrap_array
T & unwrap_array(array_wrapper &)
Definition array_wrapper.hpp:268
sparrow::visit_result_t
std::invoke_result_t< F, null_array > visit_result_t
Definition dispatch.hpp:42
sparrow::primitive_array
array_trivial_copyable< T > primitive_array
Array of values of whose type has fixed binary size.
Definition primitive_array.hpp:42
sparrow::visit
visit_result_t< F > visit(F &&func, const array_wrapper &ar)
Definition dispatch.hpp:45
nested_value_types.hpp
null_array.hpp
primitive_array.hpp
run_end_encoded_array.hpp
struct_array.hpp
time_array.hpp
timestamp_array.hpp
union_array.hpp
variable_size_binary_array.hpp