taverna-workbench-graph-model/src/main/java/net/sf/taverna/t2/workbench/models/graph/DefaultGraphEventManager.java - incubator-taverna-workbench - Git at Google

 /*******************************************************************************
  * Copyright (C) 2007 The University of Manchester
  *
  *  Modifications to the initial code base are copyright of their
  *  respective authors, or their employers as appropriate.
  *
  *  This program 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 2.1 of
  *  the License, or (at your option) any later version.
  *
  *  This program 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 this program; if not, write to the Free Software
  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
  ******************************************************************************/
 package net.sf.taverna.t2.workbench.models.graph;

 import static javax.swing.SwingUtilities.convertPointFromScreen;
 import static javax.swing.SwingUtilities.invokeLater;
 import static net.sf.taverna.t2.workbench.models.graph.GraphController.PortStyle.ALL;
 import static net.sf.taverna.t2.workbench.models.graph.GraphController.PortStyle.NONE;

 import java.awt.Component;
 import java.awt.Point;
 import java.awt.event.ActionEvent;
 import java.net.URI;
 import java.util.List;

 import javax.swing.AbstractAction;
 import javax.swing.JMenu;
 import javax.swing.JMenuItem;
 import javax.swing.JPopupMenu;

 import net.sf.taverna.t2.ui.menu.MenuManager;
 import uk.org.taverna.scufl2.api.activity.Activity;
 import uk.org.taverna.scufl2.api.common.Scufl2Tools;
 import uk.org.taverna.scufl2.api.core.Processor;
 import uk.org.taverna.scufl2.api.profiles.ProcessorBinding;

 /**
  * Manager for handling UI events on GraphElements.
  *
  * @author David Withers
  */
 public class DefaultGraphEventManager implements GraphEventManager {
 	private static final URI NESTED_WORKFLOW_URI = URI
 			.create("http://ns.taverna.org.uk/2010/activity/nested-workflow");

 	private GraphController graphController;
 	private Component component;
 	private JPopupMenu menu;
 	private MenuManager menuManager;

 	private Scufl2Tools scufl2Tools = new Scufl2Tools();

 	/**
 	 * Constructs a new instance of GraphEventManager.
 	 *
 	 * @param graphController
 	 * @param component
 	 *            component to use when displaying popup menus
 	 */
 	public DefaultGraphEventManager(GraphController graphController, Component component,
 			MenuManager menuManager) {
 		this.graphController = graphController;
 		this.component = component;
 		this.menuManager = menuManager;
 	}

 	@Override
 	public void mouseClicked(final GraphElement graphElement, short button, boolean altKey,
 			boolean ctrlKey, boolean metaKey, final int x, final int y, int screenX, int screenY) {
 		Object dataflowObject = graphElement.getWorkflowBean();

 		// For both left and right click - add to selection model
 		if (graphController.getDataflowSelectionModel() != null)
 			graphController.getDataflowSelectionModel().addSelection(dataflowObject);

 		if ((button != 2) && !ctrlKey)return;

 		// If this was a right click - show a pop-up as well
 		if (dataflowObject == null)
 			menu = menuManager.createContextMenu(graphController.getWorkflow(),
 					graphController.getWorkflow(), component);
 		else {
 			menu = menuManager.createContextMenu(graphController.getWorkflow(),
 					dataflowObject, component);
 			if (dataflowObject instanceof Processor) {
 				final Processor processor = (Processor) dataflowObject;
 				ProcessorBinding processorBinding = scufl2Tools
 						.processorBindingForProcessor(processor,
 								graphController.getProfile());
 				final Activity activity = processorBinding.getBoundActivity();
 				if (menu == null)
 					menu = new JPopupMenu();
 				if (graphElement instanceof GraphNode) {
 					defineMenuForGraphElement(graphElement, x, y, processor,
 							activity);
 				} else if (graphElement instanceof Graph) {
 					defineMenuForGraphBackground(activity);
 				}
 			}
 		}

 		if (menu != null) {
 			final Point p = new Point(screenX, screenY);
 			convertPointFromScreen(p, component);
 			invokeLater(new Runnable() {
 				@Override
 				public void run() {
 					menu.show(component, p.x, p.y);
 				}
 			});
 		}
 	}

 	@SuppressWarnings("serial")
 	private void defineMenuForGraphBackground(final Activity activity) {
 		if (activity.getType().equals(NESTED_WORKFLOW_URI)) {
 			menu.addSeparator();
 			menu.add(new JMenuItem(new AbstractAction("Hide nested workflow") {
 				@Override
 				public void actionPerformed(ActionEvent ev) {
 					graphController.setExpandNestedDataflow(activity, false);
 					graphController.redraw();
 				}
 			}));
 		}
 	}

 	@SuppressWarnings("serial")
 	private void defineMenuForGraphElement(final GraphElement graphElement,
 			final int x, final int y, final Processor processor,
 			final Activity activity) {
 		if (graphController.getPortStyle(processor).equals(NONE)) {
 			menu.addSeparator();
 			menu.add(new JMenuItem(new AbstractAction("Show ports") {
 				@Override
 				public void actionPerformed(ActionEvent ev) {
 					graphController.setPortStyle(processor, ALL);
 					graphController.redraw();
 				}
 			}));
 		} else if (graphController.getPortStyle(processor).equals(ALL)) {
 			menu.addSeparator();
 			menu.add(new JMenuItem(new AbstractAction("Hide ports") {
 				@Override
 				public void actionPerformed(ActionEvent arg0) {
 					graphController.setPortStyle(processor, NONE);
 					graphController.redraw();
 				}
 			}));
 		}

 		if (activity.getType().equals(NESTED_WORKFLOW_URI)) {
 			menu.addSeparator();
 			menu.add(new JMenuItem(new AbstractAction("Show nested workflow") {
 				@Override
 				public void actionPerformed(ActionEvent arg0) {
 					graphController.setExpandNestedDataflow(activity, true);
 					graphController.redraw();
 				}
 			}));
 		}

 		menu.addSeparator();

 		GraphNode graphNode = (GraphNode) graphElement;

 		List<GraphNode> sourceNodes = graphNode.getSourceNodes();
 		if (sourceNodes.size() == 1) {
 			final GraphNode sourceNode = sourceNodes.get(0);
 			if (sourceNode.getLabel() != null) {
 				menu.add(new JMenuItem(new AbstractAction("Link from output '"
 						+ sourceNode.getLabel() + "'") {
 					@Override
 					public void actionPerformed(ActionEvent arg0) {
 						graphController.startEdgeCreation(sourceNode,
 								new Point(x, y));
 					}
 				}));
 			}
 		} else if (sourceNodes.size() > 0) {
 			JMenu linkMenu = new JMenu("Link from output...");
 			menu.add(linkMenu);
 			for (final GraphNode sourceNode : sourceNodes) {
 				linkMenu.add(new JMenuItem(new AbstractAction(sourceNode
 						.getLabel()) {
 					@Override
 					public void actionPerformed(ActionEvent arg0) {
 						graphController.startEdgeCreation(sourceNode,
 								new Point(x, y));
 					}
 				}));
 			}
 		}

 		List<GraphNode> sinkNodes = graphNode.getSinkNodes();
 		if (sinkNodes.size() == 1) {
 			final GraphNode sinkNode = sinkNodes.get(0);
 			if (sinkNode.getLabel() != null) {
 				menu.add(new JMenuItem(new AbstractAction("Link to input '"
 						+ sinkNode.getLabel() + "'") {
 					@Override
 					public void actionPerformed(ActionEvent arg0) {
 						graphController.startEdgeCreation(sinkNode, new Point(
 								x, y));
 					}
 				}));
 			}
 		} else if (sinkNodes.size() > 0) {
 			JMenu linkMenu = new JMenu("Link to input...");
 			menu.add(linkMenu);
 			for (final GraphNode sinkNode : sinkNodes) {
 				linkMenu.add(new JMenuItem(new AbstractAction(sinkNode
 						.getLabel()) {
 					@Override
 					public void actionPerformed(ActionEvent arg0) {
 						graphController.startEdgeCreation(sinkNode, new Point(
 								x, y));
 					}
 				}));
 			}
 		}
 	}

 	@Override
 	public void mouseDown(GraphElement graphElement, short button,
 			boolean altKey, boolean ctrlKey, boolean metaKey, int x, int y,
 			int screenX, int screenY) {
 		if (button == 0)
 			graphController.startEdgeCreation(graphElement, new Point(x, y));
 	}

 	@Override
 	public void mouseUp(GraphElement graphElement, short button,
 			boolean altKey, boolean ctrlKey, boolean metaKey, final int x,
 			final int y, int screenX, int screenY) {
 		if (button == 0)
 			graphController.stopEdgeCreation(graphElement, new Point(screenX,
 					screenY));
 	}

 	@Override
 	public void mouseMoved(GraphElement graphElement, short button,
 			boolean altKey, boolean ctrlKey, boolean metaKey, int x, int y,
 			int screenX, int screenY) {
 		graphController.moveEdgeCreationTarget(graphElement, new Point(x, y));
 	}

 	@Override
 	public void mouseOver(GraphElement graphElement, short button,
 			boolean altKey, boolean ctrlKey, boolean metaKey, int x, int y,
 			int screenX, int screenY) {
 		if (graphElement.getWorkflowBean() != null)
 			graphElement.setActive(true);
 	}

 	@Override
 	public void mouseOut(GraphElement graphElement, short button,
 			boolean altKey, boolean ctrlKey, boolean metaKey, int x, int y,
 			int screenX, int screenY) {
 		if (graphElement.getWorkflowBean() != null)
 			graphElement.setActive(false);
 	}
 }
	/*******************************************************************************
	* Copyright (C) 2007 The University of Manchester
	*
	* Modifications to the initial code base are copyright of their
	* respective authors, or their employers as appropriate.
	*
	* This program 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 2.1 of
	* the License, or (at your option) any later version.
	*
	* This program 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 this program; if not, write to the Free Software
	* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
	******************************************************************************/
	package net.sf.taverna.t2.workbench.models.graph;

	import static javax.swing.SwingUtilities.convertPointFromScreen;
	import static javax.swing.SwingUtilities.invokeLater;
	import static net.sf.taverna.t2.workbench.models.graph.GraphController.PortStyle.ALL;
	import static net.sf.taverna.t2.workbench.models.graph.GraphController.PortStyle.NONE;

	import java.awt.Component;
	import java.awt.Point;
	import java.awt.event.ActionEvent;
	import java.net.URI;
	import java.util.List;

	import javax.swing.AbstractAction;
	import javax.swing.JMenu;
	import javax.swing.JMenuItem;
	import javax.swing.JPopupMenu;

	import net.sf.taverna.t2.ui.menu.MenuManager;
	import uk.org.taverna.scufl2.api.activity.Activity;
	import uk.org.taverna.scufl2.api.common.Scufl2Tools;
	import uk.org.taverna.scufl2.api.core.Processor;
	import uk.org.taverna.scufl2.api.profiles.ProcessorBinding;

	/**
	* Manager for handling UI events on GraphElements.
	*
	* @author David Withers
	*/
	public class DefaultGraphEventManager implements GraphEventManager {
	private static final URI NESTED_WORKFLOW_URI = URI
	.create("http://ns.taverna.org.uk/2010/activity/nested-workflow");

	private GraphController graphController;
	private Component component;
	private JPopupMenu menu;
	private MenuManager menuManager;

	private Scufl2Tools scufl2Tools = new Scufl2Tools();

	/**
	* Constructs a new instance of GraphEventManager.
	*
	* @param graphController
	* @param component
	* component to use when displaying popup menus
	*/
	public DefaultGraphEventManager(GraphController graphController, Component component,
	MenuManager menuManager) {
	this.graphController = graphController;
	this.component = component;
	this.menuManager = menuManager;
	}

	@Override
	public void mouseClicked(final GraphElement graphElement, short button, boolean altKey,
	boolean ctrlKey, boolean metaKey, final int x, final int y, int screenX, int screenY) {
	Object dataflowObject = graphElement.getWorkflowBean();

	// For both left and right click - add to selection model
	if (graphController.getDataflowSelectionModel() != null)
	graphController.getDataflowSelectionModel().addSelection(dataflowObject);

	if ((button != 2) && !ctrlKey)return;

	// If this was a right click - show a pop-up as well
	if (dataflowObject == null)
	menu = menuManager.createContextMenu(graphController.getWorkflow(),
	graphController.getWorkflow(), component);
	else {
	menu = menuManager.createContextMenu(graphController.getWorkflow(),
	dataflowObject, component);
	if (dataflowObject instanceof Processor) {
	final Processor processor = (Processor) dataflowObject;
	ProcessorBinding processorBinding = scufl2Tools
	.processorBindingForProcessor(processor,
	graphController.getProfile());
	final Activity activity = processorBinding.getBoundActivity();
	if (menu == null)
	menu = new JPopupMenu();
	if (graphElement instanceof GraphNode) {
	defineMenuForGraphElement(graphElement, x, y, processor,
	activity);
	} else if (graphElement instanceof Graph) {
	defineMenuForGraphBackground(activity);
	}
	}
	}

	if (menu != null) {
	final Point p = new Point(screenX, screenY);
	convertPointFromScreen(p, component);
	invokeLater(new Runnable() {
	@Override
	public void run() {
	menu.show(component, p.x, p.y);
	}
	});
	}
	}

	@SuppressWarnings("serial")
	private void defineMenuForGraphBackground(final Activity activity) {
	if (activity.getType().equals(NESTED_WORKFLOW_URI)) {
	menu.addSeparator();
	menu.add(new JMenuItem(new AbstractAction("Hide nested workflow") {
	@Override
	public void actionPerformed(ActionEvent ev) {
	graphController.setExpandNestedDataflow(activity, false);
	graphController.redraw();
	}
	}));
	}
	}

	@SuppressWarnings("serial")
	private void defineMenuForGraphElement(final GraphElement graphElement,
	final int x, final int y, final Processor processor,
	final Activity activity) {
	if (graphController.getPortStyle(processor).equals(NONE)) {
	menu.addSeparator();
	menu.add(new JMenuItem(new AbstractAction("Show ports") {
	@Override
	public void actionPerformed(ActionEvent ev) {
	graphController.setPortStyle(processor, ALL);
	graphController.redraw();
	}
	}));
	} else if (graphController.getPortStyle(processor).equals(ALL)) {
	menu.addSeparator();
	menu.add(new JMenuItem(new AbstractAction("Hide ports") {
	@Override
	public void actionPerformed(ActionEvent arg0) {
	graphController.setPortStyle(processor, NONE);
	graphController.redraw();
	}
	}));
	}

	if (activity.getType().equals(NESTED_WORKFLOW_URI)) {
	menu.addSeparator();
	menu.add(new JMenuItem(new AbstractAction("Show nested workflow") {
	@Override
	public void actionPerformed(ActionEvent arg0) {
	graphController.setExpandNestedDataflow(activity, true);
	graphController.redraw();
	}
	}));
	}

	menu.addSeparator();

	GraphNode graphNode = (GraphNode) graphElement;

	List<GraphNode> sourceNodes = graphNode.getSourceNodes();
	if (sourceNodes.size() == 1) {
	final GraphNode sourceNode = sourceNodes.get(0);
	if (sourceNode.getLabel() != null) {
	menu.add(new JMenuItem(new AbstractAction("Link from output '"
	+ sourceNode.getLabel() + "'") {
	@Override
	public void actionPerformed(ActionEvent arg0) {
	graphController.startEdgeCreation(sourceNode,
	new Point(x, y));
	}
	}));
	}
	} else if (sourceNodes.size() > 0) {
	JMenu linkMenu = new JMenu("Link from output...");
	menu.add(linkMenu);
	for (final GraphNode sourceNode : sourceNodes) {
	linkMenu.add(new JMenuItem(new AbstractAction(sourceNode
	.getLabel()) {
	@Override
	public void actionPerformed(ActionEvent arg0) {
	graphController.startEdgeCreation(sourceNode,
	new Point(x, y));
	}
	}));
	}
	}

	List<GraphNode> sinkNodes = graphNode.getSinkNodes();
	if (sinkNodes.size() == 1) {
	final GraphNode sinkNode = sinkNodes.get(0);
	if (sinkNode.getLabel() != null) {
	menu.add(new JMenuItem(new AbstractAction("Link to input '"
	+ sinkNode.getLabel() + "'") {
	@Override
	public void actionPerformed(ActionEvent arg0) {
	graphController.startEdgeCreation(sinkNode, new Point(
	x, y));
	}
	}));
	}
	} else if (sinkNodes.size() > 0) {
	JMenu linkMenu = new JMenu("Link to input...");
	menu.add(linkMenu);
	for (final GraphNode sinkNode : sinkNodes) {
	linkMenu.add(new JMenuItem(new AbstractAction(sinkNode
	.getLabel()) {
	@Override
	public void actionPerformed(ActionEvent arg0) {
	graphController.startEdgeCreation(sinkNode, new Point(
	x, y));
	}
	}));
	}
	}
	}

	@Override
	public void mouseDown(GraphElement graphElement, short button,
	boolean altKey, boolean ctrlKey, boolean metaKey, int x, int y,
	int screenX, int screenY) {
	if (button == 0)
	graphController.startEdgeCreation(graphElement, new Point(x, y));
	}

	@Override
	public void mouseUp(GraphElement graphElement, short button,
	boolean altKey, boolean ctrlKey, boolean metaKey, final int x,
	final int y, int screenX, int screenY) {
	if (button == 0)
	graphController.stopEdgeCreation(graphElement, new Point(screenX,
	screenY));
	}

	@Override
	public void mouseMoved(GraphElement graphElement, short button,
	boolean altKey, boolean ctrlKey, boolean metaKey, int x, int y,
	int screenX, int screenY) {
	graphController.moveEdgeCreationTarget(graphElement, new Point(x, y));
	}

	@Override
	public void mouseOver(GraphElement graphElement, short button,
	boolean altKey, boolean ctrlKey, boolean metaKey, int x, int y,
	int screenX, int screenY) {
	if (graphElement.getWorkflowBean() != null)
	graphElement.setActive(true);
	}

	@Override
	public void mouseOut(GraphElement graphElement, short button,
	boolean altKey, boolean ctrlKey, boolean metaKey, int x, int y,
	int screenX, int screenY) {
	if (graphElement.getWorkflowBean() != null)
	graphElement.setActive(false);
	}
	}