blob: 34ef5dbb901dfe47db3ff643b8b5d42f5b94ac68 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
|
/**
* Copyright (c) 2017, 2019 CEA LIST.
*
* All rights reserved. This program and the accompanying materials
* are made available under the terms of the Eclipse Public License 2.0
* which accompanies this distribution, and is available at
* https://www.eclipse.org/legal/epl-2.0/
*
* SPDX-License-Identifier: EPL-2.0
*
* Contributors:
* Maged Elaasar - Initial API and implementation
* Nicolas FAUVERGUE (CEA LIST) nicolas.fauvergue@cea.fr - Bug 550535
*
*/
package org.eclipse.papyrus.infra.ui.architecture.handlers;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import java.util.stream.Collectors;
import org.eclipse.emf.common.command.CompoundCommand;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.emf.ecore.resource.ResourceSet;
import org.eclipse.emf.edit.provider.ComposedAdapterFactory;
import org.eclipse.emf.edit.ui.provider.AdapterFactoryLabelProvider;
import org.eclipse.emf.edit.ui.provider.ExtendedImageRegistry;
import org.eclipse.emf.transaction.TransactionalEditingDomain;
import org.eclipse.jface.action.Action;
import org.eclipse.jface.action.ActionContributionItem;
import org.eclipse.jface.action.IContributionItem;
import org.eclipse.jface.resource.ImageDescriptor;
import org.eclipse.jface.viewers.ILabelProvider;
import org.eclipse.jface.viewers.ISelection;
import org.eclipse.jface.viewers.IStructuredSelection;
import org.eclipse.papyrus.infra.architecture.ArchitectureDescriptionUtils;
import org.eclipse.papyrus.infra.core.architecture.merged.MergedArchitectureContext;
import org.eclipse.papyrus.infra.core.architecture.merged.MergedArchitectureViewpoint;
import org.eclipse.papyrus.infra.core.resource.ModelSet;
import org.eclipse.papyrus.infra.emf.utils.EMFHelper;
import org.eclipse.papyrus.infra.ui.architecture.commands.CloseEditorsForViewpointsCommand;
import org.eclipse.ui.IWorkbenchWindow;
import org.eclipse.ui.PlatformUI;
import org.eclipse.ui.actions.CompoundContributionItem;
/**
* A command handler for changing architecture viewpoints in a model set
*
* @since 1.0
*/
public class ChangeArchitectureViewpointsHandler extends CompoundContributionItem {
// the label provider for architecture viewpoints
private ILabelProvider provider;
public ChangeArchitectureViewpointsHandler() {
ComposedAdapterFactory composedAdapterFactory = new ComposedAdapterFactory(ComposedAdapterFactory.Descriptor.Registry.INSTANCE);
provider = new AdapterFactoryLabelProvider(composedAdapterFactory);
}
@Override
protected IContributionItem[] getContributionItems() {
final EObject selection = getSelection();
if (selection == null) {
return new IContributionItem[0];
}
ResourceSet resourceSet = selection.eResource().getResourceSet();
if (!(resourceSet instanceof ModelSet)) {
return new IContributionItem[0];
}
final ArchitectureDescriptionUtils helper = new ArchitectureDescriptionUtils((ModelSet) resourceSet);
MergedArchitectureContext context = helper.getArchitectureContext();
final Set<String> viewpointIds = new HashSet<>(helper.getArchitectureViewpointIds());
List<IContributionItem> items = new ArrayList<>();
for (MergedArchitectureViewpoint viewpoint : context.getViewpoints()) {
Object imageObject = viewpoint.getImageObject();
ImageDescriptor desc = ExtendedImageRegistry.getInstance().getImageDescriptor(provider.getImage(imageObject));
items.add(new ActionContributionItem(new Action(viewpoint.getName(), desc) {
{
setChecked(viewpointIds.contains(viewpoint.getId()));
}
@Override
public void run() {
final Collection<MergedArchitectureViewpoint> viewpoints = helper.getArchitectureViewpoints();
CompoundCommand cc = new CompoundCommand();
if (!isChecked()) {
viewpointIds.remove(viewpoint.getId());
viewpoints.remove(viewpoint);
} else {
viewpointIds.add(viewpoint.getId());
viewpoints.add(viewpoint);
}
TransactionalEditingDomain ted = helper.getModelSet().getTransactionalEditingDomain();
// More than set the architecture viewpoints used, close needed editors
// It is needed to be the first action to avoid opened dialog for viewpoints selection
cc.append(new CloseEditorsForViewpointsCommand(viewpoints.stream().filter(viewpoint -> viewpointIds.contains(viewpoint.getId())).collect(Collectors.toList())));
cc.append(helper.switchArchitectureViewpointIds(viewpointIds.toArray(new String[0])));
ted.getCommandStack().execute(cc);
}
}));
}
return items.toArray(new IContributionItem[0]);
}
/**
* Gets the current selection as an EObject
*
* @return The current selection, or <code>null</code> if it is not an EObject
*/
protected EObject getSelection() {
IWorkbenchWindow window = PlatformUI.getWorkbench().getActiveWorkbenchWindow();
if (window == null) {
return null;
}
ISelection selection = window.getSelectionService().getSelection();
if (selection == null) {
return null;
}
if (selection.isEmpty()) {
return null;
}
if (selection instanceof IStructuredSelection) {
IStructuredSelection struct = (IStructuredSelection) selection;
Object obj = struct.getFirstElement();
return EMFHelper.getEObject(obj);
}
return null;
}
}
|
