Take: Simplify TakeMetaFunction even further

apache · Jun 19, 2024 · 28da5e6 · 28da5e6
1 parent 98bebf4
commit 28da5e6
Showing 1 changed file with 2 additions and 8 deletions.
diff --git a/cpp/src/arrow/compute/kernels/vector_selection_take_internal.cc b/cpp/src/arrow/compute/kernels/vector_selection_take_internal.cc
@@ -830,15 +830,9 @@ class TakeMetaFunction : public MetaFunction {
     const auto& take_opts = static_cast<const TakeOptions&>(*options);
     switch (args[0].kind()) {
       case Datum::ARRAY:
-        // "array_take" can handle AA->A and AC->C cases directly
-        // (via their VectorKernel::exec and VectorKernel::exec_chunked)
-        if (index_kind == Datum::ARRAY || index_kind == Datum::CHUNKED_ARRAY) {
-          return CallArrayTake(args, take_opts, ctx);
-        }
-        break;
       case Datum::CHUNKED_ARRAY:
-        // "array_take" can handle CA->C and CC->C cases directly
-        // (via their VectorKernel::exec_chunked)
+        // "array_take" can handle AA->A, AC->C, CA->C, CC->C cases directly
+        // (via their VectorKernel::exec and VectorKernel::exec_chunked)
         if (index_kind == Datum::ARRAY || index_kind == Datum::CHUNKED_ARRAY) {
           return CallArrayTake(args, take_opts, ctx);
         }