1.5/plugins/org.eclipse.epf.diagram.ad/src/org/eclipse/epf/diagram/ad/custom/commands/CreateOutgoingControlFlowCommand.java

//------------------------------------------------------------------------------
// Copyright (c) 2005, 2007 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.ad.custom.commands;

import org.eclipse.emf.ecore.EClass;
import org.eclipse.emf.ecore.EClassifier;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.gmf.runtime.emf.core.util.PackageUtil;
import org.eclipse.gmf.runtime.emf.type.core.commands.CreateRelationshipCommand;
import org.eclipse.gmf.runtime.emf.type.core.requests.CreateRelationshipRequest;
import org.eclipse.uml2.uml.ActivityNode;
import org.eclipse.uml2.uml.ControlFlow;
import org.eclipse.uml2.uml.UMLPackage;

/**
 * @author Shashidhar Kannoori
 *
 */
public class CreateOutgoingControlFlowCommand extends CreateRelationshipCommand {

	/**
	 * 
	 * @param request
	 * @param element
	 */
	public CreateOutgoingControlFlowCommand(CreateRelationshipRequest request,
			EObject element) {
		super(request);
		setElementToEdit(element);
	}
	/**
	 * @param request
	 */
	public CreateOutgoingControlFlowCommand(CreateRelationshipRequest request) {
		super(request);
		// TODO Auto-generated constructor stub
	}
	protected EClass getEClassToEdit() {
		return UMLPackage.eINSTANCE.getActivityNode();
	}
	
	public boolean canExecute() {
		if(getSource() == null)
			return false;
		
		if (getEClassToEdit() == null) {
			return false;
		}

		if (getContainmentFeature() != null) {
			EClassifier eClassifier = getContainmentFeature().getEType();
			boolean result = true;

			if (eClassifier instanceof EClass) {
				result = ((EClass) eClassifier).isSuperTypeOf(getElementType()
						.getEClass());
			}

			result = result
					&& PackageUtil.canReference(getEClassToEdit(),
							getContainmentFeature(), getElementType()
									.getEClass());

			return result;
		}
		return false;
		//return super.canExecute();
	}
	
	protected EObject doDefaultElementCreation() {
		// TODO: Resolve the source and target properly.
		ControlFlow newElement = (ControlFlow) super.doDefaultElementCreation();
		if (newElement != null) {
			newElement.setTarget((ActivityNode) getTarget());
			newElement.setSource((ActivityNode) getSource());
		}
		return newElement;
	}
	
}