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
Expr *	get_arg () const
std::shared_ptr< Analyzer::Expr >	get_own_arg () const
bool	get_is_distinct () const
std::shared_ptr< Analyzer::Expr >	get_arg1 () const
std::shared_ptr< Analyzer::Expr >	deep_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::Expr >	rewrite_with_targetlist (const std::vector< std::shared_ptr< TargetEntry >> &tlist) const override
std::shared_ptr< Analyzer::Expr >	rewrite_with_child_targetlist (const std::vector< std::shared_ptr< TargetEntry >> &tlist) const override
std::shared_ptr< Analyzer::Expr >	rewrite_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::Expr >	get_shared_ptr ()
const SQLTypeInfo &	get_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::Expr >	decompress ()
virtual void	get_domain (DomainSet &domain_set) const

Private Attributes
SQLAgg	aggtype
std::shared_ptr< Analyzer::Expr >	arg
bool	is_distinct
std::shared_ptr< Analyzer::Expr >	arg1

Additional Inherited Members
Protected Attributes inherited from Analyzer::Expr
SQLTypeInfo	type_info
bool	contains_agg

Detailed Description

Definition at line 1310 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 1312 of file Analyzer.h.

      : Expr(ti, true), aggtype(a), arg(g), is_distinct(d), arg1(e) {}

Analyzer::AggExpr::AggExpr ( SQLTypes  t, SQLAgg  a, Expr *  g, bool  d, std::shared_ptr< Analyzer::Expr >  e, int  idx  )

inline

Definition at line 1318 of file Analyzer.h.

      : Expr(SQLTypeInfo(t, g == nullptr ? true : g->get_type_info().get_notnull()), true)
      , aggtype(a)
      , arg(g)
      , is_distinct(d)
      , arg1(e) {}

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 1343 of file Analyzer.h.

References arg.

                                       {
    if (include_agg && arg != nullptr) {
      arg->collect_column_var(colvar_set, include_agg);
    }
  }

void Analyzer::AggExpr::collect_rte_idx ( std::set< int > &  rte_idx_set ) const

inlineoverridevirtual

Reimplemented from Analyzer::Expr.

Definition at line 1338 of file Analyzer.h.

References arg.

                                                                {
    if (arg) {
      arg->collect_rte_idx(rte_idx_set);
    }
  };

std::shared_ptr< Analyzer::Expr > Analyzer::AggExpr::deep_copy ( ) const

overridevirtual

Implements Analyzer::Expr.

Definition at line 194 of file Analyzer.cpp.

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

Referenced by rewrite_with_targetlist().

                                                     {
  return makeExpr<AggExpr>(
      type_info, aggtype, arg ? arg->deep_copy() : nullptr, is_distinct, arg1);
}

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 3402 of file Analyzer.cpp.

References Analyzer::Expr::add_unique(), arg, and f().

                                                               {
  if (f(this)) {
    add_unique(expr_list);
    return;
  }
  if (arg != nullptr) {
    arg->find_expr(f, expr_list);
  }
}

Here is the call graph for this function:

SQLAgg Analyzer::AggExpr::get_aggtype ( ) const

inline

Definition at line 1329 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().

{ return aggtype; }

Here is the caller graph for this function:

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

inline

Definition at line 1330 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().

{ return arg.get(); }

Here is the caller graph for this function:

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

inline

Definition at line 1333 of file Analyzer.h.

References arg1.

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

{ return arg1; }

Here is the caller graph for this function:

bool Analyzer::AggExpr::get_is_distinct ( ) const

inline

Definition at line 1332 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().

{ 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 1331 of file Analyzer.h.

References arg.

{ return arg; }

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 1871 of file Analyzer.cpp.

References arg.

                                                                             {
  std::set<int> rte_idx_set;
  arg->collect_rte_idx(rte_idx_set);
  if (rte_idx_set.size() > 1) {
    join_predicates.push_back(this);
  } else if (rte_idx_set.size() == 1) {
    scan_predicates.push_back(this);
  } else {
    const_predicates.push_back(this);
  }
}

bool Analyzer::AggExpr::operator== ( const Expr &  rhs ) const

overridevirtual

Implements Analyzer::Expr.

Definition at line 2560 of file Analyzer.cpp.

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

                                              {
  if (typeid(rhs) != typeid(AggExpr)) {
    return false;
  }
  const AggExpr& rhs_ae = dynamic_cast<const AggExpr&>(rhs);
  if (aggtype != rhs_ae.get_aggtype() || is_distinct != rhs_ae.get_is_distinct()) {
    return false;
  }
  if (arg.get() == rhs_ae.get_arg()) {
    return true;
  }
  if (arg == nullptr || rhs_ae.get_arg() == nullptr) {
    return false;
  }
  return *arg == *rhs_ae.get_arg();
}

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 2117 of file Analyzer.cpp.

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

                                                              {
  int varno = 1;
  for (auto tle : tlist) {
    const Expr* e = tle->get_expr();
    if (typeid(*e) == typeid(AggExpr)) {
      const AggExpr* agg_expr = dynamic_cast<const AggExpr*>(e);
      if (*this == *agg_expr) {
        return makeExpr<Var>(agg_expr->get_type_info(), Var::kINPUT_OUTER, varno);
      }
    }
    varno++;
  }
  throw std::runtime_error(
      "Internal error: cannot find AggExpr from having clause in targetlist.");
}

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 2108 of file Analyzer.cpp.

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

                                                              {
  return makeExpr<AggExpr>(type_info,
                           aggtype,
                           arg ? arg->rewrite_with_child_targetlist(tlist) : nullptr,
                           is_distinct,
                           arg1);
}

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 2094 of file Analyzer.cpp.

References deep_copy().

                                                              {
  for (auto tle : tlist) {
    const Expr* e = tle->get_expr();
    if (typeid(*e) == typeid(AggExpr)) {
      const AggExpr* agg = dynamic_cast<const AggExpr*>(e);
      if (*this == *agg) {
        return agg->deep_copy();
      }
    }
  }
  throw std::runtime_error("Internal error: cannot find AggExpr in targetlist.");
}

Here is the call graph for this function:

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

overridevirtual

Implements Analyzer::Expr.

Definition at line 3015 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.

                                  {
  std::string agg;
  switch (aggtype) {
    case kAVG:
      agg = "AVG ";
      break;
    case kMIN:
      agg = "MIN ";
      break;
    case kMAX:
      agg = "MAX ";
      break;
    case kSUM:
      agg = "SUM ";
      break;
    case kCOUNT:
      agg = "COUNT ";
      break;
    case kAPPROX_COUNT_DISTINCT:
      agg = "APPROX_COUNT_DISTINCT";
      break;
    case kAPPROX_QUANTILE:
      agg = "APPROX_PERCENTILE";
      break;
    case kSINGLE_VALUE:
      agg = "SINGLE_VALUE";
      break;
    case kSAMPLE:
      agg = "SAMPLE";
      break;
    case kMODE:
      agg = "MODE";
      break;
    case kCOUNT_IF:
      agg = "COUNT_IF";
      break;
    case kSUM_IF:
      agg = "SUM_IF";
      break;
    default:
      UNREACHABLE() << "Unhandled aggtype: " << aggtype;
      break;
  }
  std::string str{"(" + agg};
  if (is_distinct) {
    str += "DISTINCT ";
  }
  if (arg) {
    str += arg->toString();
  } else {
    str += "*";
  }
  return str + ") ";
}

Member Data Documentation

SQLAgg Analyzer::AggExpr::aggtype

private

Definition at line 1363 of file Analyzer.h.

Referenced by deep_copy(), get_aggtype(), operator==(), rewrite_with_child_targetlist(), and toString().

std::shared_ptr<Analyzer::Expr> Analyzer::AggExpr::arg

private

Definition at line 1364 of file Analyzer.h.

Referenced by collect_column_var(), collect_rte_idx(), deep_copy(), find_expr(), get_arg(), get_own_arg(), group_predicates(), operator==(), rewrite_with_child_targetlist(), and toString().

std::shared_ptr<Analyzer::Expr> Analyzer::AggExpr::arg1

private

Definition at line 1367 of file Analyzer.h.

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

bool Analyzer::AggExpr::is_distinct

private

Definition at line 1365 of file Analyzer.h.

Referenced by deep_copy(), get_is_distinct(), operator==(), heavydb.serialized_result_set.ttypes.TTargetInfo::read(), rewrite_with_child_targetlist(), toString(), and heavydb.serialized_result_set.ttypes.TTargetInfo::write().

---

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

• /home/jenkins-slave/workspace/core-os-doxygen/Analyzer/Analyzer.h
• /home/jenkins-slave/workspace/core-os-doxygen/Analyzer/Analyzer.cpp