groupnodeaction.java

对eclipse gef进行封装,可以生成图形化编辑器

/*******************************************************************************
 * $Header: /cvsroot/EOS6/work_dir/niegy/com.primeton.studio.gef.ui/src/com/primeton/studio/gef/ui/action/GroupNodeAction.java,v 1.2 2006/12/30 02:45:04 niegy Exp $
 * $Revision: 1.2 $
 * $Date: 2006/12/30 02:45:04 $
 *
 *==============================================================================
 *
 * Copyright (c) 2001-2006 Primeton Technologies, Ltd.
 * All rights reserved.
 * 
 * Created on 2006-12-6
 *******************************************************************************/


package com.primeton.studio.gef.ui.action;

import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

import org.eclipse.gef.EditPart;
import org.eclipse.gef.EditPartViewer;
import org.eclipse.gef.GraphicalViewer;
import org.eclipse.gef.editparts.AbstractGraphicalEditPart;
import org.eclipse.gef.ui.actions.EditorPartAction;
import org.eclipse.jface.viewers.StructuredSelection;
import org.eclipse.ui.IEditorPart;

import com.primeton.studio.gef.core.CoreFactory;
import com.primeton.studio.gef.core.Diagram;
import com.primeton.studio.gef.core.GroupNode;
import com.primeton.studio.gef.core.NodeElement;
import com.primeton.studio.gef.ui.Constants;
import com.primeton.studio.gef.ui.commands.GroupCommand;
import com.primeton.studio.gef.ui.editor.AbstractGraphicalEditor;
import com.primeton.studio.gef.ui.parts.NodeEditPartEx;

/**
 * TODO 此处填写 class 信息
 *
 * @author niegy (mailto:[e-mail])
 */
/*
 * 修改历史
 * $Log: GroupNodeAction.java,v $
 * Revision 1.2  2006/12/30 02:45:04  niegy
 * 重构代码
 *
 * Revision 1.1  2006/12/08 07:04:32  niegy
 * 重构代码
 * 
 */
public class GroupNodeAction extends EditorPartAction {

	/**
	 * @param editor
	 */
	public GroupNodeAction(IEditorPart editor) {
		super(editor);
		// TODO Auto-generated constructor stub
	}

    /* (non-Javadoc)
     * @see org.eclipse.gef.ui.actions.WorkbenchPartAction#calculateEnabled()
     */
    protected boolean calculateEnabled() {
        return getSelectedObjects().size() > 1;
    }

    protected void init() {
        setId(Constants.ACTION_GROUP);
        setText("group"); 
    }
    
    
    public void run() {
        List objects = getSelectedObjects();
        GroupNode groupNode = CoreFactory.eINSTANCE.createGroupNode();
        groupNode.setId("group");
        groupNode.setName("group");
        
//        groupNode.setParent(getGraphicalDiagram());
        List array = new ArrayList();
        for (Iterator iter = objects.iterator(); iter.hasNext();) {
        	AbstractGraphicalEditPart element = (AbstractGraphicalEditPart) iter.next();
            if (element instanceof NodeEditPartEx) {
            	NodeElement nodeElement = (NodeElement)element.getModel();
            	if(!nodeElement.isGrouped())
                    array.add(nodeElement);
            }
        }
        GroupCommand command = new GroupCommand(array,getGraphicalDiagram());
        command.setGroupNode(groupNode);
        getCommandStack().execute(command);
        
        AbstractGraphicalEditor editor = (AbstractGraphicalEditor)getEditorPart();
        EditPartViewer viewer = editor.getGraphicalViewer();
        EditPart editPart = (EditPart)viewer.getEditPartRegistry().get(groupNode);
        editor.getGraphicalViewer().setSelection(new StructuredSelection(editPart));       
    }
    
    public List getSelectedObjects() {
    	AbstractGraphicalEditor editor = (AbstractGraphicalEditor)getEditorPart();
        GraphicalViewer viewer = (GraphicalViewer)editor.getAdapter(GraphicalViewer.class);
        
        List result = viewer.getSelectedEditParts();
        return result;
    }
    
    public Diagram getGraphicalDiagram() {
    	AbstractGraphicalEditor editor = (AbstractGraphicalEditor)getEditorPart();
        return (Diagram)editor.getModel();
    }

}