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 void addNode(CFGNode node) {
        this.nodes.add(node);
    }

    protected void addEdge(CFGNode source, CFGNode dest) {
        MinimalLogger.info(String.format("Edge %s -> %s",
                                         source.getInstruction().sourcePos.line,
                                         dest.getInstruction().sourcePos.line));
        source.addDest(dest);
        dest.addSource(source);
    }

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

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

}