Skip to content

Commit

Permalink
fix: if target is non-null, use non-null sentinel value
Browse files Browse the repository at this point in the history
  • Loading branch information
wjones127 committed Nov 17, 2023
1 parent ba214ab commit d422617
Showing 1 changed file with 17 additions and 10 deletions.
27 changes: 17 additions & 10 deletions arrow-cast/src/cast.rs
Expand Up @@ -284,7 +284,8 @@ pub fn can_cast_types(from_type: &DataType, to_type: &DataType) -> bool {
/// * List to List: the underlying data type is cast
/// * List to FixedSizeList: the underlying data type is cast. If the list size is different
/// than the FixedSizeList size and safe casting is requested, then lists are
/// truncated or filled will nulls to match the FixedSizeList size.
/// truncated or filled will some value to achieve the requested size. If the output
/// field is nullable, the fill value is null, otherwise it is the first value.
/// * Primitive to List: a list array with 1 value per slot is created
/// * Date32 and Date64: precision lost when going to higher interval
/// * Time32 and Time64: precision lost when going to higher interval
Expand Down Expand Up @@ -3244,21 +3245,27 @@ where
// to use take to resize those slots.
if !can_zero_copy_values {
let (take_indices, null_buffer) = build_take_indices(array, size);
let values = take(cast_array.as_ref(), take_indices.as_ref(), None)?;
let values = values.to_data().into_builder().nulls(null_buffer).build()?;
cast_array = make_array(values);
cast_array = take(cast_array.as_ref(), take_indices.as_ref(), None)?;
if field.is_nullable() {
cast_array = make_array(
cast_array
.to_data()
.into_builder()
.nulls(null_buffer)
.build()?,
);
}
}

// Build the FixedSizeListArray
let dest_type = DataType::FixedSizeList(field.clone(), size);
let builder = data
let array_data = data
.into_builder()
.buffers(vec![])
.data_type(dest_type)
.child_data(vec![cast_array.into_data()]);
.child_data(vec![cast_array.into_data()])
.build()?;

// Safety
// Data was valid before
let array_data = unsafe { builder.build_unchecked() };
let list = FixedSizeListArray::from(array_data);
Ok(Arc::new(list) as ArrayRef)
}
Expand Down Expand Up @@ -3328,7 +3335,7 @@ where
/// filled with 0, so that they will be populated with the first value. This
/// means that if only the null slots are improperly sized, then the null buffer
/// returned will be None.
///
///
/// If there are valid slots in the list array that are too long, they will be
/// truncated. If there are valid slots that are too short, then they will have
/// their corresponding slots in the null buffer set to null. Thus, if there are
Expand Down

0 comments on commit d422617

Please sign in to comment.