Function to return vtable from an array

encodings/alp/public-api.lock

@@ -76,7 +76,7 @@ pub fn vortex_alp::ALP::execute(array: &Self::Array, ctx: &mut vortex_array::exe
 
 pub fn vortex_alp::ALP::execute_parent(array: &Self::Array, parent: &vortex_array::array::ArrayRef, child_idx: usize, ctx: &mut vortex_array::executor::ExecutionCtx) -> vortex_error::VortexResult<core::option::Option<vortex_array::array::ArrayRef>>
 
-pub fn vortex_alp::ALP::id(_array: &Self::Array) -> vortex_array::vtable::dyn_::ArrayId
+pub fn vortex_alp::ALP::id(&self) -> vortex_array::vtable::dyn_::ArrayId
 
 pub fn vortex_alp::ALP::len(array: &vortex_alp::ALPArray) -> usize
 
@@ -92,6 +92,8 @@ pub fn vortex_alp::ALP::serialize(metadata: Self::Metadata) -> vortex_error::Vor
 
 pub fn vortex_alp::ALP::stats(array: &vortex_alp::ALPArray) -> vortex_array::stats::array::StatsSetRef<'_>
 
+pub fn vortex_alp::ALP::vtable(_array: &Self::Array) -> &Self
+
 pub fn vortex_alp::ALP::with_children(array: &mut Self::Array, children: alloc::vec::Vec<vortex_array::array::ArrayRef>) -> vortex_error::VortexResult<()>
 
 impl vortex_array::vtable::operations::OperationsVTable<vortex_alp::ALP> for vortex_alp::ALP
@@ -232,7 +234,7 @@ pub fn vortex_alp::ALPRD::execute(array: &Self::Array, ctx: &mut vortex_array::e
 
 pub fn vortex_alp::ALPRD::execute_parent(array: &Self::Array, parent: &vortex_array::array::ArrayRef, child_idx: usize, ctx: &mut vortex_array::executor::ExecutionCtx) -> vortex_error::VortexResult<core::option::Option<vortex_array::array::ArrayRef>>
 
-pub fn vortex_alp::ALPRD::id(_array: &Self::Array) -> vortex_array::vtable::dyn_::ArrayId
+pub fn vortex_alp::ALPRD::id(&self) -> vortex_array::vtable::dyn_::ArrayId
 
 pub fn vortex_alp::ALPRD::len(array: &vortex_alp::ALPRDArray) -> usize
 
@@ -248,6 +250,8 @@ pub fn vortex_alp::ALPRD::serialize(metadata: Self::Metadata) -> vortex_error::V
 
 pub fn vortex_alp::ALPRD::stats(array: &vortex_alp::ALPRDArray) -> vortex_array::stats::array::StatsSetRef<'_>
 
+pub fn vortex_alp::ALPRD::vtable(_array: &Self::Array) -> &Self
+
 pub fn vortex_alp::ALPRD::with_children(array: &mut Self::Array, children: alloc::vec::Vec<vortex_array::array::ArrayRef>) -> vortex_error::VortexResult<()>
 
 impl vortex_array::vtable::operations::OperationsVTable<vortex_alp::ALPRD> for vortex_alp::ALPRD

encodings/alp/src/alp/array.rs

@@ -54,7 +54,11 @@ impl VTable for ALP {
     type OperationsVTable = Self;
     type ValidityVTable = ValidityVTableFromChild;
 
-    fn id(_array: &Self::Array) -> ArrayId {
+    fn vtable(_array: &Self::Array) -> &Self {
+        &ALP
+    }
+
+    fn id(&self) -> ArrayId {
         Self::ID
     }
 

encodings/alp/src/alp_rd/array.rs

@@ -71,7 +71,11 @@ impl VTable for ALPRD {
     type OperationsVTable = Self;
     type ValidityVTable = ValidityVTableFromChild;
 
-    fn id(_array: &Self::Array) -> ArrayId {
+    fn vtable(_array: &Self::Array) -> &Self {
+        &ALPRD
+    }
+
+    fn id(&self) -> ArrayId {
         Self::ID
     }
 

encodings/alp/src/lib.rs

@@ -29,8 +29,8 @@ mod alp_rd;
 
 /// Initialize ALP encoding in the given session.
 pub fn initialize(session: &mut VortexSession) {
-    session.arrays().register(ALP::ID, ALP);
-    session.arrays().register(ALPRD::ID, ALPRD);
+    session.arrays().register(ALP);
+    session.arrays().register(ALPRD);
 
     // Register the ALP-specific NaN count aggregate kernel.
     session.aggregate_fns().register_aggregate_kernel(

encodings/bytebool/public-api.lock

@@ -58,7 +58,7 @@ pub fn vortex_bytebool::ByteBool::execute(array: &Self::Array, _ctx: &mut vortex
 
 pub fn vortex_bytebool::ByteBool::execute_parent(array: &Self::Array, parent: &vortex_array::array::ArrayRef, child_idx: usize, ctx: &mut vortex_array::executor::ExecutionCtx) -> vortex_error::VortexResult<core::option::Option<vortex_array::array::ArrayRef>>
 
-pub fn vortex_bytebool::ByteBool::id(_array: &Self::Array) -> vortex_array::vtable::dyn_::ArrayId
+pub fn vortex_bytebool::ByteBool::id(&self) -> vortex_array::vtable::dyn_::ArrayId
 
 pub fn vortex_bytebool::ByteBool::len(array: &vortex_bytebool::ByteBoolArray) -> usize
 
@@ -74,6 +74,8 @@ pub fn vortex_bytebool::ByteBool::serialize(_metadata: Self::Metadata) -> vortex
 
 pub fn vortex_bytebool::ByteBool::stats(array: &vortex_bytebool::ByteBoolArray) -> vortex_array::stats::array::StatsSetRef<'_>
 
+pub fn vortex_bytebool::ByteBool::vtable(_array: &Self::Array) -> &Self
+
 pub fn vortex_bytebool::ByteBool::with_children(array: &mut Self::Array, children: alloc::vec::Vec<vortex_array::array::ArrayRef>) -> vortex_error::VortexResult<()>
 
 impl vortex_array::vtable::operations::OperationsVTable<vortex_bytebool::ByteBool> for vortex_bytebool::ByteBool

encodings/bytebool/src/array.rs

@@ -48,7 +48,11 @@ impl VTable for ByteBool {
     type OperationsVTable = Self;
     type ValidityVTable = ValidityVTableFromValidityHelper;
 
-    fn id(_array: &Self::Array) -> ArrayId {
+    fn vtable(_array: &Self::Array) -> &Self {
+        &ByteBool
+    }
+
+    fn id(&self) -> ArrayId {
         Self::ID
     }
 

encodings/datetime-parts/public-api.lock

@@ -66,7 +66,7 @@ pub fn vortex_datetime_parts::DateTimeParts::execute(array: &Self::Array, ctx: &
 
 pub fn vortex_datetime_parts::DateTimeParts::execute_parent(array: &Self::Array, parent: &vortex_array::array::ArrayRef, child_idx: usize, ctx: &mut vortex_array::executor::ExecutionCtx) -> vortex_error::VortexResult<core::option::Option<vortex_array::array::ArrayRef>>
 
-pub fn vortex_datetime_parts::DateTimeParts::id(_array: &Self::Array) -> vortex_array::vtable::dyn_::ArrayId
+pub fn vortex_datetime_parts::DateTimeParts::id(&self) -> vortex_array::vtable::dyn_::ArrayId
 
 pub fn vortex_datetime_parts::DateTimeParts::len(array: &vortex_datetime_parts::DateTimePartsArray) -> usize
 
@@ -82,6 +82,8 @@ pub fn vortex_datetime_parts::DateTimeParts::serialize(metadata: Self::Metadata)
 
 pub fn vortex_datetime_parts::DateTimeParts::stats(array: &vortex_datetime_parts::DateTimePartsArray) -> vortex_array::stats::array::StatsSetRef<'_>
 
+pub fn vortex_datetime_parts::DateTimeParts::vtable(_array: &Self::Array) -> &Self
+
 pub fn vortex_datetime_parts::DateTimeParts::with_children(array: &mut Self::Array, children: alloc::vec::Vec<vortex_array::array::ArrayRef>) -> vortex_error::VortexResult<()>
 
 impl vortex_array::vtable::operations::OperationsVTable<vortex_datetime_parts::DateTimeParts> for vortex_datetime_parts::DateTimeParts

encodings/datetime-parts/src/array.rs

@@ -78,7 +78,11 @@ impl VTable for DateTimeParts {
     type OperationsVTable = Self;
     type ValidityVTable = ValidityVTableFromChild;
 
-    fn id(_array: &Self::Array) -> ArrayId {
+    fn vtable(_array: &Self::Array) -> &Self {
+        &DateTimeParts
+    }
+
+    fn id(&self) -> ArrayId {
         Self::ID
     }
 

encodings/datetime-parts/src/lib.rs

@@ -19,7 +19,7 @@ use vortex_session::VortexSession;
 
 /// Initialize datetime-parts encoding in the given session.
 pub fn initialize(session: &mut VortexSession) {
-    session.arrays().register(DateTimeParts::ID, DateTimeParts);
+    session.arrays().register(DateTimeParts);
 
     session.aggregate_fns().register_aggregate_kernel(
         DateTimeParts::ID,

encodings/decimal-byte-parts/public-api.lock

@@ -66,7 +66,7 @@ pub fn vortex_decimal_byte_parts::DecimalByteParts::execute(array: &Self::Array,
 
 pub fn vortex_decimal_byte_parts::DecimalByteParts::execute_parent(array: &Self::Array, parent: &vortex_array::array::ArrayRef, child_idx: usize, ctx: &mut vortex_array::executor::ExecutionCtx) -> vortex_error::VortexResult<core::option::Option<vortex_array::array::ArrayRef>>
 
-pub fn vortex_decimal_byte_parts::DecimalByteParts::id(_array: &Self::Array) -> vortex_array::vtable::dyn_::ArrayId
+pub fn vortex_decimal_byte_parts::DecimalByteParts::id(&self) -> vortex_array::vtable::dyn_::ArrayId
 
 pub fn vortex_decimal_byte_parts::DecimalByteParts::len(array: &vortex_decimal_byte_parts::DecimalBytePartsArray) -> usize
 
@@ -82,6 +82,8 @@ pub fn vortex_decimal_byte_parts::DecimalByteParts::serialize(metadata: Self::Me
 
 pub fn vortex_decimal_byte_parts::DecimalByteParts::stats(array: &vortex_decimal_byte_parts::DecimalBytePartsArray) -> vortex_array::stats::array::StatsSetRef<'_>
 
+pub fn vortex_decimal_byte_parts::DecimalByteParts::vtable(_array: &Self::Array) -> &Self
+
 pub fn vortex_decimal_byte_parts::DecimalByteParts::with_children(array: &mut Self::Array, children: alloc::vec::Vec<vortex_array::array::ArrayRef>) -> vortex_error::VortexResult<()>
 
 impl vortex_array::vtable::operations::OperationsVTable<vortex_decimal_byte_parts::DecimalByteParts> for vortex_decimal_byte_parts::DecimalByteParts

encodings/decimal-byte-parts/src/decimal_byte_parts/mod.rs

@@ -65,7 +65,11 @@ impl VTable for DecimalByteParts {
     type OperationsVTable = Self;
     type ValidityVTable = ValidityVTableFromChild;
 
-    fn id(_array: &Self::Array) -> ArrayId {
+    fn vtable(_array: &Self::Array) -> &Self {
+        &DecimalByteParts
+    }
+
+    fn id(&self) -> ArrayId {
         Self::ID
     }
 

encodings/decimal-byte-parts/src/lib.rs

@@ -21,9 +21,7 @@ use vortex_session::VortexSession;
 
 /// Initialize decimal-byte-parts encoding in the given session.
 pub fn initialize(session: &mut VortexSession) {
-    session
-        .arrays()
-        .register(DecimalByteParts::ID, DecimalByteParts);
+    session.arrays().register(DecimalByteParts);
 
     session.aggregate_fns().register_aggregate_kernel(
         DecimalByteParts::ID,

encodings/fastlanes/public-api.lock

@@ -180,7 +180,7 @@ pub fn vortex_fastlanes::BitPacked::execute(array: &Self::Array, ctx: &mut vorte
 
 pub fn vortex_fastlanes::BitPacked::execute_parent(array: &Self::Array, parent: &vortex_array::array::ArrayRef, child_idx: usize, ctx: &mut vortex_array::executor::ExecutionCtx) -> vortex_error::VortexResult<core::option::Option<vortex_array::array::ArrayRef>>
 
-pub fn vortex_fastlanes::BitPacked::id(_array: &Self::Array) -> vortex_array::vtable::dyn_::ArrayId
+pub fn vortex_fastlanes::BitPacked::id(&self) -> vortex_array::vtable::dyn_::ArrayId
 
 pub fn vortex_fastlanes::BitPacked::len(array: &vortex_fastlanes::BitPackedArray) -> usize
 
@@ -196,6 +196,8 @@ pub fn vortex_fastlanes::BitPacked::serialize(metadata: Self::Metadata) -> vorte
 
 pub fn vortex_fastlanes::BitPacked::stats(array: &vortex_fastlanes::BitPackedArray) -> vortex_array::stats::array::StatsSetRef<'_>
 
+pub fn vortex_fastlanes::BitPacked::vtable(_array: &Self::Array) -> &Self
+
 pub fn vortex_fastlanes::BitPacked::with_children(array: &mut Self::Array, children: alloc::vec::Vec<vortex_array::array::ArrayRef>) -> vortex_error::VortexResult<()>
 
 impl vortex_array::vtable::operations::OperationsVTable<vortex_fastlanes::BitPacked> for vortex_fastlanes::BitPacked
@@ -326,7 +328,7 @@ pub fn vortex_fastlanes::Delta::dtype(array: &vortex_fastlanes::DeltaArray) -> &
 
 pub fn vortex_fastlanes::Delta::execute(array: &Self::Array, ctx: &mut vortex_array::executor::ExecutionCtx) -> vortex_error::VortexResult<vortex_array::executor::ExecutionStep>
 
-pub fn vortex_fastlanes::Delta::id(_array: &Self::Array) -> vortex_array::vtable::dyn_::ArrayId
+pub fn vortex_fastlanes::Delta::id(&self) -> vortex_array::vtable::dyn_::ArrayId
 
 pub fn vortex_fastlanes::Delta::len(array: &vortex_fastlanes::DeltaArray) -> usize
 
@@ -342,6 +344,8 @@ pub fn vortex_fastlanes::Delta::serialize(metadata: Self::Metadata) -> vortex_er
 
 pub fn vortex_fastlanes::Delta::stats(array: &vortex_fastlanes::DeltaArray) -> vortex_array::stats::array::StatsSetRef<'_>
 
+pub fn vortex_fastlanes::Delta::vtable(_array: &Self::Array) -> &Self
+
 pub fn vortex_fastlanes::Delta::with_children(array: &mut Self::Array, children: alloc::vec::Vec<vortex_array::array::ArrayRef>) -> vortex_error::VortexResult<()>
 
 impl vortex_array::vtable::operations::OperationsVTable<vortex_fastlanes::Delta> for vortex_fastlanes::Delta
@@ -466,7 +470,7 @@ pub fn vortex_fastlanes::FoR::execute(array: &Self::Array, ctx: &mut vortex_arra
 
 pub fn vortex_fastlanes::FoR::execute_parent(array: &Self::Array, parent: &vortex_array::array::ArrayRef, child_idx: usize, ctx: &mut vortex_array::executor::ExecutionCtx) -> vortex_error::VortexResult<core::option::Option<vortex_array::array::ArrayRef>>
 
-pub fn vortex_fastlanes::FoR::id(_array: &Self::Array) -> vortex_array::vtable::dyn_::ArrayId
+pub fn vortex_fastlanes::FoR::id(&self) -> vortex_array::vtable::dyn_::ArrayId
 
 pub fn vortex_fastlanes::FoR::len(array: &vortex_fastlanes::FoRArray) -> usize
 
@@ -482,6 +486,8 @@ pub fn vortex_fastlanes::FoR::serialize(metadata: Self::Metadata) -> vortex_erro
 
 pub fn vortex_fastlanes::FoR::stats(array: &vortex_fastlanes::FoRArray) -> vortex_array::stats::array::StatsSetRef<'_>
 
+pub fn vortex_fastlanes::FoR::vtable(_array: &Self::Array) -> &Self
+
 pub fn vortex_fastlanes::FoR::with_children(array: &mut Self::Array, children: alloc::vec::Vec<vortex_array::array::ArrayRef>) -> vortex_error::VortexResult<()>
 
 impl vortex_array::vtable::operations::OperationsVTable<vortex_fastlanes::FoR> for vortex_fastlanes::FoR
@@ -588,7 +594,7 @@ pub fn vortex_fastlanes::RLE::execute(array: &Self::Array, ctx: &mut vortex_arra
 
 pub fn vortex_fastlanes::RLE::execute_parent(array: &Self::Array, parent: &vortex_array::array::ArrayRef, child_idx: usize, ctx: &mut vortex_array::executor::ExecutionCtx) -> vortex_error::VortexResult<core::option::Option<vortex_array::array::ArrayRef>>
 
-pub fn vortex_fastlanes::RLE::id(_array: &Self::Array) -> vortex_array::vtable::dyn_::ArrayId
+pub fn vortex_fastlanes::RLE::id(&self) -> vortex_array::vtable::dyn_::ArrayId
 
 pub fn vortex_fastlanes::RLE::len(array: &vortex_fastlanes::RLEArray) -> usize
 
@@ -604,6 +610,8 @@ pub fn vortex_fastlanes::RLE::serialize(metadata: Self::Metadata) -> vortex_erro
 
 pub fn vortex_fastlanes::RLE::stats(array: &vortex_fastlanes::RLEArray) -> vortex_array::stats::array::StatsSetRef<'_>
 
+pub fn vortex_fastlanes::RLE::vtable(_array: &Self::Array) -> &Self
+
 pub fn vortex_fastlanes::RLE::with_children(array: &mut Self::Array, children: alloc::vec::Vec<vortex_array::array::ArrayRef>) -> vortex_error::VortexResult<()>
 
 impl vortex_array::vtable::operations::OperationsVTable<vortex_fastlanes::RLE> for vortex_fastlanes::RLE

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

@@ -70,7 +70,11 @@ impl VTable for BitPacked {
     type OperationsVTable = Self;
     type ValidityVTable = ValidityVTableFromValidityHelper;
 
-    fn id(_array: &Self::Array) -> ArrayId {
+    fn vtable(_array: &Self::Array) -> &Self {
+        &BitPacked
+    }
+
+    fn id(&self) -> ArrayId {
         Self::ID
     }
 

encodings/fastlanes/src/delta/vtable/mod.rs

@@ -55,7 +55,11 @@ impl VTable for Delta {
     type OperationsVTable = Self;
     type ValidityVTable = Self;
 
-    fn id(_array: &Self::Array) -> ArrayId {
+    fn vtable(_array: &Self::Array) -> &Self {
+        &Delta
+    }
+
+    fn id(&self) -> ArrayId {
         Self::ID
     }
 

encodings/fastlanes/src/for/vtable/mod.rs

@@ -48,7 +48,11 @@ impl VTable for FoR {
     type OperationsVTable = Self;
     type ValidityVTable = ValidityVTableFromChild;
 
-    fn id(_array: &Self::Array) -> ArrayId {
+    fn vtable(_array: &Self::Array) -> &Self {
+        &FoR
+    }
+
+    fn id(&self) -> ArrayId {
         Self::ID
     }
 

encodings/fastlanes/src/lib.rs

@@ -28,10 +28,10 @@ use vortex_session::VortexSession;
 
 /// Initialize fastlanes encodings in the given session.
 pub fn initialize(session: &mut VortexSession) {
-    session.arrays().register(BitPacked::ID, BitPacked);
-    session.arrays().register(Delta::ID, Delta);
-    session.arrays().register(FoR::ID, FoR);
-    session.arrays().register(RLE::ID, RLE);
+    session.arrays().register(BitPacked);
+    session.arrays().register(Delta);
+    session.arrays().register(FoR);
+    session.arrays().register(RLE);
 
     // Register the encoding-specific aggregate kernels.
     session.aggregate_fns().register_aggregate_kernel(
@@ -62,10 +62,10 @@ mod test {
 
     pub static SESSION: LazyLock<VortexSession> = LazyLock::new(|| {
         let session = VortexSession::empty();
-        session.arrays().register(BitPacked::ID, BitPacked);
-        session.arrays().register(Delta::ID, Delta);
-        session.arrays().register(FoR::ID, FoR);
-        session.arrays().register(RLE::ID, RLE);
+        session.arrays().register(BitPacked);
+        session.arrays().register(Delta);
+        session.arrays().register(FoR);
+        session.arrays().register(RLE);
         session
     });
 }

encodings/fastlanes/src/rle/vtable/mod.rs

@@ -62,7 +62,11 @@ impl VTable for RLE {
     type OperationsVTable = Self;
     type ValidityVTable = ValidityVTableFromChildSliceHelper;
 
-    fn id(_array: &Self::Array) -> ArrayId {
+    fn vtable(_array: &Self::Array) -> &Self {
+        &RLE
+    }
+
+    fn id(&self) -> ArrayId {
         Self::ID
     }
 

encodings/fsst/public-api.lock

@@ -68,7 +68,7 @@ pub fn vortex_fsst::FSST::execute(array: &Self::Array, ctx: &mut vortex_array::e
 
 pub fn vortex_fsst::FSST::execute_parent(array: &Self::Array, parent: &vortex_array::array::ArrayRef, child_idx: usize, ctx: &mut vortex_array::executor::ExecutionCtx) -> vortex_error::VortexResult<core::option::Option<vortex_array::array::ArrayRef>>
 
-pub fn vortex_fsst::FSST::id(_array: &Self::Array) -> vortex_array::vtable::dyn_::ArrayId
+pub fn vortex_fsst::FSST::id(&self) -> vortex_array::vtable::dyn_::ArrayId
 
 pub fn vortex_fsst::FSST::len(array: &vortex_fsst::FSSTArray) -> usize
 
@@ -84,6 +84,8 @@ pub fn vortex_fsst::FSST::serialize(metadata: Self::Metadata) -> vortex_error::V
 
 pub fn vortex_fsst::FSST::stats(array: &vortex_fsst::FSSTArray) -> vortex_array::stats::array::StatsSetRef<'_>
 
+pub fn vortex_fsst::FSST::vtable(_array: &Self::Array) -> &Self
+
 pub fn vortex_fsst::FSST::with_children(array: &mut Self::Array, children: alloc::vec::Vec<vortex_array::array::ArrayRef>) -> vortex_error::VortexResult<()>
 
 impl vortex_array::vtable::operations::OperationsVTable<vortex_fsst::FSST> for vortex_fsst::FSST

encodings/fsst/src/array.rs

@@ -81,7 +81,11 @@ impl VTable for FSST {
     type OperationsVTable = Self;
     type ValidityVTable = ValidityVTableFromChild;
 
-    fn id(_array: &Self::Array) -> ArrayId {
+    fn vtable(_array: &Self::Array) -> &Self {
+        &FSST
+    }
+
+    fn id(&self) -> ArrayId {
         Self::ID
     }
 

encodings/pco/public-api.lock

@@ -48,7 +48,7 @@ pub fn vortex_pco::Pco::dtype(array: &vortex_pco::PcoArray) -> &vortex_array::dt
 
 pub fn vortex_pco::Pco::execute(array: &Self::Array, ctx: &mut vortex_array::executor::ExecutionCtx) -> vortex_error::VortexResult<vortex_array::executor::ExecutionStep>
 
-pub fn vortex_pco::Pco::id(_array: &Self::Array) -> vortex_array::vtable::dyn_::ArrayId
+pub fn vortex_pco::Pco::id(&self) -> vortex_array::vtable::dyn_::ArrayId
 
 pub fn vortex_pco::Pco::len(array: &vortex_pco::PcoArray) -> usize
 
@@ -64,6 +64,8 @@ pub fn vortex_pco::Pco::serialize(metadata: Self::Metadata) -> vortex_error::Vor
 
 pub fn vortex_pco::Pco::stats(array: &vortex_pco::PcoArray) -> vortex_array::stats::array::StatsSetRef<'_>
 
+pub fn vortex_pco::Pco::vtable(_array: &Self::Array) -> &Self
+
 pub fn vortex_pco::Pco::with_children(array: &mut Self::Array, children: alloc::vec::Vec<vortex_array::array::ArrayRef>) -> vortex_error::VortexResult<()>
 
 impl vortex_array::vtable::operations::OperationsVTable<vortex_pco::Pco> for vortex_pco::Pco