package org.eclipse.papyrus.uml.diagram.sequence.edit.commands; import java.util.Collections; import java.util.List; import org.eclipse.core.commands.ExecutionException; import org.eclipse.core.runtime.IAdaptable; import org.eclipse.core.runtime.IProgressMonitor; import org.eclipse.emf.ecore.EObject; import org.eclipse.gmf.runtime.common.core.command.CommandResult; import org.eclipse.gmf.runtime.common.core.command.ICommand; import org.eclipse.gmf.runtime.emf.type.core.IElementType; import org.eclipse.gmf.runtime.emf.type.core.commands.EditElementCommand; import org.eclipse.gmf.runtime.emf.type.core.requests.ConfigureRequest; import org.eclipse.gmf.runtime.emf.type.core.requests.CreateElementRequest; import org.eclipse.gmf.runtime.emf.type.core.requests.CreateRelationshipRequest; import org.eclipse.papyrus.uml.diagram.sequence.edit.policies.UMLBaseItemSemanticEditPolicy; import org.eclipse.papyrus.uml.diagram.sequence.providers.ElementInitializers; import org.eclipse.papyrus.uml.diagram.sequence.util.SequenceRequestConstant; import org.eclipse.papyrus.uml.diagram.sequence.util.SequenceUtil; import org.eclipse.uml2.uml.GeneralOrdering; import org.eclipse.uml2.uml.InteractionFragment; import org.eclipse.uml2.uml.OccurrenceSpecification; import org.eclipse.uml2.uml.UMLFactory; /** * @generated */ public class GeneralOrderingCreateCommand extends EditElementCommand { /** * @generated */ protected final EObject source; /** * @generated */ protected final EObject target; /** * @generated */ protected InteractionFragment container; /** * Non generated, completely redefined to find correct ends and correct container * * @generated NOT find source and target events and container */ public GeneralOrderingCreateCommand(CreateRelationshipRequest request, EObject source, EObject target) { super(request.getLabel(), null, request); // check source occurrence specification List<OccurrenceSpecification> possibleSourceOcc = null; if(getRequest().getParameters().containsKey(SequenceRequestConstant.NEAREST_OCCURRENCE_SPECIFICATION)) { Object paramOccurrence1 = getRequest().getParameter(SequenceRequestConstant.NEAREST_OCCURRENCE_SPECIFICATION); List<OccurrenceSpecification> occ1List = SequenceUtil.getAsOccSpecList(paramOccurrence1); possibleSourceOcc = occ1List; } // check target occurrence specification List<OccurrenceSpecification> possibleTargetOcc = null; if(getRequest().getParameters().containsKey(SequenceRequestConstant.NEAREST_OCCURRENCE_SPECIFICATION_2)) { Object paramOccurrence2 = getRequest().getParameter(SequenceRequestConstant.NEAREST_OCCURRENCE_SPECIFICATION_2); List<OccurrenceSpecification> occ2List = SequenceUtil.getAsOccSpecList(paramOccurrence2); possibleTargetOcc = occ2List; } // disable duration constraint on a same event if(possibleSourceOcc != null && !possibleSourceOcc.isEmpty()) { this.source = possibleSourceOcc.get(0); } else { this.source = null; } if(possibleTargetOcc != null && !possibleTargetOcc.isEmpty() && Collections.disjoint(possibleSourceOcc, possibleTargetOcc)) { this.target = possibleTargetOcc.get(0); } else { // set invalid target to disable it (or null to enable in case second end is not defined) this.target = target; } container = deduceContainer(source, target); } /** * @generated */ public boolean canExecute() { if(source == null && target == null) { return false; } if(source != null && false == source instanceof OccurrenceSpecification) { return false; } if(target != null && false == target instanceof OccurrenceSpecification) { return false; } if(getSource() == null) { return true; // link creation is in progress; source is not defined yet } // target may be null here but it's possible to check constraint if(getContainer() == null) { return false; } return UMLBaseItemSemanticEditPolicy.getLinkConstraints().canCreateGeneralOrdering_4012(getContainer(), getSource(), getTarget()); } /** * @generated */ protected CommandResult doExecuteWithResult(IProgressMonitor monitor, IAdaptable info) throws ExecutionException { if(!canExecute()) { throw new ExecutionException("Invalid arguments in create link command"); //$NON-NLS-1$ } GeneralOrdering newElement = UMLFactory.eINSTANCE.createGeneralOrdering(); getContainer().getGeneralOrderings().add(newElement); newElement.setBefore(getSource()); newElement.setAfter(getTarget()); ElementInitializers.getInstance().init_GeneralOrdering_4012(newElement); doConfigure(newElement, monitor, info); ((CreateElementRequest)getRequest()).setNewElement(newElement); return CommandResult.newOKCommandResult(newElement); } /** * @generated */ protected void doConfigure(GeneralOrdering newElement, IProgressMonitor monitor, IAdaptable info) throws ExecutionException { IElementType elementType = ((CreateElementRequest)getRequest()).getElementType(); ConfigureRequest configureRequest = new ConfigureRequest(getEditingDomain(), newElement, elementType); configureRequest.setClientContext(((CreateElementRequest)getRequest()).getClientContext()); configureRequest.addParameters(getRequest().getParameters()); configureRequest.setParameter(CreateRelationshipRequest.SOURCE, getSource()); configureRequest.setParameter(CreateRelationshipRequest.TARGET, getTarget()); ICommand configureCommand = elementType.getEditCommand(configureRequest); if(configureCommand != null && configureCommand.canExecute()) { configureCommand.execute(monitor, info); } } /** * @generated */ protected void setElementToEdit(EObject element) { throw new UnsupportedOperationException(); } /** * @generated */ protected OccurrenceSpecification getSource() { return (OccurrenceSpecification)source; } /** * @generated */ protected OccurrenceSpecification getTarget() { return (OccurrenceSpecification)target; } /** * @generated */ public InteractionFragment getContainer() { return container; } /** * Default approach is to traverse ancestors of the source to find instance of container. * Modify with appropriate logic. * * @generated */ protected InteractionFragment deduceContainer(EObject source, EObject target) { // Find container element for the new link. // Climb up by containment hierarchy starting from the source // and return the first element that is instance of the container class. for(EObject element = source; element != null; element = element.eContainer()) { if(element instanceof InteractionFragment) { return (InteractionFragment)element; } } return null; } }