| /* |
| * Copyright (c) 2005, 2018 IBM Corporation, CEA, and others. |
| * All rights reserved. This program and the accompanying materials |
| * are 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: |
| * IBM - initial API and implementation |
| * Kenn Hussey - 323181, 522658, 528925, 531034 |
| * Kenn Hussey (CEA) - 281326 |
| * |
| */ |
| package org.eclipse.uml2.uml.editor.presentation; |
| |
| import java.util.ArrayList; |
| import java.util.Collection; |
| import java.util.Iterator; |
| import java.util.LinkedHashMap; |
| import java.util.Map; |
| import java.util.StringTokenizer; |
| import java.util.Collections; |
| import java.util.Comparator; |
| import java.util.List; |
| |
| import org.eclipse.core.runtime.IProgressMonitor; |
| import org.eclipse.emf.common.CommonPlugin; |
| import org.eclipse.emf.common.command.Command; |
| import org.eclipse.emf.common.command.CommandWrapper; |
| import org.eclipse.emf.common.command.CompoundCommand; |
| import org.eclipse.emf.common.notify.AdapterFactory; |
| import org.eclipse.emf.common.ui.viewer.IViewerProvider; |
| import org.eclipse.emf.common.util.Diagnostic; |
| import org.eclipse.emf.common.util.DiagnosticChain; |
| import org.eclipse.emf.common.util.TreeIterator; |
| |
| import org.eclipse.emf.ecore.EClass; |
| import org.eclipse.emf.ecore.EObject; |
| import org.eclipse.emf.ecore.resource.Resource; |
| import org.eclipse.emf.ecore.util.Diagnostician; |
| import org.eclipse.emf.ecore.util.EcoreUtil; |
| import org.eclipse.emf.edit.command.AddCommand; |
| import org.eclipse.emf.edit.command.RemoveCommand; |
| import org.eclipse.emf.edit.domain.AdapterFactoryEditingDomain; |
| import org.eclipse.emf.edit.domain.EditingDomain; |
| import org.eclipse.emf.edit.domain.IEditingDomainProvider; |
| |
| import org.eclipse.emf.edit.provider.IItemLabelProvider; |
| import org.eclipse.emf.edit.ui.EMFEditUIPlugin; |
| import org.eclipse.emf.edit.ui.action.ControlAction; |
| import org.eclipse.emf.edit.ui.action.CreateChildAction; |
| import org.eclipse.emf.edit.ui.action.CreateSiblingAction; |
| import org.eclipse.emf.edit.ui.action.EditingDomainActionBarContributor; |
| import org.eclipse.emf.edit.ui.action.LoadResourceAction; |
| import org.eclipse.emf.edit.ui.action.ValidateAction; |
| |
| import org.eclipse.jface.action.Action; |
| import org.eclipse.jface.action.ActionContributionItem; |
| import org.eclipse.jface.action.IAction; |
| import org.eclipse.jface.action.IContributionItem; |
| import org.eclipse.jface.action.IContributionManager; |
| import org.eclipse.jface.action.IMenuListener; |
| import org.eclipse.jface.action.IMenuManager; |
| import org.eclipse.jface.action.IToolBarManager; |
| import org.eclipse.jface.action.MenuManager; |
| import org.eclipse.jface.action.Separator; |
| import org.eclipse.jface.action.SubContributionItem; |
| |
| import org.eclipse.jface.viewers.ISelection; |
| import org.eclipse.jface.viewers.ISelectionChangedListener; |
| import org.eclipse.jface.viewers.ISelectionProvider; |
| import org.eclipse.jface.viewers.IStructuredSelection; |
| import org.eclipse.jface.viewers.SelectionChangedEvent; |
| import org.eclipse.jface.viewers.Viewer; |
| |
| import org.eclipse.ui.IEditorPart; |
| import org.eclipse.ui.PartInitException; |
| |
| import org.eclipse.uml2.common.edit.command.ChangeCommand; |
| import org.eclipse.uml2.uml.Element; |
| import org.eclipse.uml2.uml.editor.UMLEditorPlugin; |
| import org.eclipse.uml2.uml.util.UMLUtil; |
| |
| /** |
| * This is the action bar contributor for the UML model editor. |
| * <!-- begin-user-doc --> |
| * <!-- end-user-doc --> |
| * @generated |
| */ |
| public class UMLActionBarContributor |
| extends EditingDomainActionBarContributor |
| implements ISelectionChangedListener { |
| |
| /** |
| * This keeps track of the active editor. |
| * <!-- begin-user-doc --> |
| * <!-- end-user-doc --> |
| * @generated |
| */ |
| protected IEditorPart activeEditorPart; |
| |
| /** |
| * This keeps track of the current selection provider. |
| * <!-- begin-user-doc --> |
| * <!-- end-user-doc --> |
| * @generated |
| */ |
| protected ISelectionProvider selectionProvider; |
| |
| /** |
| * This action opens the Properties view. |
| * <!-- begin-user-doc --> |
| * <!-- end-user-doc --> |
| * @generated |
| */ |
| protected IAction showPropertiesViewAction = new Action( |
| UMLEditorPlugin.INSTANCE.getString("_UI_ShowPropertiesView_menu_item")) //$NON-NLS-1$ |
| { |
| |
| @Override |
| public void run() { |
| try { |
| getPage().showView("org.eclipse.ui.views.PropertySheet"); //$NON-NLS-1$ |
| } catch (PartInitException exception) { |
| UMLEditorPlugin.INSTANCE.log(exception); |
| } |
| } |
| }; |
| |
| /** |
| * This action refreshes the viewer of the current editor if the editor |
| * implements {@link org.eclipse.emf.common.ui.viewer.IViewerProvider}. |
| * <!-- begin-user-doc --> |
| * <!-- end-user-doc --> |
| * @generated |
| */ |
| protected IAction refreshViewerAction = new Action( |
| UMLEditorPlugin.INSTANCE.getString("_UI_RefreshViewer_menu_item")) //$NON-NLS-1$ |
| { |
| |
| @Override |
| public boolean isEnabled() { |
| return activeEditorPart instanceof IViewerProvider; |
| } |
| |
| @Override |
| public void run() { |
| if (activeEditorPart instanceof IViewerProvider) { |
| Viewer viewer = ((IViewerProvider) activeEditorPart) |
| .getViewer(); |
| if (viewer != null) { |
| viewer.refresh(); |
| } |
| } |
| } |
| }; |
| |
| /** |
| * This will contain one {@link org.eclipse.emf.edit.ui.action.CreateChildAction} corresponding to each descriptor |
| * generated for the current selection by the item provider. |
| * <!-- begin-user-doc --> |
| * <!-- end-user-doc --> |
| * @generated |
| */ |
| protected Collection<IAction> createChildActions; |
| |
| /** |
| * This will contain a map of {@link org.eclipse.emf.edit.ui.action.CreateChildAction}s, keyed by sub-menu text. |
| * <!-- begin-user-doc --> |
| * <!-- end-user-doc --> |
| * @generated |
| */ |
| protected Map<String, Collection<IAction>> createChildSubmenuActions; |
| |
| /** |
| * This is the menu manager into which menu contribution items should be added for CreateChild actions. |
| * <!-- begin-user-doc --> |
| * <!-- end-user-doc --> |
| * @generated |
| */ |
| protected IMenuManager createChildMenuManager; |
| |
| /** |
| * This will contain one {@link org.eclipse.emf.edit.ui.action.CreateSiblingAction} corresponding to each descriptor |
| * generated for the current selection by the item provider. |
| * <!-- begin-user-doc --> |
| * <!-- end-user-doc --> |
| * @generated |
| */ |
| protected Collection<IAction> createSiblingActions; |
| |
| /** |
| * This will contain a map of {@link org.eclipse.emf.edit.ui.action.CreateSiblingAction}s, keyed by submenu text. |
| * <!-- begin-user-doc --> |
| * <!-- end-user-doc --> |
| * @generated |
| */ |
| protected Map<String, Collection<IAction>> createSiblingSubmenuActions; |
| |
| /** |
| * This is the menu manager into which menu contribution items should be added for CreateSibling actions. |
| * <!-- begin-user-doc --> |
| * <!-- end-user-doc --> |
| * @generated |
| */ |
| protected IMenuManager createSiblingMenuManager; |
| |
| /** |
| * This creates an instance of the contributor. |
| * <!-- begin-user-doc --> |
| * <!-- end-user-doc --> |
| * @generated NOT |
| */ |
| public UMLActionBarContributor() { |
| super(ADDITIONS_LAST_STYLE); |
| |
| loadResourceAction = new LoadResourceAction(); |
| |
| validateAction = new UMLValidateAction(); |
| controlAction = new UMLControlAction(); |
| } |
| |
| /** |
| * This adds Separators for editor additions to the tool bar. |
| * <!-- begin-user-doc --> |
| * <!-- end-user-doc --> |
| * @generated |
| */ |
| @Override |
| public void contributeToToolBar(IToolBarManager toolBarManager) { |
| super.contributeToToolBar(toolBarManager); |
| toolBarManager.add(new Separator("uml-settings")); //$NON-NLS-1$ |
| toolBarManager.add(new Separator("uml-additions")); //$NON-NLS-1$ |
| } |
| |
| /** |
| * This adds to the menu bar a menu and some separators for editor additions, |
| * as well as the sub-menus for object creation items. |
| * <!-- begin-user-doc --> |
| * <!-- end-user-doc --> |
| * @generated |
| */ |
| @Override |
| public void contributeToMenu(IMenuManager menuManager) { |
| super.contributeToMenu(menuManager); |
| |
| IMenuManager submenuManager = new MenuManager( |
| UMLEditorPlugin.INSTANCE.getString("_UI_UMLEditor_menu"), //$NON-NLS-1$ |
| "org.eclipse.uml2.umlMenuID"); //$NON-NLS-1$ |
| menuManager.insertAfter("additions", submenuManager); //$NON-NLS-1$ |
| submenuManager.add(new Separator("settings")); //$NON-NLS-1$ |
| submenuManager.add(new Separator("actions")); //$NON-NLS-1$ |
| submenuManager.add(new Separator("additions")); //$NON-NLS-1$ |
| submenuManager.add(new Separator("additions-end")); //$NON-NLS-1$ |
| |
| // Prepare for CreateChild item addition or removal. |
| // |
| createChildMenuManager = new MenuManager( |
| UMLEditorPlugin.INSTANCE.getString("_UI_CreateChild_menu_item")); //$NON-NLS-1$ |
| submenuManager.insertBefore("additions", createChildMenuManager); //$NON-NLS-1$ |
| |
| // Prepare for CreateSibling item addition or removal. |
| // |
| createSiblingMenuManager = new MenuManager( |
| UMLEditorPlugin.INSTANCE.getString("_UI_CreateSibling_menu_item")); //$NON-NLS-1$ |
| submenuManager.insertBefore("additions", createSiblingMenuManager); //$NON-NLS-1$ |
| |
| // Force an update because Eclipse hides empty menus now. |
| // |
| submenuManager.addMenuListener(new IMenuListener() { |
| |
| public void menuAboutToShow(IMenuManager menuManager) { |
| menuManager.updateAll(true); |
| } |
| }); |
| |
| addGlobalActions(submenuManager); |
| } |
| |
| /** |
| * When the active editor changes, this remembers the change and registers with it as a selection provider. |
| * <!-- begin-user-doc --> |
| * <!-- end-user-doc --> |
| * @generated |
| */ |
| @Override |
| public void setActiveEditor(IEditorPart part) { |
| super.setActiveEditor(part); |
| activeEditorPart = part; |
| |
| // Switch to the new selection provider. |
| // |
| if (selectionProvider != null) { |
| selectionProvider.removeSelectionChangedListener(this); |
| } |
| if (part == null) { |
| selectionProvider = null; |
| } else { |
| selectionProvider = part.getSite().getSelectionProvider(); |
| selectionProvider.addSelectionChangedListener(this); |
| |
| // Fake a selection changed event to update the menus. |
| // |
| if (selectionProvider.getSelection() != null) { |
| selectionChanged(new SelectionChangedEvent(selectionProvider, |
| selectionProvider.getSelection())); |
| } |
| } |
| } |
| |
| /** |
| * This implements {@link org.eclipse.jface.viewers.ISelectionChangedListener}, |
| * handling {@link org.eclipse.jface.viewers.SelectionChangedEvent}s by querying for the children and siblings |
| * that can be added to the selected object and updating the menus accordingly. |
| * <!-- begin-user-doc --> |
| * <!-- end-user-doc --> |
| * @generated |
| */ |
| public void selectionChanged(SelectionChangedEvent event) { |
| // Remove any menu items for old selection. |
| // |
| if (createChildMenuManager != null) { |
| depopulateManager(createChildMenuManager, |
| createChildSubmenuActions); |
| depopulateManager(createChildMenuManager, createChildActions); |
| } |
| if (createSiblingMenuManager != null) { |
| depopulateManager(createSiblingMenuManager, |
| createSiblingSubmenuActions); |
| depopulateManager(createSiblingMenuManager, createSiblingActions); |
| } |
| |
| // Query the new selection for appropriate new child/sibling descriptors |
| // |
| Collection<?> newChildDescriptors = null; |
| Collection<?> newSiblingDescriptors = null; |
| |
| ISelection selection = event.getSelection(); |
| if (selection instanceof IStructuredSelection |
| && ((IStructuredSelection) selection).size() == 1) { |
| Object object = ((IStructuredSelection) selection) |
| .getFirstElement(); |
| |
| EditingDomain domain = ((IEditingDomainProvider) activeEditorPart) |
| .getEditingDomain(); |
| |
| newChildDescriptors = domain.getNewChildDescriptors(object, null); |
| newSiblingDescriptors = domain.getNewChildDescriptors(null, object); |
| } |
| |
| // Generate actions for selection; populate and redraw the menus. |
| // |
| createChildActions = generateCreateChildActions(newChildDescriptors, |
| selection); |
| createChildSubmenuActions = extractSubmenuActions(createChildActions); |
| createSiblingActions = generateCreateSiblingActions( |
| newSiblingDescriptors, selection); |
| createSiblingSubmenuActions = extractSubmenuActions( |
| createSiblingActions); |
| |
| if (createChildMenuManager != null) { |
| populateManager(createChildMenuManager, createChildSubmenuActions, |
| null); |
| populateManager(createChildMenuManager, createChildActions, null); |
| createChildMenuManager.update(true); |
| } |
| if (createSiblingMenuManager != null) { |
| populateManager(createSiblingMenuManager, |
| createSiblingSubmenuActions, null); |
| populateManager(createSiblingMenuManager, createSiblingActions, |
| null); |
| createSiblingMenuManager.update(true); |
| } |
| } |
| |
| /** |
| * This generates a {@link org.eclipse.emf.edit.ui.action.CreateChildAction} for each object in <code>descriptors</code>, |
| * and returns the collection of these actions. |
| * <!-- begin-user-doc --> |
| * <!-- end-user-doc --> |
| * @generated |
| */ |
| protected Collection<IAction> generateCreateChildActionsGen( |
| Collection<?> descriptors, ISelection selection) { |
| Collection<IAction> actions = new ArrayList<IAction>(); |
| if (descriptors != null) { |
| for (Object descriptor : descriptors) { |
| actions.add(new CreateChildAction(activeEditorPart, selection, |
| descriptor)); |
| } |
| } |
| return actions; |
| } |
| |
| protected Collection<IAction> generateCreateChildActions( |
| Collection<?> descriptors, ISelection selection) { |
| List<IAction> createChildActions = (List<IAction>) generateCreateChildActionsGen( |
| descriptors, selection); |
| |
| Collections.<IAction> sort(createChildActions, |
| new Comparator<IAction>() { |
| |
| public int compare(IAction a1, IAction a2) { |
| return CommonPlugin.INSTANCE.getComparator() |
| .compare(a1.getText(), a2.getText()); |
| } |
| }); |
| |
| return createChildActions; |
| } |
| |
| /** |
| * This generates a {@link org.eclipse.emf.edit.ui.action.CreateSiblingAction} for each object in <code>descriptors</code>, |
| * and returns the collection of these actions. |
| * <!-- begin-user-doc --> |
| * <!-- end-user-doc --> |
| * @generated |
| */ |
| protected Collection<IAction> generateCreateSiblingActionsGen( |
| Collection<?> descriptors, ISelection selection) { |
| Collection<IAction> actions = new ArrayList<IAction>(); |
| if (descriptors != null) { |
| for (Object descriptor : descriptors) { |
| actions.add(new CreateSiblingAction(activeEditorPart, selection, |
| descriptor)); |
| } |
| } |
| return actions; |
| } |
| |
| protected Collection<IAction> generateCreateSiblingActions( |
| Collection<?> descriptors, ISelection selection) { |
| List<IAction> createSiblingActions = (List<IAction>) generateCreateSiblingActionsGen( |
| descriptors, selection); |
| |
| Collections.<IAction> sort(createSiblingActions, |
| new Comparator<IAction>() { |
| |
| public int compare(IAction a1, IAction a2) { |
| return CommonPlugin.INSTANCE.getComparator() |
| .compare(a1.getText(), a2.getText()); |
| } |
| }); |
| |
| return createSiblingActions; |
| } |
| |
| /** |
| * This populates the specified <code>manager</code> with {@link org.eclipse.jface.action.ActionContributionItem}s |
| * based on the {@link org.eclipse.jface.action.IAction}s contained in the <code>actions</code> collection, |
| * by inserting them before the specified contribution item <code>contributionID</code>. |
| * If <code>contributionID</code> is <code>null</code>, they are simply added. |
| * <!-- begin-user-doc --> |
| * <!-- end-user-doc --> |
| * @generated |
| */ |
| protected void populateManager(IContributionManager manager, |
| Collection<? extends IAction> actions, String contributionID) { |
| if (actions != null) { |
| for (IAction action : actions) { |
| if (contributionID != null) { |
| manager.insertBefore(contributionID, action); |
| } else { |
| manager.add(action); |
| } |
| } |
| } |
| } |
| |
| /** |
| * This removes from the specified <code>manager</code> all {@link org.eclipse.jface.action.ActionContributionItem}s |
| * based on the {@link org.eclipse.jface.action.IAction}s contained in the <code>actions</code> collection. |
| * <!-- begin-user-doc --> |
| * <!-- end-user-doc --> |
| * @generated |
| */ |
| protected void depopulateManager(IContributionManager manager, |
| Collection<? extends IAction> actions) { |
| if (actions != null) { |
| IContributionItem[] items = manager.getItems(); |
| for (int i = 0; i < items.length; i++) { |
| // Look into SubContributionItems |
| // |
| IContributionItem contributionItem = items[i]; |
| while (contributionItem instanceof SubContributionItem) { |
| contributionItem = ((SubContributionItem) contributionItem) |
| .getInnerItem(); |
| } |
| |
| // Delete the ActionContributionItems with matching action. |
| // |
| if (contributionItem instanceof ActionContributionItem) { |
| IAction action = ((ActionContributionItem) contributionItem) |
| .getAction(); |
| if (actions.contains(action)) { |
| manager.remove(contributionItem); |
| } |
| } |
| } |
| } |
| } |
| |
| /** |
| * This extracts those actions in the <code>submenuActions</code> collection whose text is qualified and returns |
| * a map of these actions, keyed by submenu text. |
| * <!-- begin-user-doc --> |
| * <!-- end-user-doc --> |
| * @generated |
| */ |
| protected Map<String, Collection<IAction>> extractSubmenuActions( |
| Collection<IAction> createActions) { |
| Map<String, Collection<IAction>> createSubmenuActions = new LinkedHashMap<String, Collection<IAction>>(); |
| if (createActions != null) { |
| for (Iterator<IAction> actions = createActions.iterator(); actions |
| .hasNext();) { |
| IAction action = actions.next(); |
| StringTokenizer st = new StringTokenizer(action.getText(), "|"); //$NON-NLS-1$ |
| if (st.countTokens() == 2) { |
| String text = st.nextToken().trim(); |
| Collection<IAction> submenuActions = createSubmenuActions |
| .get(text); |
| if (submenuActions == null) { |
| createSubmenuActions.put(text, |
| submenuActions = new ArrayList<IAction>()); |
| } |
| action.setText(st.nextToken().trim()); |
| submenuActions.add(action); |
| actions.remove(); |
| } |
| } |
| } |
| return createSubmenuActions; |
| } |
| |
| /** |
| * This populates the specified <code>manager</code> with {@link org.eclipse.jface.action.MenuManager}s containing |
| * {@link org.eclipse.jface.action.ActionContributionItem}s based on the {@link org.eclipse.jface.action.IAction}s |
| * contained in the <code>submenuActions</code> collection, by inserting them before the specified contribution |
| * item <code>contributionID</code>. |
| * If <code>contributionID</code> is <code>null</code>, they are simply added. |
| * <!-- begin-user-doc --> |
| * <!-- end-user-doc --> |
| * @generated |
| */ |
| protected void populateManager(IContributionManager manager, |
| Map<String, Collection<IAction>> submenuActions, |
| String contributionID) { |
| if (submenuActions != null) { |
| for (Map.Entry<String, Collection<IAction>> entry : submenuActions |
| .entrySet()) { |
| MenuManager submenuManager = new MenuManager(entry.getKey()); |
| if (contributionID != null) { |
| manager.insertBefore(contributionID, submenuManager); |
| } else { |
| manager.add(submenuManager); |
| } |
| populateManager(submenuManager, entry.getValue(), null); |
| } |
| } |
| } |
| |
| /** |
| * This removes from the specified <code>manager</code> all {@link org.eclipse.jface.action.MenuManager}s and their |
| * {@link org.eclipse.jface.action.ActionContributionItem}s based on the {@link org.eclipse.jface.action.IAction}s |
| * contained in the <code>submenuActions</code> map. |
| * <!-- begin-user-doc --> |
| * <!-- end-user-doc --> |
| * @generated |
| */ |
| protected void depopulateManager(IContributionManager manager, |
| Map<String, Collection<IAction>> submenuActions) { |
| if (submenuActions != null) { |
| IContributionItem[] items = manager.getItems(); |
| for (int i = 0; i < items.length; i++) { |
| IContributionItem contributionItem = items[i]; |
| if (contributionItem instanceof MenuManager) { |
| MenuManager submenuManager = (MenuManager) contributionItem; |
| if (submenuActions |
| .containsKey(submenuManager.getMenuText())) { |
| depopulateManager(submenuManager, |
| submenuActions.get(contributionItem)); |
| manager.remove(contributionItem); |
| } |
| } |
| } |
| } |
| } |
| |
| /** |
| * This populates the pop-up menu before it appears. |
| * <!-- begin-user-doc --> |
| * <!-- end-user-doc --> |
| * @generated |
| */ |
| @Override |
| public void menuAboutToShow(IMenuManager menuManager) { |
| super.menuAboutToShow(menuManager); |
| MenuManager submenuManager = null; |
| |
| submenuManager = new MenuManager( |
| UMLEditorPlugin.INSTANCE.getString("_UI_CreateChild_menu_item")); //$NON-NLS-1$ |
| populateManager(submenuManager, createChildSubmenuActions, null); |
| populateManager(submenuManager, createChildActions, null); |
| menuManager.insertBefore("edit", submenuManager); //$NON-NLS-1$ |
| |
| submenuManager = new MenuManager( |
| UMLEditorPlugin.INSTANCE.getString("_UI_CreateSibling_menu_item")); //$NON-NLS-1$ |
| populateManager(submenuManager, createSiblingSubmenuActions, null); |
| populateManager(submenuManager, createSiblingActions, null); |
| menuManager.insertBefore("edit", submenuManager); //$NON-NLS-1$ |
| } |
| |
| /** |
| * This inserts global actions before the "additions-end" separator. |
| * <!-- begin-user-doc --> |
| * <!-- end-user-doc --> |
| * @generated |
| */ |
| @Override |
| protected void addGlobalActions(IMenuManager menuManager) { |
| menuManager.insertAfter("additions-end", new Separator("ui-actions")); //$NON-NLS-1$ //$NON-NLS-2$ |
| menuManager.insertAfter("ui-actions", showPropertiesViewAction); //$NON-NLS-1$ |
| |
| refreshViewerAction.setEnabled(refreshViewerAction.isEnabled()); |
| menuManager.insertAfter("ui-actions", refreshViewerAction); //$NON-NLS-1$ |
| |
| super.addGlobalActions(menuManager); |
| } |
| |
| /** |
| * This ensures that a delete action will clean up all references to deleted objects. |
| * <!-- begin-user-doc --> |
| * <!-- end-user-doc --> |
| * @generated |
| */ |
| @Override |
| protected boolean removeAllReferencesOnDelete() { |
| return true; |
| } |
| |
| protected static class UMLControlAction |
| extends ControlAction { |
| |
| protected List<EObject> collectAllStereotypeApplications( |
| EObject eObject, List<EObject> allStereotypeApplications) { |
| |
| if (eObject instanceof Element) { |
| |
| for (EObject stereotypeApplication : ((Element) eObject) |
| .getStereotypeApplications()) { |
| |
| allStereotypeApplications.add(stereotypeApplication); |
| |
| collectAllStereotypeApplications(stereotypeApplication, |
| allStereotypeApplications); |
| } |
| } |
| |
| for (TreeIterator<EObject> allProperContents = EcoreUtil |
| .getAllProperContents(eObject, true); allProperContents |
| .hasNext();) { |
| |
| EObject content = allProperContents.next(); |
| |
| if (content instanceof Element) { |
| |
| for (EObject stereotypeApplication : ((Element) content) |
| .getStereotypeApplications()) { |
| |
| allStereotypeApplications.add(stereotypeApplication); |
| |
| collectAllStereotypeApplications(stereotypeApplication, |
| allStereotypeApplications); |
| } |
| } |
| } |
| |
| return allStereotypeApplications; |
| } |
| |
| @Override |
| public boolean updateSelection(IStructuredSelection selection) { |
| this.selection = selection; |
| |
| if (selection.size() != 1) { |
| return false; |
| } |
| |
| Object object = AdapterFactoryEditingDomain |
| .unwrap(selection.getFirstElement()); |
| boolean result = domain.isControllable(object); |
| eObject = result |
| ? (EObject) object |
| : null; |
| |
| if (!AdapterFactoryEditingDomain.isControlled(object)) { |
| setText(EMFEditUIPlugin.INSTANCE |
| .getString("_UI_Control_menu_item")); //$NON-NLS-1$ |
| setDescription("_UI_Control_menu_item_description"); //$NON-NLS-1$ |
| |
| command = null; |
| } else { |
| setText(EMFEditUIPlugin.INSTANCE |
| .getString("_UI_Uncontrol_menu_item")); //$NON-NLS-1$ |
| setDescription("_UI_Uncontrol_menu_item_description"); //$NON-NLS-1$ |
| |
| if (result) { |
| CompoundCommand compoundCommand = new CompoundCommand( |
| EMFEditUIPlugin.INSTANCE |
| .getString("_UI_UncontrolCommand_label")); //$NON-NLS-1$ |
| |
| compoundCommand.append(new RemoveCommand(domain, |
| eObject.eResource().getContents(), eObject)); |
| |
| final List<EObject> allStereotypeApplications = collectAllStereotypeApplications( |
| eObject, new ArrayList<EObject>()); |
| |
| compoundCommand.append(new CommandWrapper() { |
| |
| @Override |
| protected Command createCommand() { |
| return new ChangeCommand(domain, new Runnable() { |
| |
| public void run() { |
| |
| for (EObject stereotypeApplication : allStereotypeApplications) { |
| UMLUtil.StereotypeApplicationHelper |
| .getInstance(null) |
| .addToContainmentList( |
| UMLUtil.getBaseElement( |
| stereotypeApplication), |
| stereotypeApplication, null); |
| } |
| } |
| }); |
| } |
| }); |
| |
| command = compoundCommand; |
| |
| result = command.canExecute(); |
| } |
| } |
| |
| return result; |
| } |
| |
| @Override |
| public void run() { |
| |
| if (command == null) { |
| |
| if (eObject == null) { |
| return; |
| } |
| |
| Resource resource = getResource(); |
| |
| if (resource == null) { |
| return; |
| } |
| |
| CompoundCommand compoundCommand = new CompoundCommand( |
| EMFEditUIPlugin.INSTANCE |
| .getString("_UI_ControlCommand_label")); //$NON-NLS-1$ |
| |
| compoundCommand.append( |
| new AddCommand(domain, resource.getContents(), eObject)); |
| |
| final List<EObject> allStereotypeApplications = collectAllStereotypeApplications( |
| eObject, new ArrayList<EObject>()); |
| |
| compoundCommand.append(new CommandWrapper() { |
| |
| @Override |
| protected Command createCommand() { |
| return new ChangeCommand(domain, new Runnable() { |
| |
| public void run() { |
| |
| for (EObject stereotypeApplication : allStereotypeApplications) { |
| UMLUtil.StereotypeApplicationHelper |
| .getInstance(null).addToContainmentList( |
| UMLUtil.getBaseElement( |
| stereotypeApplication), |
| stereotypeApplication, null); |
| } |
| } |
| }); |
| } |
| }); |
| |
| command = compoundCommand; |
| } |
| |
| super.run(); |
| } |
| |
| } |
| |
| protected static class UMLValidateAction |
| extends ValidateAction { |
| |
| protected int count(EObject eObject) { |
| int count = 1; |
| |
| if (eObject instanceof Element) { |
| |
| for (EObject stereotypeApplication : ((Element) eObject) |
| .getStereotypeApplications()) { |
| |
| count += count(stereotypeApplication); |
| } |
| } |
| |
| for (EObject eContent : eObject.eContents()) { |
| count += count(eContent); |
| } |
| |
| return count; |
| } |
| |
| @Override |
| protected Diagnostic validate(final IProgressMonitor progressMonitor) { |
| EObject eObject = (EObject) selectedObjects.iterator().next(); |
| |
| progressMonitor.beginTask("", count(eObject)); //$NON-NLS-1$ |
| |
| final AdapterFactory adapterFactory = domain instanceof AdapterFactoryEditingDomain |
| ? ((AdapterFactoryEditingDomain) domain).getAdapterFactory() |
| : null; |
| |
| Diagnostician diagnostician = new Diagnostician() { |
| |
| @Override |
| public String getObjectLabel(EObject eObject) { |
| |
| if (adapterFactory != null && !eObject.eIsProxy()) { |
| IItemLabelProvider itemLabelProvider = (IItemLabelProvider) adapterFactory |
| .adapt(eObject, IItemLabelProvider.class); |
| |
| if (itemLabelProvider != null) { |
| return itemLabelProvider.getText(eObject); |
| } |
| } |
| |
| return super.getObjectLabel(eObject); |
| } |
| |
| protected boolean doValidateStereotypeApplications( |
| EObject eObject, DiagnosticChain diagnostics, |
| Map<Object, Object> context) { |
| List<EObject> stereotypeApplications = eObject instanceof Element |
| ? ((Element) eObject).getStereotypeApplications() |
| : Collections.<EObject> emptyList(); |
| |
| if (!stereotypeApplications.isEmpty()) { |
| Iterator<EObject> i = stereotypeApplications.iterator(); |
| boolean result = validate(i.next(), diagnostics, |
| context); |
| |
| while (i.hasNext() && (result || diagnostics != null)) { |
| result &= validate(i.next(), diagnostics, context); |
| } |
| |
| return result; |
| } else { |
| return true; |
| } |
| } |
| |
| @Override |
| protected boolean doValidateContents(EObject eObject, |
| DiagnosticChain diagnostics, |
| Map<Object, Object> context) { |
| boolean result = doValidateStereotypeApplications(eObject, |
| diagnostics, context); |
| |
| if (result || diagnostics != null) { |
| result &= super.doValidateContents(eObject, diagnostics, |
| context); |
| } |
| |
| return result; |
| } |
| |
| @Override |
| public boolean validate(EClass eClass, EObject eObject, |
| DiagnosticChain diagnostics, |
| Map<Object, Object> context) { |
| progressMonitor.worked(1); |
| return super.validate(eClass, eObject, diagnostics, |
| context); |
| } |
| }; |
| |
| progressMonitor.setTaskName(EMFEditUIPlugin.INSTANCE |
| .getString("_UI_Validating_message", new Object[]{diagnostician //$NON-NLS-1$ |
| .getObjectLabel(eObject)})); |
| |
| return diagnostician.validate(eObject); |
| } |
| } |
| |
| } |
