OmniSciDB  0bd2ec9cf4
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Pages
anonymous_namespace{RelAlgDagBuilder.cpp}::RexWindowFuncReplacementVisitor Class Reference
+ Inheritance diagram for anonymous_namespace{RelAlgDagBuilder.cpp}::RexWindowFuncReplacementVisitor:
+ Collaboration diagram for anonymous_namespace{RelAlgDagBuilder.cpp}::RexWindowFuncReplacementVisitor:

Public Member Functions

 RexWindowFuncReplacementVisitor (std::unique_ptr< const RexScalar > replacement_rex)
 ~RexWindowFuncReplacementVisitor ()
- Public Member Functions inherited from RexVisitorBase< std::unique_ptr< const RexScalar > >
virtual std::unique_ptr< const
visit (const RexScalar *rex_scalar) const

Protected Member Functions

RetType visitOperator (const RexOperator *rex_operator) const final
RetType visitCase (const RexCase *rex_case) const final
- Protected Member Functions inherited from RexDeepCopyVisitor
RetType visitInput (const RexInput *input) const override
RetType visitLiteral (const RexLiteral *literal) const override
RetType visitSubQuery (const RexSubQuery *subquery) const override
RetType visitRef (const RexRef *ref) const override
RetType visitOperator (const RexOperator *rex_operator) const override
RetType visitWindowFunctionOperator (const RexWindowFunctionOperator *rex_window_function_operator) const
RetType visitCase (const RexCase *rex_case) const override

Private Member Functions

bool should_replace_operand (const RexScalar *rex) const

Private Attributes

std::unique_ptr< const RexScalarreplacement_rex_

Additional Inherited Members

- Protected Types inherited from RexDeepCopyVisitor
using RetType = std::unique_ptr< const RexScalar >

Detailed Description

Replaces the first occurrence of a WindowFunctionOperator rex with the provided replacement_rex. Typically used for splitting a complex rex into two simpler rexes, and forwarding one of the rexes to a later node. The forwarded rex is then replaced with a RexInput using this visitor. Note that for window function replacement, the overloads in this visitor must match the overloads in the detection visitor above, to ensure a detected window function expression is properly replaced.

Definition at line 1560 of file RelAlgDagBuilder.cpp.

Constructor & Destructor Documentation

anonymous_namespace{RelAlgDagBuilder.cpp}::RexWindowFuncReplacementVisitor::RexWindowFuncReplacementVisitor ( std::unique_ptr< const RexScalar replacement_rex)

Definition at line 1562 of file RelAlgDagBuilder.cpp.

1563  : replacement_rex_(std::move(replacement_rex)) {}
anonymous_namespace{RelAlgDagBuilder.cpp}::RexWindowFuncReplacementVisitor::~RexWindowFuncReplacementVisitor ( )

Definition at line 1565 of file RelAlgDagBuilder.cpp.

References CHECK().

1565 { CHECK(replacement_rex_ == nullptr); }

+ Here is the call graph for this function:

Member Function Documentation

bool anonymous_namespace{RelAlgDagBuilder.cpp}::RexWindowFuncReplacementVisitor::should_replace_operand ( const RexScalar rex) const

Definition at line 1615 of file RelAlgDagBuilder.cpp.

References anonymous_namespace{RelAlgDagBuilder.cpp}::anonymous_namespace{RelAlgDagBuilder.cpp}::is_window_function_operator().

+ Here is the call graph for this function:

RetType anonymous_namespace{RelAlgDagBuilder.cpp}::RexWindowFuncReplacementVisitor::visitCase ( const RexCase rex_case) const

Implements RexVisitorBase< std::unique_ptr< const RexScalar > >.

Definition at line 1593 of file RelAlgDagBuilder.cpp.

References RexCase::branchCount(), RexCase::getElse(), RexCase::getThen(), and RexCase::getWhen().

1593  {
1594  if (should_replace_operand(rex_case)) {
1595  return std::move(replacement_rex_);
1596  }
1598  std::vector<std::pair<RetType, RetType>> new_pair_list;
1599  for (size_t i = 0; i < rex_case->branchCount(); ++i) {
1600  auto when_operand = rex_case->getWhen(i);
1601  auto then_operand = rex_case->getThen(i);
1602  new_pair_list.emplace_back(
1603  should_replace_operand(when_operand) ? std::move(replacement_rex_)
1604  : visit(when_operand),
1605  should_replace_operand(then_operand) ? std::move(replacement_rex_)
1606  : visit(then_operand));
1607  }
1608  auto new_else = should_replace_operand(rex_case->getElse())
1609  ? std::move(replacement_rex_)
1610  : visit(rex_case->getElse());
1611  return std::make_unique<RexCase>(new_pair_list, new_else);
1612  }
const RexScalar * getThen(const size_t idx) const
const RexScalar * getElse() const
const RexScalar * getWhen(const size_t idx) const
virtual std::unique_ptr< const RexScalar > visit(const RexScalar *rex_scalar) const
Definition: RexVisitor.h:27
size_t branchCount() const

+ Here is the call graph for this function:

RetType anonymous_namespace{RelAlgDagBuilder.cpp}::RexWindowFuncReplacementVisitor::visitOperator ( const RexOperator rex_operator) const

Implements RexVisitorBase< std::unique_ptr< const RexScalar > >.

Definition at line 1568 of file RelAlgDagBuilder.cpp.

1568  {
1569  if (should_replace_operand(rex_operator)) {
1570  return std::move(replacement_rex_);
1571  }
1573  const auto rex_window_function_operator =
1574  dynamic_cast<const RexWindowFunctionOperator*>(rex_operator);
1575  if (rex_window_function_operator) {
1576  // Deep copy the embedded window function operator
1577  return visitWindowFunctionOperator(rex_window_function_operator);
1578  }
1580  const size_t operand_count = rex_operator->size();
1581  std::vector<RetType> new_opnds;
1582  for (size_t i = 0; i < operand_count; ++i) {
1583  const auto operand = rex_operator->getOperand(i);
1584  if (should_replace_operand(operand)) {
1585  new_opnds.push_back(std::move(replacement_rex_));
1586  } else {
1587  new_opnds.emplace_back(visit(rex_operator->getOperand(i)));
1588  }
1589  }
1590  return rex_operator->getDisambiguated(new_opnds);
1591  }
size_t size() const
const RexScalar * getOperand(const size_t idx) const
RetType visitWindowFunctionOperator(const RexWindowFunctionOperator *rex_window_function_operator) const
Definition: RexVisitor.h:168
virtual std::unique_ptr< const RexOperator > getDisambiguated(std::vector< std::unique_ptr< const RexScalar >> &operands) const
virtual std::unique_ptr< const RexScalar > visit(const RexScalar *rex_scalar) const
Definition: RexVisitor.h:27

Member Data Documentation

std::unique_ptr<const RexScalar> anonymous_namespace{RelAlgDagBuilder.cpp}::RexWindowFuncReplacementVisitor::replacement_rex_

Definition at line 1619 of file RelAlgDagBuilder.cpp.

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