GraphStreamViewer.java

/**
 * ****************************************************************************
 * Copyright (c) 2010-2016 by Min Cai (min.cai.china@gmail.com).
 * <p>
 * This file is part of the Archimulator multicore architectural simulator.
 * <p>
 * Archimulator is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 * <p>
 * Archimulator is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * <p>
 * You should have received a copy of the GNU General Public License
 * along with Archimulator. If not, see <http://www.gnu.org/licenses/>.
 * ****************************************************************************
 */
package archimulator.util.ai.aco;

import org.graphstream.algorithm.generator.Generator;
import org.graphstream.algorithm.generator.PetersenGraphGenerator;
import org.graphstream.graph.Edge;
import org.graphstream.graph.Graph;
import org.graphstream.graph.Node;
import org.graphstream.graph.implementations.MultiGraph;
import org.graphstream.graph.implementations.SingleGraph;

/**
 * Graph stream viewer.
 *
 * @author Min Cai
 */
public class GraphStreamViewer {
    private ACOHelper acoHelper;
    private Graph graph;

    static {
        System.setProperty("org.graphstream.ui.renderer", "org.graphstream.ui.j2dviewer.J2DGraphRenderer");
    }

    /**
     * Create a graph stream viewer.
     *
     * @param acoHelper the ACO helper
     */
    public GraphStreamViewer(ACOHelper acoHelper) {
        this.acoHelper = acoHelper;

        this.graph = new SingleGraph("ACO for TSP");
//        this.graph.addAttribute("ui.quality");
//        this.graph.addAttribute("ui.antialias");

        acoHelper.getNodes().forEach(node -> {
            Node graphNode = graph.addNode(node.getName());
            graphNode.addAttribute("xy", node.getX(), node.getY());
            graphNode.addAttribute("label", String.format("%s", node.getName()));
        });

        acoHelper.getEdges().forEach(edge -> {
            Edge graphEdge = graph.addEdge(
                    edge.getNodeFrom().getName() + "-" + edge.getNodeTo().getName(),
                    edge.getNodeFrom().getName(),
                    edge.getNodeTo().getName());
//            graphEdge.addAttribute("label", String.format("%.4f", edge.getPheromone()));
//            graphEdge.addAttribute("ui.style", "shape: angle;");
//            graphEdge.addAttribute("ui.style", "stroke-width: 10;");
//            graphEdge.addAttribute("ui.style", "fill-color: white;");
        });

        graph.addAttribute("ui.stylesheet", "url('src/main/java/archimulator/util/ai/aco/stylesheet.css')");

        this.graph.display(false);
    }

    public static void main(String[] args) {
        Graph graph = new MultiGraph("random walk");
        Generator gen = new PetersenGraphGenerator();

        gen.addSink(graph);
        gen.begin();
        for (int i = 0; i < 400; i++) {
            gen.nextEvents();
        }
        gen.end();

        graph.addAttribute("ui.quality");
        graph.addAttribute("ui.antialias");
        graph.display(false);
    }

    /**
     * Get the ACO helper.
     *
     * @return the ACO helper
     */
    public ACOHelper getAcoHelper() {
        return acoHelper;
    }

    /**
     * Get the graph.
     *
     * @return the graph
     */
    public Graph getGraph() {
        return graph;
    }
}