vortex_array/arrays/constant/compute/
invert.rs

1use vortex_error::VortexResult;
2use vortex_scalar::Scalar;
3
4use crate::arrays::{ConstantArray, ConstantVTable};
5use crate::compute::{InvertKernel, InvertKernelAdapter};
6use crate::{ArrayRef, IntoArray, register_kernel};
7
8impl InvertKernel for ConstantVTable {
9    fn invert(&self, array: &ConstantArray) -> VortexResult<ArrayRef> {
10        match array.scalar().as_bool().value() {
11            None => Ok(array.to_array()),
12            Some(b) => Ok(ConstantArray::new(
13                Scalar::bool(!b, array.dtype().nullability()),
14                array.len(),
15            )
16            .into_array()),
17        }
18    }
19}
20
21register_kernel!(InvertKernelAdapter(ConstantVTable).lift());
22
23#[cfg(test)]
24mod tests {
25    use vortex_dtype::Nullability::Nullable;
26    use vortex_scalar::Scalar;
27
28    use crate::Array;
29    use crate::arrays::ConstantArray;
30    use crate::compute::invert;
31
32    #[test]
33    fn invert_nullable_const() {
34        let constant = ConstantArray::new(Scalar::bool(false, Nullable), 10);
35
36        let inverted = invert(constant.as_ref()).unwrap();
37        assert_eq!(inverted.dtype(), constant.dtype());
38
39        let orig = invert(inverted.as_ref()).unwrap();
40
41        assert_eq!(orig.dtype(), constant.dtype());
42        assert_eq!(orig.as_constant(), constant.as_constant())
43    }
44}
vortex_array/arrays/constant/compute/invert.rs

vortex_array/arrays/constant/compute/
invert.rs
