/*
    * 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.BorderLayout;
  import java.awt.Color;
  import java.awt.Container;
  import java.awt.Dimension;
  import java.awt.GridLayout;
  import java.awt.event.ActionEvent;
  import java.awt.event.ActionListener;
  
  import javax.swing.BorderFactory;
  import javax.swing.JApplet;
  import javax.swing.JButton;
  import javax.swing.JFrame;
  import javax.swing.JPanel;
  
  import edu.uci.ics.jung.algorithms.layout.FRLayout;
  import edu.uci.ics.jung.algorithms.layout.ISOMLayout;
  import edu.uci.ics.jung.algorithms.layout.Layout;
  import edu.uci.ics.jung.graph.Graph;
  import edu.uci.ics.jung.graph.util.TestGraphs;
  import edu.uci.ics.jung.visualization.DefaultVisualizationModel;
  import edu.uci.ics.jung.visualization.GraphZoomScrollPane;
  import edu.uci.ics.jung.visualization.VisualizationModel;
  import edu.uci.ics.jung.visualization.VisualizationViewer;
  import edu.uci.ics.jung.visualization.control.CrossoverScalingControl;
  import edu.uci.ics.jung.visualization.control.DefaultModalGraphMouse;
  import edu.uci.ics.jung.visualization.control.ScalingControl;
  import edu.uci.ics.jung.visualization.decorators.PickableEdgePaintTransformer;
  import edu.uci.ics.jung.visualization.decorators.PickableVertexPaintTransformer;
  import edu.uci.ics.jung.visualization.decorators.ToStringLabeller;
  import edu.uci.ics.jung.visualization.picking.MultiPickedState;
  import edu.uci.ics.jung.visualization.picking.PickedState;
  import edu.uci.ics.jung.visualization.transform.MutableTransformer;
  
  /**
   * Demonstrates a single graph with 2 layouts in 2 views.
   * They share picking, transforms, and a pluggable renderer
   * 
   * @author Tom Nelson
   * 
   */
  @SuppressWarnings("serial")
  public class TwoModelDemo extends JApplet {
  
       /**
       * the graph
       */
      Graph<String,Number> graph;
  
      /**
       * the visual components and renderers for the graph
       */
      VisualizationViewer<String,Number> vv1;
      VisualizationViewer<String,Number> vv2;
      
      /**
       * the normal Function
       */
      MutableTransformer layoutTransformer;
      
      Dimension preferredSize = new Dimension(300,300);
      
      /**
       * create an instance of a simple graph in two views with controls to
       * demo the zoom features.
       * 
       */
      public TwoModelDemo() {
          
          // create a simple graph for the demo
          // both models will share one graph
          graph = TestGraphs.getOneComponentGraph();
          
          // create two layouts for the one graph, one layout for each model
          Layout<String,Number> layout1 = new FRLayout<String,Number>(graph);
          Layout<String,Number> layout2 = new ISOMLayout<String,Number>(graph);
  
          // create the two models, each with a different layout
          VisualizationModel<String,Number> vm1 =
              new DefaultVisualizationModel<String,Number>(layout1, preferredSize);
          VisualizationModel<String,Number> vm2 = 
              new DefaultVisualizationModel<String,Number>(layout2, preferredSize);
  
          // create the two views, one for each model
          // they share the same renderer
          vv1 = new VisualizationViewer<String,Number>(vm1, preferredSize);
          vv2 = new VisualizationViewer<String,Number>(vm2, preferredSize);
          vv1.setRenderContext(vv2.getRenderContext());
          
          // share the model Function between the two models
 //        layoutTransformer = vv1.getLayoutTransformer();
 //        vv2.setLayoutTransformer(layoutTransformer);
 //        
 //        // share the view Function between the two models
 //        vv2.setViewTransformer(vv1.getViewTransformer());
         
         vv2.getRenderContext().setMultiLayerTransformer(vv1.getRenderContext().getMultiLayerTransformer());
         vv2.getRenderContext().getMultiLayerTransformer().addChangeListener(vv1);
 
         vv1.setBackground(Color.white);
         vv2.setBackground(Color.white);
         
         // share one PickedState between the two views
         PickedState<String> ps = new MultiPickedState<String>();
         vv1.setPickedVertexState(ps);
         vv2.setPickedVertexState(ps);
         PickedState<Number> pes = new MultiPickedState<Number>();
         vv1.setPickedEdgeState(pes);
         vv2.setPickedEdgeState(pes);
         
         // set an edge paint function that will show picking for edges
         vv1.getRenderContext().setEdgeDrawPaintTransformer(new PickableEdgePaintTransformer<Number>(vv1.getPickedEdgeState(), Color.black, Color.red));
         vv1.getRenderContext().setVertexFillPaintTransformer(new PickableVertexPaintTransformer<String>(vv1.getPickedVertexState(),
                 Color.red, Color.yellow));
         // add default listeners for ToolTips
         vv1.setVertexToolTipTransformer(new ToStringLabeller());
         vv2.setVertexToolTipTransformer(new ToStringLabeller());
         
         Container content = getContentPane();
         JPanel panel = new JPanel(new GridLayout(1,0));
         panel.add(new GraphZoomScrollPane(vv1));
         panel.add(new GraphZoomScrollPane(vv2));
 
         content.add(panel);
         
         // create a GraphMouse for each view
         final DefaultModalGraphMouse<String, Number> gm1
         	= new DefaultModalGraphMouse<String, Number>();
 
         DefaultModalGraphMouse<String, Number> gm2
         	= new DefaultModalGraphMouse<String, Number>();
 
         vv1.setGraphMouse(gm1);
         vv2.setGraphMouse(gm2);
 
         // create zoom buttons for scaling the Function that is
         // shared between the two models.
         final ScalingControl scaler = new CrossoverScalingControl();
 
         JButton plus = new JButton("+");
         plus.addActionListener(new ActionListener() {
             public void actionPerformed(ActionEvent e) {
                 scaler.scale(vv1, 1.1f, vv1.getCenter());
             }
         });
         JButton minus = new JButton("-");
         minus.addActionListener(new ActionListener() {
             public void actionPerformed(ActionEvent e) {
                 scaler.scale(vv1, 1/1.1f, vv1.getCenter());
             }
         });
         
         JPanel zoomPanel = new JPanel(new GridLayout(1,2));
         zoomPanel.setBorder(BorderFactory.createTitledBorder("Zoom"));
         
         JPanel modePanel = new JPanel();
         modePanel.setBorder(BorderFactory.createTitledBorder("Mouse Mode"));
         gm1.getModeComboBox().addItemListener(gm2.getModeListener());
         modePanel.add(gm1.getModeComboBox());
 
         JPanel controls = new JPanel();
         zoomPanel.add(plus);
         zoomPanel.add(minus);
         controls.add(zoomPanel);
         controls.add(modePanel);
         content.add(controls, BorderLayout.SOUTH);
     }
 
     public static void main(String[] args) {
         JFrame f = new JFrame();
         f.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
         f.getContentPane().add(new TwoModelDemo());
         f.pack();
         f.setVisible(true);
     }
 }