summaryrefslogtreecommitdiff
path: root/submit/ast/Assignment.java
blob: 9c06cbed16d649aab1fe68eb98f13c9a95309a03 (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
/*
 * Code formatter project
 * CS 4481
 */
package submit.ast;

import submit.MIPSResult;
import submit.RegisterAllocator;
import submit.SymbolTable;

/**
 *
 * @author edwajohn
 */
public class Assignment extends AbstractNode implements Expression {

  private final Mutable mutable;
  private final AssignmentType type;
  private final Expression rhs;

  public Assignment(Mutable mutable, String assign, Expression rhs) {
    this.mutable = mutable;
    this.type = AssignmentType.fromString(assign);
    this.rhs = rhs;
  }

  public void toCminus(StringBuilder builder, final String prefix) {
    mutable.toCminus(builder, prefix);
    if (rhs != null) {
      builder.append(" ").append(type.toString()).append(" ");
      rhs.toCminus(builder, prefix);
    } else {
      builder.append(type.toString());
    }
  }

  public MIPSResult toMIPS(StringBuilder code, StringBuilder data,
                           SymbolTable symbolTable,
                           RegisterAllocator regAllocator) {
    MIPSResult mut = mutable.toMIPS(code, data, symbolTable, regAllocator);
    MIPSResult result = rhs.toMIPS(code, data, symbolTable, regAllocator);

    String registerWithAddressOfMutable = mut.getAddress();

    if (result.getRegister() != null) {
      code.append(String.format("sw %s 0(%s)\n", result.getRegister(),
                                registerWithAddressOfMutable));
      regAllocator.clear(result.getRegister());
    } else if (result.getAddress() != null) {
      regAllocator.loadIntoRegister(result, code, result.getAddress());
      regAllocator.clear(result.getAddress());
    }

    regAllocator.clear(registerWithAddressOfMutable);

    return MIPSResult.createVoidResult();
  }
}