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

Public Member Functions

 RexWindowFuncReplacementVisitor (std::unordered_set< size_t > &collected_window_func_hash, std::vector< std::unique_ptr< const RexScalar >> &new_rex_input_for_window_func, std::unordered_map< size_t, size_t > &window_func_to_new_rex_input_idx_map, RelProject *new_project, std::unordered_map< size_t, std::unique_ptr< const RexInput >> &new_rex_input_from_child_node)
 
- Public Member Functions inherited from RexVisitorBase< std::unique_ptr< const RexScalar > >
virtual std::unique_ptr< const
RexScalar
visit (const RexScalar *rex_scalar) const
 

Protected Member Functions

RetType visitInput (const RexInput *rex_input) const final
 
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

std::optional< size_t > is_collected_window_function (size_t rex_hash) const
 
std::unique_ptr< const RexScalarget_new_rex_input (size_t rex_idx) const
 

Private Attributes

std::unordered_set< size_t > & collected_window_func_hash_
 
std::vector< std::unique_ptr
< const RexScalar > > & 
new_rex_input_for_window_func_
 
std::unordered_map< size_t,
size_t > & 
window_func_to_new_rex_input_idx_map_
 
RelProjectnew_project_
 
std::unordered_map< size_t,
std::unique_ptr< const
RexInput > > & 
new_rex_input_from_child_node_
 
RexDeepCopyVisitor copier_
 

Additional Inherited Members

- Public Types inherited from RexDeepCopyVisitor
using RowValues = std::vector< std::unique_ptr< const RexScalar >>
 
- Static Public Member Functions inherited from RexDeepCopyVisitor
static std::vector< RowValuescopy (std::vector< RowValues > const &rhs)
 
- Protected Types inherited from RexDeepCopyVisitor
using RetType = std::unique_ptr< const RexScalar >
 

Detailed Description

Definition at line 2082 of file RelAlgDag.cpp.

Constructor & Destructor Documentation

anonymous_namespace{RelAlgDag.cpp}::RexWindowFuncReplacementVisitor::RexWindowFuncReplacementVisitor ( std::unordered_set< size_t > &  collected_window_func_hash,
std::vector< std::unique_ptr< const RexScalar >> &  new_rex_input_for_window_func,
std::unordered_map< size_t, size_t > &  window_func_to_new_rex_input_idx_map,
RelProject new_project,
std::unordered_map< size_t, std::unique_ptr< const RexInput >> &  new_rex_input_from_child_node 
)
inline

Definition at line 2084 of file RelAlgDag.cpp.

References CHECK, CHECK_EQ, and CHECK_LT.

2091  : collected_window_func_hash_(collected_window_func_hash)
2092  , new_rex_input_for_window_func_(new_rex_input_for_window_func)
2093  , window_func_to_new_rex_input_idx_map_(window_func_to_new_rex_input_idx_map)
2094  , new_project_(new_project)
2095  , new_rex_input_from_child_node_(new_rex_input_from_child_node) {
2098  for (auto hash : collected_window_func_hash_) {
2099  auto rex_it = window_func_to_new_rex_input_idx_map_.find(hash);
2101  CHECK_LT(rex_it->second, new_rex_input_for_window_func_.size());
2102  }
2104  }
#define CHECK_EQ(x, y)
Definition: Logger.h:230
std::vector< std::unique_ptr< const RexScalar > > & new_rex_input_for_window_func_
Definition: RelAlgDag.cpp:2208
std::unordered_map< size_t, std::unique_ptr< const RexInput > > & new_rex_input_from_child_node_
Definition: RelAlgDag.cpp:2213
#define CHECK_LT(x, y)
Definition: Logger.h:232
#define CHECK(condition)
Definition: Logger.h:222
std::unordered_map< size_t, size_t > & window_func_to_new_rex_input_idx_map_
Definition: RelAlgDag.cpp:2210

Member Function Documentation

std::unique_ptr<const RexScalar> anonymous_namespace{RelAlgDag.cpp}::RexWindowFuncReplacementVisitor::get_new_rex_input ( size_t  rex_idx) const
inlineprivate

Definition at line 2197 of file RelAlgDag.cpp.

References CHECK, CHECK_GE, and CHECK_LT.

2197  {
2198  CHECK_GE(rex_idx, 0UL);
2199  CHECK_LT(rex_idx, new_rex_input_for_window_func_.size());
2200  auto& new_rex_input = new_rex_input_for_window_func_.at(rex_idx);
2201  CHECK(new_rex_input);
2202  auto copied_rex_input = copier_.visit(new_rex_input.get());
2203  return copied_rex_input;
2204  }
#define CHECK_GE(x, y)
Definition: Logger.h:235
std::vector< std::unique_ptr< const RexScalar > > & new_rex_input_for_window_func_
Definition: RelAlgDag.cpp:2208
virtual T visit(const RexScalar *rex_scalar) const
Definition: RexVisitor.h:27
#define CHECK_LT(x, y)
Definition: Logger.h:232
#define CHECK(condition)
Definition: Logger.h:222
std::optional<size_t> anonymous_namespace{RelAlgDag.cpp}::RexWindowFuncReplacementVisitor::is_collected_window_function ( size_t  rex_hash) const
inlineprivate

Definition at line 2189 of file RelAlgDag.cpp.

2189  {
2190  auto rex_it = window_func_to_new_rex_input_idx_map_.find(rex_hash);
2191  if (rex_it != window_func_to_new_rex_input_idx_map_.end()) {
2192  return rex_it->second;
2193  }
2194  return std::nullopt;
2195  }
std::unordered_map< size_t, size_t > & window_func_to_new_rex_input_idx_map_
Definition: RelAlgDag.cpp:2210
RetType anonymous_namespace{RelAlgDag.cpp}::RexWindowFuncReplacementVisitor::visitCase ( const RexCase rex_case) const
inlinefinalprotectedvirtual

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

Definition at line 2158 of file RelAlgDag.cpp.

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

2158  {
2159  auto new_rex_idx = is_collected_window_function(rex_case->toHash());
2160  if (new_rex_idx) {
2161  return get_new_rex_input(*new_rex_idx);
2162  }
2163 
2164  std::vector<std::pair<RetType, RetType>> new_pair_list;
2165  for (size_t i = 0; i < rex_case->branchCount(); ++i) {
2166  auto when_operand = rex_case->getWhen(i);
2167  auto new_rex_idx_for_when_operand =
2168  is_collected_window_function(when_operand->toHash());
2169 
2170  auto then_operand = rex_case->getThen(i);
2171  auto new_rex_idx_for_then_operand =
2172  is_collected_window_function(then_operand->toHash());
2173 
2174  new_pair_list.emplace_back(
2175  new_rex_idx_for_when_operand ? get_new_rex_input(*new_rex_idx_for_when_operand)
2176  : visit(when_operand),
2177  new_rex_idx_for_then_operand ? get_new_rex_input(*new_rex_idx_for_then_operand)
2178  : visit(then_operand));
2179  }
2180  auto new_rex_idx_for_else_operand =
2182  auto new_else = new_rex_idx_for_else_operand
2183  ? get_new_rex_input(*new_rex_idx_for_else_operand)
2184  : visit(rex_case->getElse());
2185  return std::make_unique<RexCase>(new_pair_list, new_else);
2186  }
std::optional< size_t > is_collected_window_function(size_t rex_hash) const
Definition: RelAlgDag.cpp:2189
const RexScalar * getThen(const size_t idx) const
Definition: RelAlgDag.h:400
const RexScalar * getElse() const
Definition: RelAlgDag.h:405
std::unique_ptr< const RexScalar > get_new_rex_input(size_t rex_idx) const
Definition: RelAlgDag.cpp:2197
const RexScalar * getWhen(const size_t idx) const
Definition: RelAlgDag.h:395
virtual std::unique_ptr< const RexScalar > visit(const RexScalar *rex_scalar) const
Definition: RexVisitor.h:27
size_t branchCount() const
Definition: RelAlgDag.h:393
size_t toHash() const override
Definition: RelAlgDag.h:417
virtual size_t toHash() const =0

+ Here is the call graph for this function:

RetType anonymous_namespace{RelAlgDag.cpp}::RexWindowFuncReplacementVisitor::visitInput ( const RexInput rex_input) const
inlinefinalprotectedvirtual

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

Definition at line 2107 of file RelAlgDag.cpp.

2107  {
2108  if (rex_input->getSourceNode() != new_project_) {
2109  const auto cur_index = rex_input->getIndex();
2110  auto cur_source_node = rex_input->getSourceNode();
2111  std::string field_name = "";
2112  if (auto cur_project_node = dynamic_cast<const RelProject*>(cur_source_node)) {
2113  field_name = cur_project_node->getFieldName(cur_index);
2114  }
2115  auto rex_input_hash = rex_input->toHash();
2116  auto rex_input_it = new_rex_input_from_child_node_.find(rex_input_hash);
2117  if (rex_input_it == new_rex_input_from_child_node_.end()) {
2118  auto new_rex_input =
2119  std::make_unique<RexInput>(new_project_, new_project_->size());
2120  new_project_->appendInput(field_name, rex_input->deepCopy());
2121  new_rex_input_from_child_node_.emplace(rex_input_hash, new_rex_input->deepCopy());
2122  return new_rex_input;
2123  } else {
2124  return rex_input_it->second->deepCopy();
2125  }
2126  } else {
2127  return rex_input->deepCopy();
2128  }
2129  }
size_t size() const override
Definition: RelAlgDag.h:1068
size_t toHash() const override
Definition: RelAlgDag.cpp:3252
unsigned getIndex() const
Definition: RelAlgDag.h:72
std::unordered_map< size_t, std::unique_ptr< const RexInput > > & new_rex_input_from_child_node_
Definition: RelAlgDag.cpp:2213
std::unique_ptr< RexInput > deepCopy() const
Definition: RelAlgDag.h:367
const RelAlgNode * getSourceNode() const
Definition: RelAlgDag.h:351
RetType anonymous_namespace{RelAlgDag.cpp}::RexWindowFuncReplacementVisitor::visitOperator ( const RexOperator rex_operator) const
inlinefinalprotectedvirtual

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

Definition at line 2131 of file RelAlgDag.cpp.

2131  {
2132  auto new_rex_idx = is_collected_window_function(rex_operator->toHash());
2133  if (new_rex_idx) {
2134  return get_new_rex_input(*new_rex_idx);
2135  }
2136 
2137  const auto rex_window_function_operator =
2138  dynamic_cast<const RexWindowFunctionOperator*>(rex_operator);
2139  if (rex_window_function_operator) {
2140  // Deep copy the embedded window function operator
2141  return visitWindowFunctionOperator(rex_window_function_operator);
2142  }
2143 
2144  const size_t operand_count = rex_operator->size();
2145  std::vector<RetType> new_opnds;
2146  for (size_t i = 0; i < operand_count; ++i) {
2147  const auto operand = rex_operator->getOperand(i);
2148  auto new_rex_idx_for_operand = is_collected_window_function(operand->toHash());
2149  if (new_rex_idx_for_operand) {
2150  new_opnds.push_back(get_new_rex_input(*new_rex_idx_for_operand));
2151  } else {
2152  new_opnds.emplace_back(visit(rex_operator->getOperand(i)));
2153  }
2154  }
2155  return rex_operator->getDisambiguated(new_opnds);
2156  }
std::optional< size_t > is_collected_window_function(size_t rex_hash) const
Definition: RelAlgDag.cpp:2189
std::unique_ptr< const RexScalar > get_new_rex_input(size_t rex_idx) const
Definition: RelAlgDag.cpp:2197
size_t size() const
Definition: RelAlgDag.h:245
const RexScalar * getOperand(const size_t idx) const
Definition: RelAlgDag.h:247
RetType visitWindowFunctionOperator(const RexWindowFunctionOperator *rex_window_function_operator) const
Definition: RexVisitor.h:169
virtual std::unique_ptr< const RexOperator > getDisambiguated(std::vector< std::unique_ptr< const RexScalar >> &operands) const
Definition: RelAlgDag.h:240
virtual std::unique_ptr< const RexScalar > visit(const RexScalar *rex_scalar) const
Definition: RexVisitor.h:27
size_t toHash() const override
Definition: RelAlgDag.h:270

Member Data Documentation

std::unordered_set<size_t>& anonymous_namespace{RelAlgDag.cpp}::RexWindowFuncReplacementVisitor::collected_window_func_hash_
private

Definition at line 2206 of file RelAlgDag.cpp.

RexDeepCopyVisitor anonymous_namespace{RelAlgDag.cpp}::RexWindowFuncReplacementVisitor::copier_
private

Definition at line 2214 of file RelAlgDag.cpp.

RelProject* anonymous_namespace{RelAlgDag.cpp}::RexWindowFuncReplacementVisitor::new_project_
private

Definition at line 2211 of file RelAlgDag.cpp.

std::vector<std::unique_ptr<const RexScalar> >& anonymous_namespace{RelAlgDag.cpp}::RexWindowFuncReplacementVisitor::new_rex_input_for_window_func_
private

Definition at line 2208 of file RelAlgDag.cpp.

std::unordered_map<size_t, std::unique_ptr<const RexInput> >& anonymous_namespace{RelAlgDag.cpp}::RexWindowFuncReplacementVisitor::new_rex_input_from_child_node_
private

Definition at line 2213 of file RelAlgDag.cpp.

std::unordered_map<size_t, size_t>& anonymous_namespace{RelAlgDag.cpp}::RexWindowFuncReplacementVisitor::window_func_to_new_rex_input_idx_map_
private

Definition at line 2210 of file RelAlgDag.cpp.


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