polars_arrow/util/
macros.rs

1#[macro_export]
2macro_rules! with_match_primitive_type {(
3    $key_type:expr, | $_:tt $T:ident | $($body:tt)*
4) => ({
5    macro_rules! __with_ty__ {( $_ $T:ident ) => ( $($body)* )}
6    use polars_utils::float16::pf16;
7    use $crate::datatypes::PrimitiveType::*;
8
9    match $key_type {
10        Int8 => __with_ty__! { i8 },
11        Int16 => __with_ty__! { i16 },
12        Int32 => __with_ty__! { i32 },
13        Int64 => __with_ty__! { i64 },
14        Int128 => __with_ty__! { i128 },
15        UInt8 => __with_ty__! { u8 },
16        UInt16 => __with_ty__! { u16 },
17        UInt32 => __with_ty__! { u32 },
18        UInt64 => __with_ty__! { u64 },
19        UInt128 => __with_ty__! { u128 },
20        Float16 => __with_ty__! { pf16 },
21        Float32 => __with_ty__! { f32 },
22        Float64 => __with_ty__! { f64 },
23        _ => panic!("operator does not support primitive `{:?}`",
24            $key_type)
25    }
26})}
27
28#[macro_export]
29macro_rules! with_match_primitive_type_full {(
30    $key_type:expr, | $_:tt $T:ident | $($body:tt)*
31) => ({
32    macro_rules! __with_ty__ {( $_ $T:ident ) => ( $($body)* )}
33    use polars_utils::float16::pf16;
34    use $crate::datatypes::PrimitiveType::*;
35
36    match $key_type {
37        Int8 => __with_ty__! { i8 },
38        Int16 => __with_ty__! { i16 },
39        Int32 => __with_ty__! { i32 },
40        Int64 => __with_ty__! { i64 },
41        Int128 => __with_ty__! { i128 },
42        UInt8 => __with_ty__! { u8 },
43        UInt16 => __with_ty__! { u16 },
44        UInt32 => __with_ty__! { u32 },
45        UInt64 => __with_ty__! { u64 },
46        UInt128 => __with_ty__! { u128 },
47        Float16 => __with_ty__! { pf16 },
48        Float32 => __with_ty__! { f32 },
49        Float64 => __with_ty__! { f64 },
50        _ => panic!("operator does not support primitive `{:?}`",
51            $key_type)
52    }
53})}
54
55#[macro_export]
56macro_rules! match_integer_type {(
57    $key_type:expr, | $_:tt $T:ident | $($body:tt)*
58) => ({
59    macro_rules! __with_ty__ {( $_ $T:ident ) => ( $($body)* )}
60    use $crate::datatypes::IntegerType::*;
61    match $key_type {
62        Int8 => __with_ty__! { i8 },
63        Int16 => __with_ty__! { i16 },
64        Int32 => __with_ty__! { i32 },
65        Int64 => __with_ty__! { i64 },
66        Int128 => __with_ty__! { i128 },
67        UInt8 => __with_ty__! { u8 },
68        UInt16 => __with_ty__! { u16 },
69        UInt32 => __with_ty__! { u32 },
70        UInt64 => __with_ty__! { u64 },
71        UInt128 => __with_ty__! { u128 },
72    }
73})}
polars_arrow/util/macros.rs

polars_arrow/util/
macros.rs
