plugins/org.eclipse.bpmn2.modeler.ui/src/org/eclipse/bpmn2/modeler/ui/features/event/MoveBoundaryEventFeature.java

/******************************************************************************* 
 * Copyright (c) 2011, 2012 Red Hat, Inc. 
 *  All rights reserved. 
 * This program is 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: 
 * Red Hat, Inc. - initial API and implementation 
 *
 * @author Innar Made
 ******************************************************************************/
package org.eclipse.bpmn2.modeler.ui.features.event;

import java.util.List;

import org.eclipse.bpmn2.Activity;
import org.eclipse.bpmn2.BoundaryEvent;
import org.eclipse.bpmn2.modeler.core.features.GraphitiConstants;
import org.eclipse.bpmn2.modeler.core.features.MoveFlowNodeFeature;
import org.eclipse.bpmn2.modeler.core.utils.BoundaryEventPositionHelper;
import org.eclipse.bpmn2.modeler.core.utils.BoundaryEventPositionHelper.PositionOnLine;
import org.eclipse.bpmn2.modeler.core.utils.BusinessObjectUtil;
import org.eclipse.bpmn2.modeler.core.utils.FeatureSupport;
import org.eclipse.bpmn2.modeler.core.utils.GraphicsUtil;
import org.eclipse.bpmn2.modeler.core.utils.ModelUtil;
import org.eclipse.graphiti.datatypes.ILocation;
import org.eclipse.graphiti.features.IFeatureProvider;
import org.eclipse.graphiti.features.context.IMoveShapeContext;
import org.eclipse.graphiti.features.context.impl.MoveShapeContext;
import org.eclipse.graphiti.mm.algorithms.GraphicsAlgorithm;
import org.eclipse.graphiti.mm.pictograms.ContainerShape;
import org.eclipse.graphiti.mm.pictograms.Diagram;
import org.eclipse.graphiti.mm.pictograms.PictogramElement;
import org.eclipse.graphiti.mm.pictograms.Shape;
import org.eclipse.graphiti.services.Graphiti;
import org.eclipse.graphiti.services.IGaService;
import org.eclipse.graphiti.services.IPeService;

public class MoveBoundaryEventFeature extends MoveFlowNodeFeature {

	public MoveBoundaryEventFeature(IFeatureProvider fp) {
		super(fp);
	}

	@Override
	protected void preMoveShape(IMoveShapeContext context) {
		super.preMoveShape(context);
		
		ContainerShape targetContainer = context.getTargetContainer();
		Activity activity = BusinessObjectUtil.getFirstElementOfType(targetContainer, Activity.class);
		Object property = context.getProperty(GraphitiConstants.ACTIVITY_MOVE_PROPERTY);
		
		if (activity != null && property == null) {
			ContainerShape taskContainer = context.getTargetContainer();
			ContainerShape parentContainer = (ContainerShape) context.getPictogramElement().eContainer();

			IPeService peService = Graphiti.getPeService();

			ILocation loc = peService.getLocationRelativeToDiagram(taskContainer);
			MoveShapeContext c = (MoveShapeContext) context;
			int eventX = loc.getX() + context.getX();
			int eventY = loc.getY() + context.getY();

			if (!(parentContainer instanceof Diagram)) {
				loc = peService.getLocationRelativeToDiagram(parentContainer);
				eventX = eventX - loc.getX();
				eventY = eventY - loc.getY();
			}

			c.setLocation(eventX, eventY);
			c.setTargetContainer(parentContainer);
		}
	}

	@Override
	public boolean canMoveShape(IMoveShapeContext context) {
		Object property = context.getProperty(GraphitiConstants.ACTIVITY_MOVE_PROPERTY);
		PictogramElement[] selection = getDiagramEditor().getSelectedPictogramElements();
		boolean singleSelection = selection != null && selection.length == 1;
		
		BoundaryEvent event = (BoundaryEvent) context.getShape().getLink().getBusinessObjects().get(0);
		List<PictogramElement> activityPictogramElements = Graphiti.getLinkService().getPictogramElements(getDiagram(), event.getAttachedToRef());
		if (!singleSelection) {
			for (PictogramElement activityElement : activityPictogramElements) {
				if (ModelUtil.isElementSelected(getDiagramBehavior().getDiagramContainer(), activityElement)){
					if (!ModelUtil.isElementSelected(getDiagramBehavior().getDiagramContainer(), context.getPictogramElement())) {
						context.putProperty(GraphitiConstants.SELECTION_MOVE_PROPERTY, false);
						context.putProperty(GraphitiConstants.ACTIVITY_MOVE_PROPERTY, true);
					} else {
						context.putProperty(GraphitiConstants.SELECTION_MOVE_PROPERTY, true);
						context.putProperty(GraphitiConstants.ACTIVITY_MOVE_PROPERTY, false);
					}
					return true;
				}
			}
		}
		
		boolean movedByActivity = property != null && (Boolean) property;
		if (!movedByActivity && !BoundaryEventPositionHelper.canMoveTo(context, getDiagram())) {
			return false;
		}
		return super.canMoveShape(context);
	}

	@Override
	protected boolean onMoveAlgorithmNotFound(IMoveShapeContext context) {
		return true;
	}

	@Override
	protected void postMoveShape(IMoveShapeContext context) {
		PictogramElement pe = context.getPictogramElement();
		if (pe instanceof ContainerShape) {
			ContainerShape eventContainer = (ContainerShape) pe;
			
			Object property = context.getProperty(GraphitiConstants.ACTIVITY_MOVE_PROPERTY);
			Object selectionFlag = context.getProperty(GraphitiConstants.SELECTION_MOVE_PROPERTY);
			if (property != null && (Boolean) property) {
				IGaService gaService = Graphiti.getGaService();
				GraphicsAlgorithm ga = eventContainer.getGraphicsAlgorithm();
				// make sure the x/y coordinates are within the bounds of the
				// Activity to which this Boundary Event is attached
				int x = ga.getX() + context.getDeltaX();
				int y = ga.getY() + context.getDeltaY();
				if (!BoundaryEventPositionHelper.canMoveTo(context, getDiagram())) {
					ILocation loc = BoundaryEventPositionHelper.getLocation(context, getDiagram());
					x = loc.getX();
					y = loc.getY();
				}
				gaService.setLocation(ga, x, y);
			} 
			else if (selectionFlag != null && (Boolean) selectionFlag) {
				// do nothing, let base class do the job
			}
			else {
				BoundaryEvent event = BusinessObjectUtil.getFirstElementOfType(eventContainer, BoundaryEvent.class);
				PictogramElement activityContainer = BusinessObjectUtil.getFirstBaseElementFromDiagram(getDiagram(),
						event.getAttachedToRef());
				PositionOnLine newPos = BoundaryEventPositionHelper.getPositionOnLineUsingAbsoluteCoordinates(
						eventContainer, (Shape) activityContainer);
				BoundaryEventPositionHelper.assignPositionOnLineProperty(eventContainer, newPos);
			}
			
			GraphicsUtil.sendToFront(context.getShape());
			super.postMoveShape(context);
			FeatureSupport.updateConnections(getFeatureProvider(), eventContainer);
		}		
	}
}