vortex-arrow 0.85.0

Apache Arrow interoperability for Vortex arrays
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

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

use std::sync::Arc;

use arrow_array::ArrayRef as ArrowArrayRef;
use arrow_array::BooleanArray as ArrowBooleanArray;
use vortex_array::ArrayRef;
use vortex_array::ExecutionCtx;
use vortex_array::arrays::BoolArray;
use vortex_array::arrays::bool::BoolArrayExt;
use vortex_array::dtype::DType;
use vortex_error::VortexResult;
use vortex_error::vortex_ensure;

use crate::null_buffer::to_null_buffer;

/// Convert a canonical BoolArray directly to Arrow.
pub fn canonical_bool_to_arrow(
    array: &BoolArray,
    ctx: &mut ExecutionCtx,
) -> VortexResult<ArrowArrayRef> {
    Ok(Arc::new(ArrowBooleanArray::new(
        array.to_bit_buffer().into(),
        to_null_buffer(
            array
                .as_ref()
                .validity()?
                .execute_mask(array.as_ref().len(), ctx)?,
        ),
    )))
}

pub(super) fn to_arrow_bool(
    array: ArrayRef,
    ctx: &mut ExecutionCtx,
) -> VortexResult<ArrowArrayRef> {
    vortex_ensure!(
        matches!(array.dtype(), DType::Bool(_)),
        "Cannot convert Vortex array with dtype {} to an Arrow Boolean array",
        array.dtype()
    );
    let bool_array = array.execute::<BoolArray>(ctx)?;
    canonical_bool_to_arrow(&bool_array, ctx)
}