feat: exploit PiecewiseSequence for faster take (#8802)

1. Extension pushes take down into its children. This ensures extension
types benefit from the other PiecewiseSequence optimizations.
2. Ensure `List(List(...(List(Primitive))).take(indices)` keeps
broadening the slices all the way down to the Primitive array.
3. Implement list view rebuild using PiecewiseSequence, since rebuild is
just take anyway!

---------

Signed-off-by: Daniel King <dan@spiraldb.com>

Author: danking

vortex-array/src/arrays/extension/compute/rules.rs

@@ -11,6 +11,7 @@ use crate::arrays::ConstantArray;
 use crate::arrays::Extension;
 use crate::arrays::ExtensionArray;
 use crate::arrays::Filter;
+use crate::arrays::dict::TakeReduceAdaptor;
 use crate::arrays::extension::ExtensionArrayExt;
 use crate::arrays::filter::FilterReduceAdaptor;
 use crate::arrays::slice::SliceReduceAdaptor;
@@ -50,6 +51,7 @@ pub(crate) const PARENT_RULES: ParentRuleSet<Extension> = ParentRuleSet::new(&[
     ParentRuleSet::lift(&FilterReduceAdaptor(Extension)),
     ParentRuleSet::lift(&MaskReduceAdaptor(Extension)),
     ParentRuleSet::lift(&SliceReduceAdaptor(Extension)),
+    ParentRuleSet::lift(&TakeReduceAdaptor(Extension)),
 ]);
 
 /// Push filter operations into the storage array of an extension array.

vortex-array/src/arrays/extension/compute/take.rs

@@ -10,8 +10,24 @@ use crate::array::ArrayView;
 use crate::arrays::Extension;
 use crate::arrays::ExtensionArray;
 use crate::arrays::dict::TakeExecute;
+use crate::arrays::dict::TakeReduce;
 use crate::arrays::extension::ExtensionArrayExt;
 
+impl TakeReduce for Extension {
+    fn take(array: ArrayView<'_, Extension>, indices: &ArrayRef) -> VortexResult<Option<ArrayRef>> {
+        let taken_storage = array.storage_array().take(indices.clone())?;
+        Ok(Some(
+            ExtensionArray::new(
+                array
+                    .ext_dtype()
+                    .with_nullability(taken_storage.dtype().nullability()),
+                taken_storage,
+            )
+            .into_array(),
+        ))
+    }
+}
+
 impl TakeExecute for Extension {
     fn take(
         array: ArrayView<'_, Extension>,