com.googlecode.sarasvati.visual.jung.JungVisualizer.java Source code

Introduction

Here is the source code for com.googlecode.sarasvati.visual.jung.JungVisualizer.java

Source

/*
This file is part of Sarasvati.
    
Sarasvati is free software: you can redistribute it and/or modify
it under the terms of the GNU Lesser General Public License as
published by the Free Software Foundation, either version 3 of the
License, or (at your option) any later version.
    
Sarasvati 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 Lesser General Public License for more details.
    
You should have received a copy of the GNU Lesser General Public
License along with Sarasvati.  If not, see <http://www.gnu.org/licenses/>.
    
Copyright 2008 Paul Lorenz
*/
package com.googlecode.sarasvati.visual.jung;

import java.awt.Color;
import java.awt.Component;
import java.awt.Dimension;
import java.awt.Paint;

import javax.swing.DefaultListCellRenderer;
import javax.swing.DefaultListModel;
import javax.swing.Icon;
import javax.swing.JFrame;
import javax.swing.JList;
import javax.swing.JScrollPane;
import javax.swing.JSplitPane;
import javax.swing.ListCellRenderer;
import javax.swing.ListSelectionModel;
import javax.swing.ScrollPaneConstants;
import javax.swing.event.ListSelectionEvent;
import javax.swing.event.ListSelectionListener;

import org.apache.commons.collections15.Transformer;
import org.hibernate.Session;

import com.googlecode.sarasvati.Arc;
import com.googlecode.sarasvati.Graph;
import com.googlecode.sarasvati.Node;
import com.googlecode.sarasvati.example.db.TestSetup;
import com.googlecode.sarasvati.hib.HibEngine;
import com.googlecode.sarasvati.visual.GraphTree;

import edu.uci.ics.jung.graph.DirectedSparseMultigraph;
import edu.uci.ics.jung.visualization.BasicVisualizationServer;

public class JungVisualizer {
    protected static Graph currentGraph = null;

    @SuppressWarnings("serial")
    public static void main(String[] args) throws Exception {
        TestSetup.init();

        Session session = TestSetup.openSession();
        HibEngine engine = new HibEngine(session);

        JFrame frame = new JFrame("Workflow Visualizer");
        frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
        frame.setMinimumSize(new Dimension(800, 600));

        JSplitPane splitPane = new JSplitPane(JSplitPane.HORIZONTAL_SPLIT);
        frame.getContentPane().add(splitPane);

        DefaultListModel listModel = new DefaultListModel();
        for (Graph g : engine.getRepository().getGraphs()) {
            listModel.addElement(g);
        }

        ListCellRenderer cellRenderer = new DefaultListCellRenderer() {
            @Override
            public Component getListCellRendererComponent(JList list, Object value, int index, boolean isSelected,
                    boolean cellHasFocus) {
                super.getListCellRendererComponent(list, value, index, isSelected, cellHasFocus);

                Graph g = (Graph) value;

                setText(g.getName() + "." + g.getVersion() + "  ");
                return this;
            }
        };

        final JList graphList = new JList(listModel);
        graphList.setSelectionMode(ListSelectionModel.SINGLE_SELECTION);
        graphList.setCellRenderer(cellRenderer);

        JScrollPane listScrollPane = new JScrollPane(graphList);
        listScrollPane.setHorizontalScrollBarPolicy(ScrollPaneConstants.HORIZONTAL_SCROLLBAR_AS_NEEDED);
        listScrollPane.setVerticalScrollBarPolicy(ScrollPaneConstants.VERTICAL_SCROLLBAR_ALWAYS);

        splitPane.add(listScrollPane);

        //TreeLayout<NodeRef, Arc> layout = new TreeLayout<NodeRef, Arc>();

        DirectedSparseMultigraph<Node, Arc> graph = new DirectedSparseMultigraph<Node, Arc>();

        //final SpringLayout2<HibNodeRef, HibArc> layout = new SpringLayout2<HibNodeRef, HibArc>(graph);
        //final KKLayout<HibNodeRef, HibArc> layout = new KKLayout<HibNodeRef, HibArc>(graph);
        final TreeLayout layout = new TreeLayout(graph);
        final BasicVisualizationServer<Node, Arc> vs = new BasicVisualizationServer<Node, Arc>(layout);
        //vs.getRenderContext().setVertexLabelTransformer( new NodeLabeller() );
        //vs.getRenderContext().setEdgeLabelTransformer( new ArcLabeller() );
        vs.getRenderContext().setVertexShapeTransformer(new NodeShapeTransformer());
        vs.getRenderContext().setVertexFillPaintTransformer(new NodeColorTransformer());
        vs.getRenderContext().setLabelOffset(5);
        vs.getRenderContext().setVertexIconTransformer(new Transformer<Node, Icon>() {
            @Override
            public Icon transform(Node node) {
                return "task".equals(node.getType()) ? new TaskIcon(node) : null;
            }
        });

        Transformer<Arc, Paint> edgeColorTrans = new Transformer<Arc, Paint>() {
            private Color darkRed = new Color(128, 0, 0);

            @Override
            public Paint transform(Arc arc) {
                return "reject".equals(arc.getName()) ? darkRed : Color.black;
            }
        };

        vs.getRenderContext().setEdgeDrawPaintTransformer(edgeColorTrans);
        vs.getRenderContext().setArrowDrawPaintTransformer(edgeColorTrans);

        final JScrollPane scrollPane = new JScrollPane(vs);
        scrollPane.setHorizontalScrollBarPolicy(ScrollPaneConstants.HORIZONTAL_SCROLLBAR_AS_NEEDED);
        scrollPane.setVerticalScrollBarPolicy(ScrollPaneConstants.VERTICAL_SCROLLBAR_AS_NEEDED);

        splitPane.add(scrollPane);
        scrollPane.setBackground(Color.white);

        graphList.addListSelectionListener(new ListSelectionListener() {
            @Override
            public void valueChanged(ListSelectionEvent e) {
                if (e.getValueIsAdjusting()) {
                    return;
                }

                final Graph g = (Graph) graphList.getSelectedValue();

                if ((g == null && currentGraph == null) || (g != null && g.equals(currentGraph))) {
                    return;
                }

                currentGraph = g;

                DirectedSparseMultigraph<Node, Arc> jungGraph = new DirectedSparseMultigraph<Node, Arc>();

                for (Node ref : currentGraph.getNodes()) {
                    jungGraph.addVertex(ref);
                }

                for (Arc arc : currentGraph.getArcs()) {
                    jungGraph.addEdge(arc, arc.getStartNode(), arc.getEndNode());
                }

                GraphTree graphTree = new GraphTree(g);
                layout.setGraph(jungGraph);
                layout.setInitializer(new NodeLocationTransformer(graphTree));
                scrollPane.repaint();
            }
        });

        frame.setVisible(true);
    }
}