wip

Signed-off-by: Alexander Droste <[email protected]>

Author: 0ax1

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

@@ -81,6 +81,19 @@ fn _take<I: IntegerPType, O: IntegerPType>(
     let mut current_offset = 0u64;
     new_offsets.append_zero();
 
+    // Approximate the total element count.
+    let approx_total_count = indices
+        .iter()
+        .map(|idx| {
+            let idx = idx
+                .to_usize()
+                .unwrap_or_else(|| vortex_panic!("Failed to convert index to usize: {}", idx));
+            (offsets[idx + 1] - offsets[idx]).as_() as usize
+        })
+        .max()
+        .unwrap_or(0)
+        * array.len();
+
     for &data_idx in indices {
         let data_idx = data_idx
             .to_usize()