SelectGraphicCommand.java

/*
 * Copyright (C) Yutaka Matsuno 2010-2012 All rights reserved.
 */
package net.dependableos.dcase.diagram.editor.command;

import java.util.Set;

import net.dependableos.dcase.diagram.edit.parts.ArgumentEditPart;
import net.dependableos.dcase.diagram.editor.common.util.DcaseEditorUtil;

import org.eclipse.core.commands.ExecutionException;
import org.eclipse.core.runtime.IAdaptable;
import org.eclipse.core.runtime.IProgressMonitor;
import org.eclipse.emf.ecore.xmi.XMLResource;
import org.eclipse.gmf.runtime.common.core.command.AbstractCommand;
import org.eclipse.gmf.runtime.common.core.command.CommandResult;
import org.eclipse.gmf.runtime.diagram.ui.editparts.ConnectionNodeEditPart;
import org.eclipse.gmf.runtime.diagram.ui.editparts.ShapeNodeEditPart;

/**
 * A command to select subtree specified edit parts.
 */
public class SelectGraphicCommand extends AbstractCommand {
    
    /**
     * the argument.
     */
    private ArgumentEditPart argumentEditPart;

    /**
     * the set of IDs those will be excluded to select.
     */
    private Set<String> subSelectIdSet;


    /**
     * Creates the command and initializes it.
     * 
     * @param label the command label.
     * @param argumentEditPart the argument.
     * @param subSelectIdSet the select subtree ID set.
     */
    public SelectGraphicCommand(String label,
            ArgumentEditPart argumentEditPart, Set<String> subSelectIdSet) {
        super(label);
        this.argumentEditPart = argumentEditPart;
        this.subSelectIdSet = subSelectIdSet;
    }

    /* (non-Javadoc)
     * @see org.eclipse.gmf.runtime.common.core.command.AbstractCommand#doExecuteWithResult
     * (org.eclipse.core.runtime.IProgressMonitor, org.eclipse.core.runtime.IAdaptable)
     */
    @Override
    protected CommandResult doExecuteWithResult(
            IProgressMonitor progressMonitor, IAdaptable info)
            throws ExecutionException {
        doProcess();
        return null;
    }

    /* (non-Javadoc)
     * @see org.eclipse.gmf.runtime.common.core.command.AbstractCommand#doRedoWithResult
     * (org.eclipse.core.runtime.IProgressMonitor, org.eclipse.core.runtime.IAdaptable)
     */
    @Override
    protected CommandResult doRedoWithResult(IProgressMonitor progressMonitor,
            IAdaptable info) throws ExecutionException {
        doProcess();
        return null;
    }

    /* (non-Javadoc)
     * @see org.eclipse.gmf.runtime.common.core.command.AbstractCommand#doUndoWithResult
     * (org.eclipse.core.runtime.IProgressMonitor, org.eclipse.core.runtime.IAdaptable)
     */
    @Override
    protected CommandResult doUndoWithResult(IProgressMonitor progressMonitor,
            IAdaptable info) throws ExecutionException {
        return null;
    }

    /**
     * Selects edit parts excluding specified edit parts.
     */
    private void doProcess() {

        // select nodes
        XMLResource resource = DcaseEditorUtil.getXMLResource(argumentEditPart);
        int counter = 0;
        for (Object obj : argumentEditPart.getChildren()) {

            if (!(obj instanceof ShapeNodeEditPart)) {
                continue;
            }
            
            ShapeNodeEditPart nodeEditPart = (ShapeNodeEditPart) obj;

            String id = DcaseEditorUtil.getUUIDs(nodeEditPart, resource);
            if (!subSelectIdSet.contains(id)) {
                continue;
            }

            if (counter == 0) {
                argumentEditPart.getViewer().select(nodeEditPart);
            } else {
                argumentEditPart.getViewer().appendSelection(nodeEditPart);
            }
            counter++;
        }
        
        // select connections
        for (Object obj : argumentEditPart.getConnections()) {
            
            if (!(obj instanceof ConnectionNodeEditPart)) {
                continue;
            }
            
            ConnectionNodeEditPart nodeEditPart = (ConnectionNodeEditPart) obj;
            
            String id = DcaseEditorUtil.getUUIDs(nodeEditPart);
            
            if (!subSelectIdSet.contains(id)) {
                continue;
            }
           
            argumentEditPart.getViewer().appendSelection(nodeEditPart);
        }
        
    }
    
}