/**
* Copyright (c) 2014 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
*
* Contributors:
* CEA LIST - Initial API and implementation
*/
package org.eclipse.papyrus.uml.diagram.component.edit.policies;
import org.eclipse.emf.ecore.EObject;
import org.eclipse.gef.commands.Command;
import org.eclipse.gef.commands.UnexecutableCommand;
import org.eclipse.gmf.runtime.common.core.command.ICommand;
import org.eclipse.gmf.runtime.diagram.ui.commands.ICommandProxy;
import org.eclipse.gmf.runtime.emf.type.core.IElementType;
import org.eclipse.gmf.runtime.emf.type.core.requests.CreateRelationshipRequest;
import org.eclipse.gmf.runtime.emf.type.core.requests.DestroyElementRequest;
import org.eclipse.gmf.runtime.emf.type.core.requests.ReorientReferenceRelationshipRequest;
import org.eclipse.gmf.runtime.emf.type.core.requests.ReorientRelationshipRequest;
import org.eclipse.papyrus.infra.extendedtypes.types.IExtendedHintedElementType;
import org.eclipse.papyrus.infra.extendedtypes.util.ElementTypeUtils;
import org.eclipse.papyrus.infra.services.edit.service.ElementEditServiceUtils;
import org.eclipse.papyrus.infra.services.edit.service.IElementEditService;
import org.eclipse.papyrus.uml.diagram.component.edit.commands.AbstractionCreateCommand;
import org.eclipse.papyrus.uml.diagram.component.edit.commands.CommentAnnotatedElementCreateCommand;
import org.eclipse.papyrus.uml.diagram.component.edit.commands.CommentAnnotatedElementReorientCommand;
import org.eclipse.papyrus.uml.diagram.component.edit.commands.ComponentRealizationCreateCommand;
import org.eclipse.papyrus.uml.diagram.component.edit.commands.ConstraintConstrainedElementCreateCommand;
import org.eclipse.papyrus.uml.diagram.component.edit.commands.ConstraintConstrainedElementReorientCommand;
import org.eclipse.papyrus.uml.diagram.component.edit.commands.DependencyBranchCreateCommand;
import org.eclipse.papyrus.uml.diagram.component.edit.commands.DependencyCreateCommand;
import org.eclipse.papyrus.uml.diagram.component.edit.commands.InterfaceRealizationCreateCommand;
import org.eclipse.papyrus.uml.diagram.component.edit.commands.ManifestationCreateCommand;
import org.eclipse.papyrus.uml.diagram.component.edit.commands.SubstitutionCreateCommand;
import org.eclipse.papyrus.uml.diagram.component.edit.commands.UsageCreateCommand;
import org.eclipse.papyrus.uml.diagram.component.edit.parts.AbstractionEditPart;
import org.eclipse.papyrus.uml.diagram.component.edit.parts.CommentAnnotatedElementEditPart;
import org.eclipse.papyrus.uml.diagram.component.edit.parts.ComponentRealizationEditPart;
import org.eclipse.papyrus.uml.diagram.component.edit.parts.ConstraintConstrainedElementEditPart;
import org.eclipse.papyrus.uml.diagram.component.edit.parts.DependencyBranchEditPart;
import org.eclipse.papyrus.uml.diagram.component.edit.parts.DependencyEditPart;
import org.eclipse.papyrus.uml.diagram.component.edit.parts.InterfaceRealizationEditPart;
import org.eclipse.papyrus.uml.diagram.component.edit.parts.ManifestationEditPart;
import org.eclipse.papyrus.uml.diagram.component.edit.parts.SubstitutionEditPart;
import org.eclipse.papyrus.uml.diagram.component.edit.parts.UsageEditPart;
import org.eclipse.papyrus.uml.diagram.component.providers.UMLElementTypes;
/**
* @generated
*/
public class ComponentRealizationItemSemanticEditPolicy extends UMLBaseItemSemanticEditPolicy {
/**
* @generated
*/
public ComponentRealizationItemSemanticEditPolicy() {
super(UMLElementTypes.ComponentRealization_4007);
}
/**
* @generated
*/
@Override
protected Command getDestroyElementCommand(DestroyElementRequest req) {
EObject selectedEObject = req.getElementToDestroy();
IElementEditService provider = ElementEditServiceUtils.getCommandProvider(selectedEObject);
if (provider != null) {
// Retrieve delete command from the Element Edit service
ICommand deleteCommand = provider.getEditCommand(req);
if (deleteCommand != null) {
return new ICommandProxy(deleteCommand);
}
}
return UnexecutableCommand.INSTANCE;
}
/**
* @generated
*/
@Override
protected Command getCreateRelationshipCommand(CreateRelationshipRequest req) {
Command command = req.getTarget() == null ?
getStartCreateRelationshipCommand(req) : getCompleteCreateRelationshipCommand(req);
return command != null ? command : super.getCreateRelationshipCommand(req);
}
/**
* @generated
*/
protected Command getStartCreateRelationshipCommand(CreateRelationshipRequest req) {
IElementType requestElementType = req.getElementType();
if (requestElementType == null) {
return null;
}
IElementType baseElementType = requestElementType;
boolean isExtendedType = false;
if (requestElementType instanceof IExtendedHintedElementType) {
baseElementType = ElementTypeUtils.getClosestDiagramType(requestElementType);
if (baseElementType != null) {
isExtendedType = true;
} else {
// no reference element type ID. using the closest super element type to give more opportunities, but can lead to bugs.
baseElementType = ElementTypeUtils.findClosestNonExtendedElementType((IExtendedHintedElementType) requestElementType);
isExtendedType = true;
}
}
if (UMLElementTypes.Usage_4001 == baseElementType) {
if (isExtendedType) {
return getExtendedStartCreateRelationshipCommand(req, (IExtendedHintedElementType) requestElementType);
}
return getGEFWrapper(new UsageCreateCommand(req,
req.getSource(), req.getTarget()));
}
if (UMLElementTypes.InterfaceRealization_4006 == baseElementType) {
if (isExtendedType) {
return getExtendedStartCreateRelationshipCommand(req, (IExtendedHintedElementType) requestElementType);
}
return getGEFWrapper(new InterfaceRealizationCreateCommand(req,
req.getSource(), req.getTarget()));
}
if (UMLElementTypes.Substitution_4012 == baseElementType) {
if (isExtendedType) {
return getExtendedStartCreateRelationshipCommand(req, (IExtendedHintedElementType) requestElementType);
}
return getGEFWrapper(new SubstitutionCreateCommand(req,
req.getSource(), req.getTarget()));
}
if (UMLElementTypes.Manifestation_4014 == baseElementType) {
if (isExtendedType) {
return getExtendedStartCreateRelationshipCommand(req, (IExtendedHintedElementType) requestElementType);
}
return getGEFWrapper(new ManifestationCreateCommand(req,
req.getSource(), req.getTarget()));
}
if (UMLElementTypes.ComponentRealization_4007 == baseElementType) {
if (isExtendedType) {
return getExtendedStartCreateRelationshipCommand(req, (IExtendedHintedElementType) requestElementType);
}
return getGEFWrapper(new ComponentRealizationCreateCommand(req,
req.getSource(), req.getTarget()));
}
if (UMLElementTypes.Abstraction_4013 == baseElementType) {
if (isExtendedType) {
return getExtendedStartCreateRelationshipCommand(req, (IExtendedHintedElementType) requestElementType);
}
return getGEFWrapper(new AbstractionCreateCommand(req,
req.getSource(), req.getTarget()));
}
if (UMLElementTypes.CommentAnnotatedElement_4015 == baseElementType) {
return null;
}
if (UMLElementTypes.ConstraintConstrainedElement_4009 == baseElementType) {
return null;
}
if (UMLElementTypes.Dependency_4010 == baseElementType) {
if (isExtendedType) {
return getExtendedStartCreateRelationshipCommand(req, (IExtendedHintedElementType) requestElementType);
}
return getGEFWrapper(new DependencyCreateCommand(req,
req.getSource(), req.getTarget()));
}
if (UMLElementTypes.Dependency_4017 == baseElementType) {
if (isExtendedType) {
return getExtendedStartCreateRelationshipCommand(req, (IExtendedHintedElementType) requestElementType);
}
return getGEFWrapper(new DependencyBranchCreateCommand(req,
req.getSource(), req.getTarget()));
}
return null;
}
/**
* @generated
*/
protected Command getCompleteCreateRelationshipCommand(CreateRelationshipRequest req) {
IElementType requestElementType = req.getElementType();
if (requestElementType == null) {
return null;
}
IElementType baseElementType = requestElementType;
boolean isExtendedType = false;
if (requestElementType instanceof IExtendedHintedElementType) {
baseElementType = ElementTypeUtils.getClosestDiagramType(requestElementType);
if (baseElementType != null) {
isExtendedType = true;
} else {
// no reference element type ID. using the closest super element type to give more opportunities, but can lead to bugs.
baseElementType = ElementTypeUtils.findClosestNonExtendedElementType((IExtendedHintedElementType) requestElementType);
isExtendedType = true;
}
}
if (UMLElementTypes.Usage_4001 == baseElementType) {
if (isExtendedType) {
return getExtendedCompleteCreateRelationshipCommand(req, (IExtendedHintedElementType) requestElementType);
}
return getGEFWrapper(new UsageCreateCommand(req,
req.getSource(), req.getTarget()));
}
if (UMLElementTypes.InterfaceRealization_4006 == baseElementType) {
return null;
}
if (UMLElementTypes.Substitution_4012 == baseElementType) {
if (isExtendedType) {
return getExtendedCompleteCreateRelationshipCommand(req, (IExtendedHintedElementType) requestElementType);
}
return getGEFWrapper(new SubstitutionCreateCommand(req,
req.getSource(), req.getTarget()));
}
if (UMLElementTypes.Manifestation_4014 == baseElementType) {
if (isExtendedType) {
return getExtendedCompleteCreateRelationshipCommand(req, (IExtendedHintedElementType) requestElementType);
}
return getGEFWrapper(new ManifestationCreateCommand(req,
req.getSource(), req.getTarget()));
}
if (UMLElementTypes.ComponentRealization_4007 == baseElementType) {
if (isExtendedType) {
return getExtendedCompleteCreateRelationshipCommand(req, (IExtendedHintedElementType) requestElementType);
}
return getGEFWrapper(new ComponentRealizationCreateCommand(req,
req.getSource(), req.getTarget()));
}
if (UMLElementTypes.Abstraction_4013 == baseElementType) {
if (isExtendedType) {
return getExtendedCompleteCreateRelationshipCommand(req, (IExtendedHintedElementType) requestElementType);
}
return getGEFWrapper(new AbstractionCreateCommand(req,
req.getSource(), req.getTarget()));
}
if (UMLElementTypes.CommentAnnotatedElement_4015 == baseElementType) {
if (isExtendedType) {
return getExtendedCompleteCreateRelationshipCommand(req, (IExtendedHintedElementType) requestElementType);
}
return getGEFWrapper(new CommentAnnotatedElementCreateCommand(req,
req.getSource(), req.getTarget()));
}
if (UMLElementTypes.ConstraintConstrainedElement_4009 == baseElementType) {
if (isExtendedType) {
return getExtendedCompleteCreateRelationshipCommand(req, (IExtendedHintedElementType) requestElementType);
}
return getGEFWrapper(new ConstraintConstrainedElementCreateCommand(req,
req.getSource(), req.getTarget()));
}
if (UMLElementTypes.Dependency_4010 == baseElementType) {
if (isExtendedType) {
return getExtendedCompleteCreateRelationshipCommand(req, (IExtendedHintedElementType) requestElementType);
}
return getGEFWrapper(new DependencyCreateCommand(req,
req.getSource(), req.getTarget()));
}
if (UMLElementTypes.Dependency_4017 == baseElementType) {
if (isExtendedType) {
return getExtendedCompleteCreateRelationshipCommand(req, (IExtendedHintedElementType) requestElementType);
}
return getGEFWrapper(new DependencyBranchCreateCommand(req,
req.getSource(), req.getTarget()));
}
return null;
}
/**
* Returns command to reorient EClass based link. New link target or source
* should be the domain model element associated with this node.
*
* @generated
*/
@Override
protected Command getReorientRelationshipCommand(ReorientRelationshipRequest req) {
switch (getVisualID(req)) {
case UsageEditPart.VISUAL_ID:
case InterfaceRealizationEditPart.VISUAL_ID:
case SubstitutionEditPart.VISUAL_ID:
case ManifestationEditPart.VISUAL_ID:
case ComponentRealizationEditPart.VISUAL_ID:
case AbstractionEditPart.VISUAL_ID:
case DependencyEditPart.VISUAL_ID:
case DependencyBranchEditPart.VISUAL_ID:
IElementEditService provider = ElementEditServiceUtils.getCommandProvider(req.getRelationship());
if (provider == null) {
return UnexecutableCommand.INSTANCE;
}
// Retrieve re-orient command from the Element Edit service
ICommand reorientCommand = provider.getEditCommand(req);
if (reorientCommand == null) {
return UnexecutableCommand.INSTANCE;
}
return getGEFWrapper(reorientCommand.reduce());
}
return super.getReorientRelationshipCommand(req);
}
/**
* Returns command to reorient EReference based link. New link target or source
* should be the domain model element associated with this node.
*
* @generated
*/
@Override
protected Command getReorientReferenceRelationshipCommand(ReorientReferenceRelationshipRequest req) {
switch (getVisualID(req)) {
case CommentAnnotatedElementEditPart.VISUAL_ID:
return getGEFWrapper(new CommentAnnotatedElementReorientCommand(req));
case ConstraintConstrainedElementEditPart.VISUAL_ID:
return getGEFWrapper(new ConstraintConstrainedElementReorientCommand(req));
}
return super.getReorientReferenceRelationshipCommand(req);
}
}