net.automatalib.visualization.dot.GraphVizBrowserVisualizationProvider Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of automata-dot-visualizer Show documentation

This artifact contains a visualization implementation for the GraphVIZ DOT (http://www.graphviz.org/) tool.

The newest version!

/* Copyright (C) 2013-2023 TU Dortmund
 * This file is part of AutomataLib, http://www.automatalib.net/.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package net.automatalib.visualization.dot;

import java.awt.Desktop;
import java.io.File;
import java.io.IOException;
import java.io.Writer;
import java.util.List;
import java.util.Map;

import javax.swing.JOptionPane;

import net.automatalib.common.util.IOUtil;
import net.automatalib.graph.Graph;
import net.automatalib.serialization.dot.GraphDOT;
import net.automatalib.visualization.VisualizationHelper;
import net.automatalib.visualization.VisualizationProvider;
import org.kohsuke.MetaInfServices;

@MetaInfServices(VisualizationProvider.class)
public class GraphVizBrowserVisualizationProvider implements VisualizationProvider {

    /**
     * the {@link #getId() id} of this {@link VisualizationProvider}.
     */
    public static final String ID = "graphviz-browser";

    private static final int PRIORITY = 10;

    @Override
    public String getId() {
        return ID;
    }

    @Override
    public int getPriority() {
        return PRIORITY;
    }

    @Override
    public boolean checkUsable() {
        return DOT.checkUsable();
    }

    @Override
    public  void visualize(Graph graph,
                                 List> additionalHelpers,
                                 boolean modal,
                                 Map visOptions) {
        try {
            final StringBuilder sb = new StringBuilder();
            GraphDOT.write(graph, sb, additionalHelpers);

            final File imgTmp = File.createTempFile(ID, ".png");
            DOT.runDOT(sb.toString(), "png", imgTmp);

            final File htmlTmp = File.createTempFile(ID, ".html");
            try (Writer w = IOUtil.asBufferedUTF8Writer(htmlTmp)) {
                w.write("");
            }

            Desktop.getDesktop().browse(htmlTmp.toURI());

            if (modal) {
                JOptionPane.showMessageDialog(null, "Click OK to continue ...");
            }
        } catch (IOException ex) {
            JOptionPane.showMessageDialog(null,
                                          "Error rendering graph: " + ex.getMessage(),
                                          "Error rendering graph",
                                          JOptionPane.ERROR_MESSAGE);
        }
    }

}