Skip to main content

vortex_array/arrays/constant/compute/
not.rs

1// SPDX-License-Identifier: Apache-2.0
2// SPDX-FileCopyrightText: Copyright the Vortex contributors
3
4use vortex_error::VortexResult;
5
6use crate::ArrayRef;
7use crate::IntoArray;
8use crate::array::ArrayView;
9use crate::arrays::Constant;
10use crate::arrays::ConstantArray;
11use crate::scalar::Scalar;
12use crate::scalar_fn::fns::not::NotReduce;
13
14impl NotReduce for Constant {
15    fn invert(array: ArrayView<'_, Constant>) -> VortexResult<Option<ArrayRef>> {
16        let value = match array.scalar().as_bool().value() {
17            Some(b) => Scalar::bool(!b, array.dtype().nullability()),
18            None => Scalar::null(array.dtype().clone()),
19        };
20        Ok(Some(ConstantArray::new(value, array.len()).into_array()))
21    }
22}