OmniSciDB  72c90bc290
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
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::ExprgetOperand (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
operand_owned_
 
- Protected Attributes inherited from spatial_type::Codegen
const Analyzer::GeoOperatoroperator_
 
bool is_nullable_ {true}
 

Additional Inherited Members

- Static Public Member Functions inherited from spatial_type::Codegen
static std::unique_ptr< Codegeninit (const Analyzer::GeoOperator *geo_operator)
 

Detailed Description

Definition at line 23 of file NPoints.h.

Constructor & Destructor Documentation

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

Definition at line 25 of file NPoints.h.

25 : Codegen(geo_operator) {}
Codegen(const Analyzer::GeoOperator *geo_operator)
Definition: Codegen.h:28

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 81 of file NPoints.h.

References run_benchmark_import::args, CHECK, CHECK_EQ, geo_ti_, SQLTypeInfo::get_compression(), spatial_type::Codegen::getTypeInfo(), spatial_type::Codegen::isNullable(), and kENCODING_GEOINT.

84  {
85  CHECK_EQ(args.size(), size_t(1));
86 
87  // divide the coord size by the constant compression value and return it
88  auto& builder = cgen_state->ir_builder_;
89  llvm::Value* conversion_constant{nullptr};
91  conversion_constant = cgen_state->llInt(4);
92  } else {
93  conversion_constant = cgen_state->llInt(8);
94  }
95  CHECK(conversion_constant);
96  const auto total_num_pts = builder.CreateUDiv(args.front(), conversion_constant);
97  const auto ret = builder.CreateUDiv(total_num_pts, cgen_state->llInt(2));
98  if (isNullable()) {
99  CHECK(nullcheck_codegen);
100  return {nullcheck_codegen->finalize(cgen_state->inlineIntNull(getTypeInfo()), ret)};
101  } else {
102  return {ret};
103  }
104  }
#define CHECK_EQ(x, y)
Definition: Logger.h:301
llvm::IRBuilder ir_builder_
Definition: CgenState.h:384
auto getTypeInfo() const
Definition: Codegen.h:34
llvm::ConstantInt * inlineIntNull(const SQLTypeInfo &)
Definition: CgenState.cpp:65
auto isNullable() const
Definition: Codegen.h:32
HOST DEVICE EncodingType get_compression() const
Definition: sqltypes.h:399
SQLTypeInfo geo_ti_
Definition: NPoints.h:107
llvm::ConstantInt * llInt(const T v) const
Definition: CgenState.h:249
llvm::Value * finalize(llvm::Value *null_lv, llvm::Value *notnull_lv)
Definition: IRCodegen.cpp:1529
#define CHECK(condition)
Definition: Logger.h:291

+ Here is the call graph for this function:

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 56 of file NPoints.h.

References CHECK_EQ, SQLTypeInfo::get_elem_type(), get_int_type(), Analyzer::Expr::get_type_info(), getOperand(), spatial_type::Codegen::getTypeInfo(), spatial_type::Codegen::isNullable(), and log2_bytes().

59  {
60  CHECK_EQ(arg_lvs.size(), size_t(1));
61  auto& argument_lv = arg_lvs.front();
62  std::string fn_name("array_size");
63 
64  const auto& elem_ti = getOperand(0)->get_type_info().get_elem_type();
65  std::vector<llvm::Value*> array_size_args{
66  argument_lv,
67  pos_lvs.front(),
68  cgen_state->llInt(log2_bytes(elem_ti.get_logical_size()))};
69 
70  const bool is_nullable = isNullable();
71 
72  if (is_nullable) {
73  fn_name += "_nullable";
74  array_size_args.push_back(cgen_state->inlineIntNull(getTypeInfo()));
75  }
76  const auto coords_arr_sz_lv = cgen_state->emitExternalCall(
77  fn_name, get_int_type(32, cgen_state->context_), array_size_args);
78  return std::make_tuple(std::vector<llvm::Value*>{coords_arr_sz_lv}, coords_arr_sz_lv);
79  }
#define CHECK_EQ(x, y)
Definition: Logger.h:301
llvm::Type * get_int_type(const int width, llvm::LLVMContext &context)
auto getTypeInfo() const
Definition: Codegen.h:34
const Analyzer::Expr * getOperand(const size_t index) final
Definition: NPoints.h:31
llvm::LLVMContext & context_
Definition: CgenState.h:382
llvm::Value * emitExternalCall(const std::string &fname, llvm::Type *ret_type, const std::vector< llvm::Value * > args, const std::vector< llvm::Attribute::AttrKind > &fnattrs={}, const bool has_struct_return=false)
Definition: CgenState.cpp:395
llvm::ConstantInt * inlineIntNull(const SQLTypeInfo &)
Definition: CgenState.cpp:65
auto isNullable() const
Definition: Codegen.h:32
const SQLTypeInfo & get_type_info() const
Definition: Analyzer.h:79
llvm::ConstantInt * llInt(const T v) const
Definition: CgenState.h:249
uint32_t log2_bytes(const uint32_t bytes)
Definition: Execute.h:198
SQLTypeInfo get_elem_type() const
Definition: sqltypes.h:975

+ Here is the call graph for this function:

SQLTypeInfo spatial_type::NPoints::getNullType ( ) const
inlinefinalvirtual

Implements spatial_type::Codegen.

Definition at line 29 of file NPoints.h.

References kINT.

29 { return SQLTypeInfo(kINT); }
Definition: sqltypes.h:72
const Analyzer::Expr* spatial_type::NPoints::getOperand ( const size_t  index)
inlinefinalvirtual

Reimplemented from spatial_type::Codegen.

Definition at line 31 of file NPoints.h.

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

Referenced by codegenLoads().

31  {
32  CHECK_EQ(index, size_t(0));
33  if (operand_owned_) {
34  return operand_owned_.get();
35  }
36 
37  const auto operand = operator_->getOperand(0);
38  auto col_var = dynamic_cast<const Analyzer::ColumnVar*>(operand);
39  CHECK(col_var);
40 
41  geo_ti_ = col_var->get_type_info();
44 
45  // create a new operand which is just the coords and codegen it
46  auto column_key = col_var->getColumnKey();
47  column_key.column_id = column_key.column_id + 1; // + 1 for coords
48  const auto coords_cd = get_column_descriptor(column_key);
49  CHECK(coords_cd);
50 
51  operand_owned_ = std::make_unique<Analyzer::ColumnVar>(
52  coords_cd->columnType, column_key, col_var->get_rte_idx());
53  return operand_owned_.get();
54  }
std::unique_ptr< Analyzer::ColumnVar > operand_owned_
Definition: NPoints.h:108
#define CHECK_EQ(x, y)
Definition: Logger.h:301
const ColumnDescriptor * get_column_descriptor(const shared::ColumnKey &column_key)
Definition: Execute.h:213
const Analyzer::GeoOperator * operator_
Definition: Codegen.h:67
SQLTypeInfo geo_ti_
Definition: NPoints.h:107
#define CHECK(condition)
Definition: Logger.h:291
bool is_geometry() const
Definition: sqltypes.h:595
Analyzer::Expr * getOperand(const size_t index) const
Definition: Analyzer.cpp:4186
HOST DEVICE bool get_notnull() const
Definition: sqltypes.h:398

+ 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 27 of file NPoints.h.

27 { return 1; }

Member Data Documentation

SQLTypeInfo spatial_type::NPoints::geo_ti_
protected

Definition at line 107 of file NPoints.h.

Referenced by codegen(), and getOperand().

std::unique_ptr<Analyzer::ColumnVar> spatial_type::NPoints::operand_owned_
protected

Definition at line 108 of file NPoints.h.

Referenced by getOperand().


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