OmniSciDB  c07336695a
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 };
47 
53  switch (agg_kind) {
54  case kMIN:
55  case kMAX:
56  case kSAMPLE:
57  return true;
58  default:
59  break;
60  }
61  return false;
62 }
63 
64 template <class PointerType>
65 inline TargetInfo get_target_info(const PointerType target_expr,
66  const bool bigint_count) {
67  const auto agg_expr = cast_to_agg_expr(target_expr);
68  bool notnull = target_expr->get_type_info().get_notnull();
69  if (!agg_expr) {
70  auto target_ti = target_expr ? get_logical_type_info(target_expr->get_type_info())
71  : SQLTypeInfo(kBIGINT, notnull);
72  return {false, kMIN, target_ti, SQLTypeInfo(kNULLT, false), false, false};
73  }
74  const auto agg_type = agg_expr->get_aggtype();
75  const auto agg_arg = agg_expr->get_arg();
76  if (!agg_arg) {
77  CHECK_EQ(kCOUNT, agg_type);
78  CHECK(!agg_expr->get_is_distinct());
79  return {true,
80  kCOUNT,
81  SQLTypeInfo(bigint_count ? kBIGINT : kINT, notnull),
82  SQLTypeInfo(kNULLT, false),
83  false,
84  false};
85  }
86 
87  const auto& agg_arg_ti = agg_arg->get_type_info();
88  bool is_distinct{false};
89  if (agg_expr->get_aggtype() == kCOUNT) {
90  is_distinct = agg_expr->get_is_distinct();
91  }
92 
93  if (agg_type == kAVG) {
94  // Upcast the target type for AVG, so that the integer argument does not overflow the
95  // sum
96  return {true,
97  agg_expr->get_aggtype(),
98  agg_arg_ti.is_integer() ? SQLTypeInfo(kBIGINT, agg_arg_ti.get_notnull())
99  : agg_arg_ti,
100  agg_arg_ti,
101  !agg_arg_ti.get_notnull(),
102  is_distinct};
103  }
104 
105  return {
106  true,
107  agg_expr->get_aggtype(),
108  agg_type == kCOUNT
109  ? SQLTypeInfo((is_distinct || bigint_count) ? kBIGINT : kINT, notnull)
110  : agg_expr->get_type_info(),
111  agg_arg_ti,
112  agg_type == kCOUNT && agg_arg_ti.is_varlen() ? false : !agg_arg_ti.get_notnull(),
113  is_distinct};
114 }
115 
116 inline bool is_distinct_target(const TargetInfo& target_info) {
117  return target_info.is_distinct || target_info.agg_kind == kAPPROX_COUNT_DISTINCT;
118 }
119 
120 inline bool takes_float_argument(const TargetInfo& target_info) {
121  return target_info.is_agg &&
122  (target_info.agg_kind == kAVG || target_info.agg_kind == kSUM ||
123  target_info.agg_kind == kMIN || target_info.agg_kind == kMAX) &&
124  target_info.agg_arg_type.get_type() == kFLOAT;
125 }
126 
127 #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:195
SQLTypeInfo sql_type
Definition: TargetInfo.h:42
HOST DEVICE bool get_notnull() const
Definition: sqltypes.h:326
TargetInfo get_target_info(const PointerType target_expr, const bool bigint_count)
Definition: TargetInfo.h:65
HOST DEVICE SQLTypes get_type() const
Definition: sqltypes.h:319
SQLTypeInfo get_logical_type_info(const SQLTypeInfo &type_info)
Definition: sqltypes.h:836
Constants for Builtin SQL Types supported by MapD.
bool takes_float_argument(const TargetInfo &target_info)
Definition: TargetInfo.h:120
bool skip_null_val
Definition: TargetInfo.h:44
SQLTypeInfo agg_arg_type
Definition: TargetInfo.h:43
Definition: sqldefs.h:71
bool is_agg
Definition: TargetInfo.h:40
Definition: sqldefs.h:71
bool is_distinct_target(const TargetInfo &target_info)
Definition: TargetInfo.h:116
SQLTypeInfoCore< ArrayContextTypeSizer, ExecutorTypePackaging, DateTimeFacilities > SQLTypeInfo
Definition: sqltypes.h:819
SQLAgg agg_kind
Definition: TargetInfo.h:41
const Analyzer::AggExpr * cast_to_agg_expr(const Analyzer::Expr *target_expr)
Definition: TargetInfo.h:30
Definition: sqldefs.h:71
const SQLTypeInfo & get_type_info() const
Definition: Analyzer.h:77
#define CHECK(condition)
Definition: Logger.h:187
Common Enum definitions for SQL processing.
Definition: sqltypes.h:47
bool is_distinct
Definition: TargetInfo.h:45
Definition: sqldefs.h:71
Definition: sqldefs.h:71
bool is_agg_domain_range_equivalent(const SQLAgg &agg_kind)
Definition: TargetInfo.h:52