import org.objectstyle.wolips.eomodeler.core.model.EOModelGroup;
import org.objectstyle.wolips.eomodeler.core.model.EORelationship;
public class EOModelListContentProvider implements IStructuredContentProvider {
public Object[] getElements(Object _inputElement) {
EOModelGroup modelGroup;
if (_inputElement instanceof EORelationship) {
modelGroup = ((EORelationship) _inputElement).getEntity().getModel().getModelGroup();
} else if (_inputElement instanceof EOEntity) {
modelGroup = ((EOEntity) _inputElement).getModel().getModelGroup();
} else if (_inputElement instanceof EOModel) {
modelGroup = ((EOModel) _inputElement).getModelGroup();
} else if (_inputElement instanceof EOModelGroup) {
modelGroup = (EOModelGroup) _inputElement;
} else {
throw new IllegalArgumentException("Unknown input element: " + _inputElement);
}
Set modelsList = modelGroup.getSortedModels();
EOModel[] models = (EOModel[]) modelsList.toArray(new EOModel[modelsList.size()]);
return models;
}