/*******************************************************************************
* 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.choreography;
import static org.eclipse.bpmn2.modeler.core.features.choreography.ChoreographyProperties.TEXT_H;
import org.eclipse.bpmn2.ChoreographyActivity;
import org.eclipse.bpmn2.modeler.core.features.BusinessObjectUtil;
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.Text;
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;
import org.eclipse.graphiti.services.IPeService;
public class ChoreographyLayoutFeature extends AbstractLayoutFeature {
protected IPeService peService = Graphiti.getPeService();
protected IGaService gaService = Graphiti.getGaService();
public ChoreographyLayoutFeature(IFeatureProvider fp) {
super(fp);
}
@Override
public boolean canLayout(ILayoutContext context) {
return BusinessObjectUtil.getFirstElementOfType(context.getPictogramElement(), ChoreographyActivity.class) != null;
}
@Override
public boolean layout(ILayoutContext context) {
ContainerShape choreographyContainer = (ContainerShape) context.getPictogramElement();
GraphicsAlgorithm parentGa = choreographyContainer.getGraphicsAlgorithm();
int newWidth = parentGa.getWidth();
int newHeight = parentGa.getHeight();
for (Shape s : peService.getAllContainedShapes(choreographyContainer)) {
String property = peService.getPropertyValue(s, ChoreographyProperties.CHOREOGRAPHY_NAME);
if (property != null && new Boolean(property)) {
GraphicsAlgorithm ga = s.getGraphicsAlgorithm();
setTextLocation(choreographyContainer, (Text) ga, newWidth, newHeight);
}
property = peService.getPropertyValue(s, ChoreographyProperties.CALL_CHOREO_BORDER);
if (property != null && new Boolean(property)) {
GraphicsAlgorithm ga = s.getGraphicsAlgorithm();
gaService.setSize(ga, newWidth, newHeight);
peService.sendToFront(s);
}
// use it when property editor supports enums
// property = peService.getPropertyValue(s, ChoreographyProperties.CHOREOGRAPHY_MARKER_SHAPE);
// if (property != null && new Boolean(property)) {
// List<ContainerShape> bands = ChoreographyUtil.getParticipantBandContainerShapes(choreographyContainer);
// List<ContainerShape> bottomBands = ChoreographyUtil.getTopAndBottomBands(bands).getSecond();
// int x = (newWidth / 2) - (MARKER_H / 2);
// int y = newHeight - MARKER_H;
// if (!bottomBands.isEmpty()) {
// ContainerShape b = bottomBands.get(0);
// y = b.getGraphicsAlgorithm().getY() - MARKER_H;
// }
// gaService.setLocation(s.getGraphicsAlgorithm(), x, y);
// }
}
return true;
}
protected void setTextLocation(ContainerShape choreographyContainer, Text text, int w, int h) {
int y = (h / 2) - (TEXT_H / 2);
gaService.setLocationAndSize(text, 0, y, w, TEXT_H);
}
}