WindowExpressionRewrite.cpp File Reference

#include "WindowExpressionRewrite.h"

Include dependency graph for WindowExpressionRewrite.cpp:

Go to the source code of this file.

Namespaces
	anonymous_namespace{WindowExpressionRewrite.cpp}

Functions
bool	anonymous_namespace{WindowExpressionRewrite.cpp}::matches_else_null (const Analyzer::CaseExpr *case_expr)
bool	anonymous_namespace{WindowExpressionRewrite.cpp}::matches_gt_bigint_zero (const Analyzer::BinOper *window_gt_zero)
bool	anonymous_namespace{WindowExpressionRewrite.cpp}::window_sum_and_count_match (const Analyzer::WindowFunction *sum_window_expr, const Analyzer::WindowFunction *count_window_expr)
bool	anonymous_namespace{WindowExpressionRewrite.cpp}::is_sum_kind (const SqlWindowFunctionKind kind)
std::shared_ptr < Analyzer::WindowFunction >	rewrite_sum_window (const Analyzer::Expr *expr)
std::shared_ptr < Analyzer::WindowFunction >	rewrite_avg_window (const Analyzer::Expr *expr)

Function Documentation

std::shared_ptr<Analyzer::WindowFunction> rewrite_avg_window

(

const Analyzer::Expr *

expr

)

Definition at line 101 of file WindowExpressionRewrite.cpp.

References AVG, CHECK_EQ, COUNT, expr_list_match(), kBIGINT, kCAST, kDIVIDE, kDOUBLE, and rewrite_sum_window().

Referenced by rewrite_expr().

                                                                                   {
  const auto cast_expr = dynamic_cast<const Analyzer::UOper*>(expr);
  const auto div_expr = dynamic_cast<const Analyzer::BinOper*>(
      cast_expr && cast_expr->get_optype() == kCAST ? cast_expr->get_operand() : expr);
  if (!div_expr || div_expr->get_optype() != kDIVIDE) {
    return nullptr;
  }
  const auto sum_window_expr = rewrite_sum_window(div_expr->get_left_operand());
  if (!sum_window_expr) {
    return nullptr;
  }
  const auto cast_count_window =
      dynamic_cast<const Analyzer::UOper*>(div_expr->get_right_operand());
  if (cast_count_window && cast_count_window->get_optype() != kCAST) {
    return nullptr;
  }
  const auto count_window = dynamic_cast<const Analyzer::WindowFunction*>(
      cast_count_window ? cast_count_window->get_operand()
                        : div_expr->get_right_operand());
  if (!count_window || count_window->getKind() != SqlWindowFunctionKind::COUNT) {
    return nullptr;
  }
  CHECK_EQ(count_window->get_type_info().get_type(), kBIGINT);
  if (cast_count_window && cast_count_window->get_type_info().get_type() !=
                               sum_window_expr->get_type_info().get_type()) {
    return nullptr;
  }
  if (!expr_list_match(sum_window_expr.get()->getArgs(), count_window->getArgs())) {
    return nullptr;
  }
  return makeExpr<Analyzer::WindowFunction>(
      SQLTypeInfo(kDOUBLE),
      SqlWindowFunctionKind::AVG,
      sum_window_expr->getArgs(),
      sum_window_expr->getPartitionKeys(),
      sum_window_expr->getOrderKeys(),
      sum_window_expr->getFrameBoundType(),
      sum_window_expr->getFrameStartBound()->deep_copy(),
      sum_window_expr->getFrameEndBound()->deep_copy(),
      sum_window_expr->getCollation());
}

Here is the call graph for this function:

Here is the caller graph for this function:

std::shared_ptr<Analyzer::WindowFunction> rewrite_sum_window

(

const Analyzer::Expr *

expr

)

Definition at line 54 of file WindowExpressionRewrite.cpp.

References CHECK, COUNT, Analyzer::CaseExpr::get_expr_pair_list(), anonymous_namespace{WindowExpressionRewrite.cpp}::is_sum_kind(), kBIGINT, anonymous_namespace{WindowExpressionRewrite.cpp}::matches_else_null(), anonymous_namespace{WindowExpressionRewrite.cpp}::matches_gt_bigint_zero(), remove_cast(), SUM, and anonymous_namespace{WindowExpressionRewrite.cpp}::window_sum_and_count_match().

Referenced by rewrite_avg_window(), and rewrite_expr().

                                                                                   {
  const auto case_expr = dynamic_cast<const Analyzer::CaseExpr*>(expr);
  if (!case_expr || !matches_else_null(case_expr)) {
    return nullptr;
  }
  const auto& expr_pair_list = case_expr->get_expr_pair_list();
  if (expr_pair_list.size() != 1) {
    return nullptr;
  }
  const auto& expr_pair = expr_pair_list.front();
  const auto window_gt_zero =
      dynamic_cast<const Analyzer::BinOper*>(expr_pair.first.get());
  if (!window_gt_zero || !matches_gt_bigint_zero(window_gt_zero)) {
    return nullptr;
  }
  const auto sum_window_expr =
      std::dynamic_pointer_cast<Analyzer::WindowFunction>(remove_cast(expr_pair.second));
  if (!sum_window_expr || !is_sum_kind(sum_window_expr->getKind())) {
    return nullptr;
  }
  const auto count_window_expr =
      std::dynamic_pointer_cast<const Analyzer::WindowFunction>(
          remove_cast(window_gt_zero->get_own_left_operand()));
  if (!count_window_expr ||
      count_window_expr->getKind() != SqlWindowFunctionKind::COUNT) {
    return nullptr;
  }
  if (!window_sum_and_count_match(sum_window_expr.get(), count_window_expr.get())) {
    return nullptr;
  }
  CHECK(sum_window_expr);
  auto sum_ti = sum_window_expr->get_type_info();
  if (sum_ti.is_integer()) {
    sum_ti = SQLTypeInfo(kBIGINT, sum_ti.get_notnull());
  }
  return makeExpr<Analyzer::WindowFunction>(
      sum_ti,
      SqlWindowFunctionKind::SUM,
      sum_window_expr->getArgs(),
      sum_window_expr->getPartitionKeys(),
      sum_window_expr->getOrderKeys(),
      sum_window_expr->getFrameBoundType(),
      sum_window_expr->getFrameStartBound()->deep_copy(),
      sum_window_expr->getFrameEndBound()->deep_copy(),
      sum_window_expr->getCollation());
}

Here is the call graph for this function:

Here is the caller graph for this function: