Analyzer::MLPredictExpr Class Reference

#include <Analyzer.h>

Inheritance diagram for Analyzer::MLPredictExpr:

Collaboration diagram for Analyzer::MLPredictExpr:

Public Member Functions
	MLPredictExpr (const std::shared_ptr< Analyzer::Expr > &model, const std::vector< std::shared_ptr< Analyzer::Expr >> &regressors)
const Expr *	get_model_value () const
const std::vector < std::shared_ptr < Analyzer::Expr > > &	get_regressor_values () const
std::shared_ptr< Analyzer::Expr >	deep_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::Expr >	rewrite_with_targetlist (const std::vector< std::shared_ptr< TargetEntry >> &tlist) const override
std::shared_ptr< Analyzer::Expr >	rewrite_with_child_targetlist (const std::vector< std::shared_ptr< TargetEntry >> &tlist) const override
std::shared_ptr< Analyzer::Expr >	rewrite_agg_to_var (const std::vector< std::shared_ptr< TargetEntry >> &tlist) const override
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
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::Expr >	get_shared_ptr ()
const SQLTypeInfo &	get_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::Expr >	decompress ()
virtual void	get_domain (DomainSet &domain_set) const

Private Attributes
std::shared_ptr< Analyzer::Expr >	model_value_
std::vector< std::shared_ptr < Analyzer::Expr > >	regressor_values_

Additional Inherited Members
Protected Attributes inherited from Analyzer::Expr
SQLTypeInfo	type_info
bool	contains_agg

Detailed Description

Definition at line 707 of file Analyzer.h.

Constructor & Destructor Documentation

Analyzer::MLPredictExpr::MLPredictExpr ( const std::shared_ptr< Analyzer::Expr > &  model, const std::vector< std::shared_ptr< Analyzer::Expr >> &  regressors  )

inline

Definition at line 709 of file Analyzer.h.

      : Expr(kDOUBLE, false), model_value_(model), regressor_values_(regressors) {}

Member Function Documentation

void Analyzer::MLPredictExpr::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 729 of file Analyzer.h.

References regressor_values_.

                                       {
    for (const auto& regressor_value : regressor_values_) {
      regressor_value->collect_column_var(colvar_set, include_agg);
    }
  }

void Analyzer::MLPredictExpr::collect_rte_idx ( std::set< int > &  rte_idx_set ) const

inlineoverridevirtual

Reimplemented from Analyzer::Expr.

Definition at line 723 of file Analyzer.h.

References regressor_values_.

                                                                {
    for (const auto& regressor_value : regressor_values_) {
      regressor_value->collect_rte_idx(rte_idx_set);
    }
  }

std::shared_ptr< Analyzer::Expr > Analyzer::MLPredictExpr::deep_copy ( ) const

overridevirtual

Implements Analyzer::Expr.

Definition at line 136 of file Analyzer.cpp.

References model_value_, and regressor_values_.

Referenced by DeepCopyVisitor::visitMLPredict().

                                                           {
  std::vector<std::shared_ptr<Analyzer::Expr>> regressors_copy;
  for (auto r : regressor_values_) {
    regressors_copy.emplace_back(r->deep_copy());
  }
  return makeExpr<MLPredictExpr>(model_value_->deep_copy(), regressors_copy);
}

Here is the caller graph for this function:

void Analyzer::MLPredictExpr::find_expr ( std::function< bool(const Expr *)>  f, std::list< const Expr * > &  expr_list  ) const

overridevirtual

Reimplemented from Analyzer::Expr.

Definition at line 3286 of file Analyzer.cpp.

References Analyzer::Expr::add_unique(), f(), and regressor_values_.

                                                                     {
  if (f(this)) {
    add_unique(expr_list);
    return;
  }
  for (auto& r : regressor_values_) {
    r->find_expr(f, expr_list);
  }
}

Here is the call graph for this function:

const Expr* Analyzer::MLPredictExpr::get_model_value ( ) const

inline

Definition at line 713 of file Analyzer.h.

References model_value_.

Referenced by CodeGenerator::codegen(), operator==(), and ScalarExprVisitor< std::set< shared::TableKey > >::visitMLPredict().

{ return model_value_.get(); }

Here is the caller graph for this function:

const std::vector<std::shared_ptr<Analyzer::Expr> >& Analyzer::MLPredictExpr::get_regressor_values ( ) const

inline

Definition at line 714 of file Analyzer.h.

References regressor_values_.

Referenced by CodeGenerator::codegen(), CodeGenerator::codegenLinRegPredict(), CodeGenerator::codegenTreeRegPredict(), operator==(), and ScalarExprVisitor< std::set< shared::TableKey > >::visitMLPredict().

                                                                             {
    return regressor_values_;
  }

Here is the caller graph for this function:

void Analyzer::MLPredictExpr::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 1739 of file Analyzer.cpp.

References regressor_values_.

                                                                                   {
  std::set<int> rte_idx_set;
  for (const auto& regressor_value : regressor_values_) {
    regressor_value->collect_rte_idx(rte_idx_set);
  }
  if (rte_idx_set.size() > 1) {
    join_predicates.push_back(this);
  } else if (rte_idx_set.size() == 1) {
    scan_predicates.push_back(this);
  } else {
    const_predicates.push_back(this);
  }
}

bool Analyzer::MLPredictExpr::operator== ( const Expr &  rhs ) const

overridevirtual

Implements Analyzer::Expr.

Definition at line 2413 of file Analyzer.cpp.

References get_model_value(), get_regressor_values(), model_value_, and regressor_values_.

                                                    {
  if (typeid(rhs) != typeid(MLPredictExpr)) {
    return false;
  }
  const MLPredictExpr& rhs_cl = dynamic_cast<const MLPredictExpr&>(rhs);
  if (!(*model_value_ == *rhs_cl.get_model_value())) {
    return false;
  }
  auto rhs_regressor_values = rhs_cl.get_regressor_values();
  if (regressor_values_.size() != rhs_regressor_values.size()) {
    return false;
  }
  for (size_t regressor_idx = 0; regressor_idx < regressor_values_.size();
       ++regressor_idx) {
    if (!(*regressor_values_[regressor_idx] == *rhs_regressor_values[regressor_idx])) {
      return false;
    }
  }
  return true;
}

Here is the call graph for this function:

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

inlineoverridevirtual

Reimplemented from Analyzer::Expr.

Definition at line 758 of file Analyzer.h.

References model_value_, and regressor_values_.

                                                                         {
    std::vector<std::shared_ptr<Analyzer::Expr>> new_regressor_values;
    for (auto& r : regressor_values_) {
      new_regressor_values.push_back(r->deep_copy());
    }
    return makeExpr<MLPredictExpr>(model_value_->rewrite_agg_to_var(tlist),
                                   new_regressor_values);
  }

std::shared_ptr<Analyzer::Expr> Analyzer::MLPredictExpr::rewrite_with_child_targetlist ( const std::vector< std::shared_ptr< TargetEntry >> &  tlist ) const

inlineoverridevirtual

Reimplemented from Analyzer::Expr.

Definition at line 748 of file Analyzer.h.

References model_value_, and regressor_values_.

                                                                         {
    std::vector<std::shared_ptr<Analyzer::Expr>> new_regressor_values;
    for (auto& r : regressor_values_) {
      new_regressor_values.push_back(r->deep_copy());
    }
    return makeExpr<MLPredictExpr>(model_value_->rewrite_with_child_targetlist(tlist),
                                   new_regressor_values);
  }

std::shared_ptr<Analyzer::Expr> Analyzer::MLPredictExpr::rewrite_with_targetlist ( const std::vector< std::shared_ptr< TargetEntry >> &  tlist ) const

inlineoverridevirtual

Reimplemented from Analyzer::Expr.

Definition at line 738 of file Analyzer.h.

References model_value_, and regressor_values_.

                                                                         {
    std::vector<std::shared_ptr<Analyzer::Expr>> new_regressor_values;
    for (auto& r : regressor_values_) {
      new_regressor_values.push_back(r->deep_copy());
    }
    return makeExpr<MLPredictExpr>(model_value_->rewrite_with_targetlist(tlist),
                                   new_regressor_values);
  }

std::string Analyzer::MLPredictExpr::toString ( ) const

overridevirtual

Implements Analyzer::Expr.

Definition at line 2857 of file Analyzer.cpp.

References model_value_, and regressor_values_.

                                        {
  std::stringstream ss;
  ss << "ML_PREDICT(Model: ";
  ss << model_value_->toString();
  ss << " Regressors: ";
  for (const auto& regressor_value : regressor_values_) {
    ss << regressor_value->toString() << " ";
  }
  ss << ") ";
  return ss.str();
}

Member Data Documentation

std::shared_ptr<Analyzer::Expr> Analyzer::MLPredictExpr::model_value_

private

Definition at line 774 of file Analyzer.h.

Referenced by deep_copy(), get_model_value(), operator==(), rewrite_agg_to_var(), rewrite_with_child_targetlist(), rewrite_with_targetlist(), and toString().

std::vector<std::shared_ptr<Analyzer::Expr> > Analyzer::MLPredictExpr::regressor_values_

private

Definition at line 775 of file Analyzer.h.

Referenced by collect_column_var(), collect_rte_idx(), deep_copy(), find_expr(), get_regressor_values(), group_predicates(), operator==(), rewrite_agg_to_var(), rewrite_with_child_targetlist(), rewrite_with_targetlist(), and toString().

---

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

• /home/jenkins-slave/workspace/core-os-doxygen/Analyzer/Analyzer.h
• /home/jenkins-slave/workspace/core-os-doxygen/Analyzer/Analyzer.cpp