vortex-array 0.69.0

Vortex in memory columnar data format
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

// 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_error::VortexResult;

use crate::ArrayRef;
use crate::ExecutionCtx;
use crate::arrays::BoolArray;
use crate::arrays::bool::BoolArrayExt;
use crate::arrow::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()?
                .to_mask(array.as_ref().len(), ctx)?,
        ),
    )))
}

pub(super) fn to_arrow_bool(
    array: ArrayRef,
    ctx: &mut ExecutionCtx,
) -> VortexResult<ArrowArrayRef> {
    let bool_array = array.execute::<BoolArray>(ctx)?;
    canonical_bool_to_arrow(&bool_array, ctx)
}