vortex-sparse 0.67.0

Vortex Sparse array
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

// SPDX-License-Identifier: Apache-2.0
// SPDX-FileCopyrightText: Copyright the Vortex contributors

use vortex_array::ArrayRef;
use vortex_array::IntoArray;
use vortex_array::builtins::ArrayBuiltins;
use vortex_array::optimizer::rules::ParentRuleSet;
use vortex_array::scalar_fn::fns::cast::CastReduceAdaptor;
use vortex_array::scalar_fn::fns::not::NotReduce;
use vortex_array::scalar_fn::fns::not::NotReduceAdaptor;
use vortex_error::VortexResult;

use crate::Sparse;
use crate::SparseArray;

pub(crate) static RULES: ParentRuleSet<Sparse> = ParentRuleSet::new(&[
    ParentRuleSet::lift(&CastReduceAdaptor(Sparse)),
    ParentRuleSet::lift(&NotReduceAdaptor(Sparse)),
]);

impl NotReduce for Sparse {
    fn invert(array: &SparseArray) -> VortexResult<Option<ArrayRef>> {
        let inverted_fill = array.fill_scalar().as_bool().invert().into_scalar();
        let inverted_patches = array.patches().clone().map_values(|values| values.not())?;
        Ok(Some(
            SparseArray::try_new_from_patches(inverted_patches, inverted_fill)?.into_array(),
        ))
    }
}