ReconnectLinkCommand.java

//------------------------------------------------------------------------------
// Copyright (c) 2005, 2006 IBM Corporation and others.
// All rights reserved. This program and the accompanying materials
// are made available under the terms of the Eclipse Public License v1.0
// which accompanies this distribution, and is available at
// http://www.eclipse.org/legal/epl-v10.html
//
// Contributors:
// IBM Corporation - initial implementation
//------------------------------------------------------------------------------
package org.eclipse.epf.diagram.core.commands;

import java.util.Collections;

import org.eclipse.core.commands.ExecutionException;
import org.eclipse.draw2d.geometry.Point;
import org.eclipse.emf.transaction.Transaction;
import org.eclipse.emf.transaction.impl.InternalTransactionalEditingDomain;
import org.eclipse.epf.diagram.core.DiagramCorePlugin;
import org.eclipse.epf.diagram.core.util.DiagramCoreValidation;
import org.eclipse.epf.diagram.model.util.TxUtil;
import org.eclipse.gef.commands.Command;
import org.eclipse.gmf.runtime.notation.Edge;
import org.eclipse.gmf.runtime.notation.Node;

/**
 * @author Shashidhar Kannoori
 * @since 1.2
 */
public class ReconnectLinkCommand extends Command {

	private boolean isSource;

	private Edge link;

	private Node newNode, oldNode;

	private int viewIndex;

	private Point endPoint;

	private InternalTransactionalEditingDomain domain;

	private static final String LABEL = "Reconnect Link"; //$NON-NLS-1$

	public ReconnectLinkCommand(InternalTransactionalEditingDomain domain, Edge link, Node newNode, boolean isSource) {
		super(LABEL);
		this.domain = domain;
		this.link = link;
		this.newNode = newNode;
		this.isSource = isSource;
	}

	public void setEndPoint(Point p) {
		endPoint = p.getCopy();
	}

	public boolean canExecute() {
		boolean result = link != null && newNode != null;
		if (!result) {
			return false;
		}
		if (isSource) {
			return DiagramCoreValidation.checkReconnect(newNode, (Node)link.getTarget(), link) == null;
		} else {
			return DiagramCoreValidation.checkReconnect((Node)link.getSource(), newNode, link) == null;
		}
	}

	public void execute() {

		// Link
		try {
			TxUtil.runInTransaction(domain, new Runnable() {

				public void run() {
					if (isSource) {
						oldNode = (Node) link.getSource();
						if (oldNode != null) { // safety check to handle dangling link
							viewIndex = oldNode.getTargetEdges().indexOf(link);
						}
						link.setSource(newNode);
					} else {
						oldNode = (Node) link.getTarget();
						if (oldNode != null) { // safety check to handle dangling link
							viewIndex = oldNode.getSourceEdges().indexOf(link);
						}
						link.setTarget(newNode);
					}
				}
				
			});
		} catch (Exception e) {
			DiagramCorePlugin.getDefault().getLogger().logError("Error while re-connect the link :", e); //$NON-NLS-1$
		}
	}

	public void undo() {
		if (isSource) {
			newNode.getTargetEdges().remove(link);
			oldNode.getTargetEdges().add(viewIndex, link);
		} else {
			newNode.getSourceEdges().remove(link);
			oldNode.getSourceEdges().add(viewIndex, link);
		}
		oldNode = null;
	}

}