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

// 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::Extension;
use crate::arrays::extension::ExtensionArrayExt;
use crate::arrays::extension::ExtensionSlots;
use crate::scalar::Scalar;

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

    fn probe_scalar(
        state: &mut ProbeState<'_, Extension>,
        index: usize,
        ctx: &mut ExecutionCtx,
    ) -> VortexResult<Scalar> {
        let ext_dtype = state.array().ext_dtype().clone();
        let storage = state
            .slot(ExtensionSlots::STORAGE)?
            .ok_or_else(|| vortex_err!("Extension storage slot is missing"))?
            .execute_scalar(index, ctx)?;
        Ok(Scalar::extension_ref(ext_dtype, storage))
    }

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