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

#include <PointConstructor.h>

+ Inheritance diagram for spatial_type::PointConstructor:
+ Collaboration diagram for spatial_type::PointConstructor:

Public Member Functions

 PointConstructor (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 ()
 

Private Attributes

llvm::AllocaInst * pt_local_storage_lv_ {nullptr}
 

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 26 of file PointConstructor.h.

Constructor & Destructor Documentation

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

Definition at line 28 of file PointConstructor.h.

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

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  }
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 SQLTypeInfo & get_type_info() const
Definition: Analyzer.h:77
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::PointConstructor::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 155 of file PointConstructor.h.

References run_benchmark_import::args, CHECK, CHECK_EQ, Analyzer::Expr::get_type_info(), kENCODING_GEOINT, kPOINT, spatial_type::Codegen::operator_, and pt_local_storage_lv_.

158  {
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  }
#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::LLVMContext & context_
Definition: CgenState.h:338
const SQLTypeInfo & get_type_info() const
Definition: Analyzer.h:77
llvm::AllocaInst * pt_local_storage_lv_
const Analyzer::GeoOperator * operator_
Definition: Codegen.h:70
llvm::ConstantInt * llInt(const T v) const
Definition: CgenState.h:307
llvm::Value * finalize(llvm::Value *null_lv, llvm::Value *notnull_lv)
Definition: IRCodegen.cpp:1447
#define CHECK(condition)
Definition: Logger.h:211

+ Here is the call graph for this function:

std::tuple<std::vector<llvm::Value*>, llvm::Value*> spatial_type::PointConstructor::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 85 of file PointConstructor.h.

References CHECK, CHECK_EQ, Analyzer::Expr::get_type_info(), Analyzer::GeoOperator::getName(), spatial_type::Codegen::getOperand(), inline_fp_null_val(), inline_int_null_val(), is_null(), spatial_type::Codegen::is_nullable_, kENCODING_GEOINT, kENCODING_NONE, kPOINT, spatial_type::Codegen::operator_, pt_local_storage_lv_, and size().

88  {
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  }
#define CHECK_EQ(x, y)
Definition: Logger.h:219
llvm::IRBuilder ir_builder_
Definition: CgenState.h:340
double inline_fp_null_val(const SQL_TYPE_INFO &ti)
llvm::LLVMContext & context_
Definition: CgenState.h:338
CONSTEXPR DEVICE bool is_null(const T &value)
const std::string & getName() const
Definition: Analyzer.h:1925
llvm::ConstantFP * llFp(const float v) const
Definition: CgenState.h:311
const SQLTypeInfo & get_type_info() const
Definition: Analyzer.h:77
llvm::AllocaInst * pt_local_storage_lv_
const Analyzer::GeoOperator * operator_
Definition: Codegen.h:70
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
int64_t inline_int_null_val(const SQL_TYPE_INFO &ti)

+ Here is the call graph for this function:

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

Reimplemented from spatial_type::Codegen.

Definition at line 40 of file PointConstructor.h.

References CHECK, CHECK_EQ, Analyzer::Expr::get_type_info(), spatial_type::Codegen::getName(), getNullType(), inline_int_null_val(), spatial_type::Codegen::isNullable(), kENCODING_GEOINT, kINT, NULL_ARRAY_DOUBLE, spatial_type::Codegen::operator_, and pt_local_storage_lv_.

43  {
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  }
#define CHECK_EQ(x, y)
Definition: Logger.h:219
SQLTypeInfo getNullType() const final
auto isNullable() const
Definition: Codegen.h:35
const SQLTypeInfo & get_type_info() const
Definition: Analyzer.h:77
llvm::AllocaInst * pt_local_storage_lv_
const Analyzer::GeoOperator * operator_
Definition: Codegen.h:70
#define NULL_ARRAY_DOUBLE
#define CHECK(condition)
Definition: Logger.h:211
int64_t inline_int_null_val(const SQL_TYPE_INFO &ti)
std::string getName() const
Definition: Codegen.h:39
Definition: sqltypes.h:45

+ Here is the call graph for this function:

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

Implements spatial_type::Codegen.

Definition at line 82 of file PointConstructor.h.

References kBOOLEAN.

Referenced by getNullCheckCodegen().

82 { return SQLTypeInfo(kBOOLEAN); }

+ Here is the caller graph for this function:

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

Implements spatial_type::Codegen.

Definition at line 80 of file PointConstructor.h.

Referenced by codegenLoads().

80 { return 2; }

+ Here is the caller graph for this function:

Member Data Documentation

llvm::AllocaInst* spatial_type::PointConstructor::pt_local_storage_lv_ {nullptr}
private

Definition at line 211 of file PointConstructor.h.

Referenced by codegen(), codegenLoads(), and getNullCheckCodegen().


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