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(array: &BoolArray) -> VortexResult<ArrowArrayRef> {
18    Ok(Arc::new(ArrowBooleanArray::new(
19        array.to_bit_buffer().into(),
20        to_null_buffer(array.validity_mask()?),
21    )))
22}
23
24pub(super) fn to_arrow_bool(
25    array: ArrayRef,
26    ctx: &mut ExecutionCtx,
27) -> VortexResult<ArrowArrayRef> {
28    let bool_array = array.execute::<BoolArray>(ctx)?;
29    canonical_bool_to_arrow(&bool_array)
30}