blob: 9f32b299a1ce3a00bcec50e55aca71f92c885f98 (
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
|
/*****************************************************************************
* Copyright (c) 2010 CEA
*
*
* 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:
* Atos Origin - Initial API and implementation
*
*****************************************************************************/
package org.eclipse.papyrus.uml.diagram.sequence.edit.policies;
import java.util.Collections;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import org.eclipse.draw2d.geometry.Point;
import org.eclipse.gef.EditPart;
import org.eclipse.gef.commands.Command;
import org.eclipse.gmf.runtime.diagram.ui.editparts.ConnectionNodeEditPart;
import org.eclipse.gmf.runtime.diagram.ui.editpolicies.CreationEditPolicy;
import org.eclipse.gmf.runtime.diagram.ui.requests.CreateViewAndElementRequest;
import org.eclipse.gmf.runtime.emf.type.core.IHintedType;
import org.eclipse.papyrus.uml.diagram.sequence.edit.parts.LifelineEditPart;
import org.eclipse.papyrus.uml.diagram.sequence.providers.UMLElementTypes;
import org.eclipse.papyrus.uml.diagram.sequence.util.SequenceRequestConstant;
import org.eclipse.papyrus.uml.diagram.sequence.util.SequenceUtil;
import org.eclipse.uml2.uml.OccurrenceSpecification;
public class CreationOnMessageEditPolicy extends CreationEditPolicy {
/**
* Get the command after updating the request with appropriate occurrences
*
* @see org.eclipse.gmf.runtime.diagram.ui.editpolicies.CreationEditPolicy#getCreateElementAndViewCommand(org.eclipse.gmf.runtime.diagram.ui.requests.CreateViewAndElementRequest)
*
* @param request
* the request
* @return command or null
*/
@Override
protected Command getCreateElementAndViewCommand(CreateViewAndElementRequest request) {
Map<String, Object> extendedData = request.getExtendedData();
// record the nearest event if necessary
String requestHint = request.getViewAndElementDescriptor().getSemanticHint();
if(isCreatedOnOccurrenceSpecification(requestHint)) {
EditPart hostPart = getHost();
if(hostPart instanceof ConnectionNodeEditPart) {
LifelineEditPart sourceLifeline = SequenceUtil.getParentLifelinePart(((ConnectionNodeEditPart)hostPart).getSource());
LifelineEditPart targetLifeline = SequenceUtil.getParentLifelinePart(((ConnectionNodeEditPart)hostPart).getTarget());
Entry<Point, List<OccurrenceSpecification>> eventAndLocation = null;
if(sourceLifeline != null) {
eventAndLocation = SequenceUtil.findNearestEvent(request.getLocation(), sourceLifeline);
}
if(targetLifeline != null && eventAndLocation == null) {
eventAndLocation = SequenceUtil.findNearestEvent(request.getLocation(), targetLifeline);
}
// find an event near enough to create the constraint or observation
List<OccurrenceSpecification> events = Collections.emptyList();
Point location = null;
if(eventAndLocation != null) {
location = eventAndLocation.getKey();
events = eventAndLocation.getValue();
}
if(extendedData.containsKey(SequenceRequestConstant.NEAREST_OCCURRENCE_SPECIFICATION_2)) {
extendedData.put(SequenceRequestConstant.NEAREST_OCCURRENCE_SPECIFICATION_2, events);
} else {
extendedData.put(SequenceRequestConstant.NEAREST_OCCURRENCE_SPECIFICATION, events);
}
if(extendedData.containsKey(SequenceRequestConstant.OCCURRENCE_SPECIFICATION_LOCATION_2)) {
extendedData.put(SequenceRequestConstant.OCCURRENCE_SPECIFICATION_LOCATION_2, location);
} else {
extendedData.put(SequenceRequestConstant.OCCURRENCE_SPECIFICATION_LOCATION, location);
}
}
}
return super.getCreateElementAndViewCommand(request);
}
/**
* Return true if creation must be performed on an occurrence specification
*
* @param requestHint
* the hint of object to create
* @return true if creation on an occurrence specification
*/
private boolean isCreatedOnOccurrenceSpecification(String requestHint) {
return isTimeHint(requestHint) || isDurationHint(requestHint);
}
/**
* Return true if hint is for creating a duration observation/constraint
*
* @param requestHint
* the hint of object to create
* @return true if correct hint
*/
private boolean isDurationHint(String requestHint) {
String durCstOnLifelineHint = ((IHintedType)UMLElementTypes.DurationConstraint_3021).getSemanticHint();
String durCstOnMessage = ((IHintedType)UMLElementTypes.DurationConstraint_3023).getSemanticHint();
String durObsOnMessage = ((IHintedType)UMLElementTypes.DurationObservation_3024).getSemanticHint();
return durCstOnLifelineHint.equals(requestHint) || durCstOnMessage.equals(requestHint) || durObsOnMessage.equals(requestHint);
}
/**
* Return true if hint is for creating a time observation/constraint
*
* @param requestHint
* the hint of object to create
* @return true if correct hint
*/
private boolean isTimeHint(String requestHint) {
String timeConstraintHint = ((IHintedType)UMLElementTypes.TimeConstraint_3019).getSemanticHint();
String timeObservationHint = ((IHintedType)UMLElementTypes.TimeObservation_3020).getSemanticHint();
return timeConstraintHint.equals(requestHint) || timeObservationHint.equals(requestHint);
}
}
|