1.5/plugins/org.eclipse.epf.diagram.ad/src/org/eclipse/epf/diagram/ad/custom/policies/ActivityXYLayoutEditPolicy.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.policies;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

import org.eclipse.core.commands.ExecutionException;
import org.eclipse.core.runtime.IAdaptable;
import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.draw2d.IFigure;
import org.eclipse.draw2d.geometry.Point;
import org.eclipse.draw2d.geometry.Rectangle;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.emf.transaction.TransactionalEditingDomain;
import org.eclipse.epf.diagram.ad.edit.parts.ActivityPartitionEditPart;
import org.eclipse.epf.diagram.ad.part.ActivityDiagramEditorPlugin;
import org.eclipse.gef.EditPart;
import org.eclipse.gef.GraphicalEditPart;
import org.eclipse.gef.Request;
import org.eclipse.gef.commands.Command;
import org.eclipse.gef.commands.CompoundCommand;
import org.eclipse.gef.requests.ChangeBoundsRequest;
import org.eclipse.gmf.runtime.common.core.command.CommandResult;
import org.eclipse.gmf.runtime.common.core.command.CompositeCommand;
import org.eclipse.gmf.runtime.common.core.command.ICommand;
import org.eclipse.gmf.runtime.diagram.core.commands.AddCommand;
import org.eclipse.gmf.runtime.diagram.core.util.ViewUtil;
import org.eclipse.gmf.runtime.diagram.ui.commands.ICommandProxy;
import org.eclipse.gmf.runtime.diagram.ui.commands.SetBoundsCommand;
import org.eclipse.gmf.runtime.diagram.ui.editparts.DiagramEditPart;
import org.eclipse.gmf.runtime.diagram.ui.editparts.IGraphicalEditPart;
import org.eclipse.gmf.runtime.diagram.ui.editparts.ShapeEditPart;
import org.eclipse.gmf.runtime.diagram.ui.editparts.ShapeNodeEditPart;
import org.eclipse.gmf.runtime.diagram.ui.editpolicies.XYLayoutEditPolicy;
import org.eclipse.gmf.runtime.diagram.ui.l10n.DiagramUIMessages;
import org.eclipse.gmf.runtime.diagram.ui.requests.RequestConstants;
import org.eclipse.gmf.runtime.emf.core.util.EObjectAdapter;
import org.eclipse.gmf.runtime.emf.type.core.commands.EditElementCommand;
import org.eclipse.gmf.runtime.emf.type.core.requests.IEditCommandRequest;
import org.eclipse.gmf.runtime.emf.type.core.requests.MoveRequest;
import org.eclipse.gmf.runtime.notation.Bounds;
import org.eclipse.gmf.runtime.notation.LayoutConstraint;
import org.eclipse.gmf.runtime.notation.Node;
import org.eclipse.gmf.runtime.notation.View;
import org.eclipse.uml2.uml.Activity;
import org.eclipse.uml2.uml.ActivityNode;
import org.eclipse.uml2.uml.ActivityPartition;
import org.eclipse.uml2.uml.StructuredActivityNode;
import org.eclipse.uml2.uml.UMLPackage;

/**
 * @author Shashidhar Kannoori
 * @author Shilpa Toraskar
 * 
 */
public class ActivityXYLayoutEditPolicy extends XYLayoutEditPolicy {

	/**
	 * 
	 */
	public ActivityXYLayoutEditPolicy() {
	}

	protected Command createAddCommand(EditPart child, Object constraint) {
		Object parent  = child.getParent();		
		if ((parent instanceof ActivityPartitionEditPart) &&
			( child instanceof ShapeEditPart && constraint instanceof Rectangle)) {
			Rectangle rect = (Rectangle) constraint;
			IFigure parentFigure = ((ActivityPartitionEditPart)parent).getFigure();			
			View view = (View) child.getModel();
			boolean adjustLocationNeeded = ViewUtil
					.resolveSemanticElement(view) != null
					|| parentFigure.getBounds().contains(rect.getLocation());
			if(adjustLocationNeeded) {
				int x = rect.x - ((ActivityPartitionEditPart) parent).getLocation().x;
				int y = rect.y - ((ActivityPartitionEditPart) parent).getLocation().y;

				ICommand boundsCommand = 
					new SetBoundsCommand(((ShapeEditPart) child).getEditingDomain(),
							DiagramUIMessages.SetLocationCommand_Label_Resize,  //$NON-NLS-1$
							new EObjectAdapter(view),
									new Point(x, y)); 

					return new ICommandProxy(boundsCommand);
			}
		}
		
		return super.createAddCommand(child, constraint);
	}

	protected Command getMoveChildrenCommand(Request request) {
		return super.getMoveChildrenCommand(request);
	}


	protected Command createChangeConstraintCommand(EditPart child,
			Object constraint) {
		return super.createChangeConstraintCommand(child, constraint);
	}


	protected Command createChangeConstraintCommand(
			ChangeBoundsRequest request, EditPart child, Object constraint) {
		Command cmd = super.createChangeConstraintCommand(request, child,
				constraint);
		if (child instanceof ActivityPartitionEditPart) {
			List list = getNodesInPartition((ActivityPartitionEditPart) child);
			if (list != null && !list.isEmpty()) {
				for (Iterator iter = list.iterator(); iter.hasNext();) {
					EditPart element = (EditPart) iter.next();
					if (element instanceof ShapeNodeEditPart) {
						ChangeBoundsRequest req = new ChangeBoundsRequest(
								REQ_RESIZE_CHILDREN);
						req.setEditParts(element);
						req.setMoveDelta(request.getMoveDelta());
						req.setSizeDelta(request.getSizeDelta());
						req.setLocation(request.getLocation());
						req.setExtendedData(request.getExtendedData());
						req.setResizeDirection(request.getResizeDirection());
						cmd = cmd.chain(element.getParent().getCommand(req));
					}
				}
			}
		}else{
			EditPart parent = child.getParent();
			if(parent instanceof DiagramEditPart){
				List children = ((DiagramEditPart)parent).getChildren();
				for (Iterator iter = children.iterator(); iter.hasNext();) {
					EditPart part = (EditPart) iter.next();
					if(part instanceof ActivityPartitionEditPart){
						Node view = (Node)part.getModel();
						LayoutConstraint lConstaint = view.getLayoutConstraint();
						if(lConstaint instanceof Bounds){
							Rectangle partitionConstraint = ((ActivityPartitionEditPart)part).getFigure().getBounds();
							if(constraint instanceof Rectangle){
								if(partitionConstraint.contains((Rectangle)constraint)){
									request.setType(RequestConstants.REQ_CREATE);
									cmd = cmd.chain(part.getCommand(request));
								}
							}
						}else{
							if (ActivityDiagramEditorPlugin.getInstance().isDebugging()) {
								System.out.println("LayoutConstraint: "+ lConstaint);
							}
						}
					}
				}
			}
		}
		return cmd;
	}
	
	
	private static List getNodesInPartition(ActivityPartitionEditPart child) {
		List nodes = new ArrayList();
		ActivityPartition partition = (ActivityPartition)((Node)child.getModel()).getElement();
		List list = partition.getNodes();
		for (Iterator iter = list.iterator(); iter.hasNext();) {
			Object element = (Object) iter.next();
			nodes.add(findEditPartyByElement(child.getParent(), element));
		}
		return nodes;
	}
	
	private static EditPart findEditPartyByElement(EditPart parent,
			Object element) {
		List list = parent.getChildren();
		for (Iterator iter = list.iterator(); iter.hasNext();) {
			EditPart editPart = (EditPart) iter.next();
			if (((Node) editPart.getModel()).getElement() == element) {
				return editPart;
			}
		}
		return null;
	}

	@Override
	protected Command getAddCommand(Request request) {
		CompoundCommand command = new CompoundCommand();
		command.setDebugLabel("Add in ConstrainedLayoutEditPolicy with Reparent if necessary");//$NON-NLS-1$
		command.add(super.getAddCommand(request));
		
		IGraphicalEditPart childPart;
		if(request instanceof ChangeBoundsRequest && request.getType() == REQ_ADD){
			ChangeBoundsRequest changeRequest =(ChangeBoundsRequest)request; 
			List editParts = changeRequest.getEditParts();
			for (int i = 0; i < editParts.size(); i++) {
				childPart = (IGraphicalEditPart)editParts.get(i);
				IGraphicalEditPart parent = (IGraphicalEditPart)childPart.getParent();
				if(parent instanceof ActivityPartitionEditPart){
					Point point = childPart.getFigure().getBounds().getLocation().getCopy();
					point.translate(changeRequest.getMoveDelta());
					if(!parent.getFigure().getBounds().contains(point)){
						command.add(getReparentCommand(childPart, parent, changeRequest));
					}
				}
			}
		}
		return command.unwrap();
//		return null;
	}
	
	
	protected Command getReparentCommand(IGraphicalEditPart gep, IGraphicalEditPart oldParent,
			ChangeBoundsRequest changeRequest) {
		CompositeCommand cc = new CompositeCommand(DiagramUIMessages.AddCommand_Label); 
		View container = (View)getHost().getModel();
		EObject context = ViewUtil.resolveSemanticElement(container);
		View view = (View)gep.getModel();
		EObject element = ViewUtil.resolveSemanticElement(view);

        TransactionalEditingDomain editingDomain = ((IGraphicalEditPart) getHost())
            .getEditingDomain();
        
		if ( element != null ) {
			cc.compose(new ReparentingElementsCommand(new MoveRequest(editingDomain, context, UMLPackage.eINSTANCE.getActivity_Node(), element),
					element,view, oldParent));
			cc.compose(new AddCommand(gep.getEditingDomain(), new EObjectAdapter(container),
					  new EObjectAdapter(view)));
			cc.compose(adjustLocationCommand(gep, changeRequest));
		}
		return cc.isEmpty() ? null : new ICommandProxy(cc.reduce());
	}
	
	
	public class ReparentingElementsCommand extends EditElementCommand{

		private final EObject moveElement;
		private IGraphicalEditPart oldParent;
		private View view;
		
		public ReparentingElementsCommand(MoveRequest req, EObject element, View view, 
				IGraphicalEditPart oldParent) {
			this("reparent on move", element, req);
			this.oldParent = oldParent;
			this.view = view;
		}
		
		protected ReparentingElementsCommand(String label, EObject elementToEdit, 
				IEditCommandRequest request) {
			super(label, elementToEdit, request);
			this.moveElement = elementToEdit; 
		}

		@Override
		protected CommandResult doExecuteWithResult(IProgressMonitor monitor, IAdaptable info) throws ExecutionException {
			View parentView = (View)getHost().getModel();
			if(parentView.getElement() instanceof Activity){
				if(moveElement instanceof ActivityNode){
					View oldParentView = (View)oldParent.getModel();
					((ActivityNode)moveElement).getInPartitions().remove(oldParentView.getElement());
				}
			}
			return null;
		}
		@Override
		public boolean canExecute() {
			if(moveElement != null) return true;
			return false;
		}
	}
	
	private ICommand adjustLocationCommand(IGraphicalEditPart child,
			ChangeBoundsRequest request) {
		if (child instanceof ShapeEditPart) {
			Point location = request.getMoveDelta().getCopy();
			Point childLocation = child.getFigure().getBounds().getLocation();
			location.x+=childLocation.x;
			location.y+=childLocation.y;
			ICommand boundsCommand = new SetBoundsCommand(
					((ShapeEditPart) child).getEditingDomain(),
					DiagramUIMessages.SetLocationCommand_Label_Resize, //$NON-NLS-1$
					new EObjectAdapter((View) child.getModel()),
					location);

			return boundsCommand;
		}
		return null;
	}
}