| /******************************************************************************* |
| * Copyright (c) 2011, 2012, 2013 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.runtime.jboss.jbpm5.features; |
| |
| import org.eclipse.bpmn2.BaseElement; |
| import org.eclipse.bpmn2.modeler.core.features.activity.task.AddTaskFeature; |
| import org.eclipse.bpmn2.modeler.runtime.jboss.jbpm5.model.bpsim.ElementParameters; |
| import org.eclipse.bpmn2.modeler.runtime.jboss.jbpm5.util.JbpmModelUtil; |
| import org.eclipse.bpmn2.modeler.ui.features.activity.task.TaskFeatureContainer; |
| import org.eclipse.graphiti.features.IAddFeature; |
| import org.eclipse.graphiti.features.IFeatureProvider; |
| import org.eclipse.graphiti.features.context.IAddContext; |
| import org.eclipse.graphiti.mm.pictograms.PictogramElement; |
| |
| public class JbpmTaskFeatureContainer extends TaskFeatureContainer { |
| |
| @Override |
| public IAddFeature getAddFeature(IFeatureProvider fp) { |
| return new JbpmAddTaskFeature(fp); |
| } |
| |
| public static class JbpmAddTaskFeature extends AddTaskFeature { |
| |
| public JbpmAddTaskFeature(IFeatureProvider fp) { |
| super(fp); |
| } |
| |
| @Override |
| public PictogramElement add(IAddContext context) { |
| final BaseElement be = getBusinessObject(context); |
| final PictogramElement pe = super.add(context); |
| ElementParameters ep = JbpmModelUtil.getElementParameters(be); |
| getFeatureProvider().link(pe, ep); |
| return pe; |
| } |
| |
| } |
| } |