OmniSciDB  d2f719934e
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
spatial_type::PointN Class Reference

#include <PointN.h>

+ Inheritance diagram for spatial_type::PointN:
+ Collaboration diagram for spatial_type::PointN:

Public Member Functions

 PointN (const Analyzer::GeoOperator *geo_operator, const Catalog_Namespace::Catalog *catalog)
 
std::unique_ptr
< CodeGenerator::NullCheckCodegen
getNullCheckCodegen (llvm::Value *null_lv, CgenState *cgen_state, Executor *executor) final
 
size_t size () const final
 
SQLTypeInfo getNullType () const 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, const Catalog_Namespace::Catalog *catalog)
 
auto isNullable () const
 
auto getTypeInfo () const
 
std::string getName () const
 
virtual const Analyzer::ExprgetOperand (const size_t index)
 
virtual ~Codegen ()
 

Additional Inherited Members

- Static Public Member Functions inherited from spatial_type::Codegen
static std::unique_ptr< Codegeninit (const Analyzer::GeoOperator *geo_operator, const Catalog_Namespace::Catalog *catalog)
 
- Protected Attributes inherited from spatial_type::Codegen
const Analyzer::GeoOperatoroperator_
 
const Catalog_Namespace::Catalogcat_
 
bool is_nullable_ {true}
 

Detailed Description

Definition at line 23 of file PointN.h.

Constructor & Destructor Documentation

spatial_type::PointN::PointN ( const Analyzer::GeoOperator geo_operator,
const Catalog_Namespace::Catalog catalog 
)
inline

Definition at line 25 of file PointN.h.

References CHECK_EQ, spatial_type::Codegen::is_nullable_, spatial_type::Codegen::operator_, and Analyzer::GeoOperator::size().

27  : Codegen(geo_operator, catalog) {
28  CHECK_EQ(operator_->size(), size_t(2));
29  // set is nullable to true, because an index outside of the linestring will return
30  // null
31  // note we could probably just set this based on the operator type, as the operator
32  // type needs to match
33  this->is_nullable_ = true;
34  }
Codegen(const Analyzer::GeoOperator *geo_operator, const Catalog_Namespace::Catalog *catalog)
Definition: Codegen.h:28
#define CHECK_EQ(x, y)
Definition: Logger.h:219
const Analyzer::GeoOperator * operator_
Definition: Codegen.h:70
size_t size() const
Definition: Analyzer.cpp:3644

+ Here is the call graph for this function:

Member Function Documentation

std::vector<llvm::Value*> spatial_type::PointN::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 126 of file PointN.h.

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

129  {
130  CHECK_EQ(args.size(), size_t(3)); // ptr, size, index
131  const auto& geo_ti = getOperand(0)->get_type_info();
132  CHECK(geo_ti.is_geometry());
133 
134  llvm::Value* array_buff_cast{nullptr};
135 
136  auto& builder = cgen_state->ir_builder_;
137  if (geo_ti.get_compression() == kENCODING_GEOINT) {
138  array_buff_cast = builder.CreateBitCast(
139  args.front(), llvm::Type::getInt32PtrTy(cgen_state->context_));
140  } else {
141  array_buff_cast = builder.CreateBitCast(
142  args.front(), llvm::Type::getDoublePtrTy(cgen_state->context_));
143  }
144 
145  const auto index_lv = args.back();
146  auto array_offset_lv =
147  builder.CreateGEP(array_buff_cast, index_lv, operator_->getName() + "_Offset");
148  CHECK(nullcheck_codegen);
149  auto ret_lv = nullcheck_codegen->finalize(
150  llvm::ConstantPointerNull::get(
151  geo_ti.get_compression() == kENCODING_GEOINT
152  ? llvm::PointerType::get(llvm::Type::getInt32Ty(cgen_state->context_), 0)
153  : llvm::PointerType::get(llvm::Type::getDoubleTy(cgen_state->context_),
154  0)),
155  array_offset_lv);
156  const auto geo_size_lv = args[1];
157  return {ret_lv, geo_size_lv};
158  }
#define CHECK_EQ(x, y)
Definition: Logger.h:219
llvm::IRBuilder ir_builder_
Definition: CgenState.h:340
llvm::LLVMContext & context_
Definition: CgenState.h:338
const std::string & getName() const
Definition: Analyzer.h:1925
const SQLTypeInfo & get_type_info() const
Definition: Analyzer.h:77
const Analyzer::GeoOperator * operator_
Definition: Codegen.h:70
llvm::Value * finalize(llvm::Value *null_lv, llvm::Value *notnull_lv)
Definition: IRCodegen.cpp:1447
#define CHECK(condition)
Definition: Logger.h:211
virtual const Analyzer::Expr * getOperand(const size_t index)
Definition: Codegen.cpp:63
std::tuple<std::vector<llvm::Value*>, llvm::Value*> spatial_type::PointN::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 57 of file PointN.h.

References CHECK, CHECK_EQ, get_int_type(), spatial_type::Codegen::getOperand(), kLINESTRING, log2_bytes(), and size().

60  {
61  CHECK_EQ(pos_lvs.size(), size());
62  CHECK_EQ(pos_lvs.front(), pos_lvs.back());
63  auto operand = getOperand(0);
64  CHECK(operand);
65  const auto& geo_ti = operand->get_type_info();
66  CHECK(geo_ti.get_type() == kLINESTRING);
67 
68  auto& builder = cgen_state->ir_builder_;
69 
70  std::vector<llvm::Value*> array_operand_lvs;
71  CHECK(!arg_lvs.empty());
72  auto index_lv = builder.CreateMul(
73  builder.CreateSub(arg_lvs.back(), cgen_state->llInt(static_cast<int32_t>(1))),
74  cgen_state->llInt(static_cast<int32_t>(2)));
75  llvm::Value* is_null_lv{nullptr};
76  if (arg_lvs.size() == 2) {
77  // col byte stream from column on disk
78  array_operand_lvs.push_back(
79  cgen_state->emitExternalCall("array_buff",
80  llvm::Type::getInt8PtrTy(cgen_state->context_),
81  {arg_lvs.front(), pos_lvs.front()}));
82  const bool is_nullable = !geo_ti.get_notnull();
83  std::string size_fn_name = "array_size";
84  if (is_nullable) {
85  size_fn_name += "_nullable";
86  }
87 
88  uint32_t elem_sz = 1; // TINYINT coords array
89  std::vector<llvm::Value*> array_sz_args{
90  arg_lvs.front(), pos_lvs.front(), cgen_state->llInt(log2_bytes(elem_sz))};
91  if (is_nullable) {
92  array_sz_args.push_back(
93  cgen_state->llInt(static_cast<int32_t>(inline_int_null_value<int32_t>())));
94  }
95  array_operand_lvs.push_back(cgen_state->emitExternalCall(
96  size_fn_name, get_int_type(32, cgen_state->context_), array_sz_args));
97 
98  auto geo_size_lv = array_operand_lvs.back();
99  // convert the index to a byte index
100  const auto outside_linestring_bounds_lv = builder.CreateNot(builder.CreateICmp(
101  llvm::ICmpInst::ICMP_SLT,
102  builder.CreateMul(index_lv, cgen_state->llInt(static_cast<int32_t>(8))),
103  geo_size_lv));
104  outside_linestring_bounds_lv->setName("outside_linestring_bounds");
105  const auto input_is_null_lv = builder.CreateICmp(
106  llvm::ICmpInst::ICMP_EQ,
107  geo_size_lv,
108  cgen_state->llInt(static_cast<int32_t>(inline_int_null_value<int32_t>())));
109  input_is_null_lv->setName("input_is_null");
110  is_null_lv = builder.CreateOr(outside_linestring_bounds_lv, input_is_null_lv);
111  } else {
112  CHECK_EQ(arg_lvs.size(), size_t(3)); // ptr, size, index
113  array_operand_lvs.push_back(arg_lvs[0]);
114  array_operand_lvs.push_back(arg_lvs[1]);
115 
116  const auto geo_size_lv = arg_lvs[1];
117  // TODO: bounds indices are 64 bits but should be 32 bits, as array length is
118  // limited to 32 bits
119  is_null_lv = builder.CreateNot(
120  builder.CreateICmp(llvm::ICmpInst::ICMP_SLT, index_lv, geo_size_lv));
121  }
122  array_operand_lvs.push_back(index_lv);
123  return std::make_tuple(array_operand_lvs, is_null_lv);
124  }
#define CHECK_EQ(x, y)
Definition: Logger.h:219
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.h:228
llvm::IRBuilder ir_builder_
Definition: CgenState.h:340
llvm::Type * get_int_type(const int width, llvm::LLVMContext &context)
llvm::LLVMContext & context_
Definition: CgenState.h:338
size_t size() const final
Definition: PointN.h:49
llvm::ConstantInt * llInt(const T v) const
Definition: CgenState.h:307
#define CHECK(condition)
Definition: Logger.h:211
virtual const Analyzer::Expr * getOperand(const size_t index)
Definition: Codegen.cpp:63
uint32_t log2_bytes(const uint32_t bytes)
Definition: Execute.h:176

+ Here is the call graph for this function:

std::unique_ptr<CodeGenerator::NullCheckCodegen> spatial_type::PointN::getNullCheckCodegen ( llvm::Value *  null_lv,
CgenState cgen_state,
Executor executor 
)
inlinefinalvirtual

Reimplemented from spatial_type::Codegen.

Definition at line 36 of file PointN.h.

References CHECK, spatial_type::Codegen::getName(), getNullType(), and spatial_type::Codegen::isNullable().

39  {
40  if (isNullable()) {
41  CHECK(null_lv);
42  return std::make_unique<CodeGenerator::NullCheckCodegen>(
43  cgen_state, executor, null_lv, getNullType(), getName() + "_nullcheck");
44  } else {
45  return nullptr;
46  }
47  }
SQLTypeInfo getNullType() const final
Definition: PointN.h:51
auto isNullable() const
Definition: Codegen.h:35
#define CHECK(condition)
Definition: Logger.h:211
std::string getName() const
Definition: Codegen.h:39

+ Here is the call graph for this function:

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

Implements spatial_type::Codegen.

Definition at line 51 of file PointN.h.

References kBOOLEAN.

Referenced by getNullCheckCodegen().

51  {
52  // nullability is the expression `linestring is null OR size within bounds`
53  return SQLTypeInfo(kBOOLEAN);
54  }

+ Here is the caller graph for this function:

size_t spatial_type::PointN::size ( ) const
inlinefinalvirtual

Implements spatial_type::Codegen.

Definition at line 49 of file PointN.h.

Referenced by codegenLoads().

49 { return 2; }

+ Here is the caller graph for this function:


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