Function RegisterVectorSelection

cpp/src/arrow/compute/kernels/vector_selection.cc:323–353

}  // namespace

void RegisterVectorSelection(FunctionRegistry* registry) {
  // Filter kernels
  std::vector<SelectionKernelData> filter_kernels;
  PopulateFilterKernels(&filter_kernels);

  VectorKernel filter_base;
  filter_base.init = FilterState::Init;
  RegisterSelectionFunction("array_filter", array_filter_doc, filter_base,
                            std::move(filter_kernels), GetDefaultFilterOptions(),
                            registry);

  DCHECK_OK(registry->AddFunction(MakeFilterMetaFunction()));

  // Take kernels
  std::vector<SelectionKernelData> take_kernels;
  PopulateTakeKernels(&take_kernels);

  VectorKernel take_base;
  take_base.init = TakeState::Init;
  take_base.can_execute_chunkwise = false;
  RegisterSelectionFunction("array_take", array_take_doc, take_base,
                            std::move(take_kernels), GetDefaultTakeOptions(), registry);

  DCHECK_OK(registry->AddFunction(MakeTakeMetaFunction()));

  // DropNull kernel
  DCHECK_OK(registry->AddFunction(std::make_shared<DropNullMetaFunction>()));

  DCHECK_OK(registry->AddFunction(
      MakeIndicesNonZeroFunction("indices_nonzero", indices_nonzero_doc)));
}

}  // namespace internal
}  // namespace compute