OmniSciDB  6686921089
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
NPoints.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 namespace spatial_type {
22 
23 class NPoints : public Codegen {
24  public:
25  NPoints(const Analyzer::GeoOperator* geo_operator,
26  const Catalog_Namespace::Catalog* catalog)
27  : Codegen(geo_operator, catalog) {}
28 
29  size_t size() const final { return 1; }
30 
32 
33  const Analyzer::Expr* getOperand(const size_t index) final {
34  CHECK_EQ(index, size_t(0));
35  if (operand_owned_) {
36  return operand_owned_.get();
37  }
38 
39  const auto operand = operator_->getOperand(0);
40  auto col_var = dynamic_cast<const Analyzer::ColumnVar*>(operand);
41  CHECK(col_var);
42 
43  geo_ti_ = col_var->get_type_info();
46 
47  // create a new operand which is just the coords and codegen it
48  const auto coords_column_id = col_var->get_column_id() + 1; // + 1 for coords
49  auto coords_cd =
50  get_column_descriptor(coords_column_id, col_var->get_table_id(), *cat_);
51  CHECK(coords_cd);
52 
53  operand_owned_ = std::make_unique<Analyzer::ColumnVar>(coords_cd->columnType,
54  col_var->get_table_id(),
55  coords_column_id,
56  col_var->get_rte_idx());
57  return operand_owned_.get();
58  }
59 
60  std::tuple<std::vector<llvm::Value*>, llvm::Value*> codegenLoads(
61  const std::vector<llvm::Value*>& arg_lvs,
62  const std::vector<llvm::Value*>& pos_lvs,
63  CgenState* cgen_state) final {
64  CHECK_EQ(arg_lvs.size(), size_t(1));
65  auto& argument_lv = arg_lvs.front();
66  std::string fn_name("array_size");
67 
68  const auto& elem_ti = getOperand(0)->get_type_info().get_elem_type();
69  std::vector<llvm::Value*> array_size_args{
70  argument_lv,
71  pos_lvs.front(),
72  cgen_state->llInt(log2_bytes(elem_ti.get_logical_size()))};
73 
74  const bool is_nullable = isNullable();
75 
76  if (is_nullable) {
77  fn_name += "_nullable";
78  array_size_args.push_back(cgen_state->inlineIntNull(getTypeInfo()));
79  }
80  const auto coords_arr_sz_lv = cgen_state->emitExternalCall(
81  fn_name, get_int_type(32, cgen_state->context_), array_size_args);
82  return std::make_tuple(std::vector<llvm::Value*>{coords_arr_sz_lv}, coords_arr_sz_lv);
83  }
84 
85  std::vector<llvm::Value*> codegen(const std::vector<llvm::Value*>& args,
86  CodeGenerator::NullCheckCodegen* nullcheck_codegen,
87  CgenState* cgen_state,
88  const CompilationOptions& co) final {
89  CHECK_EQ(args.size(), size_t(1));
90 
91  // divide the coord size by the constant compression value and return it
92  auto& builder = cgen_state->ir_builder_;
93  llvm::Value* conversion_constant{nullptr};
95  conversion_constant = cgen_state->llInt(4);
96  } else {
97  conversion_constant = cgen_state->llInt(8);
98  }
99  CHECK(conversion_constant);
100  const auto total_num_pts = builder.CreateUDiv(args.front(), conversion_constant);
101  const auto ret = builder.CreateUDiv(total_num_pts, cgen_state->llInt(2));
102  if (isNullable()) {
103  CHECK(nullcheck_codegen);
104  return {nullcheck_codegen->finalize(cgen_state->inlineIntNull(getTypeInfo()), ret)};
105  } else {
106  return {ret};
107  }
108  }
109 
110  protected:
112  std::unique_ptr<Analyzer::ColumnVar> operand_owned_;
113 };
114 
115 } // namespace spatial_type
std::unique_ptr< Analyzer::ColumnVar > operand_owned_
Definition: NPoints.h:112
#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:111
#define const
size_t size() const final
Definition: NPoints.h:29
SQLTypeInfo getNullType() const final
Definition: NPoints.h:31
llvm::Type * get_int_type(const int width, llvm::LLVMContext &context)
auto getTypeInfo() const
Definition: Codegen.h:37
const Analyzer::Expr * getOperand(const size_t index) final
Definition: NPoints.h:33
NPoints(const Analyzer::GeoOperator *geo_operator, const Catalog_Namespace::Catalog *catalog)
Definition: NPoints.h:25
const Catalog_Namespace::Catalog * cat_
Definition: Codegen.h:71
auto isNullable() const
Definition: Codegen.h:35
const SQLTypeInfo & get_type_info() const
Definition: Analyzer.h:77
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
Definition: NPoints.h:60
HOST DEVICE EncodingType get_compression() const
Definition: sqltypes.h:337
std::vector< llvm::Value * > codegen(const std::vector< llvm::Value * > &args, CodeGenerator::NullCheckCodegen *nullcheck_codegen, CgenState *cgen_state, const CompilationOptions &co) final
Definition: NPoints.h:85
const Analyzer::GeoOperator * operator_
Definition: Codegen.h:70
SQLTypeInfo geo_ti_
Definition: NPoints.h:111
#define CHECK(condition)
Definition: Logger.h:209
bool is_geometry() const
Definition: sqltypes.h:521
Analyzer::Expr * getOperand(const size_t index) const
Definition: Analyzer.cpp:3644
uint32_t log2_bytes(const uint32_t bytes)
Definition: Execute.h:174
Definition: sqltypes.h:45
HOST DEVICE bool get_notnull() const
Definition: sqltypes.h:336
SQLTypeInfo get_elem_type() const
Definition: sqltypes.h:850
const ColumnDescriptor * get_column_descriptor(const int col_id, const int table_id, const Catalog_Namespace::Catalog &cat)
Definition: Execute.h:189