Skip to content
CFGGraph.java 709 B
Newer Older
Javier Costa's avatar
Javier Costa committed
package tfm.graphs;
Javier Costa's avatar
Javier Costa committed

import edg.graphlib.Arrow;
Javier Costa's avatar
Javier Costa committed
import tfm.arcs.cfg.ControlFlowArc;
Javier Costa's avatar
Javier Costa committed
import tfm.nodes.CFGNode;
Javier Costa's avatar
Javier Costa committed

Javier Costa's avatar
Javier Costa committed
public abstract class CFGGraph extends Graph<CFGNode> {
Javier Costa's avatar
Javier Costa committed

    public CFGGraph() {
        super();
Javier Costa's avatar
Javier Costa committed
        setRootVertex(new CFGNode(NodeId.getVertexId(), getRootNodeData()));
Javier Costa's avatar
Javier Costa committed
    }

    @Override
Javier Costa's avatar
Javier Costa committed
    public CFGNode addNode(String instruction) {
        CFGNode vertex = new CFGNode(NodeId.getVertexId(), instruction);
Javier Costa's avatar
Javier Costa committed
        this.addVertex(vertex);

        return vertex;
    }

    protected abstract String getRootNodeData();

    @SuppressWarnings("unchecked")
Javier Costa's avatar
Javier Costa committed
    public void addControlFlowEdge(CFGNode from, CFGNode to) {
Javier Costa's avatar
Javier Costa committed
        super.addEdge((Arrow) new ControlFlowArc(from, to));
    }
}