Function asPyArgs

bolt/python/Utils.cpp:144–186

}

pybind11::list asPyArgs(
    const size_t numExpectedArgs,
    const std::vector<VectorPtr>& args,
    const std::vector<pybind11::object>& defaultArgs) {
  BOLT_CHECK_LE(args.size(), numExpectedArgs);
  BOLT_CHECK_GE(args.size() + defaultArgs.size(), numExpectedArgs);

  pybind11::list pyArgs;
  auto arg = args.cbegin();

  // Append arguments without existing default value as is.
  for (; arg != args.cbegin() + (numExpectedArgs - defaultArgs.size()); ++arg) {
    (*arg)->loadedVector();
    appendGenericArg(pyArgs, *arg);
  }

  // Arguments with a default value that is a constant type are expected to
  // be a python scalar by the user python function. However, when providing
  // a scalar value as part of a UDF expression, the value is converted into
  // a constant vector. Therefore, we need to recast the constant vector value
  // into a python scalar value
  //
  // This loop will iterate through arguments provided by the user which also
  // have a default value.
  for (; arg != args.cend(); ++arg) {
    (*arg)->loadedVector();
    if ((*arg)->encoding() == VectorEncoding::Simple::CONSTANT) {
      BOLT_DYNAMIC_SCALAR_TYPE_DISPATCH(
          appendConstantValue, (*arg)->type()->kind(), pyArgs, **arg);
    } else {
      appendGenericArg(pyArgs, *arg);
    }
  }

  // Iterate through remaining arguments that use the default value.
  for (auto defaultArg = defaultArgs.cend() - (numExpectedArgs - args.size());
       defaultArg != defaultArgs.cend();
       ++defaultArg) {
    pyArgs.append(*defaultArg);
  }

  return pyArgs;
}

std::shared_ptr<RowVector> combineInputColumns(
    const std::vector<VectorPtr>& args,