OmniSciDB  3a86f6ec37
TargetInfo.h
Go to the documentation of this file.
1 /*
2  * Copyright 2017 MapD Technologies, 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 
22 #ifndef QUERYENGINE_TARGETINFO_H
23 #define QUERYENGINE_TARGETINFO_H
24 
25 #include "sqldefs.h"
26 #include "sqltypes.h"
27 
28 #include "../Analyzer/Analyzer.h"
29 
30 inline const Analyzer::AggExpr* cast_to_agg_expr(const Analyzer::Expr* target_expr) {
31  return dynamic_cast<const Analyzer::AggExpr*>(target_expr);
32 }
33 
35  const std::shared_ptr<Analyzer::Expr> target_expr) {
36  return dynamic_cast<const Analyzer::AggExpr*>(target_expr.get());
37 }
38 
39 struct TargetInfo {
40  bool is_agg;
46 #ifndef __CUDACC__
47  public:
48  inline std::string toString() const {
49  auto result = std::string("TargetInfo(");
50  result += "is_agg=" + std::string(is_agg ? "true" : "false") + ", ";
51  result += "agg_kind=" + ::toString(agg_kind) + ", ";
52  result += "sql_type=" + sql_type.to_string() + ", ";
53  result += "agg_arg_type=" + agg_arg_type.to_string() + ", ";
54  result += "skip_null_val=" + std::string(skip_null_val ? "true" : "false") + ", ";
55  result += "is_distinct=" + std::string(is_distinct ? "true" : "false") + ")";
56  return result;
57  }
58 #endif
59 };
60 
66  switch (agg_kind) {
67  case kMIN:
68  case kMAX:
69  case kSINGLE_VALUE:
70  case kSAMPLE:
71  return true;
72  default:
73  break;
74  }
75  return false;
76 }
77 
78 template <class PointerType>
79 inline TargetInfo get_target_info(const PointerType target_expr,
80  const bool bigint_count) {
81  const auto agg_expr = cast_to_agg_expr(target_expr);
82  bool notnull = target_expr->get_type_info().get_notnull();
83  if (!agg_expr) {
84  auto target_ti = target_expr ? get_logical_type_info(target_expr->get_type_info())
85  : SQLTypeInfo(kBIGINT, notnull);
86  return {false, kMIN, target_ti, SQLTypeInfo(kNULLT, false), false, false};
87  }
88  const auto agg_type = agg_expr->get_aggtype();
89  const auto agg_arg = agg_expr->get_arg();
90  if (!agg_arg) {
91  CHECK_EQ(kCOUNT, agg_type);
92  CHECK(!agg_expr->get_is_distinct());
93  return {true,
94  kCOUNT,
95  SQLTypeInfo(bigint_count ? kBIGINT : kINT, notnull),
96  SQLTypeInfo(kNULLT, false),
97  false,
98  false};
99  }
100 
101  const auto& agg_arg_ti = agg_arg->get_type_info();
102  bool is_distinct{false};
103  if (agg_expr->get_aggtype() == kCOUNT) {
104  is_distinct = agg_expr->get_is_distinct();
105  }
106 
107  if (agg_type == kAVG) {
108  // Upcast the target type for AVG, so that the integer argument does not overflow the
109  // sum
110  return {true,
111  agg_expr->get_aggtype(),
112  agg_arg_ti.is_integer() ? SQLTypeInfo(kBIGINT, agg_arg_ti.get_notnull())
113  : agg_arg_ti,
114  agg_arg_ti,
115  !agg_arg_ti.get_notnull(),
116  is_distinct};
117  }
118 
119  return {
120  true,
121  agg_expr->get_aggtype(),
122  agg_type == kCOUNT
123  ? SQLTypeInfo((is_distinct || bigint_count) ? kBIGINT : kINT, notnull)
124  : agg_expr->get_type_info(),
125  agg_arg_ti,
126  agg_type == kCOUNT && agg_arg_ti.is_varlen() ? false : !agg_arg_ti.get_notnull(),
127  is_distinct};
128 }
129 
130 inline bool is_distinct_target(const TargetInfo& target_info) {
131  return target_info.is_distinct || target_info.agg_kind == kAPPROX_COUNT_DISTINCT;
132 }
133 
134 inline bool takes_float_argument(const TargetInfo& target_info) {
135  return target_info.is_agg &&
136  (target_info.agg_kind == kAVG || target_info.agg_kind == kSUM ||
137  target_info.agg_kind == kMIN || target_info.agg_kind == kMAX ||
138  target_info.agg_kind == kSINGLE_VALUE) &&
139  target_info.agg_arg_type.get_type() == kFLOAT;
140 }
141 
142 #endif // QUERYENGINE_TARGETINFO_H
const Analyzer::Expr * agg_arg(const Analyzer::Expr *expr)
SQLAgg
Definition: sqldefs.h:71
#define CHECK_EQ(x, y)
Definition: Logger.h:205
SQLTypeInfo sql_type
Definition: TargetInfo.h:42
TargetInfo get_target_info(const PointerType target_expr, const bool bigint_count)
Definition: TargetInfo.h:79
SQLTypeInfo get_logical_type_info(const SQLTypeInfo &type_info)
Definition: sqltypes.h:893
Constants for Builtin SQL Types supported by OmniSci.
bool takes_float_argument(const TargetInfo &target_info)
Definition: TargetInfo.h:134
bool skip_null_val
Definition: TargetInfo.h:44
SQLTypeInfo agg_arg_type
Definition: TargetInfo.h:43
Definition: sqldefs.h:73
HOST DEVICE bool get_notnull() const
Definition: sqltypes.h:318
bool is_agg
Definition: TargetInfo.h:40
std::string to_string() const
Definition: sqltypes.h:446
Definition: sqldefs.h:75
bool is_distinct_target(const TargetInfo &target_info)
Definition: TargetInfo.h:130
SQLAgg agg_kind
Definition: TargetInfo.h:41
std::string toString() const
Definition: TargetInfo.h:48
const Analyzer::AggExpr * cast_to_agg_expr(const Analyzer::Expr *target_expr)
Definition: TargetInfo.h:30
Definition: sqldefs.h:76
const SQLTypeInfo & get_type_info() const
Definition: Analyzer.h:78
#define CHECK(condition)
Definition: Logger.h:197
HOST DEVICE SQLTypes get_type() const
Definition: sqltypes.h:311
Common Enum definitions for SQL processing.
Definition: sqltypes.h:44
bool is_distinct
Definition: TargetInfo.h:45
Definition: sqldefs.h:74
Definition: sqldefs.h:72
bool is_agg_domain_range_equivalent(const SQLAgg &agg_kind)
Definition: TargetInfo.h:65