-
Notifications
You must be signed in to change notification settings - Fork 85
/
FunctionalExchangeStyleConfiguration.java
76 lines (69 loc) · 3.52 KB
/
FunctionalExchangeStyleConfiguration.java
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
/*******************************************************************************
* Copyright (c) 2021 THALES GLOBAL SERVICES.
*
* This program and the accompanying materials are made available under the
* terms of the Eclipse Public License 2.0 which is available at
* http://www.eclipse.org/legal/epl-2.0
*
* SPDX-License-Identifier: EPL-2.0
*
* Contributors:
* Thales - initial API and implementation
*******************************************************************************/
package org.polarsys.capella.core.sirius.analysis.extension.style;
import java.util.HashMap;
import java.util.Set;
import java.util.stream.Collectors;
import org.eclipse.gmf.runtime.diagram.ui.editparts.IGraphicalEditPart;
import org.eclipse.sirius.diagram.DDiagramElement;
import org.eclipse.sirius.diagram.DEdge;
import org.eclipse.sirius.diagram.DNode;
import org.eclipse.sirius.diagram.description.DiagramElementMapping;
import org.eclipse.sirius.diagram.description.EdgeMapping;
import org.eclipse.sirius.diagram.ui.internal.edit.parts.DEdgeBeginNameEditPart;
import org.eclipse.sirius.diagram.ui.internal.edit.parts.DEdgeEndNameEditPart;
import org.eclipse.sirius.diagram.ui.tools.api.graphical.edit.styles.IStyleConfigurationProvider;
import org.eclipse.sirius.diagram.ui.tools.api.graphical.edit.styles.SimpleStyleConfiguration;
import org.eclipse.sirius.diagram.ui.tools.api.graphical.edit.styles.StyleConfiguration;
import org.eclipse.sirius.viewpoint.Style;
import org.eclipse.swt.graphics.Image;
import org.polarsys.capella.core.data.fa.FunctionalChain;
import org.polarsys.capella.core.sirius.analysis.FunctionalChainServices;
import org.polarsys.capella.core.sirius.analysis.cache.DEdgeIconCache;
/**
* The style configuration for functional exchanges
*/
@SuppressWarnings("restriction")
public class FunctionalExchangeStyleConfiguration extends SimpleStyleConfiguration implements IStyleConfigurationProvider {
@Override
public StyleConfiguration createStyleConfiguration(DiagramElementMapping mapping, Style style) {
return this;
}
@Override
public Image getLabelIcon(DDiagramElement representationElement, IGraphicalEditPart editPart) {
if (representationElement instanceof DEdge
&& (editPart instanceof DEdgeBeginNameEditPart || editPart instanceof DEdgeEndNameEditPart)
&& isShowIcon(representationElement, editPart)) {
Image labelIcon = DEdgeIconCache.getInstance().getIcon((DEdge) representationElement);
if (labelIcon == null && representationElement.getSemanticElements().size() > 1) {
DEdge edge = (DEdge) representationElement;
Set<FunctionalChain> chains = edge.getSemanticElements().stream().filter(FunctionalChain.class::isInstance).map(FunctionalChain.class::cast).collect(Collectors.toSet());
HashMap<FunctionalChain, DNode> functionalChains = FunctionalChainServices.getFunctionalChainServices()
.getDisplayedFunctionalChains(representationElement.getParentDiagram());
FunctionalChainServices.getFunctionalChainServices().updateFunctionalExchangePieIcon(edge, chains,
functionalChains);
labelIcon = DEdgeIconCache.getInstance().getIcon((DEdge) representationElement);
}
return labelIcon;
}
return super.getLabelIcon(representationElement, editPart);
}
@Override
public boolean provides(DiagramElementMapping mapping, Style style) {
if (mapping instanceof EdgeMapping) {
String domainClass = ((EdgeMapping) mapping).getDomainClass();
return "FunctionalExchange".equals(domainClass);
}
return false;
}
}