OmniSciDB  c1a53651b2
 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 2182 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 2184 of file RelAlgDag.cpp.

References CHECK, CHECK_EQ, and CHECK_LT.

2191  : collected_window_func_hash_(collected_window_func_hash)
2192  , new_rex_input_for_window_func_(new_rex_input_for_window_func)
2193  , window_func_to_new_rex_input_idx_map_(window_func_to_new_rex_input_idx_map)
2194  , new_project_(new_project)
2195  , new_rex_input_from_child_node_(new_rex_input_from_child_node) {
2198  for (auto hash : collected_window_func_hash_) {
2199  auto rex_it = window_func_to_new_rex_input_idx_map_.find(hash);
2201  CHECK_LT(rex_it->second, new_rex_input_for_window_func_.size());
2202  }
2204  }
#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:2308
std::unordered_map< size_t, std::unique_ptr< const RexInput > > & new_rex_input_from_child_node_
Definition: RelAlgDag.cpp:2313
#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:2310

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

References CHECK, CHECK_GE, and CHECK_LT.

2297  {
2298  CHECK_GE(rex_idx, 0UL);
2299  CHECK_LT(rex_idx, new_rex_input_for_window_func_.size());
2300  auto& new_rex_input = new_rex_input_for_window_func_.at(rex_idx);
2301  CHECK(new_rex_input);
2302  auto copied_rex_input = copier_.visit(new_rex_input.get());
2303  return copied_rex_input;
2304  }
#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:2308
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 2289 of file RelAlgDag.cpp.

2289  {
2290  auto rex_it = window_func_to_new_rex_input_idx_map_.find(rex_hash);
2291  if (rex_it != window_func_to_new_rex_input_idx_map_.end()) {
2292  return rex_it->second;
2293  }
2294  return std::nullopt;
2295  }
std::unordered_map< size_t, size_t > & window_func_to_new_rex_input_idx_map_
Definition: RelAlgDag.cpp:2310
RetType anonymous_namespace{RelAlgDag.cpp}::RexWindowFuncReplacementVisitor::visitCase ( const RexCase rex_case) const
inlinefinalprotectedvirtual

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

Definition at line 2258 of file RelAlgDag.cpp.

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

2258  {
2259  auto new_rex_idx = is_collected_window_function(rex_case->toHash());
2260  if (new_rex_idx) {
2261  return get_new_rex_input(*new_rex_idx);
2262  }
2263 
2264  std::vector<std::pair<RetType, RetType>> new_pair_list;
2265  for (size_t i = 0; i < rex_case->branchCount(); ++i) {
2266  auto when_operand = rex_case->getWhen(i);
2267  auto new_rex_idx_for_when_operand =
2268  is_collected_window_function(when_operand->toHash());
2269 
2270  auto then_operand = rex_case->getThen(i);
2271  auto new_rex_idx_for_then_operand =
2272  is_collected_window_function(then_operand->toHash());
2273 
2274  new_pair_list.emplace_back(
2275  new_rex_idx_for_when_operand ? get_new_rex_input(*new_rex_idx_for_when_operand)
2276  : visit(when_operand),
2277  new_rex_idx_for_then_operand ? get_new_rex_input(*new_rex_idx_for_then_operand)
2278  : visit(then_operand));
2279  }
2280  auto new_rex_idx_for_else_operand =
2282  auto new_else = new_rex_idx_for_else_operand
2283  ? get_new_rex_input(*new_rex_idx_for_else_operand)
2284  : visit(rex_case->getElse());
2285  return std::make_unique<RexCase>(new_pair_list, new_else);
2286  }
std::optional< size_t > is_collected_window_function(size_t rex_hash) const
Definition: RelAlgDag.cpp:2289
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:2297
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 2207 of file RelAlgDag.cpp.

2207  {
2208  if (rex_input->getSourceNode() != new_project_) {
2209  const auto cur_index = rex_input->getIndex();
2210  auto cur_source_node = rex_input->getSourceNode();
2211  std::string field_name = "";
2212  if (auto cur_project_node = dynamic_cast<const RelProject*>(cur_source_node)) {
2213  field_name = cur_project_node->getFieldName(cur_index);
2214  }
2215  auto rex_input_hash = rex_input->toHash();
2216  auto rex_input_it = new_rex_input_from_child_node_.find(rex_input_hash);
2217  if (rex_input_it == new_rex_input_from_child_node_.end()) {
2218  auto new_rex_input =
2219  std::make_unique<RexInput>(new_project_, new_project_->size());
2220  new_project_->appendInput(field_name, rex_input->deepCopy());
2221  new_rex_input_from_child_node_.emplace(rex_input_hash, new_rex_input->deepCopy());
2222  return new_rex_input;
2223  } else {
2224  return rex_input_it->second->deepCopy();
2225  }
2226  } else {
2227  return rex_input->deepCopy();
2228  }
2229  }
size_t size() const override
Definition: RelAlgDag.h:1172
size_t toHash() const override
Definition: RelAlgDag.cpp:3392
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:2313
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 2231 of file RelAlgDag.cpp.

2231  {
2232  auto new_rex_idx = is_collected_window_function(rex_operator->toHash());
2233  if (new_rex_idx) {
2234  return get_new_rex_input(*new_rex_idx);
2235  }
2236 
2237  const auto rex_window_function_operator =
2238  dynamic_cast<const RexWindowFunctionOperator*>(rex_operator);
2239  if (rex_window_function_operator) {
2240  // Deep copy the embedded window function operator
2241  return visitWindowFunctionOperator(rex_window_function_operator);
2242  }
2243 
2244  const size_t operand_count = rex_operator->size();
2245  std::vector<RetType> new_opnds;
2246  for (size_t i = 0; i < operand_count; ++i) {
2247  const auto operand = rex_operator->getOperand(i);
2248  auto new_rex_idx_for_operand = is_collected_window_function(operand->toHash());
2249  if (new_rex_idx_for_operand) {
2250  new_opnds.push_back(get_new_rex_input(*new_rex_idx_for_operand));
2251  } else {
2252  new_opnds.emplace_back(visit(rex_operator->getOperand(i)));
2253  }
2254  }
2255  return rex_operator->getDisambiguated(new_opnds);
2256  }
std::optional< size_t > is_collected_window_function(size_t rex_hash) const
Definition: RelAlgDag.cpp:2289
std::unique_ptr< const RexScalar > get_new_rex_input(size_t rex_idx) const
Definition: RelAlgDag.cpp:2297
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 2306 of file RelAlgDag.cpp.

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

Definition at line 2314 of file RelAlgDag.cpp.

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

Definition at line 2311 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 2308 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 2313 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 2310 of file RelAlgDag.cpp.


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