why not

Signed-off-by: Adam Gutglick <adam@spiraldb.com>

Author: AdamGS

vortex-array/src/arrays/bool/vtable/visitor.rs

@@ -3,9 +3,12 @@
 
 use crate::ArrayBufferVisitor;
 use crate::ArrayChildVisitor;
+use crate::ArrayRef;
 use crate::arrays::BoolArray;
 use crate::arrays::BoolVTable;
 use crate::vtable::VisitorVTable;
+use crate::vtable::validity_nchildren;
+use crate::vtable::validity_to_child;
 
 impl VisitorVTable<BoolVTable> for BoolVTable {
     fn visit_buffers(array: &BoolArray, visitor: &mut dyn ArrayBufferVisitor) {
@@ -15,4 +18,15 @@ impl VisitorVTable<BoolVTable> for BoolVTable {
     fn visit_children(array: &BoolArray, visitor: &mut dyn ArrayChildVisitor) {
         visitor.visit_validity(&array.validity, array.len());
     }
+
+    fn nchildren(array: &BoolArray) -> usize {
+        validity_nchildren(&array.validity)
+    }
+
+    fn nth_child(array: &BoolArray, idx: usize) -> Option<ArrayRef> {
+        match idx {
+            0 => validity_to_child(&array.validity, array.len()),
+            _ => None,
+        }
+    }
 }

vortex-array/src/arrays/chunked/vtable/visitor.rs

@@ -26,4 +26,15 @@ impl VisitorVTable<ChunkedVTable> for ChunkedVTable {
             visitor.visit_child(chunk);
         }
     }
+
+    fn nchildren(array: &ChunkedArray) -> usize {
+        1 + array.chunks().len()
+    }
+
+    fn nth_child(array: &ChunkedArray, idx: usize) -> Option<crate::ArrayRef> {
+        match idx {
+            0 => Some(array.chunk_offsets.to_array()),
+            n => array.chunks().get(n - 1).cloned(),
+        }
+    }
 }

vortex-array/src/arrays/constant/vtable/visitor.rs

@@ -5,6 +5,7 @@ use vortex_buffer::ByteBufferMut;
 
 use crate::ArrayBufferVisitor;
 use crate::ArrayChildVisitor;
+use crate::ArrayRef;
 use crate::arrays::ConstantArray;
 use crate::arrays::ConstantVTable;
 use crate::buffer::BufferHandle;
@@ -21,4 +22,12 @@ impl VisitorVTable<ConstantVTable> for ConstantVTable {
     }
 
     fn visit_children(_array: &ConstantArray, _visitor: &mut dyn ArrayChildVisitor) {}
+
+    fn nchildren(_array: &ConstantArray) -> usize {
+        0
+    }
+
+    fn nth_child(_array: &ConstantArray, _idx: usize) -> Option<ArrayRef> {
+        None
+    }
 }

vortex-array/src/arrays/decimal/vtable/visitor.rs

@@ -3,10 +3,13 @@
 
 use crate::ArrayBufferVisitor;
 use crate::ArrayChildVisitor;
+use crate::ArrayRef;
 use crate::arrays::DecimalArray;
 use crate::arrays::DecimalVTable;
 use crate::vtable::ValidityHelper;
 use crate::vtable::VisitorVTable;
+use crate::vtable::validity_nchildren;
+use crate::vtable::validity_to_child;
 
 impl VisitorVTable<DecimalVTable> for DecimalVTable {
     fn visit_buffers(array: &DecimalArray, visitor: &mut dyn ArrayBufferVisitor) {
@@ -16,4 +19,15 @@ impl VisitorVTable<DecimalVTable> for DecimalVTable {
     fn visit_children(array: &DecimalArray, visitor: &mut dyn ArrayChildVisitor) {
         visitor.visit_validity(array.validity(), array.len())
     }
+
+    fn nchildren(array: &DecimalArray) -> usize {
+        validity_nchildren(array.validity())
+    }
+
+    fn nth_child(array: &DecimalArray, idx: usize) -> Option<ArrayRef> {
+        match idx {
+            0 => validity_to_child(array.validity(), array.len()),
+            _ => None,
+        }
+    }
 }

vortex-array/src/arrays/dict/vtable/visitor.rs

@@ -4,6 +4,7 @@
 use super::DictVTable;
 use crate::ArrayBufferVisitor;
 use crate::ArrayChildVisitor;
+use crate::ArrayRef;
 use crate::arrays::dict::DictArray;
 use crate::vtable::VisitorVTable;
 
@@ -14,4 +15,16 @@ impl VisitorVTable<DictVTable> for DictVTable {
         visitor.visit_child("codes", array.codes());
         visitor.visit_child("values", array.values());
     }
+
+    fn nchildren(_array: &DictArray) -> usize {
+        2
+    }
+
+    fn nth_child(array: &DictArray, idx: usize) -> Option<ArrayRef> {
+        match idx {
+            0 => Some(array.codes().clone()),
+            1 => Some(array.values().clone()),
+            _ => None,
+        }
+    }
 }

vortex-array/src/arrays/extension/vtable/visitor.rs

@@ -3,6 +3,7 @@
 
 use crate::ArrayBufferVisitor;
 use crate::ArrayChildVisitor;
+use crate::ArrayRef;
 use crate::arrays::extension::ExtensionArray;
 use crate::arrays::extension::ExtensionVTable;
 use crate::vtable::VisitorVTable;
@@ -13,4 +14,15 @@ impl VisitorVTable<ExtensionVTable> for ExtensionVTable {
     fn visit_children(array: &ExtensionArray, visitor: &mut dyn ArrayChildVisitor) {
         visitor.visit_child("storage", &array.storage);
     }
+
+    fn nchildren(_array: &ExtensionArray) -> usize {
+        1
+    }
+
+    fn nth_child(array: &ExtensionArray, idx: usize) -> Option<ArrayRef> {
+        match idx {
+            0 => Some(array.storage.clone()),
+            _ => None,
+        }
+    }
 }

vortex-array/src/arrays/filter/vtable.rs

@@ -176,6 +176,17 @@ impl VisitorVTable<FilterVTable> for FilterVTable {
     fn visit_children(array: &FilterArray, visitor: &mut dyn ArrayChildVisitor) {
         visitor.visit_child("child", &array.child);
     }
+
+    fn nchildren(_array: &FilterArray) -> usize {
+        1
+    }
+
+    fn nth_child(array: &FilterArray, idx: usize) -> Option<ArrayRef> {
+        match idx {
+            0 => Some(array.child.clone()),
+            _ => None,
+        }
+    }
 }
 
 pub struct FilterMetadata(pub(super) Mask);

vortex-array/src/arrays/fixed_size_list/vtable/visitor.rs

@@ -3,10 +3,13 @@
 
 use crate::ArrayBufferVisitor;
 use crate::ArrayChildVisitor;
+use crate::ArrayRef;
 use crate::arrays::FixedSizeListArray;
 use crate::arrays::FixedSizeListVTable;
 use crate::vtable::ValidityHelper;
 use crate::vtable::VisitorVTable;
+use crate::vtable::validity_nchildren;
+use crate::vtable::validity_to_child;
 
 impl VisitorVTable<FixedSizeListVTable> for FixedSizeListVTable {
     fn visit_buffers(_array: &FixedSizeListArray, _visitor: &mut dyn ArrayBufferVisitor) {
@@ -18,4 +21,16 @@ impl VisitorVTable<FixedSizeListVTable> for FixedSizeListVTable {
         visitor.visit_child("elements", array.elements());
         visitor.visit_validity(array.validity(), array.len());
     }
+
+    fn nchildren(array: &FixedSizeListArray) -> usize {
+        1 + validity_nchildren(array.validity())
+    }
+
+    fn nth_child(array: &FixedSizeListArray, idx: usize) -> Option<ArrayRef> {
+        match idx {
+            0 => Some(array.elements().clone()),
+            1 => validity_to_child(array.validity(), array.len()),
+            _ => None,
+        }
+    }
 }

vortex-array/src/arrays/list/vtable/visitor.rs

@@ -3,10 +3,13 @@
 
 use crate::ArrayBufferVisitor;
 use crate::ArrayChildVisitor;
+use crate::ArrayRef;
 use crate::arrays::ListArray;
 use crate::arrays::ListVTable;
 use crate::vtable::ValidityHelper;
 use crate::vtable::VisitorVTable;
+use crate::vtable::validity_nchildren;
+use crate::vtable::validity_to_child;
 
 impl VisitorVTable<ListVTable> for ListVTable {
     fn visit_buffers(_array: &ListArray, _visitor: &mut dyn ArrayBufferVisitor) {}
@@ -16,4 +19,17 @@ impl VisitorVTable<ListVTable> for ListVTable {
         visitor.visit_child("offsets", array.offsets());
         visitor.visit_validity(array.validity(), array.len());
     }
+
+    fn nchildren(array: &ListArray) -> usize {
+        2 + validity_nchildren(array.validity())
+    }
+
+    fn nth_child(array: &ListArray, idx: usize) -> Option<ArrayRef> {
+        match idx {
+            0 => Some(array.elements().clone()),
+            1 => Some(array.offsets().clone()),
+            2 => validity_to_child(array.validity(), array.len()),
+            _ => None,
+        }
+    }
 }

vortex-array/src/arrays/listview/vtable/visitor.rs

@@ -3,10 +3,13 @@
 
 use crate::ArrayBufferVisitor;
 use crate::ArrayChildVisitor;
+use crate::ArrayRef;
 use crate::arrays::ListViewArray;
 use crate::arrays::ListViewVTable;
 use crate::vtable::ValidityHelper;
 use crate::vtable::VisitorVTable;
+use crate::vtable::validity_nchildren;
+use crate::vtable::validity_to_child;
 
 impl VisitorVTable<ListViewVTable> for ListViewVTable {
     fn visit_buffers(_array: &ListViewArray, _visitor: &mut dyn ArrayBufferVisitor) {
@@ -19,4 +22,18 @@ impl VisitorVTable<ListViewVTable> for ListViewVTable {
         visitor.visit_child("sizes", array.sizes());
         visitor.visit_validity(array.validity(), array.len());
     }
+
+    fn nchildren(array: &ListViewArray) -> usize {
+        3 + validity_nchildren(array.validity())
+    }
+
+    fn nth_child(array: &ListViewArray, idx: usize) -> Option<ArrayRef> {
+        match idx {
+            0 => Some(array.elements().clone()),
+            1 => Some(array.offsets().clone()),
+            2 => Some(array.sizes().clone()),
+            3 => validity_to_child(array.validity(), array.len()),
+            _ => None,
+        }
+    }
 }