@Override
public IResizeShapeFeature getResizeShapeFeature(IResizeShapeContext context) {
Shape shape = context.getShape();
Object bo = getBusinessObjectForPictogramElement(shape);
if (bo instanceof SubProcess || bo instanceof Pool || bo instanceof Lane) {
return new ContainerResizeFeature(this);
} else if (bo instanceof Task || bo instanceof CallActivity) {
return new TaskResizeFeature(this);
}
return super.getResizeShapeFeature(context);
}