spatial_type::NPoints Class Reference

#include <NPoints.h>

Inheritance diagram for spatial_type::NPoints:

Collaboration diagram for spatial_type::NPoints:

Public Member Functions
	NPoints (const Analyzer::GeoOperator *geo_operator)
size_t	size () const final
SQLTypeInfo	getNullType () const final
const Analyzer::Expr *	getOperand (const size_t index) final
std::tuple< std::vector < llvm::Value * >, llvm::Value * >	codegenLoads (const std::vector< llvm::Value * > &arg_lvs, const std::vector< llvm::Value * > &pos_lvs, CgenState *cgen_state) final
std::vector< llvm::Value * >	codegen (const std::vector< llvm::Value * > &args, CodeGenerator::NullCheckCodegen *nullcheck_codegen, CgenState *cgen_state, const CompilationOptions &co) final
Public Member Functions inherited from spatial_type::Codegen
	Codegen (const Analyzer::GeoOperator *geo_operator)
auto	isNullable () const
auto	getTypeInfo () const
std::string	getName () const
virtual std::unique_ptr < CodeGenerator::NullCheckCodegen >	getNullCheckCodegen (llvm::Value *null_lv, CgenState *cgen_state, Executor *executor)
virtual	~Codegen ()

Protected Attributes
SQLTypeInfo	geo_ti_
std::unique_ptr < Analyzer::ColumnVar >	col_var_owned_
Protected Attributes inherited from spatial_type::Codegen
const Analyzer::GeoOperator *	operator_
bool	is_nullable_ {true}

Additional Inherited Members
Static Public Member Functions inherited from spatial_type::Codegen
static std::unique_ptr< Codegen >	init (const Analyzer::GeoOperator *geo_operator)
static char const *	pointIsNullFunctionName (SQLTypeInfo const &)

Detailed Description

Definition at line 24 of file NPoints.h.

Constructor & Destructor Documentation

spatial_type::NPoints::NPoints ( const Analyzer::GeoOperator *  geo_operator )

inline

Definition at line 26 of file NPoints.h.

: Codegen(geo_operator) {}

Member Function Documentation

std::vector<llvm::Value*> spatial_type::NPoints::codegen ( const std::vector< llvm::Value * > &  args, CodeGenerator::NullCheckCodegen *  nullcheck_codegen, CgenState *  cgen_state, const CompilationOptions &  co  )

inlinefinalvirtual

Implements spatial_type::Codegen.

Definition at line 111 of file NPoints.h.

References run_benchmark_import::args, CHECK, CHECK_EQ, and kENCODING_GEOINT.

                                                                        {
    CHECK_EQ(args.size(), size_t(1));

    // divide the coord size by the constant compression value and return it
    auto& builder = cgen_state->ir_builder_;
    llvm::Value* conversion_constant{nullptr};
    if (geo_ti_.get_compression() == kENCODING_GEOINT) {
      conversion_constant = cgen_state->llInt(4);
    } else {
      conversion_constant = cgen_state->llInt(8);
    }
    CHECK(conversion_constant);
    const auto total_num_pts = builder.CreateUDiv(args.front(), conversion_constant);
    const auto ret = builder.CreateUDiv(total_num_pts, cgen_state->llInt(2));
    if (isNullable()) {
      CHECK(nullcheck_codegen);
      return {nullcheck_codegen->finalize(cgen_state->inlineIntNull(getTypeInfo()), ret)};
    } else {
      return {ret};
    }
  }

std::tuple<std::vector<llvm::Value*>, llvm::Value*> spatial_type::NPoints::codegenLoads ( const std::vector< llvm::Value * > &  arg_lvs, const std::vector< llvm::Value * > &  pos_lvs, CgenState *  cgen_state  )

inlinefinalvirtual

Implements spatial_type::Codegen.

Definition at line 54 of file NPoints.h.

References CHECK, CHECK_GT, get_column_descriptor(), SQLTypeInfo::get_elem_type(), get_int_type(), Analyzer::Expr::get_type_info(), getOperand(), Analyzer::GeoOperator::getOperand(), spatial_type::Codegen::getTypeInfo(), spatial_type::Codegen::isNullable(), kPOINT, log2_bytes(), and spatial_type::Codegen::operator_.

                                   {
    llvm::Value* coords_arr_sz_lv{nullptr};
    if (auto geo_constant = dynamic_cast<const Analyzer::GeoConstant*>(getOperand(0))) {
      // count points defined in the WKTString, i.e., POLYGON(1 1, 2 2, 3 3, 1 1)
      // the validation of the WKTString must be checked before entering this logic
      std::regex regex("-?[0-9]*\\.?[0-9]+\\s+-?[0-9]*\\.?[0-9]+");
      auto target = geo_constant->getWKTString();
      auto pt_begin = std::sregex_iterator(target.begin(), target.end(), regex);
      auto pt_end = std::sregex_iterator();
      auto num_pts = std::distance(pt_begin, pt_end);
      CHECK_GT(num_pts, 0);
      coords_arr_sz_lv = cgen_state->llInt(16 * num_pts);
    } else if (arg_lvs.size() == size_t(1)) {
      std::string fn_name("array_size");
      bool is_nullable = isNullable();
      if (auto col_var = dynamic_cast<const Analyzer::ColumnVar*>(getOperand(0))) {
        auto column_key = col_var->getColumnKey();
        column_key.column_id = column_key.column_id - 1;
        const auto type_cd = get_column_descriptor(column_key);
        CHECK(type_cd);
        if (type_cd->columnType.get_type() == kPOINT) {
          fn_name = "point_coord_array_size";
        }
      }
      auto& argument_lv = arg_lvs.front();
      const auto& elem_ti = getOperand(0)->get_type_info().get_elem_type();
      std::vector<llvm::Value*> array_size_args{
          argument_lv,
          pos_lvs.front(),
          cgen_state->llInt(log2_bytes(elem_ti.get_logical_size()))};
      if (is_nullable) {
        fn_name += "_nullable";
        array_size_args.push_back(cgen_state->inlineIntNull(getTypeInfo()));
      }
      coords_arr_sz_lv = cgen_state->emitExternalCall(
          fn_name, get_int_type(32, cgen_state->context_), array_size_args);
    } else if (arg_lvs.size() == size_t(2)) {
      auto child_geo_oper =
          dynamic_cast<const Analyzer::GeoOperator*>(operator_->getOperand(0));
      CHECK(child_geo_oper);
      if (child_geo_oper->getName() == "ST_Point") {
        coords_arr_sz_lv = cgen_state->ir_builder_.CreateSelect(
            cgen_state->emitCall("point_double_is_null", {arg_lvs.front()}),
            cgen_state->inlineIntNull(getTypeInfo()),
            cgen_state->llInt(static_cast<int32_t>(16)));
      } else {
        CHECK(false) << "Not supported geo operator w/ ST_NPoints: "
                     << child_geo_oper->getName();
      }
    }
    CHECK(coords_arr_sz_lv);
    return std::make_tuple(std::vector<llvm::Value*>{coords_arr_sz_lv}, coords_arr_sz_lv);
  }

Here is the call graph for this function:

SQLTypeInfo spatial_type::NPoints::getNullType ( ) const

inlinefinalvirtual

Implements spatial_type::Codegen.

Definition at line 30 of file NPoints.h.

References kINT.

{ return SQLTypeInfo(kINT); }

const Analyzer::Expr* spatial_type::NPoints::getOperand ( const size_t  index )

inlinefinalvirtual

Reimplemented from spatial_type::Codegen.

Definition at line 32 of file NPoints.h.

References CHECK, CHECK_EQ, col_var_owned_, geo_ti_, get_column_descriptor(), SQLTypeInfo::get_notnull(), Analyzer::GeoOperator::getOperand(), SQLTypeInfo::is_geometry(), spatial_type::Codegen::is_nullable_, and spatial_type::Codegen::operator_.

Referenced by codegenLoads().

                                                           {
    CHECK_EQ(index, size_t(0));
    if (col_var_owned_) {
      return col_var_owned_.get();
    }
    const auto operand = operator_->getOperand(0);
    geo_ti_ = operand->get_type_info();
    CHECK(geo_ti_.is_geometry());
    is_nullable_ = !geo_ti_.get_notnull();
    if (auto col_var = dynamic_cast<const Analyzer::ColumnVar*>(operand)) {
      // create a new operand which is just the coords and codegen it
      auto column_key = col_var->getColumnKey();
      column_key.column_id = column_key.column_id + 1;  // + 1 for coords
      const auto coords_cd = get_column_descriptor(column_key);
      CHECK(coords_cd);
      col_var_owned_ = std::make_unique<Analyzer::ColumnVar>(
          coords_cd->columnType, column_key, col_var->get_rte_idx());
      return col_var_owned_.get();
    }
    return operand;
  }

Here is the call graph for this function:

Here is the caller graph for this function:

size_t spatial_type::NPoints::size ( ) const

inlinefinalvirtual

Implements spatial_type::Codegen.

Definition at line 28 of file NPoints.h.

{ return 1; }

Member Data Documentation

std::unique_ptr<Analyzer::ColumnVar> spatial_type::NPoints::col_var_owned_

protected

Definition at line 138 of file NPoints.h.

Referenced by getOperand().

SQLTypeInfo spatial_type::NPoints::geo_ti_

protected

Definition at line 137 of file NPoints.h.

Referenced by getOperand().

---

The documentation for this class was generated from the following file:

• /home/jenkins-slave/workspace/core-os-doxygen/QueryEngine/GeoOperators/NPoints.h