1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

use vortex_error::VortexResult;

use crate::array::{SparseArray, SparseEncoding};
use crate::compute::{invert, InvertFn};
use crate::{ArrayData, ArrayLen, IntoArrayData};

impl InvertFn<SparseArray> for SparseEncoding {
    fn invert(&self, array: &SparseArray) -> VortexResult<ArrayData> {
        let inverted_fill = array.fill_scalar().as_bool().invert().into_scalar();
        let inverted_patches = array.patches().map_values(|values| invert(&values))?;
        SparseArray::try_new_from_patches(
            inverted_patches,
            array.len(),
            array.indices_offset(),
            inverted_fill,
        )
        .map(|a| a.into_array())
    }
}