OmniSciDB  72c90bc290
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
Analyzer::TrimStringOper Class Reference

#include <Analyzer.h>

+ Inheritance diagram for Analyzer::TrimStringOper:
+ Collaboration diagram for Analyzer::TrimStringOper:

Public Member Functions

 TrimStringOper (const SqlStringOpKind trim_op_kind, const std::shared_ptr< Analyzer::Expr > &operand, const std::shared_ptr< Analyzer::Expr > &trim_chars)
 
 getMinArgs ()
 
 getExpectedTypeFamilies ()
 
 getArgNames ())
 
 TrimStringOper (const SqlStringOpKind trim_op_kind, const std::vector< std::shared_ptr< Analyzer::Expr >> &operands)
 
 TrimStringOper (const std::shared_ptr< Analyzer::StringOper > &string_oper)
 
std::shared_ptr< Analyzer::Exprdeep_copy () const override
 
size_t getMinArgs () const override
 
std::vector< OperandTypeFamilygetExpectedTypeFamilies () const override
 
std::vector< std::string > getArgNames () const override
 
- Public Member Functions inherited from Analyzer::StringOper
 StringOper (const SqlStringOpKind kind, const std::vector< std::shared_ptr< Analyzer::Expr >> &args)
 
 StringOper (const SqlStringOpKind kind, const SQLTypeInfo &return_ti, const std::vector< std::shared_ptr< Analyzer::Expr >> &args)
 
 StringOper (const SqlStringOpKind kind, const std::vector< std::shared_ptr< Analyzer::Expr >> &args, const size_t min_args, const std::vector< OperandTypeFamily > &expected_type_families, const std::vector< std::string > &arg_names)
 
 StringOper (const SqlStringOpKind kind, const SQLTypeInfo &return_ti, const std::vector< std::shared_ptr< Analyzer::Expr >> &args, const size_t min_args, const std::vector< OperandTypeFamily > &expected_type_families, const std::vector< std::string > &arg_names)
 
 StringOper (const SqlStringOpKind kind, const SQLTypeInfo &return_ti, const std::vector< std::shared_ptr< Analyzer::Expr >> &args, const std::vector< std::shared_ptr< Analyzer::Expr >> &chained_string_op_exprs)
 
 StringOper (const StringOper &other_string_oper)
 
 StringOper (const std::shared_ptr< StringOper > &other_string_oper)
 
SqlStringOpKind get_kind () const
 
size_t getArity () const
 
size_t getLiteralsArity () const
 
size_t getNonLiteralsArity () const
 
const ExprgetArg (const size_t i) const
 
std::shared_ptr< Analyzer::ExprgetOwnArg (const size_t i) const
 
std::vector< std::shared_ptr
< Analyzer::Expr > > 
getOwnArgs () const
 
std::vector< std::shared_ptr
< Analyzer::Expr > > 
getChainedStringOpExprs () const
 
bool requiresPerRowTranslation () const
 
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
 
bool hasNoneEncodedTextArg () const
 
bool hasSingleDictEncodedColInput () const
 returns whether we have one and only one column involved in this StringOper and all its descendents, and that that column is a dictionary-encoded text type More...
 
std::vector< size_t > getLiteralArgIndexes () const
 
LiteralArgMap getLiteralArgs () const
 
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
 
void group_predicates (std::list< const Expr * > &scan_predicates, std::list< const Expr * > &join_predicates, std::list< const Expr * > &const_predicates) 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::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
 

Static Private Member Functions

static SqlStringOpKind validate_trim_op_kind (const SqlStringOpKind trim_op_kind)
 
static SqlStringOpKind get_and_validate_trim_op_kind (const SqlStringOpKind trim_op_kind_maybe, const std::vector< std::shared_ptr< Analyzer::Expr >> &args)
 
static std::vector
< std::shared_ptr
< Analyzer::Expr > > 
normalize_operands (const std::vector< std::shared_ptr< Analyzer::Expr >> &operands, const SqlStringOpKind string_op_kind)
 

Additional Inherited Members

- Public Types inherited from Analyzer::StringOper
enum  OperandTypeFamily { OperandTypeFamily::STRING_FAMILY, OperandTypeFamily::INT_FAMILY }
 
using LiteralArgMap = std::map< size_t, std::pair< SQLTypes, Datum >>
 
- Protected Attributes inherited from Analyzer::Expr
SQLTypeInfo type_info
 
bool contains_agg
 

Detailed Description

Definition at line 2053 of file Analyzer.h.

Constructor & Destructor Documentation

Analyzer::TrimStringOper::TrimStringOper ( const SqlStringOpKind  trim_op_kind,
const std::shared_ptr< Analyzer::Expr > &  operand,
const std::shared_ptr< Analyzer::Expr > &  trim_chars 
)
inline

Definition at line 2055 of file Analyzer.h.

2059  {operand, trim_chars},
StringOper(const SqlStringOpKind kind, const std::vector< std::shared_ptr< Analyzer::Expr >> &args)
Definition: Analyzer.h:1607
static SqlStringOpKind validate_trim_op_kind(const SqlStringOpKind trim_op_kind)
Definition: Analyzer.h:2087
Analyzer::TrimStringOper::TrimStringOper ( const SqlStringOpKind  trim_op_kind,
const std::vector< std::shared_ptr< Analyzer::Expr >> &  operands 
)
inline

Definition at line 2064 of file Analyzer.h.

2067  TrimStringOper::normalize_operands(operands, trim_op_kind),
2068  getMinArgs(),
2070  getArgNames()) {}
static SqlStringOpKind get_and_validate_trim_op_kind(const SqlStringOpKind trim_op_kind_maybe, const std::vector< std::shared_ptr< Analyzer::Expr >> &args)
Definition: Analyzer.cpp:4459
StringOper(const SqlStringOpKind kind, const std::vector< std::shared_ptr< Analyzer::Expr >> &args)
Definition: Analyzer.h:1607
static std::vector< std::shared_ptr< Analyzer::Expr > > normalize_operands(const std::vector< std::shared_ptr< Analyzer::Expr >> &operands, const SqlStringOpKind string_op_kind)
Definition: Analyzer.cpp:4490
Analyzer::TrimStringOper::TrimStringOper ( const std::shared_ptr< Analyzer::StringOper > &  string_oper)
inline

Definition at line 2072 of file Analyzer.h.

2073  : StringOper(string_oper) {}
StringOper(const SqlStringOpKind kind, const std::vector< std::shared_ptr< Analyzer::Expr >> &args)
Definition: Analyzer.h:1607

Member Function Documentation

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

Reimplemented from Analyzer::StringOper.

Definition at line 3726 of file Analyzer.cpp.

References Analyzer::StringOper::deep_copy().

3726  {
3727  return makeExpr<Analyzer::TrimStringOper>(
3728  std::dynamic_pointer_cast<Analyzer::StringOper>(StringOper::deep_copy()));
3729 }
std::shared_ptr< Analyzer::Expr > deep_copy() const override
Definition: Analyzer.cpp:3681
Expression class for string functions The &quot;arg&quot; constructor parameter must be an expression that reso...
Definition: Analyzer.h:1601

+ Here is the call graph for this function:

SqlStringOpKind Analyzer::TrimStringOper::get_and_validate_trim_op_kind ( const SqlStringOpKind  trim_op_kind_maybe,
const std::vector< std::shared_ptr< Analyzer::Expr >> &  args 
)
staticprivate

Definition at line 4459 of file Analyzer.cpp.

References run_benchmark_import::args, INVALID, LTRIM, RTRIM, to_upper(), TRIM, and validate_trim_op_kind().

4461  {
4462  auto get_trim_type_if_exists = [&args]() {
4463  if (args.empty()) {
4464  return SqlStringOpKind::INVALID;
4465  }
4466  const auto trim_type_arg = dynamic_cast<const Analyzer::Constant*>(args[0].get());
4467  if (!trim_type_arg) {
4468  return SqlStringOpKind::INVALID;
4469  }
4470  const auto trim_type_str = to_upper(*trim_type_arg->get_constval().stringval);
4471  if (trim_type_str == "BOTH") {
4472  return SqlStringOpKind::TRIM;
4473  }
4474  if (trim_type_str == "LEADING") {
4475  return SqlStringOpKind::LTRIM;
4476  }
4477  if (trim_type_str == "TRAILING") {
4478  return SqlStringOpKind::RTRIM;
4479  }
4480  return SqlStringOpKind::INVALID;
4481  };
4482 
4483  const auto trim_op_kind = trim_op_kind_maybe == SqlStringOpKind::TRIM
4484  ? get_trim_type_if_exists()
4485  : trim_op_kind_maybe;
4487  return trim_op_kind;
4488 }
static SqlStringOpKind validate_trim_op_kind(const SqlStringOpKind trim_op_kind)
Definition: Analyzer.h:2087
std::string to_upper(const std::string &str)

+ Here is the call graph for this function:

Analyzer::TrimStringOper::getArgNames ( )
inline

Definition at line 2062 of file Analyzer.h.

2062 {}
std::vector<std::string> Analyzer::TrimStringOper::getArgNames ( ) const
inlineoverridevirtual

Reimplemented from Analyzer::StringOper.

Definition at line 2082 of file Analyzer.h.

2082  {
2083  return {"operand", "trim_chars"};
2084  }
Analyzer::TrimStringOper::getExpectedTypeFamilies ( )
std::vector<OperandTypeFamily> Analyzer::TrimStringOper::getExpectedTypeFamilies ( ) const
inlineoverridevirtual
Analyzer::TrimStringOper::getMinArgs ( )
size_t Analyzer::TrimStringOper::getMinArgs ( ) const
inlineoverridevirtual

Reimplemented from Analyzer::StringOper.

Definition at line 2077 of file Analyzer.h.

2077 { return 2UL; }
std::vector< std::shared_ptr< Analyzer::Expr > > Analyzer::TrimStringOper::normalize_operands ( const std::vector< std::shared_ptr< Analyzer::Expr >> &  operands,
const SqlStringOpKind  string_op_kind 
)
staticprivate

Definition at line 4490 of file Analyzer.cpp.

References CHECK_EQ, LTRIM, and RTRIM.

4492  {
4493  if (operands.size() < 2UL) {
4494  throw std::runtime_error("Trim operator expects 2 arguments.");
4495  }
4496 
4497  // Calcite is returning LTRIM/RTRIM(string_literal1, string_literal2) with the
4498  // arguments in that order, but LTRIM/TRIM(string_var, string_literal) in the
4499  // reverse order that the main TRIM operator uses as well
4500  // Until we can look into that more, reverse order only if we have a const then
4501  // non-const input order
4502 
4503  if (string_op_kind == SqlStringOpKind::LTRIM ||
4504  string_op_kind == SqlStringOpKind::RTRIM) {
4505  CHECK_EQ(operands.size(), 2UL);
4506  if (dynamic_cast<const Analyzer::Constant*>(operands[0].get()) &&
4507  !dynamic_cast<const Analyzer::Constant*>(operands[1].get())) {
4508  return {operands[1], operands[0]};
4509  }
4510  return operands;
4511  }
4512  CHECK_EQ(operands.size(), 3UL);
4513  return {operands[2], operands[1]};
4514 }
#define CHECK_EQ(x, y)
Definition: Logger.h:301
static SqlStringOpKind Analyzer::TrimStringOper::validate_trim_op_kind ( const SqlStringOpKind  trim_op_kind)
inlinestaticprivate

Definition at line 2087 of file Analyzer.h.

References LTRIM, RTRIM, and TRIM.

Referenced by get_and_validate_trim_op_kind().

2087  {
2088  if (!(trim_op_kind == SqlStringOpKind::TRIM ||
2089  trim_op_kind == SqlStringOpKind::LTRIM ||
2090  trim_op_kind == SqlStringOpKind::RTRIM)) {
2091  // Arguably should CHECK here
2092  throw std::runtime_error("Invalid trim type supplied to TRIM operator");
2093  }
2094  return trim_op_kind;
2095  }

+ Here is the caller graph for this function:


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