summaryrefslogtreecommitdiff
path: root/vaporize/library/ControlFlowGraph.java
blob: 5d589fb623edef30f35243b351331daa4d490a25 (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
package vaporize.library;

import cs132.vapor.ast.*;
import misc.*;

import java.util.ArrayList;

public class ControlFlowGraph {

    private ArrayList<CFGNode> nodes;
    private CFGNode start;

    protected ControlFlowGraph() {
        this.nodes = new ArrayList<>();
        this.start = null;
    }

    protected CFGNode getNode(Object a) {
        CFGNode ret = null;
        for (CFGNode n : this.nodes) {
            if (n.equals(a)) {
                ret = n;
                break;
            }
        }

        if (ret == null) {
            String str = (a instanceof Node) ? ((Node) a).sourcePos.toString() :
                a.toString();
            MinimalLogger.severe(String.format("Could not find a node matching %s",
                                               str));
        }
        return ret;
    }

    protected CFGNode getNextNode(CFGNode a) {
        CFGNode ret = null;
        // we return null if the passed node is the last in the list
        for (int i = 0; i < this.nodes.size()-1; ++i) {
            if (this.nodes.get(i).equals(a)) {
                ret = this.nodes.get(i+1);
                break;
            }
        }

        if (ret == null)
            MinimalLogger.severe(String.format("Could not find the next node for %s",
                                               a.toString()));

        return ret;

    }

    protected void addNode(CFGNode node) {
        this.nodes.add(node);
    }

    protected String addEdge(CFGNode source, CFGNode dest) {
        /**
         * Iff the CFGNodes are different, construct an edge between them.
         * All variables reaching the parent also reach the children.
         *
         * Returns a string capable of being manipulated by graphviz.
         */
        String ret = "";
        if (!source.equals(dest)) {
            ret = String.format("%d -> %d",
                                source.getInstruction().sourcePos.line,
                                dest.getInstruction().sourcePos.line);
            MinimalLogger.info(String.format("Edge %s",
                                             ret));

            source.addDest(dest);
            dest.addSource(source);

            ret += ";";
        } else {
            MinimalLogger.info(String.format("Skipping duplicate edge for %d",
                                             source.getInstruction().sourcePos.line));
        }

        MinimalLogger.info(String.format("Spilling variables: %s",
                                         source.getReaching().toString()));
        dest.getReaching().addAll(source.getReaching());

        return ret;
    }

    protected void setStart(CFGNode start) {
        this.start = start;
    }

    protected ArrayList<CFGNode> getNodes() {
        return this.nodes;
    }

}