blob: c9c9191a6fb1ac44f19a7f490a9aa6a71bdc4807 (
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
|
/*******************************************************************************
* Copyright (c) 2012 CEA LIST.
*
* 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
*******************************************************************************/
package org.eclipse.papyrus.uml.diagram.timing.custom.edit.policies;
import org.eclipse.gef.commands.Command;
import org.eclipse.gmf.runtime.emf.type.core.IElementType;
import org.eclipse.gmf.runtime.emf.type.core.requests.CreateRelationshipRequest;
import org.eclipse.papyrus.uml.diagram.timing.custom.edit.commands.CustomMessageCreateCommand;
import org.eclipse.papyrus.uml.diagram.timing.custom.utils.MessageUtils;
import org.eclipse.papyrus.uml.diagram.timing.edit.policies.CompactLifelineItemSemanticEditPolicyCN;
public class CustomCompactLifelineItemSemanticEditPolicyCN extends CompactLifelineItemSemanticEditPolicyCN {
/** Allows creating a a lost Message that targets a Lifeline, or a found Message that originates from a Lifeline */
@Override
protected Command getCreateRelationshipCommand(final CreateRelationshipRequest req) {
final IElementType requestElementType = req.getElementType();
if(requestElementType == null) {
return null;
}
if(MessageUtils.isMessage(requestElementType)) {
return getGEFWrapper(new CustomMessageCreateCommand(req));
}
return super.getCreateRelationshipCommand(req);
}
}
|
