nl.uu.cs.ape.solver.solutionStructure.graphviz.SolutionGraph Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of APE Show documentation

APE is a command line tool and an API for the automated exploration of possible computational pipelines (workflows) from large collections of computational tools.

There is a newer version: 2.3.0

Show newest version

package nl.uu.cs.ape.solver.solutionStructure.graphviz;

import guru.nidi.graphviz.engine.Format;
import guru.nidi.graphviz.engine.Graphviz;
import guru.nidi.graphviz.engine.Renderer;
import guru.nidi.graphviz.model.Graph;
import nl.uu.cs.ape.utils.APEUtils;

import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;

import static guru.nidi.graphviz.model.Factory.graph;

/**
 * The {@code SolutionGraph} class is used to represent the graphical
 * representation of the solution.
 *
 * @author Vedran Kasalica
 */
public class SolutionGraph {

    private Graph graph;

    /**
     * Instantiates a new Solution graph.
     *
     * @param graph the graph
     */
    public SolutionGraph(Graph graph) {
        this.graph = graph;
    }

    /**
     * Get the PNG rendered file, based on the workflow structure.
     *
     * @param format The format in which the graph should be rendered (e.g., PNG,
     *               SVG, etc.).
     *
     * @return The {@link Renderer} object.
     */
    public Renderer getRenderer(Format format) {
        return Graphviz.fromGraph(graph).scale(2).render(format);
    }

    /**
     * Get the PNG file, depicting the workflow structure.
     * 
     * @param format The format in which the graph should be rendered (e.g., PNG,
     *               SVG, etc.)
     * @param debug  true if the debugging mode is ON
     * 
     * @return The {@link BufferedImage} object that correspond to the workflow.
     */
    public BufferedImage getImage(Format format, boolean debug) {
        final Renderer renderer = getRenderer(format);
        if (true) {
            APEUtils.disableErr();
        }
        final BufferedImage image = renderer.toImage();
        if (true) {
            APEUtils.enableErr();
        }
        return image;
    }

    /**
     * Write to a file the PNG version of the graph.
     *
     * @param file   The file that should be written to.
     * @param format The format in which the graph should be rendered (e.g., PNG,
     *               SVG, etc.)
     * @param debug  true if the debugging mode is ON
     * @throws IOException Exception in case of error in file handling.
     */
    public void write2File(File file, Format format, boolean debug) throws IOException {
        final Renderer renderer = getRenderer(format);
        if (true) {
            APEUtils.disableErr();
        }
        renderer.toFile(file);
        if (true) {
            APEUtils.enableErr();
        }
    }

    /**
     * Change title of the graph.
     *
     * @param title The new title of the graph.
     * @return A graph with the changed title.
     */
    public SolutionGraph changeTitle(String title) {
        this.graph = graph(title).with(this.graph);
        return this;
    }
}