PackageItemSemanticEditPolicy.java

Go to the documentation of this file.

package org.ros.model.ros.diagram.edit.policies;

import java.util.Iterator;

import org.eclipse.emf.ecore.EAnnotation;
import org.eclipse.gef.commands.Command;
import org.eclipse.gmf.runtime.common.core.command.ICompositeCommand;
import org.eclipse.gmf.runtime.diagram.core.commands.DeleteCommand;
import org.eclipse.gmf.runtime.emf.commands.core.command.CompositeTransactionalCommand;
import org.eclipse.gmf.runtime.emf.type.core.commands.DestroyElementCommand;
import org.eclipse.gmf.runtime.emf.type.core.requests.DestroyElementRequest;
import org.eclipse.gmf.runtime.emf.type.core.requests.DestroyRequest;
import org.eclipse.gmf.runtime.emf.type.core.requests.IEditCommandRequest;
import org.eclipse.gmf.runtime.notation.Node;
import org.eclipse.gmf.runtime.notation.View;
import org.ros.model.ros.diagram.edit.parts.NodeEditPart;
import org.ros.model.ros.diagram.edit.parts.PackagePackageContentCompartmentEditPart;
import org.ros.model.ros.diagram.part.RosVisualIDRegistry;
import org.ros.model.ros.diagram.providers.RosElementTypes;

public class PackageItemSemanticEditPolicy extends
                RosBaseItemSemanticEditPolicy {

        public PackageItemSemanticEditPolicy() {
                super(RosElementTypes.Package_2001);
        }

        protected Command getSemanticCommand(IEditCommandRequest request) {
                if (request instanceof DestroyRequest) {
                        return null;
                }
                return super.getSemanticCommand(request);
        }

        protected Command getDestroyElementCommand(DestroyElementRequest req) {
                View view = (View) getHost().getModel();
                CompositeTransactionalCommand cmd = new CompositeTransactionalCommand(
                                getEditingDomain(), null);
                cmd.setTransactionNestingEnabled(false);
                EAnnotation annotation = view.getEAnnotation("Shortcut"); //$NON-NLS-1$
                if (annotation == null) {
                        // there are indirectly referenced children, need extra commands: false
                        addDestroyChildNodesCommand(cmd);
                        addDestroyShortcutsCommand(cmd, view);
                        // delete host element
                        cmd.add(new DestroyElementCommand(req));
                } else {
                        cmd.add(new DeleteCommand(getEditingDomain(), view));
                }
                return getGEFWrapper(cmd.reduce());
        }

        private void addDestroyChildNodesCommand(ICompositeCommand cmd) {
                View view = (View) getHost().getModel();
                for (Iterator<?> nit = view.getChildren().iterator(); nit.hasNext();) {
                        Node node = (Node) nit.next();
                        switch (RosVisualIDRegistry.getVisualID(node)) {
                        case PackagePackageContentCompartmentEditPart.VISUAL_ID:
                                for (Iterator<?> cit = node.getChildren().iterator(); cit
                                                .hasNext();) {
                                        Node cnode = (Node) cit.next();
                                        switch (RosVisualIDRegistry.getVisualID(cnode)) {
                                        case NodeEditPart.VISUAL_ID:
                                                cmd.add(new DestroyElementCommand(
                                                                new DestroyElementRequest(getEditingDomain(),
                                                                                cnode.getElement(), false))); // directlyOwned: true
                                                // don't need explicit deletion of cnode as parent's view deletion would clean child views as well 
                                                // cmd.add(new org.eclipse.gmf.runtime.diagram.core.commands.DeleteCommand(getEditingDomain(), cnode));
                                                break;
                                        }
                                }
                                break;
                        }
                }
        }

}