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::arrow::null_buffer::to_null_buffer;
14
15/// Convert a canonical BoolArray directly to Arrow.
16pub fn canonical_bool_to_arrow(array: &BoolArray) -> VortexResult<ArrowArrayRef> {
17    Ok(Arc::new(ArrowBooleanArray::new(
18        array.to_bit_buffer().into(),
19        to_null_buffer(array.validity_mask()?),
20    )))
21}
22
23pub(super) fn to_arrow_bool(
24    array: ArrayRef,
25    ctx: &mut ExecutionCtx,
26) -> VortexResult<ArrowArrayRef> {
27    let bool_array = array.execute::<BoolArray>(ctx)?;
28    canonical_bool_to_arrow(&bool_array)
29}