import org.jbpm.workflow.core.NodeExtension;
public class DefaultNodeEditPart extends ElementEditPart {
protected IFigure createFigure() {
DefaultNodeFigure figure = new DefaultNodeFigure();
Node node = ((NodeWrapper) getElementWrapper()).getNode();
if (node instanceof NodeExtension) {
figure.setIcon(((NodeExtension) node).getIcon());
}
return figure;
}