vortex-array 0.88.0

Vortex in memory columnar data format
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

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

use vortex_error::VortexResult;
use vortex_error::vortex_err;

use crate::ExecutionCtx;
use crate::array::ArrayView;
use crate::array::OperationsVTable;
use crate::array::ProbeState;
use crate::arrays::Masked;
use crate::arrays::masked::MaskedSlots;
use crate::scalar::Scalar;

impl OperationsVTable<Masked> for Masked {
    type ProbeState = ();

    fn probe_scalar(
        state: &mut ProbeState<'_, Masked>,
        index: usize,
        ctx: &mut ExecutionCtx,
    ) -> VortexResult<Scalar> {
        if !state.is_valid(index, ctx)? {
            return Ok(Scalar::null(state.array().dtype().clone()));
        }
        Ok(state
            .slot(MaskedSlots::CHILD)?
            .ok_or_else(|| vortex_err!("Masked child slot is missing"))?
            .execute_scalar(index, ctx)?
            .into_nullable())
    }

    fn scalar_at(
        array: ArrayView<'_, Masked>,
        index: usize,
        ctx: &mut ExecutionCtx,
    ) -> VortexResult<Scalar> {
        Self::probe_scalar(&mut ProbeState::once(array), index, ctx)
    }
}