/*
    * Copyright (c) 2005, 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.
    * Created on Mar 8, 2005
   *
   */
  package edu.uci.ics.jung.visualization.control;
  
  import java.awt.Cursor;
  import java.awt.event.InputEvent;
  import java.awt.event.MouseEvent;
  import java.awt.event.MouseListener;
  import java.awt.geom.Point2D;
  
  import javax.swing.JOptionPane;
  
  import com.google.common.base.Function;
  
  import edu.uci.ics.jung.algorithms.layout.GraphElementAccessor;
  import edu.uci.ics.jung.algorithms.layout.Layout;
  import edu.uci.ics.jung.algorithms.util.MapSettableTransformer;
  import edu.uci.ics.jung.visualization.Layer;
  import edu.uci.ics.jung.visualization.VisualizationViewer;
  
  /** 
   * 
   * 
   * @author Tom Nelson
   */
  public class LabelEditingGraphMousePlugin<V, E> extends AbstractGraphMousePlugin
      implements MouseListener {
  
  	/**
  	 * the picked Vertex, if any
  	 */
      protected V vertex;
      
      /**
       * the picked Edge, if any
       */
      protected E edge;
      
      /**
  	 * create an instance with default settings
  	 */
  	public LabelEditingGraphMousePlugin() {
  	    this(InputEvent.BUTTON1_MASK);
  	}
  
  	/**
  	 * create an instance with overrides
  	 * @param selectionModifiers for primary selection
  	 */
      public LabelEditingGraphMousePlugin(int selectionModifiers) {
          super(selectionModifiers);
          this.cursor = Cursor.getPredefinedCursor(Cursor.HAND_CURSOR);
      }
      
  	/**
  	 * For primary modifiers (default, MouseButton1):
  	 * pick a single Vertex or Edge that
       * is under the mouse pointer. If no Vertex or edge is under
       * the pointer, unselect all picked Vertices and edges, and
       * set up to draw a rectangle for multiple selection
       * of contained Vertices.
       * For additional selection (default Shift+MouseButton1):
       * Add to the selection, a single Vertex or Edge that is
       * under the mouse pointer. If a previously picked Vertex
       * or Edge is under the pointer, it is un-picked.
       * If no vertex or Edge is under the pointer, set up
       * to draw a multiple selection rectangle (as above)
       * but do not unpick previously picked elements.
  	 * 
  	 * @param e the event
  	 */
      @SuppressWarnings("unchecked")
      public void mouseClicked(MouseEvent e) {
      	if (e.getModifiers() == modifiers && e.getClickCount() == 2) {
      		VisualizationViewer<V,E> vv = (VisualizationViewer<V, E>)e.getSource();
      		GraphElementAccessor<V,E> pickSupport = vv.getPickSupport();
      		if (pickSupport != null) {
      			Function<? super V,String> vs = vv.getRenderContext().getVertexLabelTransformer();
      			if (vs instanceof MapSettableTransformer) {
      				MapSettableTransformer<? super V, String> mst =
      					(MapSettableTransformer<? super V, String>)vs;
      				Layout<V,E> layout = vv.getGraphLayout();
      				// p is the screen point for the mouse event
      				Point2D p = e.getPoint();
  
      				V vertex = pickSupport.getVertex(layout, p.getX(), p.getY());
      				if(vertex != null) {
      					String newLabel = vs.apply(vertex);
      					newLabel = JOptionPane.showInputDialog("New Vertex Label for "+vertex);
      					if(newLabel != null) {
     						mst.set(vertex, newLabel);
     						vv.repaint();
     					}
     					return;
     				}
     			}
     			Function<? super E,String> es = vv.getRenderContext().getEdgeLabelTransformer();
     			if (es instanceof MapSettableTransformer) {
     				MapSettableTransformer<? super E, String> mst =
     					(MapSettableTransformer<? super E, String>)es;
     				Layout<V,E> layout = vv.getGraphLayout();
     				// p is the screen point for the mouse event
     				Point2D p = e.getPoint();
     				// take away the view transform
     				Point2D ip = vv.getRenderContext().getMultiLayerTransformer().inverseTransform(Layer.VIEW, p);
     				E edge = pickSupport.getEdge(layout, ip.getX(), ip.getY());
     				if(edge != null) {
     					String newLabel = JOptionPane.showInputDialog("New Edge Label for "+edge);
     					if(newLabel != null) {
     						mst.set(edge, newLabel);
     						vv.repaint();
     					}
     					return;
     				}
     			}
     		}
      		e.consume();
      	}
     }
 
     /**
 	 * If the mouse is dragging a rectangle, pick the
 	 * Vertices contained in that rectangle
 	 * 
 	 * clean up settings from mousePressed
 	 */
     public void mouseReleased(MouseEvent e) {
     }
     
     /**
 	 * If the mouse is over a picked vertex, drag all picked
 	 * vertices with the mouse.
 	 * If the mouse is not over a Vertex, draw the rectangle
 	 * to select multiple Vertices
 	 * 
 	 */
 
     public void mousePressed(MouseEvent e) {
     }
 
 	public void mouseEntered(MouseEvent e) {
 	}
 
 	public void mouseExited(MouseEvent e) {
 	}
 
 }