ExtensionFunctionsBinding.h File Reference

#include "ExtensionFunctionsWhitelist.h"
#include "TableFunctions/TableFunctionsFactory.h"
#include "../Analyzer/Analyzer.h"
#include "../Shared/sqltypes.h"
#include <vector>

Include dependency graph for ExtensionFunctionsBinding.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
class	ExtensionFunctionBindingError

Namespaces
	Analyzer

Functions
ExtensionFunction	bind_function (std::string name, Analyzer::ExpressionPtrVector func_args, const bool is_gpu)
ExtensionFunction	bind_function (std::string name, Analyzer::ExpressionPtrVector func_args)
ExtensionFunction	bind_function (const Analyzer::FunctionOper *function_oper, const bool is_gpu)
const table_functions::TableFunction	bind_table_function (std::string name, Analyzer::ExpressionPtrVector input_args, const bool is_gpu)
bool	is_ext_arg_type_column (const ExtArgumentType ext_arg_type)
bool	is_ext_arg_type_array (const ExtArgumentType ext_arg_type)
bool	is_ext_arg_type_geo (const ExtArgumentType ext_arg_type)
bool	is_ext_arg_type_pointer (const ExtArgumentType ext_arg_type)
bool	is_ext_arg_type_scalar (const ExtArgumentType ext_arg_type)

Function Documentation

ExtensionFunction bind_function	(	std::string	name,
		Analyzer::ExpressionPtrVector	func_args,
		const bool	is_gpu
	)

Definition at line 518 of file ExtensionFunctionsBinding.cpp.

References ExtensionFunctionsWhitelist::get_ext_funcs(), and setup::name.

                                                   {
  // used below
  std::vector<ExtensionFunction> ext_funcs =
      ExtensionFunctionsWhitelist::get_ext_funcs(name, is_gpu);
  std::string processor = (is_gpu ? "GPU" : "CPU");
  return bind_function<ExtensionFunction>(name, func_args, ext_funcs, processor);
}

Here is the call graph for this function:

ExtensionFunction bind_function	(	std::string	name,
		Analyzer::ExpressionPtrVector	func_args
	)

Definition at line 492 of file ExtensionFunctionsBinding.cpp.

References ExtensionFunctionsWhitelist::get_ext_funcs(), and setup::name.

                                                                       {
  // used in RelAlgTranslator.cpp, first try GPU UDFs, then fall back
  // to CPU UDFs.
  bool is_gpu = true;
  std::string processor = "GPU";
  auto ext_funcs = ExtensionFunctionsWhitelist::get_ext_funcs(name, is_gpu);
  if (!ext_funcs.size()) {
    is_gpu = false;
    processor = "CPU";
    ext_funcs = ExtensionFunctionsWhitelist::get_ext_funcs(name, is_gpu);
  }
  try {
    return bind_function<ExtensionFunction>(name, func_args, ext_funcs, processor);
  } catch (ExtensionFunctionBindingError& e) {
    if (is_gpu) {
      is_gpu = false;
      processor = "GPU|CPU";
      ext_funcs = ExtensionFunctionsWhitelist::get_ext_funcs(name, is_gpu);
      return bind_function<ExtensionFunction>(name, func_args, ext_funcs, processor);
    } else {
      throw;
    }
  }
}

Here is the call graph for this function:

ExtensionFunction bind_function	(	const Analyzer::FunctionOper *	function_oper,
		const bool	is_gpu
	)

Definition at line 528 of file ExtensionFunctionsBinding.cpp.

References bind_function(), Analyzer::FunctionOper::getArity(), Analyzer::FunctionOper::getName(), Analyzer::FunctionOper::getOwnArg(), and setup::name.

                                                   {
  // used in ExtensionsIR.cpp
  auto name = function_oper->getName();
  Analyzer::ExpressionPtrVector func_args = {};
  for (size_t i = 0; i < function_oper->getArity(); ++i) {
    func_args.push_back(function_oper->getOwnArg(i));
  }
  return bind_function(name, func_args, is_gpu);
}

Here is the call graph for this function:

const table_functions::TableFunction bind_table_function	(	std::string	name,
		Analyzer::ExpressionPtrVector	input_args,
		const bool	is_gpu
	)

Definition at line 539 of file ExtensionFunctionsBinding.cpp.

References bind_table_function(), and table_functions::TableFunctionsFactory::get_table_funcs().

                       {
  // used in RelAlgExecutor.cpp
  std::vector<table_functions::TableFunction> table_funcs =
      table_functions::TableFunctionsFactory::get_table_funcs(name, is_gpu);
  return bind_table_function(name, input_args, table_funcs, is_gpu);
}

Here is the call graph for this function:

bool is_ext_arg_type_array

(

const ExtArgumentType

ext_arg_type

)

Definition at line 549 of file ExtensionFunctionsBinding.cpp.

References ArrayBool, ArrayDouble, ArrayFloat, ArrayInt16, ArrayInt32, ArrayInt64, and ArrayInt8.

Referenced by CodeGenerator::codegenFunctionOperCastArgs(), ExtensionFunctionsWhitelist::getLLVMDeclarations(), and anonymous_namespace{ExtensionFunctionsBinding.cpp}::match_arguments().

                                                               {
  switch (ext_arg_type) {
    case ExtArgumentType::ArrayInt8:
    case ExtArgumentType::ArrayInt16:
    case ExtArgumentType::ArrayInt32:
    case ExtArgumentType::ArrayInt64:
    case ExtArgumentType::ArrayFloat:
    case ExtArgumentType::ArrayDouble:
    case ExtArgumentType::ArrayBool:
      return true;

    default:
      return false;
  }
}

Here is the caller graph for this function:

bool is_ext_arg_type_column

(

const ExtArgumentType

ext_arg_type

)

Definition at line 565 of file ExtensionFunctionsBinding.cpp.

References ColumnBool, ColumnDouble, ColumnFloat, ColumnInt16, ColumnInt32, ColumnInt64, and ColumnInt8.

Referenced by anonymous_namespace{ExtensionFunctionsBinding.cpp}::match_arguments().

                                                                {
  switch (ext_arg_type) {
    case ExtArgumentType::ColumnInt8:
    case ExtArgumentType::ColumnInt16:
    case ExtArgumentType::ColumnInt32:
    case ExtArgumentType::ColumnInt64:
    case ExtArgumentType::ColumnFloat:
    case ExtArgumentType::ColumnDouble:
    case ExtArgumentType::ColumnBool:
      return true;

    default:
      return false;
  }
}

Here is the caller graph for this function:

bool is_ext_arg_type_geo

(

const ExtArgumentType

ext_arg_type

)

Definition at line 581 of file ExtensionFunctionsBinding.cpp.

References GeoLineString, GeoMultiPolygon, GeoPoint, and GeoPolygon.

Referenced by CodeGenerator::codegenFunctionOperCastArgs().

                                                             {
  switch (ext_arg_type) {
    case ExtArgumentType::GeoPoint:
    case ExtArgumentType::GeoLineString:
    case ExtArgumentType::GeoPolygon:
    case ExtArgumentType::GeoMultiPolygon:
      return true;

    default:
      return false;
  }
}

Here is the caller graph for this function:

bool is_ext_arg_type_pointer

(

const ExtArgumentType

ext_arg_type

)

Definition at line 594 of file ExtensionFunctionsBinding.cpp.

References PBool, PDouble, PFloat, PInt16, PInt32, PInt64, and PInt8.

Referenced by CodeGenerator::codegenFunctionOperCastArgs().

                                                                 {
  switch (ext_arg_type) {
    case ExtArgumentType::PInt8:
    case ExtArgumentType::PInt16:
    case ExtArgumentType::PInt32:
    case ExtArgumentType::PInt64:
    case ExtArgumentType::PFloat:
    case ExtArgumentType::PDouble:
    case ExtArgumentType::PBool:
      return true;

    default:
      return false;
  }
}

Here is the caller graph for this function:

bool is_ext_arg_type_scalar

(

const ExtArgumentType

ext_arg_type

)

Definition at line 610 of file ExtensionFunctionsBinding.cpp.

References Bool, Double, Float, Int16, Int32, Int64, and Int8.

                                                                {
  switch (ext_arg_type) {
    case ExtArgumentType::Int8:
    case ExtArgumentType::Int16:
    case ExtArgumentType::Int32:
    case ExtArgumentType::Int64:
    case ExtArgumentType::Float:
    case ExtArgumentType::Double:
    case ExtArgumentType::Bool:
      return true;

    default:
      return false;
  }
}