| /******************************************************************************* |
| * Copyright (c) 2006 Borland Software Corporation and others. |
| * 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: |
| * Borland Software Corporation - initial API and implementation |
| *******************************************************************************/ |
| package org.eclipse.gmf.examples.mindmap.diagram.edit.policies; |
| |
| import java.util.Collection; |
| import java.util.Collections; |
| import java.util.HashMap; |
| import java.util.Iterator; |
| import java.util.LinkedList; |
| import java.util.List; |
| |
| import org.eclipse.core.runtime.IAdaptable; |
| import org.eclipse.emf.ecore.EClass; |
| import org.eclipse.emf.ecore.EObject; |
| import org.eclipse.gef.EditPart; |
| import org.eclipse.gef.commands.Command; |
| import org.eclipse.gmf.examples.mindmap.Map; |
| import org.eclipse.gmf.examples.mindmap.MindmapPackage; |
| import org.eclipse.gmf.examples.mindmap.Relationship; |
| import org.eclipse.gmf.examples.mindmap.Topic; |
| import org.eclipse.gmf.examples.mindmap.diagram.edit.parts.MapEditPart; |
| import org.eclipse.gmf.examples.mindmap.diagram.edit.parts.Relationship2EditPart; |
| import org.eclipse.gmf.examples.mindmap.diagram.edit.parts.Relationship3EditPart; |
| import org.eclipse.gmf.examples.mindmap.diagram.edit.parts.RelationshipEditPart; |
| import org.eclipse.gmf.examples.mindmap.diagram.edit.parts.ResourceEditPart; |
| import org.eclipse.gmf.examples.mindmap.diagram.edit.parts.SubtopicsEditPart; |
| import org.eclipse.gmf.examples.mindmap.diagram.edit.parts.ThreadEditPart; |
| import org.eclipse.gmf.examples.mindmap.diagram.edit.parts.ThreadItemEditPart; |
| import org.eclipse.gmf.examples.mindmap.diagram.edit.parts.TopicEditPart; |
| import org.eclipse.gmf.examples.mindmap.diagram.part.MindmapVisualIDRegistry; |
| import org.eclipse.gmf.examples.mindmap.diagram.providers.MindmapElementTypes; |
| import org.eclipse.gmf.runtime.diagram.core.util.ViewUtil; |
| import org.eclipse.gmf.runtime.diagram.ui.commands.SetViewMutabilityCommand; |
| import org.eclipse.gmf.runtime.diagram.ui.editparts.IGraphicalEditPart; |
| import org.eclipse.gmf.runtime.diagram.ui.editpolicies.CanonicalConnectionEditPolicy; |
| import org.eclipse.gmf.runtime.diagram.ui.requests.CreateConnectionViewRequest; |
| import org.eclipse.gmf.runtime.diagram.ui.requests.RequestConstants; |
| import org.eclipse.gmf.runtime.emf.core.util.EObjectAdapter; |
| import org.eclipse.gmf.runtime.emf.type.core.IElementType; |
| import org.eclipse.gmf.runtime.notation.Diagram; |
| import org.eclipse.gmf.runtime.notation.Edge; |
| import org.eclipse.gmf.runtime.notation.View; |
| |
| /** |
| * @generated |
| */ |
| public class MapCanonicalEditPolicy extends CanonicalConnectionEditPolicy { |
| |
| /** |
| * @generated |
| */ |
| protected List getSemanticChildrenList() { |
| List result = new LinkedList(); |
| EObject modelObject = ((View) getHost().getModel()).getElement(); |
| View viewObject = (View) getHost().getModel(); |
| EObject nextValue; |
| int nodeVID; |
| for (Iterator values = ((Map) modelObject).getRootTopics().iterator(); values.hasNext();) { |
| nextValue = (EObject) values.next(); |
| nodeVID = MindmapVisualIDRegistry.getNodeVisualID(viewObject, nextValue); |
| if (TopicEditPart.VISUAL_ID == nodeVID) { |
| result.add(nextValue); |
| } |
| } |
| for (Iterator values = ((Map) modelObject).getResources().iterator(); values.hasNext();) { |
| nextValue = (EObject) values.next(); |
| nodeVID = MindmapVisualIDRegistry.getNodeVisualID(viewObject, nextValue); |
| if (ResourceEditPart.VISUAL_ID == nodeVID) { |
| result.add(nextValue); |
| } |
| } |
| return result; |
| } |
| |
| /** |
| * @generated |
| */ |
| protected boolean shouldDeleteView(View view) { |
| return view.isSetElement() && view.getElement() != null; |
| } |
| |
| /** |
| * @generated |
| */ |
| protected String getDefaultFactoryHint() { |
| return null; |
| } |
| |
| /** |
| * @generated |
| */ |
| protected List getSemanticConnectionsList() { |
| return Collections.EMPTY_LIST; |
| } |
| |
| /** |
| * @generated |
| */ |
| protected EObject getSourceElement(EObject relationship) { |
| return null; |
| } |
| |
| /** |
| * @generated |
| */ |
| protected EObject getTargetElement(EObject relationship) { |
| return null; |
| } |
| |
| /** |
| * @generated |
| */ |
| protected boolean shouldIncludeConnection(Edge connector, Collection children) { |
| return false; |
| } |
| |
| /** |
| * @generated |
| */ |
| protected void refreshSemantic() { |
| super.refreshSemantic(); |
| refreshConnections(); |
| } |
| |
| /** |
| * @generated |
| */ |
| private Collection myLinkDescriptors = new LinkedList(); |
| |
| /** |
| * @generated |
| */ |
| private java.util.Map myEObject2ViewMap = new HashMap(); |
| |
| /** |
| * @generated |
| */ |
| private void refreshConnections() { |
| try { |
| collectAllLinks(getDiagram()); |
| Collection existingLinks = new LinkedList(getDiagram().getEdges()); |
| for (Iterator diagramLinks = existingLinks.iterator(); diagramLinks.hasNext();) { |
| Edge nextDiagramLink = (Edge) diagramLinks.next(); |
| EObject diagramLinkObject = nextDiagramLink.getElement(); |
| EObject diagramLinkSrc = nextDiagramLink.getSource().getElement(); |
| EObject diagramLinkDst = nextDiagramLink.getTarget().getElement(); |
| int diagramLinkVisualID = MindmapVisualIDRegistry.getVisualID(nextDiagramLink); |
| for (Iterator modelLinkDescriptors = myLinkDescriptors.iterator(); modelLinkDescriptors.hasNext();) { |
| LinkDescriptor nextLinkDescriptor = (LinkDescriptor) modelLinkDescriptors.next(); |
| if (diagramLinkObject == nextLinkDescriptor.getLinkElement() && diagramLinkSrc == nextLinkDescriptor.getSource() && diagramLinkDst == nextLinkDescriptor.getDestination() |
| && diagramLinkVisualID == nextLinkDescriptor.getVisualID()) { |
| diagramLinks.remove(); |
| modelLinkDescriptors.remove(); |
| } |
| } |
| } |
| deleteViews(existingLinks.iterator()); |
| createConnections(myLinkDescriptors); |
| } finally { |
| myLinkDescriptors.clear(); |
| myEObject2ViewMap.clear(); |
| } |
| } |
| |
| /** |
| * @generated |
| */ |
| private void collectAllLinks(View view) { |
| EObject modelElement = view.getElement(); |
| int diagramElementVisualID = MindmapVisualIDRegistry.getVisualID(view); |
| switch (diagramElementVisualID) { |
| case TopicEditPart.VISUAL_ID: |
| case ResourceEditPart.VISUAL_ID: |
| case ThreadEditPart.VISUAL_ID: |
| case ThreadItemEditPart.VISUAL_ID: |
| case MapEditPart.VISUAL_ID: { |
| myEObject2ViewMap.put(modelElement, view); |
| storeLinks(modelElement, getDiagram()); |
| } |
| default: { |
| } |
| for (Iterator children = view.getChildren().iterator(); children.hasNext();) { |
| View childView = (View) children.next(); |
| collectAllLinks(childView); |
| } |
| } |
| } |
| |
| /** |
| * @generated |
| */ |
| private void createConnections(Collection linkDescriptors) { |
| if (linkDescriptors.isEmpty()) { |
| return; |
| } |
| for (Iterator linkDescriptorsIterator = linkDescriptors.iterator(); linkDescriptorsIterator.hasNext();) { |
| final LinkDescriptor nextLinkDescriptor = (LinkDescriptor) linkDescriptorsIterator.next(); |
| EditPart sourceEditPart = getEditPartFor(nextLinkDescriptor.getSource()); |
| EditPart targetEditPart = getEditPartFor(nextLinkDescriptor.getDestination()); |
| if (sourceEditPart == null || targetEditPart == null) { |
| continue; |
| } |
| CreateConnectionViewRequest.ConnectionViewDescriptor descriptor = new CreateConnectionViewRequest.ConnectionViewDescriptor(nextLinkDescriptor.getSemanticAdapter(), null, ViewUtil.APPEND, |
| false, ((IGraphicalEditPart) getHost()).getDiagramPreferencesHint()); |
| CreateConnectionViewRequest ccr = new CreateConnectionViewRequest(descriptor); |
| ccr.setType(RequestConstants.REQ_CONNECTION_START); |
| ccr.setSourceEditPart(sourceEditPart); |
| sourceEditPart.getCommand(ccr); |
| ccr.setTargetEditPart(targetEditPart); |
| ccr.setType(RequestConstants.REQ_CONNECTION_END); |
| Command cmd = targetEditPart.getCommand(ccr); |
| if (cmd != null && cmd.canExecute()) { |
| executeCommand(cmd); |
| IAdaptable viewAdapter = (IAdaptable) ccr.getNewObject(); |
| SetViewMutabilityCommand.makeImmutable(viewAdapter).execute(); |
| } |
| } |
| } |
| |
| /** |
| * @generated |
| */ |
| private EditPart getEditPartFor(EObject modelElement) { |
| View view = (View) myEObject2ViewMap.get(modelElement); |
| if (view != null) { |
| return (EditPart) getHost().getViewer().getEditPartRegistry().get(view); |
| } |
| return null; |
| } |
| |
| /** |
| * @generated |
| */ |
| private void storeLinks(EObject container, Diagram diagram) { |
| EClass containerMetaclass = container.eClass(); |
| storeFeatureModelFacetLinks(container, containerMetaclass, diagram); |
| storeTypeModelFacetLinks(container, containerMetaclass); |
| } |
| |
| /** |
| * @generated |
| */ |
| private void storeTypeModelFacetLinks(EObject container, EClass containerMetaclass) { |
| if (MindmapPackage.eINSTANCE.getMap().isSuperTypeOf(containerMetaclass)) { |
| for (Iterator values = ((Map) container).getRelations().iterator(); values.hasNext();) { |
| EObject nextValue = ((EObject) values.next()); |
| int linkVID = MindmapVisualIDRegistry.getLinkWithClassVisualID(nextValue); |
| if (RelationshipEditPart.VISUAL_ID == linkVID) { |
| Object structuralFeatureResult = ((Relationship) nextValue).getTarget(); |
| if (structuralFeatureResult instanceof EObject) { |
| EObject dst = (EObject) structuralFeatureResult; |
| structuralFeatureResult = ((Relationship) nextValue).getSource(); |
| if (structuralFeatureResult instanceof EObject) { |
| EObject src = (EObject) structuralFeatureResult; |
| myLinkDescriptors.add(new LinkDescriptor(src, dst, nextValue, linkVID)); |
| } |
| } |
| } |
| } |
| } |
| if (MindmapPackage.eINSTANCE.getMap().isSuperTypeOf(containerMetaclass)) { |
| for (Iterator values = ((Map) container).getRelations().iterator(); values.hasNext();) { |
| EObject nextValue = ((EObject) values.next()); |
| int linkVID = MindmapVisualIDRegistry.getLinkWithClassVisualID(nextValue); |
| if (Relationship2EditPart.VISUAL_ID == linkVID) { |
| Object structuralFeatureResult = ((Relationship) nextValue).getTarget(); |
| if (structuralFeatureResult instanceof EObject) { |
| EObject dst = (EObject) structuralFeatureResult; |
| structuralFeatureResult = ((Relationship) nextValue).getSource(); |
| if (structuralFeatureResult instanceof EObject) { |
| EObject src = (EObject) structuralFeatureResult; |
| myLinkDescriptors.add(new LinkDescriptor(src, dst, nextValue, linkVID)); |
| } |
| } |
| } |
| } |
| } |
| if (MindmapPackage.eINSTANCE.getMap().isSuperTypeOf(containerMetaclass)) { |
| for (Iterator values = ((Map) container).getRelations().iterator(); values.hasNext();) { |
| EObject nextValue = ((EObject) values.next()); |
| int linkVID = MindmapVisualIDRegistry.getLinkWithClassVisualID(nextValue); |
| if (Relationship3EditPart.VISUAL_ID == linkVID) { |
| Object structuralFeatureResult = ((Relationship) nextValue).getTarget(); |
| if (structuralFeatureResult instanceof EObject) { |
| EObject dst = (EObject) structuralFeatureResult; |
| structuralFeatureResult = ((Relationship) nextValue).getSource(); |
| if (structuralFeatureResult instanceof EObject) { |
| EObject src = (EObject) structuralFeatureResult; |
| myLinkDescriptors.add(new LinkDescriptor(src, dst, nextValue, linkVID)); |
| } |
| } |
| } |
| } |
| } |
| } |
| |
| /** |
| * @generated |
| */ |
| private void storeFeatureModelFacetLinks(EObject container, EClass containerMetaclass, Diagram diagram) { |
| |
| if (MindmapPackage.eINSTANCE.getTopic().isSuperTypeOf(containerMetaclass)) { |
| for (Iterator destinations = ((Topic) container).getSubtopics().iterator(); destinations.hasNext();) { |
| EObject nextDestination = (EObject) destinations.next(); |
| myLinkDescriptors.add(new LinkDescriptor(container, nextDestination, MindmapElementTypes.TopicSubtopics_3001, SubtopicsEditPart.VISUAL_ID)); |
| |
| } |
| } |
| |
| } |
| |
| /** |
| * @generated |
| */ |
| private Diagram getDiagram() { |
| return ((View) getHost().getModel()).getDiagram(); |
| } |
| |
| /** |
| * @generated |
| */ |
| private class LinkDescriptor { |
| |
| /** |
| * @generated |
| */ |
| private EObject mySource; |
| |
| /** |
| * @generated |
| */ |
| private EObject myDestination; |
| |
| /** |
| * @generated |
| */ |
| private EObject myLinkElement; |
| |
| /** |
| * @generated |
| */ |
| private int myVisualID; |
| |
| /** |
| * @generated |
| */ |
| private IAdaptable mySemanticAdapter; |
| |
| /** |
| * @generated |
| */ |
| protected LinkDescriptor(EObject source, EObject destination, EObject linkElement, int linkVID) { |
| this(source, destination, linkVID); |
| myLinkElement = linkElement; |
| mySemanticAdapter = new EObjectAdapter(linkElement); |
| } |
| |
| /** |
| * @generated |
| */ |
| protected LinkDescriptor(EObject source, EObject destination, IElementType elementType, int linkVID) { |
| this(source, destination, linkVID); |
| myLinkElement = null; |
| final IElementType elementTypeCopy = elementType; |
| mySemanticAdapter = new IAdaptable() { |
| |
| public Object getAdapter(Class adapter) { |
| if (IElementType.class.equals(adapter)) { |
| return elementTypeCopy; |
| } |
| return null; |
| } |
| }; |
| } |
| |
| /** |
| * @generated |
| */ |
| private LinkDescriptor(EObject source, EObject destination, int linkVID) { |
| mySource = source; |
| myDestination = destination; |
| myVisualID = linkVID; |
| } |
| |
| /** |
| * @generated |
| */ |
| protected EObject getSource() { |
| return mySource; |
| } |
| |
| /** |
| * @generated |
| */ |
| protected EObject getDestination() { |
| return myDestination; |
| } |
| |
| /** |
| * @generated |
| */ |
| protected EObject getLinkElement() { |
| return myLinkElement; |
| } |
| |
| /** |
| * @generated |
| */ |
| protected int getVisualID() { |
| return myVisualID; |
| } |
| |
| /** |
| * @generated |
| */ |
| protected IAdaptable getSemanticAdapter() { |
| return mySemanticAdapter; |
| } |
| } |
| |
| } |
