summaryrefslogtreecommitdiff
path: root/submit/SymbolTable.java
blob: 09251bc263c861bfdb8e69b450a1eaac96e60cdc (plain)
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
package submit;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import submit.ast.VarType;

/*
 * Code formatter project
 * CS 4481
 */
/**
 *
 */
public class SymbolTable {

  private final HashMap<String, SymbolInfo> table;
  private SymbolTable parent;
  private final List<SymbolTable> children;

  private int offset;

  public static int LABEL_IDENTIFIER = 0;

  public static String nextId() {
    return Integer.toString(SymbolTable.LABEL_IDENTIFIER++);
  }

  public SymbolTable() {
    offset = 0;
    table = new HashMap<>();
    parent = null;
    children = new ArrayList<>();

    this.addGlobalSymbols();
  }

  public List<String> symbolNames() { return new ArrayList<>(table.keySet()); }

  public void addGlobalSymbols() {
    SymbolInfo println = new SymbolInfo("println", null, true);
    this.addSymbol("println", println);
  }

  public void addSymbol(String id, SymbolInfo symbol) { table.put(id, symbol); }

  public int addOffset(int n) {
    offset -= 4 * n;
    return offset;
  }

  public int getOffset() { return this.offset; }

  // Add symbols in before and reorder offsets such that symbols in before have
  // a "higher" offset
  public void addOtherTableBefore(SymbolTable before) {
    offset = 0;
    List<String> thisSymbols = symbolNames();

    for (String id : before.symbolNames()) {
      SymbolInfo symbol = before.find(id);
      if (!symbol.isFunction()) {
        addOffset(1);
        symbol.setOffset(offset);
      }
      addSymbol(id, symbol);
    }

    for (String id : thisSymbols) {
      SymbolInfo symbol = table.get(id);
      if (!symbol.isFunction()) {
        addOffset(1);
        table.get(id).setOffset(offset);
      }
    }
  }

  public int offsetOf(String id) {
    if (table.containsKey(id)) {
      return table.get(id).getOffset();
    }
    if (parent != null) {
      return -parent.getOffset() + parent.offsetOf(id);
    }
    return 0; // This shouldn't happen :D
  }

  /**
   * Returns null if no symbol with that id is in this symbol table or an
   * ancestor table.
   *
   * @param id
   * @return
   */
  public SymbolInfo find(String id) {
    if (table.containsKey(id)) {
      return table.get(id);
    }
    if (parent != null) {
      return parent.find(id);
    }
    return null;
  }

  /**
   * Returns the new child.
   *
   * @return
   */
  public SymbolTable createChild() {
    SymbolTable child = new SymbolTable();
    children.add(child);
    child.parent = this;
    return child;
  }

  public SymbolTable getParent() { return parent; }
}