ActivityLayoutFeature.java

/******************************************************************************* 
 * Copyright (c) 2011 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 
 ******************************************************************************/
package org.eclipse.bpmn2.modeler.core.features.activity;

import java.util.Iterator;

import org.eclipse.bpmn2.Activity;
import org.eclipse.bpmn2.BaseElement;
import org.eclipse.bpmn2.modeler.core.di.DIUtils;
import org.eclipse.bpmn2.modeler.core.features.BusinessObjectUtil;
import org.eclipse.bpmn2.modeler.core.features.event.AbstractBoundaryEventOperation;
import org.eclipse.bpmn2.modeler.core.utils.FeatureSupport;
import org.eclipse.bpmn2.modeler.core.utils.GraphicsUtil;
import org.eclipse.graphiti.features.IFeatureProvider;
import org.eclipse.graphiti.features.context.ILayoutContext;
import org.eclipse.graphiti.features.impl.AbstractLayoutFeature;
import org.eclipse.graphiti.mm.algorithms.GraphicsAlgorithm;
import org.eclipse.graphiti.mm.algorithms.RoundedRectangle;
import org.eclipse.graphiti.mm.pictograms.ContainerShape;
import org.eclipse.graphiti.mm.pictograms.Shape;
import org.eclipse.graphiti.services.Graphiti;
import org.eclipse.graphiti.services.IGaService;

public class ActivityLayoutFeature extends AbstractLayoutFeature {

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

	@Override
	public boolean canLayout(ILayoutContext context) {
		Object bo = BusinessObjectUtil.getFirstElementOfType(context.getPictogramElement(), BaseElement.class);
		return bo != null && bo instanceof Activity;
	}

	@Override
	public boolean layout(ILayoutContext context) {
		ContainerShape containerShape = (ContainerShape) context.getPictogramElement();
		GraphicsAlgorithm parentGa = containerShape.getGraphicsAlgorithm();

		Iterator<Shape> iterator = Graphiti.getPeService().getAllContainedShapes(containerShape).iterator();
		while (iterator.hasNext()) {
			Shape shape = iterator.next();
			GraphicsAlgorithm ga = shape.getGraphicsAlgorithm();
			IGaService gaService = Graphiti.getGaService();

			int newWidth = parentGa.getWidth();
			int newHeight = parentGa.getHeight();

			String markerProperty = Graphiti.getPeService().getPropertyValue(shape,
					GraphicsUtil.ACTIVITY_MARKER_CONTAINER);
			if (markerProperty != null && new Boolean(markerProperty)) {
				int x = (newWidth / 2) - (ga.getWidth() / 2);
				int y = newHeight - ga.getHeight() - 3 - getMarkerContainerOffset();
				gaService.setLocation(ga, x, y);
				continue;
			}

			Shape rectShape = FeatureSupport.getShape(containerShape, "activity", Boolean.toString(true));
			gaService.setSize(rectShape.getGraphicsAlgorithm(), newWidth, newHeight);
			layoutInRectangle((RoundedRectangle) rectShape.getGraphicsAlgorithm());

			Object[] objects = getAllBusinessObjectsForPictogramElement(shape);
			for (Object bo : objects) {
				layoutHook(shape, ga, bo, newWidth, newHeight);
			}
		}

		Activity activity = BusinessObjectUtil.getFirstElementOfType(containerShape, Activity.class);
		new AbstractBoundaryEventOperation() {
			@Override
			protected void doWorkInternal(ContainerShape container) {
				layoutPictogramElement(container);
			}
		}.doWork(activity, getDiagram());

		DIUtils.updateDIShape(containerShape);
		return true;
	}

	protected int getMarkerContainerOffset() {
		return 0;
	}

	protected void layoutInRectangle(RoundedRectangle rect) {
	}

	protected boolean layoutHook(Shape shape, GraphicsAlgorithm ga, Object bo, int newWidth, int newHeight) {
		return false;
	}
}