@Override
public IDirectEditingFeature getDirectEditingFeature(IDirectEditingContext context) {
PictogramElement pe = context.getPictogramElement();
Object bo = getBusinessObjectForPictogramElement(pe);
if (bo instanceof FlowElement) {
return new DirectEditFlowElementFeature(this);
} else if (bo instanceof TextAnnotation) {
return new DirectEditTextAnnotationFeature(this);
}
return super.getDirectEditingFeature(context);
}