OmniSciDB  a987f07e93
 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 2190 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 2192 of file RelAlgDag.cpp.

References CHECK, CHECK_EQ, and CHECK_LT.

2199  : collected_window_func_hash_(collected_window_func_hash)
2200  , new_rex_input_for_window_func_(new_rex_input_for_window_func)
2201  , window_func_to_new_rex_input_idx_map_(window_func_to_new_rex_input_idx_map)
2202  , new_project_(new_project)
2203  , new_rex_input_from_child_node_(new_rex_input_from_child_node) {
2206  for (auto hash : collected_window_func_hash_) {
2207  auto rex_it = window_func_to_new_rex_input_idx_map_.find(hash);
2209  CHECK_LT(rex_it->second, new_rex_input_for_window_func_.size());
2210  }
2212  }
#define CHECK_EQ(x, y)
Definition: Logger.h:297
std::vector< std::unique_ptr< const RexScalar > > & new_rex_input_for_window_func_
Definition: RelAlgDag.cpp:2316
std::unordered_map< size_t, std::unique_ptr< const RexInput > > & new_rex_input_from_child_node_
Definition: RelAlgDag.cpp:2321
#define CHECK_LT(x, y)
Definition: Logger.h:299
#define CHECK(condition)
Definition: Logger.h:289
std::unordered_map< size_t, size_t > & window_func_to_new_rex_input_idx_map_
Definition: RelAlgDag.cpp:2318

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 2305 of file RelAlgDag.cpp.

References CHECK, CHECK_GE, and CHECK_LT.

2305  {
2306  CHECK_GE(rex_idx, 0UL);
2307  CHECK_LT(rex_idx, new_rex_input_for_window_func_.size());
2308  auto& new_rex_input = new_rex_input_for_window_func_.at(rex_idx);
2309  CHECK(new_rex_input);
2310  auto copied_rex_input = copier_.visit(new_rex_input.get());
2311  return copied_rex_input;
2312  }
#define CHECK_GE(x, y)
Definition: Logger.h:302
std::vector< std::unique_ptr< const RexScalar > > & new_rex_input_for_window_func_
Definition: RelAlgDag.cpp:2316
virtual T visit(const RexScalar *rex_scalar) const
Definition: RexVisitor.h:27
#define CHECK_LT(x, y)
Definition: Logger.h:299
#define CHECK(condition)
Definition: Logger.h:289
std::optional<size_t> anonymous_namespace{RelAlgDag.cpp}::RexWindowFuncReplacementVisitor::is_collected_window_function ( size_t  rex_hash) const
inlineprivate

Definition at line 2297 of file RelAlgDag.cpp.

2297  {
2298  auto rex_it = window_func_to_new_rex_input_idx_map_.find(rex_hash);
2299  if (rex_it != window_func_to_new_rex_input_idx_map_.end()) {
2300  return rex_it->second;
2301  }
2302  return std::nullopt;
2303  }
std::unordered_map< size_t, size_t > & window_func_to_new_rex_input_idx_map_
Definition: RelAlgDag.cpp:2318
RetType anonymous_namespace{RelAlgDag.cpp}::RexWindowFuncReplacementVisitor::visitCase ( const RexCase rex_case) const
inlinefinalprotectedvirtual

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

Definition at line 2266 of file RelAlgDag.cpp.

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

2266  {
2267  auto new_rex_idx = is_collected_window_function(rex_case->toHash());
2268  if (new_rex_idx) {
2269  return get_new_rex_input(*new_rex_idx);
2270  }
2271 
2272  std::vector<std::pair<RetType, RetType>> new_pair_list;
2273  for (size_t i = 0; i < rex_case->branchCount(); ++i) {
2274  auto when_operand = rex_case->getWhen(i);
2275  auto new_rex_idx_for_when_operand =
2276  is_collected_window_function(when_operand->toHash());
2277 
2278  auto then_operand = rex_case->getThen(i);
2279  auto new_rex_idx_for_then_operand =
2280  is_collected_window_function(then_operand->toHash());
2281 
2282  new_pair_list.emplace_back(
2283  new_rex_idx_for_when_operand ? get_new_rex_input(*new_rex_idx_for_when_operand)
2284  : visit(when_operand),
2285  new_rex_idx_for_then_operand ? get_new_rex_input(*new_rex_idx_for_then_operand)
2286  : visit(then_operand));
2287  }
2288  auto new_rex_idx_for_else_operand =
2290  auto new_else = new_rex_idx_for_else_operand
2291  ? get_new_rex_input(*new_rex_idx_for_else_operand)
2292  : visit(rex_case->getElse());
2293  return std::make_unique<RexCase>(new_pair_list, new_else);
2294  }
std::optional< size_t > is_collected_window_function(size_t rex_hash) const
Definition: RelAlgDag.cpp:2297
const RexScalar * getThen(const size_t idx) const
Definition: RelAlgDag.h:443
const RexScalar * getElse() const
Definition: RelAlgDag.h:448
std::unique_ptr< const RexScalar > get_new_rex_input(size_t rex_idx) const
Definition: RelAlgDag.cpp:2305
const RexScalar * getWhen(const size_t idx) const
Definition: RelAlgDag.h:438
virtual std::unique_ptr< const RexScalar > visit(const RexScalar *rex_scalar) const
Definition: RexVisitor.h:27
size_t branchCount() const
Definition: RelAlgDag.h:436
size_t toHash() const override
Definition: RelAlgDag.h:460
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 2215 of file RelAlgDag.cpp.

2215  {
2216  if (rex_input->getSourceNode() != new_project_) {
2217  const auto cur_index = rex_input->getIndex();
2218  auto cur_source_node = rex_input->getSourceNode();
2219  std::string field_name = "";
2220  if (auto cur_project_node = dynamic_cast<const RelProject*>(cur_source_node)) {
2221  field_name = cur_project_node->getFieldName(cur_index);
2222  }
2223  auto rex_input_hash = rex_input->toHash();
2224  auto rex_input_it = new_rex_input_from_child_node_.find(rex_input_hash);
2225  if (rex_input_it == new_rex_input_from_child_node_.end()) {
2226  auto new_rex_input =
2227  std::make_unique<RexInput>(new_project_, new_project_->size());
2228  new_project_->appendInput(field_name, rex_input->deepCopy());
2229  new_rex_input_from_child_node_.emplace(rex_input_hash, new_rex_input->deepCopy());
2230  return new_rex_input;
2231  } else {
2232  return rex_input_it->second->deepCopy();
2233  }
2234  } else {
2235  return rex_input->deepCopy();
2236  }
2237  }
size_t size() const override
Definition: RelAlgDag.h:1156
size_t toHash() const override
Definition: RelAlgDag.cpp:3395
unsigned getIndex() const
Definition: RelAlgDag.h:77
std::unordered_map< size_t, std::unique_ptr< const RexInput > > & new_rex_input_from_child_node_
Definition: RelAlgDag.cpp:2321
std::unique_ptr< RexInput > deepCopy() const
Definition: RelAlgDag.h:405
const RelAlgNode * getSourceNode() const
Definition: RelAlgDag.h:389
RetType anonymous_namespace{RelAlgDag.cpp}::RexWindowFuncReplacementVisitor::visitOperator ( const RexOperator rex_operator) const
inlinefinalprotectedvirtual

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

Definition at line 2239 of file RelAlgDag.cpp.

2239  {
2240  auto new_rex_idx = is_collected_window_function(rex_operator->toHash());
2241  if (new_rex_idx) {
2242  return get_new_rex_input(*new_rex_idx);
2243  }
2244 
2245  const auto rex_window_function_operator =
2246  dynamic_cast<const RexWindowFunctionOperator*>(rex_operator);
2247  if (rex_window_function_operator) {
2248  // Deep copy the embedded window function operator
2249  return visitWindowFunctionOperator(rex_window_function_operator);
2250  }
2251 
2252  const size_t operand_count = rex_operator->size();
2253  std::vector<RetType> new_opnds;
2254  for (size_t i = 0; i < operand_count; ++i) {
2255  const auto operand = rex_operator->getOperand(i);
2256  auto new_rex_idx_for_operand = is_collected_window_function(operand->toHash());
2257  if (new_rex_idx_for_operand) {
2258  new_opnds.push_back(get_new_rex_input(*new_rex_idx_for_operand));
2259  } else {
2260  new_opnds.emplace_back(visit(rex_operator->getOperand(i)));
2261  }
2262  }
2263  return rex_operator->getDisambiguated(new_opnds);
2264  }
std::optional< size_t > is_collected_window_function(size_t rex_hash) const
Definition: RelAlgDag.cpp:2297
std::unique_ptr< const RexScalar > get_new_rex_input(size_t rex_idx) const
Definition: RelAlgDag.cpp:2305
size_t size() const
Definition: RelAlgDag.h:270
const RexScalar * getOperand(const size_t idx) const
Definition: RelAlgDag.h:272
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:265
virtual std::unique_ptr< const RexScalar > visit(const RexScalar *rex_scalar) const
Definition: RexVisitor.h:27
size_t toHash() const override
Definition: RelAlgDag.h:295

Member Data Documentation

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

Definition at line 2314 of file RelAlgDag.cpp.

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

Definition at line 2322 of file RelAlgDag.cpp.

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

Definition at line 2319 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 2316 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 2321 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 2318 of file RelAlgDag.cpp.


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