CreateActivityNodeInActivityPartition.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.core.commands.ExecutionException;
import org.eclipse.core.runtime.IAdaptable;
import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.core.runtime.IStatus;
import org.eclipse.emf.ecore.EClass;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.emf.ecore.EReference;
import org.eclipse.epf.diagram.ad.providers.UMLElementTypes;
import org.eclipse.epf.diagram.core.bridge.BridgeHelper;
import org.eclipse.gmf.runtime.emf.core.util.PackageUtil;
import org.eclipse.gmf.runtime.emf.type.core.IElementType;
import org.eclipse.gmf.runtime.emf.type.core.commands.CreateElementCommand;
import org.eclipse.gmf.runtime.emf.type.core.requests.CreateElementRequest;
import org.eclipse.gmf.runtime.notation.View;
import org.eclipse.uml2.uml.ActivityNode;
import org.eclipse.uml2.uml.ActivityParameterNode;
import org.eclipse.uml2.uml.ActivityPartition;
import org.eclipse.uml2.uml.StructuredActivityNode;
import org.eclipse.uml2.uml.UMLPackage;

/**
 * @author Shashidhar Kannoori
 *
 */
public class CreateActivityNodeInActivityPartition extends CreateElementCommand {

	

	/**
	 * @param request
	 */
	public CreateActivityNodeInActivityPartition(CreateElementRequest request) {
		super(request);
	}
	
	/**
	 * 
	 */
	protected EClass getEClassToEdit() {
		return UMLPackage.eINSTANCE.getActivity();
	};

	/**
	 * @modified
	 */
	protected EObject getElementToEdit() {
		EObject container = ((CreateElementRequest) getRequest())
				.getContainer();
		if (container instanceof View) {
			container = ((View) container).getElement();
		}
		if(container instanceof ActivityPartition){
			container = container.eContainer();
		}
		return container;
	}
	public boolean canExecute() {
		return super.canExecute();
	}
	
	/**
	 * (non-Javadoc)
	 * @see org.eclipse.gmf.runtime.emf.commands.core.command.AbstractTransactionalCommand#doExecute(org.eclipse.core.runtime.IProgressMonitor, org.eclipse.core.runtime.IAdaptable)
	 * @modified
	 */
	protected IStatus doExecute(IProgressMonitor monitor, IAdaptable info) throws ExecutionException {
		IStatus status = null;
		status = super.doExecute(monitor, info);
		EObject newElement = getNewElement();
		
		if (newElement instanceof ActivityNode) 
		{
			EObject obj = getOriginalContainer();
			if (obj instanceof ActivityPartition) {
				// set inpartition 
				((ActivityNode) newElement).getInPartitions().add((ActivityPartition) obj);
			}
		}
		return status;
	}
	
	

	/**
	 * @custom
	 */
	private EObject getOriginalContainer()
	{
		return ((CreateElementRequest) getRequest()).getContainer();
	}
	
	/**
	 * Gets the containment feature for the new element.
	 * 
	 * @return the containment feature
	 * @modified
	 */
	protected EReference getContainmentFeature() {

		/**
		 * The containment feature in which the new element will be created.
		 */
		EReference containmentFeature=null;
		if (containmentFeature == null) {
			EClass classToEdit = getEClassToEdit();

			if (classToEdit != null) {
				IElementType type = getElementType();

				if (type != null && type.getEClass() != null) {
					containmentFeature = PackageUtil.findFeature(classToEdit,
							type.getEClass());
				}
			}
		}

		return containmentFeature;
	}
	
	@Override
	protected EObject doDefaultElementCreation() {
		// TODO Auto-generated method stub
		EObject newElement = super.doDefaultElementCreation();
		if(newElement instanceof StructuredActivityNode
				|| newElement instanceof ActivityParameterNode){
			if (newElement != null) {
				BridgeHelper.setDefaultName((ActivityNode)newElement);
			}
		}
		CreateElementRequest request = getCreateRequest();
		
		if(UMLElementTypes.StructuredActivityNode_1007 == request.getElementType()){
			if (newElement != null) {
				UMLElementTypes.Initializers.StructuredActivityNode_1007
						.init(newElement);
			}
		}else if(UMLElementTypes.StructuredActivityNode_1010 == request
						.getElementType()){
			if (newElement != null) {
				UMLElementTypes.Initializers.StructuredActivityNode_1010
						.init(newElement);
			}
		}else if(UMLElementTypes.StructuredActivityNode_1011 == request
						.getElementType()){
			if (newElement != null) {
				UMLElementTypes.Initializers.StructuredActivityNode_1011
						.init(newElement);
			}
		}else if(UMLElementTypes.ActivityParameterNode_1009 == request
						.getElementType()){
			if (newElement != null) {
				UMLElementTypes.Initializers.ActivityParameterNode_1009
						.init(newElement);
			}
		}else if(UMLElementTypes.ActivityParameterNode_1012 == request
						.getElementType()){
			if (newElement != null) {
				UMLElementTypes.Initializers.ActivityParameterNode_1012
						.init(newElement);
			}
		}
		
		return newElement;
	}
}