Skip to main content

vortex_fsst/
ops.rs

1// SPDX-License-Identifier: Apache-2.0
2// SPDX-FileCopyrightText: Copyright the Vortex contributors
3
4use vortex_array::ArrayView;
5use vortex_array::ExecutionCtx;
6use vortex_array::arrays::varbin::varbin_scalar;
7use vortex_array::scalar::Scalar;
8use vortex_array::vtable::OperationsVTable;
9use vortex_buffer::ByteBuffer;
10use vortex_error::VortexExpect;
11use vortex_error::VortexResult;
12
13use crate::FSST;
14use crate::FSSTArrayExt;
15
16impl OperationsVTable<FSST> for FSST {
17    fn scalar_at(
18        array: ArrayView<'_, FSST>,
19        index: usize,
20        ctx: &mut ExecutionCtx,
21    ) -> VortexResult<Scalar> {
22        let compressed = array.codes().execute_scalar(index, ctx)?;
23        let binary_datum = compressed.as_binary().value().vortex_expect("non-null");
24
25        let decoded_buffer = ByteBuffer::from(array.decompressor().decompress(binary_datum));
26        Ok(varbin_scalar(decoded_buffer, array.dtype()))
27    }
28}