Method Codegen

be/src/runtime/sorter.cc:1628–1657

A method to code-gen for TupleSorter::SortHelper().

// A method to code-gen for TupleSorter::SortHelper().
Status Sorter::TupleSorter::Codegen(FragmentState* state, llvm::Function* compare_fn,
    int tuple_size, CodegenFnPtr<SortHelperFn>* codegened_fn) {
  LlvmCodeGen* codegen = state->codegen();
  DCHECK(codegen != nullptr);

  llvm::Function* fn = codegen->GetFunction(IRFunction::TUPLE_SORTER_SORT_HELPER, true);
  DCHECK(fn != NULL);

  // There are 8 calls to Less() and Compare() which calls
  // comparator_.Less() and comparator_.Compare() respectively once in each.
  int replaced =
      codegen->ReplaceCallSites(fn, compare_fn, TupleRowComparator::COMPARE_SYMBOL);
  DCHECK_REPLACE_COUNT(replaced, 8) << LlvmCodeGen::Print(fn);

  // There are 2 recursive calls within SorterHelper() to replace with.
  replaced = codegen->ReplaceCallSites(fn, fn, SORTER_HELPER_SYMBOL);
  DCHECK_REPLACE_COUNT(replaced, 2) << LlvmCodeGen::Print(fn);

  replaced = codegen->ReplaceCallSitesWithValue(fn,
      codegen->GetI32Constant(tuple_size), "get_tuple_size");
  DCHECK_REPLACE_COUNT(replaced, 3) << LlvmCodeGen::Print(fn);

  fn = codegen->FinalizeFunction(fn);
  if (fn == nullptr) {
    return Status("Sorter::TupleSorter::Codegen(): failed to finalize function");
  }
  codegen->AddFunctionToJit(fn, codegened_fn);

  return Status::OK();
}

} // namespace impala