Skip to main content

vortex_array/arrays/extension/compute/
min_max.rs

1// SPDX-License-Identifier: Apache-2.0
2// SPDX-FileCopyrightText: Copyright the Vortex contributors
3
4use vortex_error::VortexResult;
5
6use crate::arrays::ExtensionArray;
7use crate::arrays::ExtensionVTable;
8use crate::compute::MinMaxKernel;
9use crate::compute::MinMaxKernelAdapter;
10use crate::compute::MinMaxResult;
11use crate::compute::{self};
12use crate::dtype::Nullability;
13use crate::register_kernel;
14use crate::scalar::Scalar;
15
16impl MinMaxKernel for ExtensionVTable {
17    fn min_max(&self, array: &ExtensionArray) -> VortexResult<Option<MinMaxResult>> {
18        let non_nullable_ext_dtype = array.ext_dtype().with_nullability(Nullability::NonNullable);
19        Ok(
20            compute::min_max(array.storage_array())?.map(|MinMaxResult { min, max }| {
21                MinMaxResult {
22                    min: Scalar::extension_ref(non_nullable_ext_dtype.clone(), min),
23                    max: Scalar::extension_ref(non_nullable_ext_dtype, max),
24                }
25            }),
26        )
27    }
28}
29
30register_kernel!(MinMaxKernelAdapter(ExtensionVTable).lift());