vortex-btrblocks 0.67.0

BtrBlocks style compressor
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

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

//! Specialized compressor for DateTimeParts metadata.

use vortex_array::ArrayRef;
use vortex_array::Canonical;
use vortex_array::IntoArray;
use vortex_array::ToCanonical;
use vortex_array::arrays::TemporalArray;
use vortex_datetime_parts::DateTimePartsArray;
use vortex_datetime_parts::TemporalParts;
use vortex_datetime_parts::split_temporal;
use vortex_error::VortexResult;

use crate::BtrBlocksCompressor;
use crate::CanonicalCompressor;
use crate::CompressorContext;
use crate::Excludes;

/// Compress a temporal array into a `DateTimePartsArray`.
pub fn compress_temporal(
    compressor: &BtrBlocksCompressor,
    array: TemporalArray,
) -> VortexResult<ArrayRef> {
    let dtype = array.dtype().clone();
    let TemporalParts {
        days,
        seconds,
        subseconds,
    } = split_temporal(array)?;

    let ctx = CompressorContext::default().descend();

    let days = compressor.compress_canonical(
        Canonical::Primitive(days.to_primitive().narrow()?),
        ctx,
        Excludes::none(),
    )?;
    let seconds = compressor.compress_canonical(
        Canonical::Primitive(seconds.to_primitive().narrow()?),
        ctx,
        Excludes::none(),
    )?;
    let subseconds = compressor.compress_canonical(
        Canonical::Primitive(subseconds.to_primitive().narrow()?),
        ctx,
        Excludes::none(),
    )?;

    Ok(DateTimePartsArray::try_new(dtype, days, seconds, subseconds)?.into_array())
}