Examples with Colour org.mapleir.dot4j.attr.builtin.Colour used on opensource projects

Example 1 with Colour

use of org.mapleir.dot4j.attr.builtin.Colour in project maple-ir by LLVM-but-worse.

the class CFGExporterUtils method makeDotGraph.

public static DotGraph makeDotGraph(ControlFlowGraph cfg, IPropertyDictionary properties) {
    boolean showStmts = PropertyHelper.isSet(properties, OPT_STMTS);
    boolean labelEdges = PropertyHelper.isSet(properties, OPT_EDGES);
    boolean simplifyEdges = PropertyHelper.isSet(properties, OPT_SIMPLE_EDGES);
    boolean hideHandlerEdges = PropertyHelper.isSet(properties, OPT_HIDE_HANDLER_EDGES);
    DotGraph sourceGraph = Factory.graph().getGraphAttr().with(Rank.SOURCE);
    DotGraph sinkGraph = Factory.graph().getGraphAttr().with(Rank.SINK);
    DotGraph g = GraphUtils.makeDotSkeleton(cfg, (block, node) -> {
        StringBuilder sb = new StringBuilder();
        sb.append("<table>");
        {
            sb.append("<tr><td><font point-size=\"12\">").append(block.getDisplayName()).append("</font></td></tr>");
        }
        {
            sb.append("<tr><td>");
            if (showStmts) {
                StringBuilder stmtsStr = new StringBuilder();
                outputBlock(block, stmtsStr);
                sb.append(stmtsStr.toString());
            }
            sb.append("</td></tr>");
        }
        sb.append("</table>");
        Attrs colour;
        DotGraph rankGraph;
        if (cfg.getEntries().contains(block)) {
            colour = Colour.RED.fill();
            rankGraph = sourceGraph;
        } else if (cfg.getEdges(block).isEmpty()) {
            colour = Colour.GREEN.fill();
            rankGraph = sinkGraph;
        } else {
            colour = Attrs.attrs();
            rankGraph = null;
        }
        if (rankGraph != null) {
            Context.use(rankGraph, (ctx) -> {
                return rankGraph.addSource(Factory.node(node.getName()));
            });
        }
        node.with(Shape.BOX, Style.FILLED, colour, ComplexLabel.html(sb.toString()).justify(Justification.LEFT));
        return true;
    }, (e, dotE) -> {
        if (hideHandlerEdges && e instanceof TryCatchEdge) {
            // dotE.with(Style.INVIS);
            return false;
        }
        if (labelEdges) {
            dotE.with(ComplexLabel.of(simplifyEdges ? e.getClass().getSimpleName().replace("Edge", "") : e.toGraphString()));
        }
        return true;
    }).setDirected(true);
    g.addSource(sourceGraph);
    g.addSource(sinkGraph);
    return g;
}