Code_comp_op.hpp
上传用户:romrleung
上传日期:2022-05-23
资源大小:18897k
文件大小:4k
源码类别:

MySQL数据库

开发平台:

Visual C++

  1. /* Copyright (C) 2003 MySQL AB
  2.    This program is free software; you can redistribute it and/or modify
  3.    it under the terms of the GNU General Public License as published by
  4.    the Free Software Foundation; either version 2 of the License, or
  5.    (at your option) any later version.
  6.    This program is distributed in the hope that it will be useful,
  7.    but WITHOUT ANY WARRANTY; without even the implied warranty of
  8.    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  9.    GNU General Public License for more details.
  10.    You should have received a copy of the GNU General Public License
  11.    along with this program; if not, write to the Free Software
  12.    Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA */
  13. #ifndef ODBC_CODEGEN_Code_comp_op_hpp
  14. #define ODBC_CODEGEN_Code_comp_op_hpp
  15. #include <common/common.hpp>
  16. #include <common/DataField.hpp>
  17. #include "Code_pred.hpp"
  18. #include "Code_expr.hpp"
  19. #include "Code_expr_column.hpp"
  20. /**
  21.  * @class Comp_op
  22.  * @brief Comparison operations
  23.  */
  24. struct Comp_op {
  25.     enum Opcode {
  26. Eq = 1, // binary
  27. Noteq,
  28. Lt,
  29. Lteq,
  30. Gt,
  31. Gteq,
  32. Like,
  33. Notlike,
  34. Isnull, // unary
  35. Isnotnull
  36.     };
  37.     Comp_op(Opcode opcode);
  38.     const char* name() const;
  39.     unsigned arity() const;
  40.     Opcode m_opcode;
  41. };
  42. inline
  43. Comp_op::Comp_op(Opcode opcode) :
  44.     m_opcode(opcode)
  45. {
  46. }
  47. /**
  48.  * @class Plan_comp_op
  49.  * @brief Comparison operator node in PlanTree
  50.  */
  51. class Plan_comp_op : public Plan_pred {
  52. public:
  53.     Plan_comp_op(Plan_root* root, Comp_op op);
  54.     virtual ~Plan_comp_op();
  55.     void setExpr(unsigned i, Plan_expr* expr);
  56.     Plan_base* analyze(Ctx& ctx, Ctl& ctl);
  57.     Exec_base* codegen(Ctx& ctx, Ctl& ctl);
  58.     void print(Ctx& ctx);
  59.     virtual bool isGroupBy(const Plan_expr_row* row) const;
  60. protected:
  61.     Comp_op m_op;
  62.     Plan_expr* m_expr[1 + 2];
  63.     Plan_expr_column* m_interpColumn[1 + 2]; // candidates
  64. };
  65. inline
  66. Plan_comp_op::Plan_comp_op(Plan_root* root, Comp_op op) :
  67.     Plan_pred(root, TypeComp),
  68.     m_op(op)
  69. {
  70.     m_expr[0] = m_expr[1] = m_expr[2] = 0;
  71.     m_interpColumn[0] = m_interpColumn[1] = m_interpColumn[2] = 0;
  72. }
  73. inline void
  74. Plan_comp_op::setExpr(unsigned i, Plan_expr* expr)
  75. {
  76.     ctx_assert(1 <= i && i <= 2);
  77.     m_expr[i] = expr;
  78. }
  79. /**
  80.  * @class Exec_comp_op
  81.  * @brief Comparison operator node in ExecTree
  82.  */
  83. class Exec_comp_op : public Exec_pred {
  84. public:
  85.     class Code : public Exec_pred::Code {
  86.     public:
  87. Code(Comp_op op);
  88. virtual ~Code();
  89.     protected:
  90. friend class Plan_comp_op;
  91. friend class Exec_comp_op;
  92. Comp_op m_op;
  93. unsigned m_interpColumn; // 1 or 2 if interpreted column, 0 if both constant
  94. NdbAttrId m_interpAttrId;
  95.     };
  96.     class Data : public Exec_pred::Data {
  97.     public:
  98. Data();
  99. virtual ~Data();
  100.     protected:
  101. friend class Exec_comp_op;
  102.     };
  103.     Exec_comp_op(Exec_root* root);
  104.     virtual ~Exec_comp_op();
  105.     void alloc(Ctx& ctx, Ctl& ctl);
  106.     void execInterp(Ctx& ctx, Ctl& ctl);
  107.     void evaluate(Ctx& ctx, Ctl& ctl);
  108.     void close(Ctx& ctx);
  109.     void print(Ctx& ctx);
  110.     // children
  111.     const Code& getCode() const;
  112.     Data& getData() const;
  113.     void setExpr(unsigned i, Exec_expr* expr);
  114. protected:
  115.     Exec_expr* m_expr[1 + 2];
  116. };
  117. inline
  118. Exec_comp_op::Code::Code(Comp_op op) :
  119.     m_op(op),
  120.     m_interpColumn(0),
  121.     m_interpAttrId((NdbAttrId)-1)
  122. {
  123. }
  124. inline
  125. Exec_comp_op::Data::Data()
  126. {
  127. }
  128. inline
  129. Exec_comp_op::Exec_comp_op(Exec_root* root) :
  130.     Exec_pred(root)
  131. {
  132.     m_expr[0] = m_expr[1] = m_expr[2] = 0;
  133. }
  134. // children
  135. inline const Exec_comp_op::Code&
  136. Exec_comp_op::getCode() const
  137. {
  138.     const Code* code = static_cast<const Code*>(m_code);
  139.     return *code;
  140. }
  141. inline Exec_comp_op::Data&
  142. Exec_comp_op::getData() const
  143. {
  144.     Data* data = static_cast<Data*>(m_data);
  145.     return *data;
  146. }
  147. inline void
  148. Exec_comp_op::setExpr(unsigned i, Exec_expr* expr)
  149. {
  150.     ctx_assert(1 <= i && i <= 2 && m_expr[i] == 0);
  151.     m_expr[i] = expr;
  152. }
  153. #endif