polars-expr 0.54.3

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

use std::sync::Arc;

use polars_core::error::PolarsResult;
use polars_core::prelude::Column;
use polars_ops::series::Roll;
use polars_plan::dsl::{ColumnsUdf, SpecialEq};
use polars_plan::plans::IRBusinessFunction;

pub fn function_expr_to_udf(func: IRBusinessFunction) -> SpecialEq<Arc<dyn ColumnsUdf>> {
    use IRBusinessFunction::*;
    match func {
        BusinessDayCount { week_mask } => {
            map_as_slice!(business_day_count, week_mask)
        },
        AddBusinessDay { week_mask, roll } => {
            map_as_slice!(add_business_days, week_mask, roll)
        },
        IsBusinessDay { week_mask } => {
            map_as_slice!(is_business_day, week_mask)
        },
    }
}

pub(super) fn business_day_count(s: &[Column], week_mask: [bool; 7]) -> PolarsResult<Column> {
    let start = &s[0];
    let end = &s[1];
    let holidays = &s[2];
    polars_ops::prelude::business_day_count(
        start.as_materialized_series(),
        end.as_materialized_series(),
        week_mask,
        holidays.as_materialized_series(),
    )
    .map(Column::from)
}

pub(super) fn add_business_days(
    s: &[Column],
    week_mask: [bool; 7],
    roll: Roll,
) -> PolarsResult<Column> {
    let start = &s[0];
    let n = &s[1];
    let holidays = &s[2];
    polars_ops::prelude::add_business_days(
        start.as_materialized_series(),
        n.as_materialized_series(),
        week_mask,
        holidays.as_materialized_series(),
        roll,
    )
    .map(Column::from)
}

pub(super) fn is_business_day(s: &[Column], week_mask: [bool; 7]) -> PolarsResult<Column> {
    let dates = &s[0];
    let holidays = &s[1];
    polars_ops::prelude::is_business_day(
        dates.as_materialized_series(),
        week_mask,
        holidays.as_materialized_series(),
    )
    .map(Column::from)
}