fql/expr.cc at master · dccmx/fql · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
#include <stdio.h>

#include "parser.h"
#include "expr.h"

UnaryExpr::UnaryExpr(int op, Expr *right) : op_(op), right_(right) {
  Value *v = dynamic_cast<Value*>(right_);
  if (v && v->id() != TK_STRING && v->id() != TK_INTEGER && v->id() != TK_FLOAT) {
    v->set_is_attr(true);
  }
}

BinaryExpr::BinaryExpr(int op, Expr *left, Expr *right) : op_(op), left_(left), right_(right) {
  Value *v = dynamic_cast<Value*>(left_);
  if (v && v->id() != TK_STRING && v->id() != TK_INTEGER && v->id() != TK_FLOAT) {
    v->set_is_attr(true);
  }
}

#define PREPARE_PARAM  \
  if (left_) left = left_->Evaluate(row); \
  if (right_) right = right_->Evaluate(row);

#define BINARY_OP(OP) do { \
  PREPARE_PARAM \
  if (left && right) { \
    ret = left->OP(right); \
  }\
} while (0)

#define LOGIC_OP(OP) do { \
  PREPARE_PARAM \
  if (left && right) { \
    ret = new Bool(left->OP(right)); \
  }\
} while (0)

#define COMPARE_OP(OP) do {\
  PREPARE_PARAM \
  if (left && right) { \
    ret = new Bool(left->Compare(right) OP 0); \
  }\
} while (0)

Variant *BinaryExpr::Evaluate(Row *row) {
  Variant *ret = NULL;
  Variant *left = NULL;
  Variant *right = NULL;

  switch (op_) {
    case TK_PLUS:
      BINARY_OP(Plus);
      break;
    case TK_MINUS:
      BINARY_OP(Minus);
      break;
    case TK_DIV:
      BINARY_OP(Div);
      break;
    case TK_MOD:
      BINARY_OP(Mod);
      break;
    case TK_STAR:
      BINARY_OP(Mult);
      break;
    case TK_OR:
      LOGIC_OP(Or);
      break;
    case TK_AND:
      LOGIC_OP(And);
      break;
    case TK_LIKE:
      LOGIC_OP(Like);
      break;
    case TK_GT:
      COMPARE_OP(>);
      break;
    case TK_LT:
      COMPARE_OP(<);
      break;
    case TK_EQ:
      COMPARE_OP(==);
      break;
    case TK_GE:
      COMPARE_OP(>=);
      break;
    case TK_LE:
      COMPARE_OP(<=);
      break;
    case TK_NE:
      COMPARE_OP(!=);
      break;
    default:
      break;
  }
  if (left) delete left;
  if (right) delete right;
  return ret;
}

void BinaryExpr::Print(int indent) {
  for (int i = 0; i < indent; i++) printf(" ");
  printf("(%2d,\n", op_);
  if (left_) left_->Print(indent + 2);
  for (int i = 0; i < indent + 2; i++) printf(" ");
  printf(", ");
  if (right_) right_->Print(indent + 2);
  for (int i = 0; i < indent; i++) printf(" ");
  printf(")\n");
}

Variant *UnaryExpr::Evaluate(Row *row) {
  Variant *ret = NULL;
  Variant *right = NULL;

  switch (op_) {
    case TK_NOT:
      if (right_) right = right_->Evaluate(row);
      if (right) {
        ret = new Bool(!right->c_bool());
      }
      break;
    default:
      break;
  }
  if (right) delete right;
  return ret;
}