package org.activiti.designer.features;
import org.activiti.bpmn.model.Gateway;
import org.activiti.bpmn.model.InclusiveGateway;
import org.activiti.bpmn.model.Lane;
import org.activiti.bpmn.model.SubProcess;
import org.activiti.designer.util.eclipse.ActivitiUiUtil;
import org.activiti.designer.util.style.StyleUtil;
import org.eclipse.graphiti.features.IFeatureProvider;
import org.eclipse.graphiti.features.context.IAddContext;
import org.eclipse.graphiti.features.impl.AbstractAddShapeFeature;
import org.eclipse.graphiti.mm.algorithms.Ellipse;
import org.eclipse.graphiti.mm.algorithms.Polygon;
import org.eclipse.graphiti.mm.pictograms.BoxRelativeAnchor;
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.services.Graphiti;
import org.eclipse.graphiti.services.IGaService;
import org.eclipse.graphiti.services.IPeCreateService;
/**
* A AddInclusiveGatewayFeature is an "Add Shape" feature, which adds a
* rhombus with a circle inside to represent the Inclusive Gateway.
*
* @author tvanbusk
*/
public class AddInclusiveGatewayFeature extends AbstractAddShapeFeature {
public AddInclusiveGatewayFeature(IFeatureProvider fp) {
super(fp);
}
@Override
public PictogramElement add(IAddContext context) {
final InclusiveGateway addedGateway = (InclusiveGateway) context.getNewObject();
final ContainerShape parent = context.getTargetContainer();
// CONTAINER SHAPE WITH CIRCLE
final IPeCreateService peCreateService = Graphiti.getPeCreateService();
final ContainerShape containerShape = peCreateService.createContainerShape(parent, true);
// check whether the context has a size (e.g. from a create feature)
// otherwise define a default size for the shape
final int width = context.getWidth() <= 0 ? 40 : context.getWidth();
final int height = context.getHeight() <= 0 ? 40 : context.getHeight();
final IGaService gaService = Graphiti.getGaService();
Polygon polygon;
{
int xy[] = new int[] { 0, 20, 20, 0, 40, 20, 20, 40, 0, 20 };
final Polygon invisiblePolygon = gaService.createPolygon(containerShape, xy);
invisiblePolygon.setFilled(false);
invisiblePolygon.setLineVisible(false);
gaService.setLocationAndSize(invisiblePolygon, context.getX(), context.getY(), width, height);
// create and set visible polygon inside invisible polygon
polygon = gaService.createPolygon(invisiblePolygon, xy);
polygon.setParentGraphicsAlgorithm(invisiblePolygon);
polygon.setStyle(StyleUtil.getStyleForEvent(getDiagram()));
gaService.setLocationAndSize(polygon, 0, 0, width, height);
// create link and wire it
link(containerShape, addedGateway);
}
// This draws the circle inside the rhombus
{
// create and set the circle inside the polygon
final Ellipse circle = gaService.createEllipse(polygon);
circle.setStyle(StyleUtil.getStyleForEvent(getDiagram()));
circle.setLineWidth(3);
circle.setParentGraphicsAlgorithm(polygon);
gaService.setLocationAndSize(circle, 10, 10, 20, 20);
}
{
// add a chopbox anchor to the shape
peCreateService.createChopboxAnchor(containerShape);
final BoxRelativeAnchor boxAnchor = peCreateService.createBoxRelativeAnchor(containerShape);
boxAnchor.setRelativeWidth(0.51);
boxAnchor.setRelativeHeight(0.10);
boxAnchor.setReferencedGraphicsAlgorithm(polygon);
final Ellipse ellipse = ActivitiUiUtil.createInvisibleEllipse(boxAnchor, gaService);
gaService.setLocationAndSize(ellipse, 0, 0, 0, 0);
}
{
// add a another chopbox anchor to the shape
peCreateService.createChopboxAnchor(containerShape);
final BoxRelativeAnchor boxAnchor2 = peCreateService.createBoxRelativeAnchor(containerShape);
boxAnchor2.setRelativeWidth(0.51);
boxAnchor2.setRelativeHeight(0.93);
boxAnchor2.setReferencedGraphicsAlgorithm(polygon);
final Ellipse ellipse = ActivitiUiUtil.createInvisibleEllipse(boxAnchor2, gaService);
gaService.setLocationAndSize(ellipse, 0, 0, 0, 0);
}
// call the layout feature
layoutPictogramElement(containerShape);
return containerShape;
}
@Override
public boolean canAdd(IAddContext context) {
if (context.getNewObject() instanceof Gateway) {
Object parentObject = getBusinessObjectForPictogramElement(context.getTargetContainer());
if (context.getTargetContainer() instanceof Diagram ||
parentObject instanceof SubProcess || parentObject instanceof Lane) {
return true;
}
}
return false;
}
}