LLVMDemo/AST.cc at master · joverbey/LLVMDemo · 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
#include "AST.h"
#include "Printer.h"
#include <cassert>

ostream &operator<<(ostream &out, const Expr *node) {
  Printer::Print(node, out);
  return out;
}

const class Type &IntegerLiteral::Type() const {
  return Type::INTEGER;
}

void IntegerLiteral::Accept(Visitor *v) const {
  v->Visit(this);
}

const class Type &RealLiteral::Type() const {
  return Type::REAL;
}

void RealLiteral::Accept(Visitor *v) const {
  v->Visit(this);
}

const class Type &InputExpr::Type() const {
  return Type::INTEGER;
}

void InputExpr::Accept(Visitor *v) const {
  v->Visit(this);
}

const class Type &CastExpr::Type() const {
  return this->type;
}

void CastExpr::Accept(Visitor *v) const {
  v->Visit(this);
}

const class Type &BinExpr::Type() const {
  if (this->lhs->Type() != this->rhs->Type()
      || this->lhs->Type().IsError()
      || this->rhs->Type().IsError()) {
    return Type::ERROR;
  }

  switch (this->op) {
    case ADD:
    case SUB:
    case MUL:
    case DIV:
      return this->lhs->Type();
    case LT:
    case LTE:
    case GT:
    case GTE:
    case EQ:
    case NEQ:
      return Type::INTEGER;
  }

  assert(false);
  return Type::ERROR;
}

void BinExpr::Accept(Visitor *v) const {
  v->Visit(this);
}

const class Type &IfExpr::Type() const {
  if (this->condExpr->Type() == Type::INTEGER
      && this->thenExpr->Type() == this->elseExpr->Type())
    return this->thenExpr->Type();

  return Type::ERROR;
}

void IfExpr::Accept(Visitor *v) const {
  v->Visit(this);
}