OmniSciDB  f17484ade4
 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 2253 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 2255 of file RelAlgDag.cpp.

References CHECK, CHECK_EQ, and CHECK_LT.

2262  : collected_window_func_hash_(collected_window_func_hash)
2263  , new_rex_input_for_window_func_(new_rex_input_for_window_func)
2264  , window_func_to_new_rex_input_idx_map_(window_func_to_new_rex_input_idx_map)
2265  , new_project_(new_project)
2266  , new_rex_input_from_child_node_(new_rex_input_from_child_node) {
2269  for (auto hash : collected_window_func_hash_) {
2270  auto rex_it = window_func_to_new_rex_input_idx_map_.find(hash);
2272  CHECK_LT(rex_it->second, new_rex_input_for_window_func_.size());
2273  }
2275  }
#define CHECK_EQ(x, y)
Definition: Logger.h:301
std::vector< std::unique_ptr< const RexScalar > > & new_rex_input_for_window_func_
Definition: RelAlgDag.cpp:2379
std::unordered_map< size_t, std::unique_ptr< const RexInput > > & new_rex_input_from_child_node_
Definition: RelAlgDag.cpp:2384
#define CHECK_LT(x, y)
Definition: Logger.h:303
#define CHECK(condition)
Definition: Logger.h:291
std::unordered_map< size_t, size_t > & window_func_to_new_rex_input_idx_map_
Definition: RelAlgDag.cpp:2381

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

References CHECK, CHECK_GE, and CHECK_LT.

2368  {
2369  CHECK_GE(rex_idx, 0UL);
2370  CHECK_LT(rex_idx, new_rex_input_for_window_func_.size());
2371  auto& new_rex_input = new_rex_input_for_window_func_.at(rex_idx);
2372  CHECK(new_rex_input);
2373  auto copied_rex_input = copier_.visit(new_rex_input.get());
2374  return copied_rex_input;
2375  }
#define CHECK_GE(x, y)
Definition: Logger.h:306
std::vector< std::unique_ptr< const RexScalar > > & new_rex_input_for_window_func_
Definition: RelAlgDag.cpp:2379
virtual T visit(const RexScalar *rex_scalar) const
Definition: RexVisitor.h:27
#define CHECK_LT(x, y)
Definition: Logger.h:303
#define CHECK(condition)
Definition: Logger.h:291
std::optional<size_t> anonymous_namespace{RelAlgDag.cpp}::RexWindowFuncReplacementVisitor::is_collected_window_function ( size_t  rex_hash) const
inlineprivate

Definition at line 2360 of file RelAlgDag.cpp.

2360  {
2361  auto rex_it = window_func_to_new_rex_input_idx_map_.find(rex_hash);
2362  if (rex_it != window_func_to_new_rex_input_idx_map_.end()) {
2363  return rex_it->second;
2364  }
2365  return std::nullopt;
2366  }
std::unordered_map< size_t, size_t > & window_func_to_new_rex_input_idx_map_
Definition: RelAlgDag.cpp:2381
RetType anonymous_namespace{RelAlgDag.cpp}::RexWindowFuncReplacementVisitor::visitCase ( const RexCase rex_case) const
inlinefinalprotectedvirtual

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

Definition at line 2329 of file RelAlgDag.cpp.

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

2329  {
2330  auto new_rex_idx = is_collected_window_function(rex_case->toHash());
2331  if (new_rex_idx) {
2332  return get_new_rex_input(*new_rex_idx);
2333  }
2334 
2335  std::vector<std::pair<RetType, RetType>> new_pair_list;
2336  for (size_t i = 0; i < rex_case->branchCount(); ++i) {
2337  auto when_operand = rex_case->getWhen(i);
2338  auto new_rex_idx_for_when_operand =
2339  is_collected_window_function(when_operand->toHash());
2340 
2341  auto then_operand = rex_case->getThen(i);
2342  auto new_rex_idx_for_then_operand =
2343  is_collected_window_function(then_operand->toHash());
2344 
2345  new_pair_list.emplace_back(
2346  new_rex_idx_for_when_operand ? get_new_rex_input(*new_rex_idx_for_when_operand)
2347  : visit(when_operand),
2348  new_rex_idx_for_then_operand ? get_new_rex_input(*new_rex_idx_for_then_operand)
2349  : visit(then_operand));
2350  }
2351  auto new_rex_idx_for_else_operand =
2353  auto new_else = new_rex_idx_for_else_operand
2354  ? get_new_rex_input(*new_rex_idx_for_else_operand)
2355  : visit(rex_case->getElse());
2356  return std::make_unique<RexCase>(new_pair_list, new_else);
2357  }
std::optional< size_t > is_collected_window_function(size_t rex_hash) const
Definition: RelAlgDag.cpp:2360
const RexScalar * getThen(const size_t idx) const
Definition: RelAlgDag.h:440
const RexScalar * getElse() const
Definition: RelAlgDag.h:445
std::unique_ptr< const RexScalar > get_new_rex_input(size_t rex_idx) const
Definition: RelAlgDag.cpp:2368
const RexScalar * getWhen(const size_t idx) const
Definition: RelAlgDag.h:435
virtual std::unique_ptr< const RexScalar > visit(const RexScalar *rex_scalar) const
Definition: RexVisitor.h:27
virtual size_t toHash() const override
Definition: RelAlgDag.h:461
size_t branchCount() const
Definition: RelAlgDag.h:433
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 2278 of file RelAlgDag.cpp.

2278  {
2279  if (rex_input->getSourceNode() != new_project_) {
2280  const auto cur_index = rex_input->getIndex();
2281  auto cur_source_node = rex_input->getSourceNode();
2282  std::string field_name = "";
2283  if (auto cur_project_node = dynamic_cast<const RelProject*>(cur_source_node)) {
2284  field_name = cur_project_node->getFieldName(cur_index);
2285  }
2286  auto rex_input_hash = rex_input->toHash();
2287  auto rex_input_it = new_rex_input_from_child_node_.find(rex_input_hash);
2288  if (rex_input_it == new_rex_input_from_child_node_.end()) {
2289  auto new_rex_input =
2290  std::make_unique<RexInput>(new_project_, new_project_->size());
2291  new_project_->appendInput(field_name, rex_input->deepCopy());
2292  new_rex_input_from_child_node_.emplace(rex_input_hash, new_rex_input->deepCopy());
2293  return new_rex_input;
2294  } else {
2295  return rex_input_it->second->deepCopy();
2296  }
2297  } else {
2298  return rex_input->deepCopy();
2299  }
2300  }
size_t size() const override
Definition: RelAlgDag.h:1320
unsigned getIndex() const
Definition: RelAlgDag.h:174
std::unordered_map< size_t, std::unique_ptr< const RexInput > > & new_rex_input_from_child_node_
Definition: RelAlgDag.cpp:2384
std::unique_ptr< RexInput > deepCopy() const
Definition: RelAlgDag.h:1070
const RelAlgNode * getSourceNode() const
Definition: RelAlgDag.h:1056
virtual size_t toHash() const override
Definition: RelAlgDag.h:1074
RetType anonymous_namespace{RelAlgDag.cpp}::RexWindowFuncReplacementVisitor::visitOperator ( const RexOperator rex_operator) const
inlinefinalprotectedvirtual

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

Definition at line 2302 of file RelAlgDag.cpp.

2302  {
2303  auto new_rex_idx = is_collected_window_function(rex_operator->toHash());
2304  if (new_rex_idx) {
2305  return get_new_rex_input(*new_rex_idx);
2306  }
2307 
2308  const auto rex_window_function_operator =
2309  dynamic_cast<const RexWindowFunctionOperator*>(rex_operator);
2310  if (rex_window_function_operator) {
2311  // Deep copy the embedded window function operator
2312  return visitWindowFunctionOperator(rex_window_function_operator);
2313  }
2314 
2315  const size_t operand_count = rex_operator->size();
2316  std::vector<RetType> new_opnds;
2317  for (size_t i = 0; i < operand_count; ++i) {
2318  const auto operand = rex_operator->getOperand(i);
2319  auto new_rex_idx_for_operand = is_collected_window_function(operand->toHash());
2320  if (new_rex_idx_for_operand) {
2321  new_opnds.push_back(get_new_rex_input(*new_rex_idx_for_operand));
2322  } else {
2323  new_opnds.emplace_back(visit(rex_operator->getOperand(i)));
2324  }
2325  }
2326  return rex_operator->getDisambiguated(new_opnds);
2327  }
std::optional< size_t > is_collected_window_function(size_t rex_hash) const
Definition: RelAlgDag.cpp:2360
std::unique_ptr< const RexScalar > get_new_rex_input(size_t rex_idx) const
Definition: RelAlgDag.cpp:2368
size_t size() const
Definition: RelAlgDag.h:364
const RexScalar * getOperand(const size_t idx) const
Definition: RelAlgDag.h:366
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:359
virtual std::unique_ptr< const RexScalar > visit(const RexScalar *rex_scalar) const
Definition: RexVisitor.h:27
virtual size_t toHash() const override
Definition: RelAlgDag.h:394

Member Data Documentation

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

Definition at line 2377 of file RelAlgDag.cpp.

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

Definition at line 2385 of file RelAlgDag.cpp.

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

Definition at line 2382 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 2379 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 2384 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 2381 of file RelAlgDag.cpp.


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