/* * */ package stateAnalysis.diagram.edit.policies; import java.util.ArrayList; 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.EObject; import org.eclipse.emf.ecore.EStructuralFeature; import org.eclipse.gef.EditPart; import org.eclipse.gef.commands.Command; import org.eclipse.gmf.runtime.diagram.core.util.ViewUtil; import org.eclipse.gmf.runtime.diagram.ui.commands.DeferredLayoutCommand; import org.eclipse.gmf.runtime.diagram.ui.commands.ICommandProxy; 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.CanonicalEditPolicy; import org.eclipse.gmf.runtime.diagram.ui.requests.CreateConnectionViewRequest; import org.eclipse.gmf.runtime.diagram.ui.requests.CreateViewRequest; import org.eclipse.gmf.runtime.diagram.ui.requests.RequestConstants; import org.eclipse.gmf.runtime.emf.core.util.EObjectAdapter; import org.eclipse.gmf.runtime.notation.Diagram; import org.eclipse.gmf.runtime.notation.Edge; import org.eclipse.gmf.runtime.notation.Node; import org.eclipse.gmf.runtime.notation.View; import org.eclipse.gmf.tooling.runtime.update.UpdaterLinkDescriptor; /** * @generated */ public class SystemCanonicalEditPolicy extends CanonicalEditPolicy { /** * @generated */ protected void refreshOnActivate() { // Need to activate editpart children before invoking the canonical refresh for EditParts to add event listeners List<?> c = getHost().getChildren(); for (int i = 0; i < c.size(); i++) { ((EditPart) c.get(i)).activate(); } super.refreshOnActivate(); } /** * @generated */ protected EStructuralFeature getFeatureToSynchronize() { return stateAnalysis.StateAnalysisPackage.eINSTANCE .getSystem_Subsystems(); } /** * @generated */ @SuppressWarnings("rawtypes") protected List getSemanticChildrenList() { View viewObject = (View) getHost().getModel(); LinkedList<EObject> result = new LinkedList<EObject>(); List<stateAnalysis.diagram.part.StateAnalysisNodeDescriptor> childDescriptors = stateAnalysis.diagram.part.StateAnalysisDiagramUpdater .getSystem_1000SemanticChildren(viewObject); for (stateAnalysis.diagram.part.StateAnalysisNodeDescriptor d : childDescriptors) { result.add(d.getModelElement()); } return result; } /** * @generated */ protected boolean isOrphaned(Collection<EObject> semanticChildren, final View view) { if (isShortcut(view)) { return stateAnalysis.diagram.part.StateAnalysisDiagramUpdater .isShortcutOrphaned(view); } return isMyDiagramElement(view) && !semanticChildren.contains(view.getElement()); } /** * @generated */ private boolean isMyDiagramElement(View view) { return stateAnalysis.diagram.edit.parts.SubsystemEditPart.VISUAL_ID == stateAnalysis.diagram.part.StateAnalysisVisualIDRegistry .getVisualID(view); } /** * @generated */ protected static boolean isShortcut(View view) { return view.getEAnnotation("Shortcut") != null; //$NON-NLS-1$ } /** * @generated */ protected void refreshSemantic() { if (resolveSemanticElement() == null) { return; } LinkedList<IAdaptable> createdViews = new LinkedList<IAdaptable>(); List<stateAnalysis.diagram.part.StateAnalysisNodeDescriptor> childDescriptors = stateAnalysis.diagram.part.StateAnalysisDiagramUpdater .getSystem_1000SemanticChildren((View) getHost().getModel()); LinkedList<View> orphaned = new LinkedList<View>(); // we care to check only views we recognize as ours and not shortcuts LinkedList<View> knownViewChildren = new LinkedList<View>(); for (View v : getViewChildren()) { if (isShortcut(v)) { if (stateAnalysis.diagram.part.StateAnalysisDiagramUpdater .isShortcutOrphaned(v)) { orphaned.add(v); } continue; } if (isMyDiagramElement(v)) { knownViewChildren.add(v); } } // alternative to #cleanCanonicalSemanticChildren(getViewChildren(), semanticChildren) // // iteration happens over list of desired semantic elements, trying to find best matching View, while original CEP // iterates views, potentially losing view (size/bounds) information - i.e. if there are few views to reference same EObject, only last one // to answer isOrphaned == true will be used for the domain element representation, see #cleanCanonicalSemanticChildren() for (Iterator<stateAnalysis.diagram.part.StateAnalysisNodeDescriptor> descriptorsIterator = childDescriptors .iterator(); descriptorsIterator.hasNext();) { stateAnalysis.diagram.part.StateAnalysisNodeDescriptor next = descriptorsIterator .next(); String hint = stateAnalysis.diagram.part.StateAnalysisVisualIDRegistry .getType(next.getVisualID()); LinkedList<View> perfectMatch = new LinkedList<View>(); // both semanticElement and hint match that of NodeDescriptor for (View childView : getViewChildren()) { EObject semanticElement = childView.getElement(); if (next.getModelElement().equals(semanticElement)) { if (hint.equals(childView.getType())) { perfectMatch.add(childView); // actually, can stop iteration over view children here, but // may want to use not the first view but last one as a 'real' match (the way original CEP does // with its trick with viewToSemanticMap inside #cleanCanonicalSemanticChildren } } } if (perfectMatch.size() > 0) { descriptorsIterator.remove(); // precise match found no need to create anything for the NodeDescriptor // use only one view (first or last?), keep rest as orphaned for further consideration knownViewChildren.remove(perfectMatch.getFirst()); } } // those left in knownViewChildren are subject to removal - they are our diagram elements we didn't find match to, // or those we have potential matches to, and thus need to be recreated, preserving size/location information. orphaned.addAll(knownViewChildren); // ArrayList<CreateViewRequest.ViewDescriptor> viewDescriptors = new ArrayList<CreateViewRequest.ViewDescriptor>( childDescriptors.size()); for (stateAnalysis.diagram.part.StateAnalysisNodeDescriptor next : childDescriptors) { String hint = stateAnalysis.diagram.part.StateAnalysisVisualIDRegistry .getType(next.getVisualID()); IAdaptable elementAdapter = new CanonicalElementAdapter( next.getModelElement(), hint); CreateViewRequest.ViewDescriptor descriptor = new CreateViewRequest.ViewDescriptor( elementAdapter, Node.class, hint, ViewUtil.APPEND, false, host().getDiagramPreferencesHint()); viewDescriptors.add(descriptor); } boolean changed = deleteViews(orphaned.iterator()); // CreateViewRequest request = getCreateViewRequest(viewDescriptors); Command cmd = getCreateViewCommand(request); if (cmd != null && cmd.canExecute()) { SetViewMutabilityCommand.makeMutable( new EObjectAdapter(host().getNotationView())).execute(); executeCommand(cmd); @SuppressWarnings("unchecked") List<IAdaptable> nl = (List<IAdaptable>) request.getNewObject(); createdViews.addAll(nl); } if (changed || createdViews.size() > 0) { postProcessRefreshSemantic(createdViews); } Collection<IAdaptable> createdConnectionViews = refreshConnections(); if (createdViews.size() > 1) { // perform a layout of the container DeferredLayoutCommand layoutCmd = new DeferredLayoutCommand(host() .getEditingDomain(), createdViews, host()); executeCommand(new ICommandProxy(layoutCmd)); } createdViews.addAll(createdConnectionViews); makeViewsImmutable(createdViews); } /** * @generated */ private Collection<IAdaptable> refreshConnections() { Domain2Notation domain2NotationMap = new Domain2Notation(); Collection<stateAnalysis.diagram.part.StateAnalysisLinkDescriptor> linkDescriptors = collectAllLinks( getDiagram(), domain2NotationMap); Collection existingLinks = new LinkedList(getDiagram().getEdges()); for (Iterator linksIterator = existingLinks.iterator(); linksIterator .hasNext();) { Edge nextDiagramLink = (Edge) linksIterator.next(); int diagramLinkVisualID = stateAnalysis.diagram.part.StateAnalysisVisualIDRegistry .getVisualID(nextDiagramLink); if (diagramLinkVisualID == -1) { if (nextDiagramLink.getSource() != null && nextDiagramLink.getTarget() != null) { linksIterator.remove(); } continue; } EObject diagramLinkObject = nextDiagramLink.getElement(); EObject diagramLinkSrc = nextDiagramLink.getSource().getElement(); EObject diagramLinkDst = nextDiagramLink.getTarget().getElement(); for (Iterator<stateAnalysis.diagram.part.StateAnalysisLinkDescriptor> linkDescriptorsIterator = linkDescriptors .iterator(); linkDescriptorsIterator.hasNext();) { stateAnalysis.diagram.part.StateAnalysisLinkDescriptor nextLinkDescriptor = linkDescriptorsIterator .next(); if (diagramLinkObject == nextLinkDescriptor.getModelElement() && diagramLinkSrc == nextLinkDescriptor.getSource() && diagramLinkDst == nextLinkDescriptor .getDestination() && diagramLinkVisualID == nextLinkDescriptor .getVisualID()) { linksIterator.remove(); linkDescriptorsIterator.remove(); break; } } } deleteViews(existingLinks.iterator()); return createConnections(linkDescriptors, domain2NotationMap); } /** * @generated */ @SuppressWarnings("rawtypes") private Collection<stateAnalysis.diagram.part.StateAnalysisLinkDescriptor> collectAllLinks( View view, Domain2Notation domain2NotationMap) { if (!stateAnalysis.diagram.edit.parts.SystemEditPart.MODEL_ID .equals(stateAnalysis.diagram.part.StateAnalysisVisualIDRegistry .getModelID(view))) { return Collections.emptyList(); } LinkedList<stateAnalysis.diagram.part.StateAnalysisLinkDescriptor> result = new LinkedList<stateAnalysis.diagram.part.StateAnalysisLinkDescriptor>(); switch (stateAnalysis.diagram.part.StateAnalysisVisualIDRegistry .getVisualID(view)) { case stateAnalysis.diagram.edit.parts.SystemEditPart.VISUAL_ID: { if (!domain2NotationMap.containsKey(view.getElement())) { result.addAll(stateAnalysis.diagram.part.StateAnalysisDiagramUpdater .getSystem_1000ContainedLinks(view)); } domain2NotationMap.putView(view.getElement(), view); break; } case stateAnalysis.diagram.edit.parts.SubsystemEditPart.VISUAL_ID: { if (!domain2NotationMap.containsKey(view.getElement())) { result.addAll(stateAnalysis.diagram.part.StateAnalysisDiagramUpdater .getSubsystem_2001ContainedLinks(view)); } domain2NotationMap.putView(view.getElement(), view); break; } case stateAnalysis.diagram.edit.parts.DeploymentSetEditPart.VISUAL_ID: { if (!domain2NotationMap.containsKey(view.getElement())) { result.addAll(stateAnalysis.diagram.part.StateAnalysisDiagramUpdater .getDeploymentSet_3001ContainedLinks(view)); } domain2NotationMap.putView(view.getElement(), view); break; } case stateAnalysis.diagram.edit.parts.DeploymentEditPart.VISUAL_ID: { if (!domain2NotationMap.containsKey(view.getElement())) { result.addAll(stateAnalysis.diagram.part.StateAnalysisDiagramUpdater .getDeployment_3002ContainedLinks(view)); } domain2NotationMap.putView(view.getElement(), view); break; } case stateAnalysis.diagram.edit.parts.StateVariableEditPart.VISUAL_ID: { if (!domain2NotationMap.containsKey(view.getElement())) { result.addAll(stateAnalysis.diagram.part.StateAnalysisDiagramUpdater .getStateVariable_3027ContainedLinks(view)); } domain2NotationMap.putView(view.getElement(), view); break; } case stateAnalysis.diagram.edit.parts.StateVariable2EditPart.VISUAL_ID: { if (!domain2NotationMap.containsKey(view.getElement())) { result.addAll(stateAnalysis.diagram.part.StateAnalysisDiagramUpdater .getStateVariable_3011ContainedLinks(view)); } domain2NotationMap.putView(view.getElement(), view); break; } case stateAnalysis.diagram.edit.parts.StateConstraintEditPart.VISUAL_ID: { if (!domain2NotationMap.containsKey(view.getElement())) { result.addAll(stateAnalysis.diagram.part.StateAnalysisDiagramUpdater .getStateConstraint_3008ContainedLinks(view)); } domain2NotationMap.putView(view.getElement(), view); break; } case stateAnalysis.diagram.edit.parts.StateConstraint2EditPart.VISUAL_ID: { if (!domain2NotationMap.containsKey(view.getElement())) { result.addAll(stateAnalysis.diagram.part.StateAnalysisDiagramUpdater .getStateConstraint_3009ContainedLinks(view)); } domain2NotationMap.putView(view.getElement(), view); break; } case stateAnalysis.diagram.edit.parts.StateUpdateEditPart.VISUAL_ID: { if (!domain2NotationMap.containsKey(view.getElement())) { result.addAll(stateAnalysis.diagram.part.StateAnalysisDiagramUpdater .getStateUpdate_3010ContainedLinks(view)); } domain2NotationMap.putView(view.getElement(), view); break; } case stateAnalysis.diagram.edit.parts.ControllerEditPart.VISUAL_ID: { if (!domain2NotationMap.containsKey(view.getElement())) { result.addAll(stateAnalysis.diagram.part.StateAnalysisDiagramUpdater .getController_3028ContainedLinks(view)); } domain2NotationMap.putView(view.getElement(), view); break; } case stateAnalysis.diagram.edit.parts.StateConstraint3EditPart.VISUAL_ID: { if (!domain2NotationMap.containsKey(view.getElement())) { result.addAll(stateAnalysis.diagram.part.StateAnalysisDiagramUpdater .getStateConstraint_3013ContainedLinks(view)); } domain2NotationMap.putView(view.getElement(), view); break; } case stateAnalysis.diagram.edit.parts.HardwareCommandEditPart.VISUAL_ID: { if (!domain2NotationMap.containsKey(view.getElement())) { result.addAll(stateAnalysis.diagram.part.StateAnalysisDiagramUpdater .getHardwareCommand_3014ContainedLinks(view)); } domain2NotationMap.putView(view.getElement(), view); break; } case stateAnalysis.diagram.edit.parts.EstimatorEditPart.VISUAL_ID: { if (!domain2NotationMap.containsKey(view.getElement())) { result.addAll(stateAnalysis.diagram.part.StateAnalysisDiagramUpdater .getEstimator_3029ContainedLinks(view)); } domain2NotationMap.putView(view.getElement(), view); break; } case stateAnalysis.diagram.edit.parts.StateConstraint4EditPart.VISUAL_ID: { if (!domain2NotationMap.containsKey(view.getElement())) { result.addAll(stateAnalysis.diagram.part.StateAnalysisDiagramUpdater .getStateConstraint_3016ContainedLinks(view)); } domain2NotationMap.putView(view.getElement(), view); break; } case stateAnalysis.diagram.edit.parts.StateUpdate2EditPart.VISUAL_ID: { if (!domain2NotationMap.containsKey(view.getElement())) { result.addAll(stateAnalysis.diagram.part.StateAnalysisDiagramUpdater .getStateUpdate_3017ContainedLinks(view)); } domain2NotationMap.putView(view.getElement(), view); break; } case stateAnalysis.diagram.edit.parts.HardwareMeasurementEditPart.VISUAL_ID: { if (!domain2NotationMap.containsKey(view.getElement())) { result.addAll(stateAnalysis.diagram.part.StateAnalysisDiagramUpdater .getHardwareMeasurement_3018ContainedLinks(view)); } domain2NotationMap.putView(view.getElement(), view); break; } case stateAnalysis.diagram.edit.parts.HardwareAdapterEditPart.VISUAL_ID: { if (!domain2NotationMap.containsKey(view.getElement())) { result.addAll(stateAnalysis.diagram.part.StateAnalysisDiagramUpdater .getHardwareAdapter_3030ContainedLinks(view)); } domain2NotationMap.putView(view.getElement(), view); break; } case stateAnalysis.diagram.edit.parts.HardwareCommand2EditPart.VISUAL_ID: { if (!domain2NotationMap.containsKey(view.getElement())) { result.addAll(stateAnalysis.diagram.part.StateAnalysisDiagramUpdater .getHardwareCommand_3020ContainedLinks(view)); } domain2NotationMap.putView(view.getElement(), view); break; } case stateAnalysis.diagram.edit.parts.HardwareMeasurement2EditPart.VISUAL_ID: { if (!domain2NotationMap.containsKey(view.getElement())) { result.addAll(stateAnalysis.diagram.part.StateAnalysisDiagramUpdater .getHardwareMeasurement_3021ContainedLinks(view)); } domain2NotationMap.putView(view.getElement(), view); break; } case stateAnalysis.diagram.edit.parts.DeviceEditPart.VISUAL_ID: { if (!domain2NotationMap.containsKey(view.getElement())) { result.addAll(stateAnalysis.diagram.part.StateAnalysisDiagramUpdater .getDevice_3024ContainedLinks(view)); } domain2NotationMap.putView(view.getElement(), view); break; } case stateAnalysis.diagram.edit.parts.DeviceCommandEditPart.VISUAL_ID: { if (!domain2NotationMap.containsKey(view.getElement())) { result.addAll(stateAnalysis.diagram.part.StateAnalysisDiagramUpdater .getDeviceCommand_3025ContainedLinks(view)); } domain2NotationMap.putView(view.getElement(), view); break; } case stateAnalysis.diagram.edit.parts.DeviceMeasurementEditPart.VISUAL_ID: { if (!domain2NotationMap.containsKey(view.getElement())) { result.addAll(stateAnalysis.diagram.part.StateAnalysisDiagramUpdater .getDeviceMeasurement_3026ContainedLinks(view)); } domain2NotationMap.putView(view.getElement(), view); break; } case stateAnalysis.diagram.edit.parts.DataCollectionEditPart.VISUAL_ID: { if (!domain2NotationMap.containsKey(view.getElement())) { result.addAll(stateAnalysis.diagram.part.StateAnalysisDiagramUpdater .getDataCollection_3004ContainedLinks(view)); } domain2NotationMap.putView(view.getElement(), view); break; } case stateAnalysis.diagram.edit.parts.QosPolicyEditPart.VISUAL_ID: { if (!domain2NotationMap.containsKey(view.getElement())) { result.addAll(stateAnalysis.diagram.part.StateAnalysisDiagramUpdater .getQosPolicy_3005ContainedLinks(view)); } domain2NotationMap.putView(view.getElement(), view); break; } case stateAnalysis.diagram.edit.parts.DataProductEditPart.VISUAL_ID: { if (!domain2NotationMap.containsKey(view.getElement())) { result.addAll(stateAnalysis.diagram.part.StateAnalysisDiagramUpdater .getDataProduct_3006ContainedLinks(view)); } domain2NotationMap.putView(view.getElement(), view); break; } } for (Iterator children = view.getChildren().iterator(); children .hasNext();) { result.addAll(collectAllLinks((View) children.next(), domain2NotationMap)); } for (Iterator edges = view.getSourceEdges().iterator(); edges.hasNext();) { result.addAll(collectAllLinks((View) edges.next(), domain2NotationMap)); } return result; } /** * @generated */ private Collection<IAdaptable> createConnections( Collection<stateAnalysis.diagram.part.StateAnalysisLinkDescriptor> linkDescriptors, Domain2Notation domain2NotationMap) { LinkedList<IAdaptable> adapters = new LinkedList<IAdaptable>(); for (stateAnalysis.diagram.part.StateAnalysisLinkDescriptor nextLinkDescriptor : linkDescriptors) { EditPart sourceEditPart = getSourceEditPart(nextLinkDescriptor, domain2NotationMap); EditPart targetEditPart = getTargetEditPart(nextLinkDescriptor, domain2NotationMap); if (sourceEditPart == null || targetEditPart == null) { continue; } CreateConnectionViewRequest.ConnectionViewDescriptor descriptor = new CreateConnectionViewRequest.ConnectionViewDescriptor( nextLinkDescriptor.getSemanticAdapter(), stateAnalysis.diagram.part.StateAnalysisVisualIDRegistry .getType(nextLinkDescriptor.getVisualID()), 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(); if (viewAdapter != null) { adapters.add(viewAdapter); } } } return adapters; } /** * @generated */ private EditPart getEditPart(EObject domainModelElement, Domain2Notation domain2NotationMap) { View view = (View) domain2NotationMap.get(domainModelElement); if (view != null) { return (EditPart) getHost().getViewer().getEditPartRegistry() .get(view); } return null; } /** * @generated */ private Diagram getDiagram() { return ((View) getHost().getModel()).getDiagram(); } /** * @generated */ private EditPart getSourceEditPart(UpdaterLinkDescriptor descriptor, Domain2Notation domain2NotationMap) { return getEditPart(descriptor.getSource(), domain2NotationMap); } /** * @generated */ private EditPart getTargetEditPart(UpdaterLinkDescriptor descriptor, Domain2Notation domain2NotationMap) { return getEditPart(descriptor.getDestination(), domain2NotationMap); } /** * @generated */ protected final EditPart getHintedEditPart(EObject domainModelElement, Domain2Notation domain2NotationMap, int hintVisualId) { View view = (View) domain2NotationMap.getHinted(domainModelElement, stateAnalysis.diagram.part.StateAnalysisVisualIDRegistry .getType(hintVisualId)); if (view != null) { return (EditPart) getHost().getViewer().getEditPartRegistry() .get(view); } return null; } /** * @generated */ @SuppressWarnings("serial") protected static class Domain2Notation extends HashMap<EObject, View> { /** * @generated */ public boolean containsDomainElement(EObject domainElement) { return this.containsKey(domainElement); } /** * @generated */ public View getHinted(EObject domainEObject, String hint) { return this.get(domainEObject); } /** * @generated */ public void putView(EObject domainElement, View view) { if (!containsKey(view.getElement()) || !isShortcut(view)) { this.put(domainElement, view); } } } }