OmniSciDB  c1a53651b2
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
Analyzer::WidthBucketExpr Class Reference

#include <Analyzer.h>

+ Inheritance diagram for Analyzer::WidthBucketExpr:
+ Collaboration diagram for Analyzer::WidthBucketExpr:

Public Member Functions

 WidthBucketExpr (const std::shared_ptr< Analyzer::Expr > target_value, const std::shared_ptr< Analyzer::Expr > lower_bound, const std::shared_ptr< Analyzer::Expr > upper_bound, const std::shared_ptr< Analyzer::Expr > partition_count)
 
const Exprget_target_value () const
 
const Exprget_lower_bound () const
 
const Exprget_upper_bound () const
 
const Exprget_partition_count () const
 
std::shared_ptr< Analyzer::Exprdeep_copy () const override
 
void group_predicates (std::list< const Expr * > &scan_predicates, std::list< const Expr * > &join_predicates, std::list< const Expr * > &const_predicates) const override
 
void collect_rte_idx (std::set< int > &rte_idx_set) const override
 
void collect_column_var (std::set< const ColumnVar *, bool(*)(const ColumnVar *, const ColumnVar *)> &colvar_set, bool include_agg) const override
 
std::shared_ptr< Analyzer::Exprrewrite_with_targetlist (const std::vector< std::shared_ptr< TargetEntry >> &tlist) const override
 
std::shared_ptr< Analyzer::Exprrewrite_with_child_targetlist (const std::vector< std::shared_ptr< TargetEntry >> &tlist) const override
 
std::shared_ptr< Analyzer::Exprrewrite_agg_to_var (const std::vector< std::shared_ptr< TargetEntry >> &tlist) const override
 
double get_bound_val (const Analyzer::Expr *bound_expr) const
 
int32_t get_partition_count_val () const
 
template<typename T >
int32_t compute_bucket (T target_const_val, SQLTypeInfo &ti) const
 
bool operator== (const Expr &rhs) const override
 
std::string toString () const override
 
void find_expr (std::function< bool(const Expr *)> f, std::list< const Expr * > &expr_list) const override
 
bool can_skip_out_of_bound_check () const
 
void skip_out_of_bound_check () const
 
void set_constant_expr () const
 
bool is_constant_expr () const
 
- Public Member Functions inherited from Analyzer::Expr
 Expr (SQLTypes t, bool notnull)
 
 Expr (SQLTypes t, int d, bool notnull)
 
 Expr (SQLTypes t, int d, int s, bool notnull)
 
 Expr (const SQLTypeInfo &ti, bool has_agg=false)
 
virtual ~Expr ()
 
std::shared_ptr< Analyzer::Exprget_shared_ptr ()
 
const SQLTypeInfoget_type_info () const
 
void set_type_info (const SQLTypeInfo &ti)
 
bool get_contains_agg () const
 
void set_contains_agg (bool a)
 
virtual std::shared_ptr
< Analyzer::Expr
add_cast (const SQLTypeInfo &new_type_info)
 
virtual void check_group_by (const std::list< std::shared_ptr< Analyzer::Expr >> &groupby) const
 
virtual std::shared_ptr
< Analyzer::Expr
normalize_simple_predicate (int &rte_idx) const
 
virtual size_t get_num_column_vars (const bool include_agg) const
 
virtual void print () const
 
virtual void add_unique (std::list< const Expr * > &expr_list) const
 
std::shared_ptr< Analyzer::Exprdecompress ()
 
virtual void get_domain (DomainSet &domain_set) const
 

Private Attributes

std::shared_ptr< Analyzer::Exprtarget_value_
 
std::shared_ptr< Analyzer::Exprlower_bound_
 
std::shared_ptr< Analyzer::Exprupper_bound_
 
std::shared_ptr< Analyzer::Exprpartition_count_
 
bool constant_expr_
 
bool skip_out_of_bound_check_
 

Additional Inherited Members

- Protected Attributes inherited from Analyzer::Expr
SQLTypeInfo type_info
 
bool contains_agg
 

Detailed Description

Definition at line 1033 of file Analyzer.h.

Constructor & Destructor Documentation

Analyzer::WidthBucketExpr::WidthBucketExpr ( const std::shared_ptr< Analyzer::Expr target_value,
const std::shared_ptr< Analyzer::Expr lower_bound,
const std::shared_ptr< Analyzer::Expr upper_bound,
const std::shared_ptr< Analyzer::Expr partition_count 
)
inline

Definition at line 1035 of file Analyzer.h.

1039  : Expr(kINT, target_value->get_type_info().get_notnull())
1040  , target_value_(target_value)
1043  , partition_count_(partition_count)
1044  , constant_expr_(false)
1045  , skip_out_of_bound_check_(false) {}
DEVICE auto upper_bound(ARGS &&...args)
Definition: gpu_enabled.h:123
Expr(SQLTypes t, bool notnull)
Definition: Analyzer.h:70
std::shared_ptr< Analyzer::Expr > upper_bound_
Definition: Analyzer.h:1129
std::shared_ptr< Analyzer::Expr > partition_count_
Definition: Analyzer.h:1130
std::shared_ptr< Analyzer::Expr > lower_bound_
Definition: Analyzer.h:1128
DEVICE auto lower_bound(ARGS &&...args)
Definition: gpu_enabled.h:78
Definition: sqltypes.h:62
std::shared_ptr< Analyzer::Expr > target_value_
Definition: Analyzer.h:1127

Member Function Documentation

bool Analyzer::WidthBucketExpr::can_skip_out_of_bound_check ( ) const
inline

Definition at line 1121 of file Analyzer.h.

References skip_out_of_bound_check_.

Referenced by CodeGenerator::codegen(), CodeGenerator::codegenConstantWidthBucketExpr(), and CodeGenerator::codegenWidthBucketExpr().

1121 { return skip_out_of_bound_check_; }

+ Here is the caller graph for this function:

void Analyzer::WidthBucketExpr::collect_column_var ( std::set< const ColumnVar *, bool(*)(const ColumnVar *, const ColumnVar *)> &  colvar_set,
bool  include_agg 
) const
inlineoverridevirtual

Reimplemented from Analyzer::Expr.

Definition at line 1057 of file Analyzer.h.

References target_value_.

1060  {
1061  target_value_->collect_column_var(colvar_set, include_agg);
1062  }
std::shared_ptr< Analyzer::Expr > target_value_
Definition: Analyzer.h:1127
void Analyzer::WidthBucketExpr::collect_rte_idx ( std::set< int > &  rte_idx_set) const
inlineoverridevirtual

Reimplemented from Analyzer::Expr.

Definition at line 1054 of file Analyzer.h.

References target_value_.

1054  {
1055  target_value_->collect_rte_idx(rte_idx_set);
1056  }
std::shared_ptr< Analyzer::Expr > target_value_
Definition: Analyzer.h:1127
template<typename T >
int32_t Analyzer::WidthBucketExpr::compute_bucket ( target_const_val,
SQLTypeInfo ti 
) const
inline

Definition at line 1087 of file Analyzer.h.

References get_bound_val(), get_partition_count_val(), inline_fp_null_val(), inline_int_null_val(), SQLTypeInfo::is_integer(), lower_bound_, run_benchmark_import::res, heavydb.dtypes::T, and upper_bound_.

Referenced by getExpressionRange().

1087  {
1088  // this utility function is useful for optimizing expression range decision
1089  // for an expression depending on width_bucket expr
1090  T null_val = ti.is_integer() ? inline_int_null_val(ti) : inline_fp_null_val(ti);
1091  double lower_bound_val = get_bound_val(lower_bound_.get());
1092  double upper_bound_val = get_bound_val(upper_bound_.get());
1093  auto partition_count_val = get_partition_count_val();
1094  if (target_const_val == null_val) {
1095  return INT32_MIN;
1096  }
1097  float res;
1098  if (lower_bound_val < upper_bound_val) {
1099  if (target_const_val < lower_bound_val) {
1100  return 0;
1101  } else if (target_const_val >= upper_bound_val) {
1102  return partition_count_val + 1;
1103  }
1104  double dividend = upper_bound_val - lower_bound_val;
1105  res = ((partition_count_val * (target_const_val - lower_bound_val)) / dividend) + 1;
1106  } else {
1107  if (target_const_val > lower_bound_val) {
1108  return 0;
1109  } else if (target_const_val <= upper_bound_val) {
1110  return partition_count_val + 1;
1111  }
1112  double dividend = lower_bound_val - upper_bound_val;
1113  res = ((partition_count_val * (lower_bound_val - target_const_val)) / dividend) + 1;
1114  }
1115  return res;
1116  }
std::shared_ptr< Analyzer::Expr > upper_bound_
Definition: Analyzer.h:1129
double inline_fp_null_val(const SQL_TYPE_INFO &ti)
double get_bound_val(const Analyzer::Expr *bound_expr) const
Definition: Analyzer.cpp:3752
std::shared_ptr< Analyzer::Expr > lower_bound_
Definition: Analyzer.h:1128
bool is_integer() const
Definition: sqltypes.h:582
int32_t get_partition_count_val() const
Definition: Analyzer.cpp:3762
int64_t inline_int_null_val(const SQL_TYPE_INFO &ti)

+ Here is the call graph for this function:

+ Here is the caller graph for this function:

std::shared_ptr< Analyzer::Expr > Analyzer::WidthBucketExpr::deep_copy ( ) const
overridevirtual

Implements Analyzer::Expr.

Definition at line 166 of file Analyzer.cpp.

References lower_bound_, partition_count_, target_value_, and upper_bound_.

166  {
167  return makeExpr<WidthBucketExpr>(target_value_->deep_copy(),
168  lower_bound_->deep_copy(),
169  upper_bound_->deep_copy(),
170  partition_count_->deep_copy());
171 }
std::shared_ptr< Analyzer::Expr > upper_bound_
Definition: Analyzer.h:1129
std::shared_ptr< Analyzer::Expr > partition_count_
Definition: Analyzer.h:1130
std::shared_ptr< Analyzer::Expr > lower_bound_
Definition: Analyzer.h:1128
std::shared_ptr< Analyzer::Expr > target_value_
Definition: Analyzer.h:1127
void Analyzer::WidthBucketExpr::find_expr ( std::function< bool(const Expr *)>  f,
std::list< const Expr * > &  expr_list 
) const
overridevirtual

Reimplemented from Analyzer::Expr.

Definition at line 3230 of file Analyzer.cpp.

References Analyzer::Expr::add_unique(), anonymous_namespace{Utm.h}::f, lower_bound_, partition_count_, target_value_, and upper_bound_.

3231  {
3232  if (f(this)) {
3233  add_unique(expr_list);
3234  return;
3235  }
3236  target_value_->find_expr(f, expr_list);
3237  lower_bound_->find_expr(f, expr_list);
3238  upper_bound_->find_expr(f, expr_list);
3239  partition_count_->find_expr(f, expr_list);
3240 }
virtual void add_unique(std::list< const Expr * > &expr_list) const
Definition: Analyzer.cpp:3116
std::shared_ptr< Analyzer::Expr > upper_bound_
Definition: Analyzer.h:1129
std::shared_ptr< Analyzer::Expr > partition_count_
Definition: Analyzer.h:1130
constexpr double f
Definition: Utm.h:31
std::shared_ptr< Analyzer::Expr > lower_bound_
Definition: Analyzer.h:1128
std::shared_ptr< Analyzer::Expr > target_value_
Definition: Analyzer.h:1127

+ Here is the call graph for this function:

double Analyzer::WidthBucketExpr::get_bound_val ( const Analyzer::Expr bound_expr) const

Definition at line 3752 of file Analyzer.cpp.

References CHECK, Analyzer::Expr::deep_copy(), and kDOUBLE.

Referenced by CodeGenerator::codegenConstantWidthBucketExpr(), and compute_bucket().

3752  {
3753  CHECK(bound_expr);
3754  auto copied_expr = bound_expr->deep_copy();
3755  auto casted_expr = copied_expr->add_cast(SQLTypeInfo(kDOUBLE, false));
3756  CHECK(casted_expr);
3757  auto casted_constant = std::dynamic_pointer_cast<const Analyzer::Constant>(casted_expr);
3758  CHECK(casted_constant);
3759  return casted_constant->get_constval().doubleval;
3760 }
virtual std::shared_ptr< Analyzer::Expr > deep_copy() const =0
#define CHECK(condition)
Definition: Logger.h:291

+ Here is the call graph for this function:

+ Here is the caller graph for this function:

const Expr* Analyzer::WidthBucketExpr::get_lower_bound ( ) const
inline

Definition at line 1047 of file Analyzer.h.

References lower_bound_.

Referenced by CodeGenerator::codegen(), CodeGenerator::codegenConstantWidthBucketExpr(), CodeGenerator::codegenWidthBucketExpr(), operator==(), DeepCopyVisitor::visitWidthBucket(), and ScalarExprVisitor< std::set< shared::TableKey > >::visitWidthBucket().

1047 { return lower_bound_.get(); }
std::shared_ptr< Analyzer::Expr > lower_bound_
Definition: Analyzer.h:1128

+ Here is the caller graph for this function:

const Expr* Analyzer::WidthBucketExpr::get_partition_count ( ) const
inline

Definition at line 1049 of file Analyzer.h.

References partition_count_.

Referenced by CodeGenerator::codegen(), CodeGenerator::codegenConstantWidthBucketExpr(), CodeGenerator::codegenWidthBucketExpr(), getExpressionRange(), operator==(), DeepCopyVisitor::visitWidthBucket(), and ScalarExprVisitor< std::set< shared::TableKey > >::visitWidthBucket().

1049 { return partition_count_.get(); }
std::shared_ptr< Analyzer::Expr > partition_count_
Definition: Analyzer.h:1130

+ Here is the caller graph for this function:

int32_t Analyzer::WidthBucketExpr::get_partition_count_val ( ) const

Definition at line 3762 of file Analyzer.cpp.

References Analyzer::Constant::get_constval(), kBIGINT, kINT, kSMALLINT, kTINYINT, and partition_count_.

Referenced by CodeGenerator::codegen(), CodeGenerator::codegenConstantWidthBucketExpr(), compute_bucket(), and getExpressionRange().

3762  {
3763  auto const_partition_count_expr =
3764  dynamic_cast<const Analyzer::Constant*>(partition_count_.get());
3765  if (!const_partition_count_expr) {
3766  return -1;
3767  }
3768  auto d = const_partition_count_expr->get_constval();
3769  switch (const_partition_count_expr->get_type_info().get_type()) {
3770  case kTINYINT:
3771  return d.tinyintval;
3772  case kSMALLINT:
3773  return d.smallintval;
3774  case kINT:
3775  return d.intval;
3776  case kBIGINT: {
3777  auto bi = d.bigintval;
3778  if (bi < 1 || bi > INT32_MAX) {
3779  return -1;
3780  }
3781  return bi;
3782  }
3783  default:
3784  return -1;
3785  }
3786 }
std::shared_ptr< Analyzer::Expr > partition_count_
Definition: Analyzer.h:1130
Datum get_constval() const
Definition: Analyzer.h:348
Definition: sqltypes.h:62

+ Here is the call graph for this function:

+ Here is the caller graph for this function:

const Expr* Analyzer::WidthBucketExpr::get_target_value ( ) const
inline

Definition at line 1046 of file Analyzer.h.

References target_value_.

Referenced by CodeGenerator::codegen(), CodeGenerator::codegenConstantWidthBucketExpr(), CodeGenerator::codegenWidthBucketExpr(), getExpressionRange(), operator==(), DeepCopyVisitor::visitWidthBucket(), and ScalarExprVisitor< std::set< shared::TableKey > >::visitWidthBucket().

1046 { return target_value_.get(); }
std::shared_ptr< Analyzer::Expr > target_value_
Definition: Analyzer.h:1127

+ Here is the caller graph for this function:

const Expr* Analyzer::WidthBucketExpr::get_upper_bound ( ) const
inline

Definition at line 1048 of file Analyzer.h.

References upper_bound_.

Referenced by CodeGenerator::codegen(), CodeGenerator::codegenConstantWidthBucketExpr(), CodeGenerator::codegenWidthBucketExpr(), operator==(), DeepCopyVisitor::visitWidthBucket(), and ScalarExprVisitor< std::set< shared::TableKey > >::visitWidthBucket().

1048 { return upper_bound_.get(); }
std::shared_ptr< Analyzer::Expr > upper_bound_
Definition: Analyzer.h:1129

+ Here is the caller graph for this function:

void Analyzer::WidthBucketExpr::group_predicates ( std::list< const Expr * > &  scan_predicates,
std::list< const Expr * > &  join_predicates,
std::list< const Expr * > &  const_predicates 
) const
overridevirtual

Reimplemented from Analyzer::Expr.

Definition at line 1784 of file Analyzer.cpp.

References target_value_.

1786  {
1787  std::set<int> rte_idx_set;
1788  target_value_->collect_rte_idx(rte_idx_set);
1789  if (rte_idx_set.size() > 1) {
1790  join_predicates.push_back(this);
1791  } else if (rte_idx_set.size() == 1) {
1792  scan_predicates.push_back(this);
1793  } else {
1794  const_predicates.push_back(this);
1795  }
1796 }
std::shared_ptr< Analyzer::Expr > target_value_
Definition: Analyzer.h:1127
bool Analyzer::WidthBucketExpr::is_constant_expr ( ) const
inline

Definition at line 1124 of file Analyzer.h.

References constant_expr_.

Referenced by CodeGenerator::codegen(), and getExpressionRange().

1124 { return constant_expr_; }

+ Here is the caller graph for this function:

bool Analyzer::WidthBucketExpr::operator== ( const Expr rhs) const
overridevirtual

Implements Analyzer::Expr.

Definition at line 2402 of file Analyzer.cpp.

References get_lower_bound(), get_partition_count(), get_target_value(), get_upper_bound(), lower_bound_, partition_count_, target_value_, and upper_bound_.

2402  {
2403  if (typeid(rhs) != typeid(WidthBucketExpr)) {
2404  return false;
2405  }
2406  const WidthBucketExpr& rhs_l = dynamic_cast<const WidthBucketExpr&>(rhs);
2407  if (!(*target_value_ == *rhs_l.get_target_value())) {
2408  return false;
2409  }
2410  if (!(*lower_bound_ == *rhs_l.get_lower_bound())) {
2411  return false;
2412  }
2413  if (!(*upper_bound_ == *rhs_l.get_upper_bound())) {
2414  return false;
2415  }
2416  if (!(*partition_count_ == *rhs_l.get_partition_count())) {
2417  return false;
2418  }
2419  return true;
2420 }
std::shared_ptr< Analyzer::Expr > upper_bound_
Definition: Analyzer.h:1129
std::shared_ptr< Analyzer::Expr > partition_count_
Definition: Analyzer.h:1130
std::shared_ptr< Analyzer::Expr > lower_bound_
Definition: Analyzer.h:1128
WidthBucketExpr(const std::shared_ptr< Analyzer::Expr > target_value, const std::shared_ptr< Analyzer::Expr > lower_bound, const std::shared_ptr< Analyzer::Expr > upper_bound, const std::shared_ptr< Analyzer::Expr > partition_count)
Definition: Analyzer.h:1035
std::shared_ptr< Analyzer::Expr > target_value_
Definition: Analyzer.h:1127

+ Here is the call graph for this function:

std::shared_ptr<Analyzer::Expr> Analyzer::WidthBucketExpr::rewrite_agg_to_var ( const std::vector< std::shared_ptr< TargetEntry >> &  tlist) const
inlineoverridevirtual

Reimplemented from Analyzer::Expr.

Definition at line 1077 of file Analyzer.h.

References lower_bound_, partition_count_, target_value_, and upper_bound_.

1078  {
1079  return makeExpr<WidthBucketExpr>(target_value_->rewrite_agg_to_var(tlist),
1080  lower_bound_,
1081  upper_bound_,
1083  }
std::shared_ptr< Analyzer::Expr > upper_bound_
Definition: Analyzer.h:1129
std::shared_ptr< Analyzer::Expr > partition_count_
Definition: Analyzer.h:1130
std::shared_ptr< Analyzer::Expr > lower_bound_
Definition: Analyzer.h:1128
std::shared_ptr< Analyzer::Expr > target_value_
Definition: Analyzer.h:1127
std::shared_ptr<Analyzer::Expr> Analyzer::WidthBucketExpr::rewrite_with_child_targetlist ( const std::vector< std::shared_ptr< TargetEntry >> &  tlist) const
inlineoverridevirtual

Reimplemented from Analyzer::Expr.

Definition at line 1070 of file Analyzer.h.

References lower_bound_, partition_count_, target_value_, and upper_bound_.

1071  {
1072  return makeExpr<WidthBucketExpr>(target_value_->rewrite_with_child_targetlist(tlist),
1073  lower_bound_,
1074  upper_bound_,
1076  }
std::shared_ptr< Analyzer::Expr > upper_bound_
Definition: Analyzer.h:1129
std::shared_ptr< Analyzer::Expr > partition_count_
Definition: Analyzer.h:1130
std::shared_ptr< Analyzer::Expr > lower_bound_
Definition: Analyzer.h:1128
std::shared_ptr< Analyzer::Expr > target_value_
Definition: Analyzer.h:1127
std::shared_ptr<Analyzer::Expr> Analyzer::WidthBucketExpr::rewrite_with_targetlist ( const std::vector< std::shared_ptr< TargetEntry >> &  tlist) const
inlineoverridevirtual

Reimplemented from Analyzer::Expr.

Definition at line 1063 of file Analyzer.h.

References lower_bound_, partition_count_, target_value_, and upper_bound_.

1064  {
1065  return makeExpr<WidthBucketExpr>(target_value_->rewrite_with_targetlist(tlist),
1066  lower_bound_,
1067  upper_bound_,
1069  }
std::shared_ptr< Analyzer::Expr > upper_bound_
Definition: Analyzer.h:1129
std::shared_ptr< Analyzer::Expr > partition_count_
Definition: Analyzer.h:1130
std::shared_ptr< Analyzer::Expr > lower_bound_
Definition: Analyzer.h:1128
std::shared_ptr< Analyzer::Expr > target_value_
Definition: Analyzer.h:1127
void Analyzer::WidthBucketExpr::set_constant_expr ( ) const
inline

Definition at line 1123 of file Analyzer.h.

References constant_expr_.

Referenced by CodeGenerator::codegen().

1123 { constant_expr_ = true; }

+ Here is the caller graph for this function:

void Analyzer::WidthBucketExpr::skip_out_of_bound_check ( ) const
inline

Definition at line 1122 of file Analyzer.h.

References skip_out_of_bound_check_.

Referenced by CodeGenerator::codegen().

1122 { skip_out_of_bound_check_ = true; }

+ Here is the caller graph for this function:

std::string Analyzer::WidthBucketExpr::toString ( ) const
overridevirtual

Implements Analyzer::Expr.

Definition at line 2871 of file Analyzer.cpp.

References lower_bound_, partition_count_, target_value_, and upper_bound_.

2871  {
2872  std::string str{"(WIDTH_BUCKET "};
2873  str += target_value_->toString();
2874  str += lower_bound_->toString();
2875  str += upper_bound_->toString();
2876  str += partition_count_->toString();
2877  return str + ") ";
2878 }
std::shared_ptr< Analyzer::Expr > upper_bound_
Definition: Analyzer.h:1129
std::shared_ptr< Analyzer::Expr > partition_count_
Definition: Analyzer.h:1130
std::shared_ptr< Analyzer::Expr > lower_bound_
Definition: Analyzer.h:1128
std::shared_ptr< Analyzer::Expr > target_value_
Definition: Analyzer.h:1127

Member Data Documentation

bool Analyzer::WidthBucketExpr::constant_expr_
mutableprivate

Definition at line 1132 of file Analyzer.h.

Referenced by is_constant_expr(), and set_constant_expr().

std::shared_ptr<Analyzer::Expr> Analyzer::WidthBucketExpr::lower_bound_
private
std::shared_ptr<Analyzer::Expr> Analyzer::WidthBucketExpr::partition_count_
private
bool Analyzer::WidthBucketExpr::skip_out_of_bound_check_
mutableprivate

Definition at line 1134 of file Analyzer.h.

Referenced by can_skip_out_of_bound_check(), and skip_out_of_bound_check().

std::shared_ptr<Analyzer::Expr> Analyzer::WidthBucketExpr::upper_bound_
private

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