OmniSciDB  eb3a3d0a03
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
PointConstructor.h
Go to the documentation of this file.
1 /*
2  * Copyright 2021 OmniSci, Inc.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 #pragma once
18 
20 
21 #include "Shared/SqlTypesLayout.h"
22 
23 namespace spatial_type {
24 
25 // ST_Point
26 class PointConstructor : public Codegen {
27  public:
29  const Catalog_Namespace::Catalog* catalog)
30  : Codegen(geo_operator, catalog) {
31  CHECK_EQ(operator_->size(), size_t(2));
32  const auto& ti = geo_operator->get_type_info();
33  if (ti.get_notnull()) {
34  is_nullable_ = false;
35  } else {
36  is_nullable_ = true;
37  }
38  }
39 
40  std::unique_ptr<CodeGenerator::NullCheckCodegen> getNullCheckCodegen(
41  llvm::Value* null_lv,
42  CgenState* cgen_state,
43  Executor* executor) final {
44  if (isNullable()) {
45  // do the standard nullcheck codegen, but modify the null basic block to emplace the
46  // null sentinel into the array
47  auto nullcheck_codegen = std::make_unique<CodeGenerator::NullCheckCodegen>(
48  cgen_state, executor, null_lv, getNullType(), getName() + "_nullcheck");
49 
50  auto& builder = cgen_state->ir_builder_;
52 
53  auto prev_insert_block = builder.GetInsertBlock();
54  auto crt_insert_block = nullcheck_codegen->null_check->cond_true_;
55  CHECK(crt_insert_block);
56  auto& instruction_list = crt_insert_block->getInstList();
57  CHECK_EQ(instruction_list.size(), size_t(1));
58  builder.SetInsertPoint(crt_insert_block, instruction_list.begin());
59 
60  auto x_coord_ptr = builder.CreateGEP(pt_local_storage_lv_,
61  {cgen_state->llInt(0), cgen_state->llInt(0)},
62  "x_coord_ptr");
63  const auto& geo_ti = operator_->get_type_info();
64  if (geo_ti.get_compression() == kENCODING_GEOINT) {
65  // TODO: probably wrong
66  builder.CreateStore(cgen_state->llInt(inline_int_null_val(SQLTypeInfo(kINT))),
67  x_coord_ptr);
68  } else {
69  builder.CreateStore(cgen_state->llFp(static_cast<double>(NULL_ARRAY_DOUBLE)),
70  x_coord_ptr);
71  }
72  builder.SetInsertPoint(prev_insert_block);
73 
74  return nullcheck_codegen;
75  } else {
76  return nullptr;
77  }
78  }
79 
80  size_t size() const final { return 2; }
81 
83 
84  // returns arguments lvs and null lv
85  std::tuple<std::vector<llvm::Value*>, llvm::Value*> codegenLoads(
86  const std::vector<llvm::Value*>& arg_lvs,
87  const std::vector<llvm::Value*>& pos_lvs,
88  CgenState* cgen_state) final {
89  CHECK_EQ(pos_lvs.size(),
90  size()); // note both pos arguments should be the same, as both inputs are
91  // expected to be from the same table, though this is moot in this
92  // function as position argumnts are not used here
93  CHECK_EQ(arg_lvs.size(), size_t(2));
94 
95  auto& builder = cgen_state->ir_builder_;
96 
97  llvm::Value* is_null{nullptr};
98  auto x_operand = getOperand(0);
99  const auto& x_ti = x_operand->get_type_info();
100  if (!x_ti.get_notnull()) {
101  CHECK(x_ti.is_integer() || x_ti.is_fp());
102  // TODO: centralize nullcheck logic for all sqltypes
103  is_null = x_ti.is_integer()
104  ? builder.CreateICmp(llvm::CmpInst::ICMP_EQ,
105  arg_lvs.front(),
106  cgen_state->llInt(inline_int_null_val(x_ti)))
107  : builder.CreateFCmp(llvm::FCmpInst::FCMP_OEQ,
108  arg_lvs.front(),
109  cgen_state->llFp(inline_fp_null_val(x_ti)));
110  }
111 
112  auto y_operand = getOperand(0);
113  const auto& y_ti = y_operand->get_type_info();
114  if (!y_ti.get_notnull()) {
115  auto y_is_null =
116  y_ti.is_integer()
117  ? builder.CreateICmp(llvm::CmpInst::ICMP_EQ,
118  arg_lvs.front(),
119  cgen_state->llInt(inline_int_null_val(y_ti)))
120  : builder.CreateFCmp(llvm::FCmpInst::FCMP_OEQ,
121  arg_lvs.front(),
122  cgen_state->llFp(inline_fp_null_val(y_ti)));
123  if (is_null) {
124  is_null = builder.CreateAnd(is_null, y_is_null);
125  } else {
126  is_null = y_is_null;
127  }
128  }
129 
130  if (is_nullable_ && !is_null) {
131  // if the inputs are not null, set the output to be not null
132  // TODO: we should do this in the translator and just confirm it here
133  is_nullable_ = false;
134  }
135 
136  // do the alloca before nullcheck codegen, as either way we will return a point array
137  const auto& geo_ti = operator_->get_type_info();
138  CHECK(geo_ti.get_type() == kPOINT);
139 
140  llvm::ArrayType* arr_type{nullptr};
141  if (geo_ti.get_compression() == kENCODING_GEOINT) {
142  auto elem_ty = llvm::Type::getInt32Ty(cgen_state->context_);
143  arr_type = llvm::ArrayType::get(elem_ty, 2);
144  } else {
145  CHECK(geo_ti.get_compression() == kENCODING_NONE);
146  auto elem_ty = llvm::Type::getDoubleTy(cgen_state->context_);
147  arr_type = llvm::ArrayType::get(elem_ty, 2);
148  }
150  builder.CreateAlloca(arr_type, nullptr, operator_->getName() + "_Local_Storage");
151 
152  return std::make_tuple(arg_lvs, is_null);
153  }
154 
155  std::vector<llvm::Value*> codegen(const std::vector<llvm::Value*>& args,
156  CodeGenerator::NullCheckCodegen* nullcheck_codegen,
157  CgenState* cgen_state,
158  const CompilationOptions& co) final {
159  CHECK_EQ(args.size(), size_t(2));
160 
161  const auto& geo_ti = operator_->get_type_info();
162  CHECK(geo_ti.get_type() == kPOINT);
163 
164  auto& builder = cgen_state->ir_builder_;
166 
167  const bool is_compressed = geo_ti.get_compression() == kENCODING_GEOINT;
168 
169  // store x coord
170  auto x_coord_ptr = builder.CreateGEP(pt_local_storage_lv_,
171  {cgen_state->llInt(0), cgen_state->llInt(0)},
172  "x_coord_ptr");
173  if (is_compressed) {
174  auto compressed_lv =
175  cgen_state->emitExternalCall("compress_x_coord_geoint",
176  llvm::Type::getInt32Ty(cgen_state->context_),
177  {args.front()});
178  builder.CreateStore(compressed_lv, x_coord_ptr);
179  } else {
180  builder.CreateStore(args.front(), x_coord_ptr);
181  }
182 
183  // store y coord
184  auto y_coord_ptr = builder.CreateGEP(pt_local_storage_lv_,
185  {cgen_state->llInt(0), cgen_state->llInt(1)},
186  "y_coord_ptr");
187  if (is_compressed) {
188  auto compressed_lv =
189  cgen_state->emitExternalCall("compress_y_coord_geoint",
190  llvm::Type::getInt32Ty(cgen_state->context_),
191  {args.back()});
192  builder.CreateStore(compressed_lv, y_coord_ptr);
193  } else {
194  builder.CreateStore(args.back(), y_coord_ptr);
195  }
196 
197  llvm::Value* ret = pt_local_storage_lv_;
198  if (is_nullable_) {
199  CHECK(nullcheck_codegen);
200  ret = nullcheck_codegen->finalize(ret, ret);
201  }
202  return {builder.CreateBitCast(ret,
203  geo_ti.get_compression() == kENCODING_GEOINT
204  ? llvm::Type::getInt32PtrTy(cgen_state->context_)
205  : llvm::Type::getDoublePtrTy(cgen_state->context_)),
206  cgen_state->llInt(static_cast<int32_t>(
207  geo_ti.get_compression() == kENCODING_GEOINT ? 8 : 16))};
208  }
209 
210  private:
211  llvm::AllocaInst* pt_local_storage_lv_{nullptr};
212 };
213 
214 } // namespace spatial_type
std::vector< llvm::Value * > codegen(const std::vector< llvm::Value * > &args, CodeGenerator::NullCheckCodegen *nullcheck_codegen, CgenState *cgen_state, const CompilationOptions &co) final
#define CHECK_EQ(x, y)
Definition: Logger.h:217
class for a per-database catalog. also includes metadata for the current database and the current use...
Definition: Catalog.h:102
SQLTypeInfo getNullType() const final
#define const
PointConstructor(const Analyzer::GeoOperator *geo_operator, const Catalog_Namespace::Catalog *catalog)
double inline_fp_null_val(const SQL_TYPE_INFO &ti)
CONSTEXPR DEVICE bool is_null(const T &value)
const std::string & getName() const
Definition: Analyzer.h:1912
auto isNullable() const
Definition: Codegen.h:35
const SQLTypeInfo & get_type_info() const
Definition: Analyzer.h:78
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
llvm::AllocaInst * pt_local_storage_lv_
const Analyzer::GeoOperator * operator_
Definition: Codegen.h:70
size_t size() const
Definition: Analyzer.cpp:3629
#define NULL_ARRAY_DOUBLE
#define CHECK(condition)
Definition: Logger.h:209
virtual const Analyzer::Expr * getOperand(const size_t index)
Definition: Codegen.cpp:63
int64_t inline_int_null_val(const SQL_TYPE_INFO &ti)
std::string getName() const
Definition: Codegen.h:39
std::unique_ptr< CodeGenerator::NullCheckCodegen > getNullCheckCodegen(llvm::Value *null_lv, CgenState *cgen_state, Executor *executor) final
Definition: sqltypes.h:45