Skip to main content

Module take

Module take 

Source
Expand description

Defines take kernel for [Array]

MacrosΒ§

to_indices_identity πŸ”’
to_indices_reinterpret πŸ”’
to_indices_widening πŸ”’

StructsΒ§

TakeOptions
Options that define how take should behave

TraitsΒ§

ToIndices πŸ”’
To avoid generating take implementations for every index type, instead we only generate for UInt32 and UInt64 and coerce inputs to these types

FunctionsΒ§

check_bounds πŸ”’
Verifies that the non-null values of indices are all < len
copy_bit_if_set πŸ”’ ⚠
Read the bit at src_bit_idx from src and, if it is set, write a 1 to dst_bit_idx in dst. Leaves dst_bit_idx unchanged (zero) when the source bit is unset.
is_new_run_take πŸ”’
pack_bit πŸ”’ ⚠
Read the bit at bit_idx from src and return it shifted to out_pos, ready to be OR’d into an output byte accumulator.
physical_indices_for_take πŸ”’
Physical run index for each logical take slot.
take
Take elements by index from [Array], creating a new [Array] from those indexes.
take_arrays
For each [ArrayRef] in the Vec<ArrayRef>, take elements by index and create a new Vec<ArrayRef> from those indices.
take_bits πŸ”’
take_bits_with_validity πŸ”’
Gather value bits and validity bits from two boolean buffers in a single pass. Used when the values array itself has nulls, avoiding two separate take_bits calls.
take_boolean πŸ”’
take implementation for boolean arrays
take_byte_view πŸ”’
take implementation for byte view arrays
take_bytes πŸ”’
take implementation for string arrays
take_dict πŸ”’
take implementation for dictionary arrays
take_fixed_size πŸ”’
Implements the take kernel semantics over a flat [Buffer], interpreting it as a slice of &[[u8; N]], where N is a compile-time constant. The usage of a flat [Buffer] allows using this kernel without an available [ArrowPrimitiveType] (e.g., for [u8; 5]).
take_fixed_size_binary πŸ”’
The take kernel implementation for FixedSizeBinaryArray.
take_fixed_size_list πŸ”’
take implementation for FixedSizeListArray
take_fixed_size_list_primitive πŸ”’
take_impl πŸ”’
take_list πŸ”’
take implementation for list arrays
take_list_view πŸ”’
take_native πŸ”’
take_nulls πŸ”’
take_primitive πŸ”’
take implementation for all primitive arrays
take_record_batch
Take rows by index from [RecordBatch] and returns a new [RecordBatch] from those indexes.
take_run πŸ”’
take implementation for run arrays
take_union_type_ids πŸ”’
Takes union type ids, substituting a valid type id for null take indices.
take_value_indices_from_fixed_size_list πŸ”’
Takes/filters a fixed size list array’s inner data using the offsets of the list array.