OmniSciDB  c1a53651b2
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
Analyzer::AggExpr Class Reference

#include <Analyzer.h>

+ Inheritance diagram for Analyzer::AggExpr:
+ Collaboration diagram for Analyzer::AggExpr:

Public Member Functions

 AggExpr (const SQLTypeInfo &ti, SQLAgg a, std::shared_ptr< Analyzer::Expr > g, bool d, std::shared_ptr< Analyzer::Expr > e)
 
 AggExpr (SQLTypes t, SQLAgg a, Expr *g, bool d, std::shared_ptr< Analyzer::Expr > e, int idx)
 
SQLAgg get_aggtype () const
 
Exprget_arg () const
 
std::shared_ptr< Analyzer::Exprget_own_arg () const
 
bool get_is_distinct () const
 
std::shared_ptr< Analyzer::Exprget_arg1 () const
 
std::shared_ptr< Analyzer::Exprdeep_copy () const override
 
void group_predicates (std::list< const Expr * > &scan_predicates, std::list< const Expr * > &join_predicates, std::list< const Expr * > &const_predicates) const override
 
void collect_rte_idx (std::set< int > &rte_idx_set) const override
 
void collect_column_var (std::set< const ColumnVar *, bool(*)(const ColumnVar *, const ColumnVar *)> &colvar_set, bool include_agg) const override
 
std::shared_ptr< Analyzer::Exprrewrite_with_targetlist (const std::vector< std::shared_ptr< TargetEntry >> &tlist) const override
 
std::shared_ptr< Analyzer::Exprrewrite_with_child_targetlist (const std::vector< std::shared_ptr< TargetEntry >> &tlist) const override
 
std::shared_ptr< Analyzer::Exprrewrite_agg_to_var (const std::vector< std::shared_ptr< TargetEntry >> &tlist) const override
 
bool operator== (const Expr &rhs) const override
 
std::string toString () const override
 
void find_expr (std::function< bool(const Expr *)> f, std::list< const Expr * > &expr_list) const override
 
- Public Member Functions inherited from Analyzer::Expr
 Expr (SQLTypes t, bool notnull)
 
 Expr (SQLTypes t, int d, bool notnull)
 
 Expr (SQLTypes t, int d, int s, bool notnull)
 
 Expr (const SQLTypeInfo &ti, bool has_agg=false)
 
virtual ~Expr ()
 
std::shared_ptr< Analyzer::Exprget_shared_ptr ()
 
const SQLTypeInfoget_type_info () const
 
void set_type_info (const SQLTypeInfo &ti)
 
bool get_contains_agg () const
 
void set_contains_agg (bool a)
 
virtual std::shared_ptr
< Analyzer::Expr
add_cast (const SQLTypeInfo &new_type_info)
 
virtual void check_group_by (const std::list< std::shared_ptr< Analyzer::Expr >> &groupby) const
 
virtual std::shared_ptr
< Analyzer::Expr
normalize_simple_predicate (int &rte_idx) const
 
virtual size_t get_num_column_vars (const bool include_agg) const
 
virtual void print () const
 
virtual void add_unique (std::list< const Expr * > &expr_list) const
 
std::shared_ptr< Analyzer::Exprdecompress ()
 
virtual void get_domain (DomainSet &domain_set) const
 

Private Attributes

SQLAgg aggtype
 
std::shared_ptr< Analyzer::Exprarg
 
bool is_distinct
 
std::shared_ptr< Analyzer::Exprarg1
 

Additional Inherited Members

- Protected Attributes inherited from Analyzer::Expr
SQLTypeInfo type_info
 
bool contains_agg
 

Detailed Description

Definition at line 1188 of file Analyzer.h.

Constructor & Destructor Documentation

Analyzer::AggExpr::AggExpr ( const SQLTypeInfo ti,
SQLAgg  a,
std::shared_ptr< Analyzer::Expr g,
bool  d,
std::shared_ptr< Analyzer::Expr e 
)
inline

Definition at line 1190 of file Analyzer.h.

1195  : Expr(ti, true), aggtype(a), arg(g), is_distinct(d), arg1(e) {}
Expr(SQLTypes t, bool notnull)
Definition: Analyzer.h:70
std::shared_ptr< Analyzer::Expr > arg
Definition: Analyzer.h:1242
constexpr double a
Definition: Utm.h:32
std::shared_ptr< Analyzer::Expr > arg1
Definition: Analyzer.h:1245
Analyzer::AggExpr::AggExpr ( SQLTypes  t,
SQLAgg  a,
Expr g,
bool  d,
std::shared_ptr< Analyzer::Expr e,
int  idx 
)
inline

Definition at line 1196 of file Analyzer.h.

1202  : Expr(SQLTypeInfo(t, g == nullptr ? true : g->get_type_info().get_notnull()), true)
1203  , aggtype(a)
1204  , arg(g)
1205  , is_distinct(d)
1206  , arg1(e) {}
Expr(SQLTypes t, bool notnull)
Definition: Analyzer.h:70
std::shared_ptr< Analyzer::Expr > arg
Definition: Analyzer.h:1242
constexpr double a
Definition: Utm.h:32
std::shared_ptr< Analyzer::Expr > arg1
Definition: Analyzer.h:1245

Member Function Documentation

void Analyzer::AggExpr::collect_column_var ( std::set< const ColumnVar *, bool(*)(const ColumnVar *, const ColumnVar *)> &  colvar_set,
bool  include_agg 
) const
inlineoverridevirtual

Reimplemented from Analyzer::Expr.

Definition at line 1221 of file Analyzer.h.

References arg.

1224  {
1225  if (include_agg && arg != nullptr) {
1226  arg->collect_column_var(colvar_set, include_agg);
1227  }
1228  }
std::shared_ptr< Analyzer::Expr > arg
Definition: Analyzer.h:1242
void Analyzer::AggExpr::collect_rte_idx ( std::set< int > &  rte_idx_set) const
inlineoverridevirtual

Reimplemented from Analyzer::Expr.

Definition at line 1216 of file Analyzer.h.

References arg.

1216  {
1217  if (arg) {
1218  arg->collect_rte_idx(rte_idx_set);
1219  }
1220  };
std::shared_ptr< Analyzer::Expr > arg
Definition: Analyzer.h:1242
std::shared_ptr< Analyzer::Expr > Analyzer::AggExpr::deep_copy ( ) const
overridevirtual

Implements Analyzer::Expr.

Definition at line 177 of file Analyzer.cpp.

References aggtype, arg, arg1, is_distinct, and Analyzer::Expr::type_info.

Referenced by rewrite_with_targetlist().

177  {
178  return makeExpr<AggExpr>(
179  type_info, aggtype, arg ? arg->deep_copy() : nullptr, is_distinct, arg1);
180 }
std::shared_ptr< Analyzer::Expr > arg
Definition: Analyzer.h:1242
SQLTypeInfo type_info
Definition: Analyzer.h:180
std::shared_ptr< Analyzer::Expr > arg1
Definition: Analyzer.h:1245

+ Here is the caller graph for this function:

void Analyzer::AggExpr::find_expr ( std::function< bool(const Expr *)>  f,
std::list< const Expr * > &  expr_list 
) const
overridevirtual

Reimplemented from Analyzer::Expr.

Definition at line 3251 of file Analyzer.cpp.

References Analyzer::Expr::add_unique(), arg, and anonymous_namespace{Utm.h}::f.

3252  {
3253  if (f(this)) {
3254  add_unique(expr_list);
3255  return;
3256  }
3257  if (arg != nullptr) {
3258  arg->find_expr(f, expr_list);
3259  }
3260 }
virtual void add_unique(std::list< const Expr * > &expr_list) const
Definition: Analyzer.cpp:3116
std::shared_ptr< Analyzer::Expr > arg
Definition: Analyzer.h:1242
constexpr double f
Definition: Utm.h:31

+ Here is the call graph for this function:

SQLAgg Analyzer::AggExpr::get_aggtype ( ) const
inline

Definition at line 1207 of file Analyzer.h.

References aggtype.

Referenced by anonymous_namespace{SerializeToSql.cpp}::agg_to_string(), anonymous_namespace{RelAlgExecutor.cpp}::is_agg(), operator==(), and DeepCopyVisitor::visitAggExpr().

1207 { return aggtype; }

+ Here is the caller graph for this function:

Expr* Analyzer::AggExpr::get_arg ( ) const
inline

Definition at line 1208 of file Analyzer.h.

References arg.

Referenced by agg_arg(), anonymous_namespace{SerializeToSql.cpp}::agg_to_string(), GroupByAndAggregate::gpuCanHandleOrderEntries(), anonymous_namespace{GroupByAndAggregate.cpp}::init_count_distinct_descriptors(), operator==(), DeepCopyVisitor::visitAggExpr(), and ScalarExprVisitor< std::set< shared::TableKey > >::visitAggExpr().

1208 { return arg.get(); }
std::shared_ptr< Analyzer::Expr > arg
Definition: Analyzer.h:1242

+ Here is the caller graph for this function:

std::shared_ptr<Analyzer::Expr> Analyzer::AggExpr::get_arg1 ( ) const
inline

Definition at line 1211 of file Analyzer.h.

References arg1.

Referenced by QueryMemoryInitializer::allocateTDigests(), and DeepCopyVisitor::visitAggExpr().

1211 { return arg1; }
std::shared_ptr< Analyzer::Expr > arg1
Definition: Analyzer.h:1245

+ Here is the caller graph for this function:

bool Analyzer::AggExpr::get_is_distinct ( ) const
inline

Definition at line 1210 of file Analyzer.h.

References is_distinct.

Referenced by anonymous_namespace{SerializeToSql.cpp}::agg_to_string(), anonymous_namespace{RelAlgExecutor.cpp}::is_count_distinct(), operator==(), and DeepCopyVisitor::visitAggExpr().

1210 { return is_distinct; }

+ Here is the caller graph for this function:

std::shared_ptr<Analyzer::Expr> Analyzer::AggExpr::get_own_arg ( ) const
inline

Definition at line 1209 of file Analyzer.h.

References arg.

1209 { return arg; }
std::shared_ptr< Analyzer::Expr > arg
Definition: Analyzer.h:1242
void Analyzer::AggExpr::group_predicates ( std::list< const Expr * > &  scan_predicates,
std::list< const Expr * > &  join_predicates,
std::list< const Expr * > &  const_predicates 
) const
overridevirtual

Reimplemented from Analyzer::Expr.

Definition at line 1812 of file Analyzer.cpp.

References arg.

1814  {
1815  std::set<int> rte_idx_set;
1816  arg->collect_rte_idx(rte_idx_set);
1817  if (rte_idx_set.size() > 1) {
1818  join_predicates.push_back(this);
1819  } else if (rte_idx_set.size() == 1) {
1820  scan_predicates.push_back(this);
1821  } else {
1822  const_predicates.push_back(this);
1823  }
1824 }
std::shared_ptr< Analyzer::Expr > arg
Definition: Analyzer.h:1242
bool Analyzer::AggExpr::operator== ( const Expr rhs) const
overridevirtual

Implements Analyzer::Expr.

Definition at line 2457 of file Analyzer.cpp.

References aggtype, arg, get_aggtype(), get_arg(), get_is_distinct(), and is_distinct.

2457  {
2458  if (typeid(rhs) != typeid(AggExpr)) {
2459  return false;
2460  }
2461  const AggExpr& rhs_ae = dynamic_cast<const AggExpr&>(rhs);
2462  if (aggtype != rhs_ae.get_aggtype() || is_distinct != rhs_ae.get_is_distinct()) {
2463  return false;
2464  }
2465  if (arg.get() == rhs_ae.get_arg()) {
2466  return true;
2467  }
2468  if (arg == nullptr || rhs_ae.get_arg() == nullptr) {
2469  return false;
2470  }
2471  return *arg == *rhs_ae.get_arg();
2472 }
std::shared_ptr< Analyzer::Expr > arg
Definition: Analyzer.h:1242
AggExpr(const SQLTypeInfo &ti, SQLAgg a, std::shared_ptr< Analyzer::Expr > g, bool d, std::shared_ptr< Analyzer::Expr > e)
Definition: Analyzer.h:1190

+ Here is the call graph for this function:

std::shared_ptr< Analyzer::Expr > Analyzer::AggExpr::rewrite_agg_to_var ( const std::vector< std::shared_ptr< TargetEntry >> &  tlist) const
overridevirtual

Reimplemented from Analyzer::Expr.

Definition at line 2058 of file Analyzer.cpp.

References Analyzer::Expr::get_type_info(), and Analyzer::Var::kINPUT_OUTER.

2059  {
2060  int varno = 1;
2061  for (auto tle : tlist) {
2062  const Expr* e = tle->get_expr();
2063  if (typeid(*e) == typeid(AggExpr)) {
2064  const AggExpr* agg_expr = dynamic_cast<const AggExpr*>(e);
2065  if (*this == *agg_expr) {
2066  return makeExpr<Var>(agg_expr->get_type_info(), Var::kINPUT_OUTER, varno);
2067  }
2068  }
2069  varno++;
2070  }
2071  throw std::runtime_error(
2072  "Internal error: cannot find AggExpr from having clause in targetlist.");
2073 }
Expr(SQLTypes t, bool notnull)
Definition: Analyzer.h:70
AggExpr(const SQLTypeInfo &ti, SQLAgg a, std::shared_ptr< Analyzer::Expr > g, bool d, std::shared_ptr< Analyzer::Expr > e)
Definition: Analyzer.h:1190

+ Here is the call graph for this function:

std::shared_ptr< Analyzer::Expr > Analyzer::AggExpr::rewrite_with_child_targetlist ( const std::vector< std::shared_ptr< TargetEntry >> &  tlist) const
overridevirtual

Reimplemented from Analyzer::Expr.

Definition at line 2049 of file Analyzer.cpp.

References aggtype, arg, arg1, is_distinct, and Analyzer::Expr::type_info.

2050  {
2051  return makeExpr<AggExpr>(type_info,
2052  aggtype,
2053  arg ? arg->rewrite_with_child_targetlist(tlist) : nullptr,
2054  is_distinct,
2055  arg1);
2056 }
std::shared_ptr< Analyzer::Expr > arg
Definition: Analyzer.h:1242
SQLTypeInfo type_info
Definition: Analyzer.h:180
std::shared_ptr< Analyzer::Expr > arg1
Definition: Analyzer.h:1245
std::shared_ptr< Analyzer::Expr > Analyzer::AggExpr::rewrite_with_targetlist ( const std::vector< std::shared_ptr< TargetEntry >> &  tlist) const
overridevirtual

Reimplemented from Analyzer::Expr.

Definition at line 2035 of file Analyzer.cpp.

References deep_copy().

2036  {
2037  for (auto tle : tlist) {
2038  const Expr* e = tle->get_expr();
2039  if (typeid(*e) == typeid(AggExpr)) {
2040  const AggExpr* agg = dynamic_cast<const AggExpr*>(e);
2041  if (*this == *agg) {
2042  return agg->deep_copy();
2043  }
2044  }
2045  }
2046  throw std::runtime_error("Internal error: cannot find AggExpr in targetlist.");
2047 }
Expr(SQLTypes t, bool notnull)
Definition: Analyzer.h:70
AggExpr(const SQLTypeInfo &ti, SQLAgg a, std::shared_ptr< Analyzer::Expr > g, bool d, std::shared_ptr< Analyzer::Expr > e)
Definition: Analyzer.h:1190

+ Here is the call graph for this function:

std::string Analyzer::AggExpr::toString ( ) const
overridevirtual

Implements Analyzer::Expr.

Definition at line 2886 of file Analyzer.cpp.

References aggtype, arg, is_distinct, kAPPROX_COUNT_DISTINCT, kAPPROX_QUANTILE, kAVG, kCOUNT, kCOUNT_IF, kMAX, kMIN, kMODE, kSAMPLE, kSINGLE_VALUE, kSUM, kSUM_IF, and UNREACHABLE.

2886  {
2887  std::string agg;
2888  switch (aggtype) {
2889  case kAVG:
2890  agg = "AVG ";
2891  break;
2892  case kMIN:
2893  agg = "MIN ";
2894  break;
2895  case kMAX:
2896  agg = "MAX ";
2897  break;
2898  case kSUM:
2899  agg = "SUM ";
2900  break;
2901  case kCOUNT:
2902  agg = "COUNT ";
2903  break;
2905  agg = "APPROX_COUNT_DISTINCT";
2906  break;
2907  case kAPPROX_QUANTILE:
2908  agg = "APPROX_PERCENTILE";
2909  break;
2910  case kSINGLE_VALUE:
2911  agg = "SINGLE_VALUE";
2912  break;
2913  case kSAMPLE:
2914  agg = "SAMPLE";
2915  break;
2916  case kMODE:
2917  agg = "MODE";
2918  break;
2919  case kCOUNT_IF:
2920  agg = "COUNT_IF";
2921  break;
2922  case kSUM_IF:
2923  agg = "SUM_IF";
2924  break;
2925  default:
2926  UNREACHABLE() << "Unhandled aggtype: " << aggtype;
2927  break;
2928  }
2929  std::string str{"(" + agg};
2930  if (is_distinct) {
2931  str += "DISTINCT ";
2932  }
2933  if (arg) {
2934  str += arg->toString();
2935  } else {
2936  str += "*";
2937  }
2938  return str + ") ";
2939 }
#define UNREACHABLE()
Definition: Logger.h:337
std::shared_ptr< Analyzer::Expr > arg
Definition: Analyzer.h:1242
Definition: sqldefs.h:75
Definition: sqldefs.h:77
Definition: sqldefs.h:78
Definition: sqldefs.h:76
Definition: sqldefs.h:74
Definition: sqldefs.h:83

Member Data Documentation

SQLAgg Analyzer::AggExpr::aggtype
private
std::shared_ptr<Analyzer::Expr> Analyzer::AggExpr::arg1
private

Definition at line 1245 of file Analyzer.h.

Referenced by deep_copy(), get_arg1(), and rewrite_with_child_targetlist().


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