arrow2/array/boolean/
ffi.rs

1use crate::{
2    array::{FromFfi, ToFfi},
3    bitmap::align,
4    ffi,
5};
6
7use crate::error::Result;
8
9use super::BooleanArray;
10
11unsafe impl ToFfi for BooleanArray {
12    fn buffers(&self) -> Vec<Option<*const u8>> {
13        vec![
14            self.validity.as_ref().map(|x| x.as_ptr()),
15            Some(self.values.as_ptr()),
16        ]
17    }
18
19    fn offset(&self) -> Option<usize> {
20        let offset = self.values.offset();
21        if let Some(bitmap) = self.validity.as_ref() {
22            if bitmap.offset() == offset {
23                Some(offset)
24            } else {
25                None
26            }
27        } else {
28            Some(offset)
29        }
30    }
31
32    fn to_ffi_aligned(&self) -> Self {
33        let offset = self.values.offset();
34
35        let validity = self.validity.as_ref().map(|bitmap| {
36            if bitmap.offset() == offset {
37                bitmap.clone()
38            } else {
39                align(bitmap, offset)
40            }
41        });
42
43        Self {
44            data_type: self.data_type.clone(),
45            validity,
46            values: self.values.clone(),
47        }
48    }
49}
50
51impl<A: ffi::ArrowArrayRef> FromFfi<A> for BooleanArray {
52    unsafe fn try_from_ffi(array: A) -> Result<Self> {
53        let data_type = array.data_type().clone();
54        let validity = unsafe { array.validity() }?;
55        let values = unsafe { array.bitmap(1) }?;
56        Self::try_new(data_type, values, validity)
57    }
58}
arrow2/array/boolean/ffi.rs

arrow2/array/boolean/
ffi.rs
