Method apply

bolt/python/PythonScalarFunction.cpp:90–126

        returnType_(returnType) {}

  void apply(
      const SelectivityVector& rows,
      std::vector<VectorPtr>& args,
      const TypePtr& outputType,
      exec::EvalCtx& context,
      VectorPtr& result) const override {
    BOLT_CHECK(numArgs_ >= args.size());
    for (auto& arg : args) {
      // The argument may be flat or constant.
      BOLT_CHECK(arg->isFlatEncoding() || arg->isConstantEncoding());
    }

    BaseVector::ensureWritable(rows, returnType_, context.pool(), result);
    result->clearNulls(rows);

    rows.applyToSelected([&](auto row) {
      py::list pyArgs;
      for (int i = 0; i < args.size(); ++i) {
        auto& arg = args[i];
        py::object val = py::none();
        if (arg->isConstantEncoding()) {
          val = BOLT_DYNAMIC_SCALAR_TYPE_DISPATCH(
              getPyObjectFromVector, arg->typeKind(), arg, 0);
        } else {
          val = BOLT_DYNAMIC_SCALAR_TYPE_DISPATCH(
              getPyObjectFromVector, arg->typeKind(), arg, row);
        }
        pyArgs.append(val);
      }

      py::object res =
          bolt::python::pyTry<py::object>([&]() { return function_(*pyArgs); });

      BOLT_DYNAMIC_SCALAR_TYPE_DISPATCH(
          setPyObjectInFlatVector, result->typeKind(), result, row, res);
    });
  }

 private:
  int numArgs_;