vortex_array/arrays/extension/compute/
min_max.rs

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