/*
    * Copyright (c) 2003, The JUNG Authors
    * All rights reserved.
    * 
    * This software is open-source under the BSD license; see either "license.txt"
    * or https://github.com/jrtom/jung/blob/master/LICENSE for a description.
    * 
    */
   package edu.uci.ics.jung.samples;
  
  import java.awt.Color;
  import java.awt.Container;
  import java.awt.Dimension;
  import java.awt.Image;
  import java.awt.Paint;
  import java.awt.geom.Point2D;
  
  import javax.swing.Icon;
  import javax.swing.ImageIcon;
  import javax.swing.JFrame;
  import javax.swing.JLabel;
  
  import com.google.common.base.Functions;
  
  import edu.uci.ics.jung.algorithms.layout.KKLayout;
  import edu.uci.ics.jung.graph.DirectedSparseMultigraph;
  import edu.uci.ics.jung.graph.util.EdgeType;
  import edu.uci.ics.jung.visualization.VisualizationImageServer;
  import edu.uci.ics.jung.visualization.decorators.ToStringLabeller;
  import edu.uci.ics.jung.visualization.renderers.GradientVertexRenderer;
  import edu.uci.ics.jung.visualization.renderers.Renderer;
  import edu.uci.ics.jung.visualization.renderers.BasicVertexLabelRenderer.InsidePositioner;
  
  
  /**
   * Demonstrates VisualizationImageServer.
   * @author Tom Nelson
   * 
   */
  public class VisualizationImageServerDemo {
  
      /**
       * the graph
       */
      DirectedSparseMultigraph<String, Number> graph;
  
      /**
       * the visual component and renderer for the graph
       */
      VisualizationImageServer<String, Number> vv;
      
      /**
       * 
       */
      public VisualizationImageServerDemo() {
          
          // create a simple graph for the demo
          graph = new DirectedSparseMultigraph<String, Number>();
          String[] v = createVertices(10);
          createEdges(v);
  
          vv =  new VisualizationImageServer<String,Number>(new KKLayout<String,Number>(graph), new Dimension(600,600));
  
          vv.getRenderer().setVertexRenderer(
          		new GradientVertexRenderer<String,Number>(
          				Color.white, Color.red, 
          				Color.white, Color.blue,
          				vv.getPickedVertexState(),
          				false));
          vv.getRenderContext().setEdgeDrawPaintTransformer(Functions.<Paint>constant(Color.lightGray));
          vv.getRenderContext().setArrowFillPaintTransformer(Functions.<Paint>constant(Color.lightGray));
          vv.getRenderContext().setArrowDrawPaintTransformer(Functions.<Paint>constant(Color.lightGray));
          
          vv.getRenderContext().setVertexLabelTransformer(new ToStringLabeller());
          vv.getRenderer().getVertexLabelRenderer().setPositioner(new InsidePositioner());
          vv.getRenderer().getVertexLabelRenderer().setPosition(Renderer.VertexLabel.Position.AUTO);
  
          // create a frome to hold the graph
          final JFrame frame = new JFrame();
          Container content = frame.getContentPane();
  
          frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
          
          Image im = vv.getImage(new Point2D.Double(300,300), new Dimension(600,600));
          Icon icon = new ImageIcon(im);
          JLabel label = new JLabel(icon);
          content.add(label);
          frame.pack();
          frame.setVisible(true);
      }
      
      /**
       * create some vertices
       * @param count how many to create
       * @return the Vertices in an array
       */
      private String[] createVertices(int count) {
          String[] v = new String[count];
          for (int i = 0; i < count; i++) {
         	v[i] = "V"+i;
             graph.addVertex(v[i]);
         }
         return v;
     }
 
     /**
      * create edges for this demo graph
      * @param v an array of Vertices to connect
      */
     void createEdges(String[] v) {
         graph.addEdge(new Double(Math.random()), v[0], v[1], EdgeType.DIRECTED);
         graph.addEdge(new Double(Math.random()), v[0], v[3], EdgeType.DIRECTED);
         graph.addEdge(new Double(Math.random()), v[0], v[4], EdgeType.DIRECTED);
         graph.addEdge(new Double(Math.random()), v[4], v[5], EdgeType.DIRECTED);
         graph.addEdge(new Double(Math.random()), v[3], v[5], EdgeType.DIRECTED);
         graph.addEdge(new Double(Math.random()), v[1], v[2], EdgeType.DIRECTED);
         graph.addEdge(new Double(Math.random()), v[1], v[4], EdgeType.DIRECTED);
         graph.addEdge(new Double(Math.random()), v[8], v[2], EdgeType.DIRECTED);
         graph.addEdge(new Double(Math.random()), v[3], v[8], EdgeType.DIRECTED);
         graph.addEdge(new Double(Math.random()), v[6], v[7], EdgeType.DIRECTED);
         graph.addEdge(new Double(Math.random()), v[7], v[5], EdgeType.DIRECTED);
         graph.addEdge(new Double(Math.random()), v[0], v[9], EdgeType.DIRECTED);
         graph.addEdge(new Double(Math.random()), v[9], v[8], EdgeType.DIRECTED);
         graph.addEdge(new Double(Math.random()), v[7], v[6], EdgeType.DIRECTED);
         graph.addEdge(new Double(Math.random()), v[6], v[5], EdgeType.DIRECTED);
         graph.addEdge(new Double(Math.random()), v[4], v[2], EdgeType.DIRECTED);
         graph.addEdge(new Double(Math.random()), v[5], v[4], EdgeType.DIRECTED);
     }
 
     public static void main(String[] args) 
     {
         new VisualizationImageServerDemo();
         
     }
 }