vortex_array/arrays/constant/compute/
cast.rs

1// SPDX-License-Identifier: Apache-2.0
2// SPDX-FileCopyrightText: Copyright the Vortex contributors
3
4use vortex_dtype::DType;
5use vortex_error::VortexResult;
6
7use crate::arrays::{ConstantArray, ConstantVTable};
8use crate::compute::{CastKernel, CastKernelAdapter};
9use crate::{ArrayRef, IntoArray, register_kernel};
10
11impl CastKernel for ConstantVTable {
12    fn cast(&self, array: &ConstantArray, dtype: &DType) -> VortexResult<Option<ArrayRef>> {
13        match array.scalar().cast(dtype) {
14            Ok(scalar) => Ok(Some(ConstantArray::new(scalar, array.len()).into_array())),
15            Err(_e) => Ok(None),
16        }
17    }
18}
19
20register_kernel!(CastKernelAdapter(ConstantVTable).lift());