Skip to main content

vortex_array/arrow/executor/
bool.rs

1// SPDX-License-Identifier: Apache-2.0
2// SPDX-FileCopyrightText: Copyright the Vortex contributors
3
4use std::sync::Arc;
5
6use arrow_array::ArrayRef as ArrowArrayRef;
7use arrow_array::BooleanArray as ArrowBooleanArray;
8use vortex_error::VortexResult;
9
10use crate::ArrayRef;
11use crate::ExecutionCtx;
12use crate::arrays::BoolArray;
13use crate::arrays::bool::BoolArrayExt;
14use crate::arrow::null_buffer::to_null_buffer;
15
16/// Convert a canonical BoolArray directly to Arrow.
17pub fn canonical_bool_to_arrow(
18    array: &BoolArray,
19    ctx: &mut ExecutionCtx,
20) -> VortexResult<ArrowArrayRef> {
21    Ok(Arc::new(ArrowBooleanArray::new(
22        array.to_bit_buffer().into(),
23        to_null_buffer(
24            array
25                .as_ref()
26                .validity()?
27                .to_mask(array.as_ref().len(), ctx)?,
28        ),
29    )))
30}
31
32pub(super) fn to_arrow_bool(
33    array: ArrayRef,
34    ctx: &mut ExecutionCtx,
35) -> VortexResult<ArrowArrayRef> {
36    let bool_array = array.execute::<BoolArray>(ctx)?;
37    canonical_bool_to_arrow(&bool_array, ctx)
38}