Class DotDumper

src/main/java/pascal/taie/util/graph/DotDumper.java:40–187

Configurable dot dumper. @param type of graph nodes

 * @param <N> type of graph nodes
 */
public class DotDumper<N> {

    private static final Logger logger = LogManager.getLogger(DotDumper.class);

    private static final String INDENT = "  ";

    /**
     * The output stream for dumping dot graph.
     */
    private PrintStream out;

    /**
     * The function that converts a node to its string representation.
     */
    private Function<N, String> nodeToString = Objects::toString;

    /**
     * Global node attributes.
     */
    private DotAttributes globalNodeAttrs = DotAttributes.of();

    /**
     * The labeler for nodes.
     */
    private Function<N, String> nodeLabeler = node -> null;

    /**
     * The node attributes.
     */
    private Function<N, DotAttributes> nodeAttributer = node -> null;

    /**
     * The labeler for edges.
     */
    private Function<Edge<N>, String> edgeLabeler = edge -> null;

    /**
     * Global edge attributes.
     */
    private DotAttributes globalEdgeAttrs = DotAttributes.of();

    /**
     * The function that maps an edge to its attributes.
     */
    private Function<Edge<N>, DotAttributes> edgeAttributer = edge -> null;

    public DotDumper<N> setNodeToString(Function<N, String> nodeToString) {
        this.nodeToString = nodeToString;
        return this;
    }

    public DotDumper<N> setGlobalNodeAttributes(DotAttributes attrs) {
        globalNodeAttrs = attrs;
        return this;
    }

    public DotDumper<N> setNodeLabeler(Function<N, String> nodeLabeler) {
        this.nodeLabeler = nodeLabeler;