polars-expr 0.54.2

Physical expression implementation of the Polars project.
Documentation 
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

use std::sync::Arc;

use polars_plan::dsl::{ColumnsUdf, SpecialEq};
use polars_plan::plans::IRRangeFunction;

#[cfg(feature = "dtype-datetime")]
mod datetime_range;
mod int_range;
mod linear_space;
#[cfg(feature = "dtype-time")]
mod time_range;
mod utils;

pub fn function_expr_to_udf(func: IRRangeFunction) -> SpecialEq<Arc<dyn ColumnsUdf>> {
    use IRRangeFunction::*;
    match func {
        IntRange { step, dtype } => {
            map_as_slice!(int_range::int_range, step, dtype.clone())
        },
        IntRanges { dtype } => {
            map_as_slice!(int_range::int_ranges, dtype.clone())
        },
        LinearSpace { closed } => {
            map_as_slice!(linear_space::linear_space, closed)
        },
        LinearSpaces {
            closed,
            array_width,
        } => {
            map_as_slice!(linear_space::linear_spaces, closed, array_width)
        },
        #[cfg(feature = "dtype-date")]
        DateRange {
            interval,
            closed,
            arg_type,
        } => {
            map_as_slice!(datetime_range::date_range, interval, closed, arg_type) // TODO! num_samples
        },
        #[cfg(feature = "dtype-date")]
        DateRanges {
            interval,
            closed,
            arg_type,
        } => {
            map_as_slice!(datetime_range::date_ranges, interval, closed, arg_type) // TODO! num_samples
        },
        #[cfg(feature = "dtype-datetime")]
        DatetimeRange {
            interval,
            closed,
            time_unit: _,
            time_zone: _,
            arg_type,
        } => {
            map_as_slice!(datetime_range::datetime_range, interval, closed, arg_type) // TODO! num_samples
        },
        #[cfg(feature = "dtype-datetime")]
        DatetimeRanges {
            interval,
            closed,
            time_unit: _,
            time_zone: _,
            arg_type,
        } => {
            map_as_slice!(datetime_range::datetime_ranges, interval, closed, arg_type) // TODO! num_samples
        },
        #[cfg(feature = "dtype-time")]
        TimeRange { interval, closed } => {
            map_as_slice!(time_range::time_range, interval, closed)
        },
        #[cfg(feature = "dtype-time")]
        TimeRanges { interval, closed } => {
            map_as_slice!(time_range::time_ranges, interval, closed)
        },
    }
}