vortex-datetime-parts 0.66.0

Vortex physical encoding that compresses temporal components individually
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

// SPDX-License-Identifier: Apache-2.0
// SPDX-FileCopyrightText: Copyright the Vortex contributors

use vortex_array::ArrayRef;
use vortex_array::IntoArray;
use vortex_array::builtins::ArrayBuiltins;
use vortex_array::scalar_fn::fns::mask::MaskReduce;
use vortex_error::VortexResult;

use crate::DateTimeParts;
use crate::DateTimePartsArray;
use crate::DateTimePartsArrayParts;

impl MaskReduce for DateTimeParts {
    fn mask(array: &DateTimePartsArray, mask: &ArrayRef) -> VortexResult<Option<ArrayRef>> {
        let DateTimePartsArrayParts {
            dtype,
            days,
            seconds,
            subseconds,
        } = array.clone().into_parts();
        let masked_days = days.mask(mask.clone())?;
        Ok(Some(
            DateTimePartsArray::try_new(dtype.as_nullable(), masked_days, seconds, subseconds)?
                .into_array(),
        ))
    }
}