Skip to content

Commit df9edda

Browse files
authored
Chore: rename unpack functions (#5428)
Just so the new unpack functions that work with vectors dont name collide Signed-off-by: Connor Tsui <connor.tsui20@gmail.com>
1 parent 3438bca commit df9edda

File tree

2 files changed

+19
-19
lines changed

2 files changed

+19
-19
lines changed

encodings/fastlanes/src/bitpacking/array/bitpack_decompress.rs

Lines changed: 16 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -17,18 +17,18 @@ use vortex_scalar::Scalar;
1717
use crate::BitPackedArray;
1818
use crate::unpack_iter::BitPacked;
1919

20-
pub fn unpack(array: &BitPackedArray) -> PrimitiveArray {
21-
match_each_integer_ptype!(array.ptype(), |P| { unpack_primitive::<P>(array) })
20+
pub fn unpack_array(array: &BitPackedArray) -> PrimitiveArray {
21+
match_each_integer_ptype!(array.ptype(), |P| { unpack_primitive_array::<P>(array) })
2222
}
2323

24-
pub fn unpack_primitive<T: BitPacked>(array: &BitPackedArray) -> PrimitiveArray {
24+
pub fn unpack_primitive_array<T: BitPacked>(array: &BitPackedArray) -> PrimitiveArray {
2525
let mut builder = PrimitiveBuilder::with_capacity(array.dtype().nullability(), array.len());
26-
unpack_into::<T>(array, &mut builder);
26+
unpack_into_primitive_builder::<T>(array, &mut builder);
2727
assert_eq!(builder.len(), array.len());
2828
builder.finish_into_primitive()
2929
}
3030

31-
pub(crate) fn unpack_into<T: BitPacked>(
31+
pub(crate) fn unpack_into_primitive_builder<T: BitPacked>(
3232
array: &BitPackedArray,
3333
// TODO(ngates): do we want to use fastlanes alignment for this buffer?
3434
builder: &mut PrimitiveBuilder<T>,
@@ -207,23 +207,23 @@ mod tests {
207207
fn test_all_zeros() {
208208
let zeros = buffer![0u16, 0, 0, 0].into_array().to_primitive();
209209
let bitpacked = bitpack_encode(&zeros, 0, None).unwrap();
210-
let actual = unpack(&bitpacked);
210+
let actual = unpack_array(&bitpacked);
211211
assert_arrays_eq!(actual, PrimitiveArray::from_iter([0u16, 0, 0, 0]));
212212
}
213213

214214
#[test]
215215
fn test_simple_patches() {
216216
let zeros = buffer![0u16, 1, 0, 1].into_array().to_primitive();
217217
let bitpacked = bitpack_encode(&zeros, 0, None).unwrap();
218-
let actual = unpack(&bitpacked);
218+
let actual = unpack_array(&bitpacked);
219219
assert_arrays_eq!(actual, PrimitiveArray::from_iter([0u16, 1, 0, 1]));
220220
}
221221

222222
#[test]
223223
fn test_one_full_chunk() {
224224
let zeros = BufferMut::from_iter(0u16..1024).into_array().to_primitive();
225225
let bitpacked = bitpack_encode(&zeros, 10, None).unwrap();
226-
let actual = unpack(&bitpacked);
226+
let actual = unpack_array(&bitpacked);
227227
assert_arrays_eq!(actual, PrimitiveArray::from_iter(0u16..1024));
228228
}
229229

@@ -234,7 +234,7 @@ mod tests {
234234
.to_primitive();
235235
let bitpacked = bitpack_encode(&zeros, 10, None).unwrap();
236236
assert!(bitpacked.patches().is_some());
237-
let actual = unpack(&bitpacked);
237+
let actual = unpack_array(&bitpacked);
238238
assert_arrays_eq!(
239239
actual,
240240
PrimitiveArray::from_iter((5u16..1029).chain(5u16..1029).chain(5u16..1029))
@@ -246,7 +246,7 @@ mod tests {
246246
let zeros = BufferMut::from_iter(0u16..1025).into_array().to_primitive();
247247
let bitpacked = bitpack_encode(&zeros, 11, None).unwrap();
248248
assert!(bitpacked.patches().is_none());
249-
let actual = unpack(&bitpacked);
249+
let actual = unpack_array(&bitpacked);
250250
assert_arrays_eq!(actual, PrimitiveArray::from_iter(0u16..1025));
251251
}
252252

@@ -258,7 +258,7 @@ mod tests {
258258
let bitpacked = bitpack_encode(&zeros, 10, None).unwrap();
259259
assert_eq!(bitpacked.len(), 1025);
260260
assert!(bitpacked.patches().is_some());
261-
let actual = unpack(&bitpacked);
261+
let actual = unpack_array(&bitpacked);
262262
assert_arrays_eq!(actual, PrimitiveArray::from_iter(512u16..1537));
263263
}
264264

@@ -271,7 +271,7 @@ mod tests {
271271
assert_eq!(bitpacked.len(), 1025);
272272
assert!(bitpacked.patches().is_some());
273273
let bitpacked = bitpacked.slice(1023..1025);
274-
let actual = unpack(bitpacked.as_::<BitPackedVTable>());
274+
let actual = unpack_array(bitpacked.as_::<BitPackedVTable>());
275275
assert_arrays_eq!(actual, PrimitiveArray::from_iter([1535u16, 1536]));
276276
}
277277

@@ -284,7 +284,7 @@ mod tests {
284284
assert_eq!(bitpacked.len(), 2229);
285285
assert!(bitpacked.patches().is_some());
286286
let bitpacked = bitpacked.slice(1023..2049);
287-
let actual = unpack(bitpacked.as_::<BitPackedVTable>());
287+
let actual = unpack_array(bitpacked.as_::<BitPackedVTable>());
288288
assert_arrays_eq!(
289289
actual,
290290
PrimitiveArray::from_iter((1023u16..2049).map(|x| x + 512))
@@ -297,7 +297,7 @@ mod tests {
297297
let bitpacked = bitpack_encode(&empty, 0, None).unwrap();
298298

299299
let mut builder = PrimitiveBuilder::<u32>::new(Nullability::NonNullable);
300-
unpack_into(&bitpacked, &mut builder);
300+
unpack_into_primitive_builder(&bitpacked, &mut builder);
301301

302302
let result = builder.finish_into_primitive();
303303
assert_eq!(
@@ -320,7 +320,7 @@ mod tests {
320320

321321
// Unpack into a new builder.
322322
let mut builder = PrimitiveBuilder::<u32>::with_capacity(Nullability::Nullable, 5);
323-
unpack_into(&bitpacked, &mut builder);
323+
unpack_into_primitive_builder(&bitpacked, &mut builder);
324324

325325
let result = builder.finish_into_primitive();
326326

@@ -351,7 +351,7 @@ mod tests {
351351

352352
// Unpack into a new builder.
353353
let mut builder = PrimitiveBuilder::<u32>::with_capacity(Nullability::NonNullable, 100);
354-
unpack_into(&bitpacked, &mut builder);
354+
unpack_into_primitive_builder(&bitpacked, &mut builder);
355355

356356
let result = builder.finish_into_primitive();
357357

encodings/fastlanes/src/bitpacking/vtable/canonical.rs

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -7,17 +7,17 @@ use vortex_array::vtable::CanonicalVTable;
77
use vortex_dtype::match_each_integer_ptype;
88
use vortex_error::VortexExpect;
99

10-
use crate::bitpack_decompress::{unpack, unpack_into};
10+
use crate::bitpack_decompress::{unpack_array, unpack_into_primitive_builder};
1111
use crate::{BitPackedArray, BitPackedVTable};
1212

1313
impl CanonicalVTable<BitPackedVTable> for BitPackedVTable {
1414
fn canonicalize(array: &BitPackedArray) -> Canonical {
15-
Canonical::Primitive(unpack(array))
15+
Canonical::Primitive(unpack_array(array))
1616
}
1717

1818
fn append_to_builder(array: &BitPackedArray, builder: &mut dyn ArrayBuilder) {
1919
match_each_integer_ptype!(array.ptype(), |T| {
20-
unpack_into::<T>(
20+
unpack_into_primitive_builder::<T>(
2121
array,
2222
builder
2323
.as_any_mut()

0 commit comments

Comments
 (0)