pastenodeaction.java

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

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

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

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

import org.eclipse.draw2d.geometry.Dimension;
import org.eclipse.draw2d.geometry.Point;
import org.eclipse.draw2d.geometry.Rectangle;
import org.eclipse.gef.EditPart;
import org.eclipse.gef.EditPartViewer;
import org.eclipse.gef.commands.Command;
import org.eclipse.gef.commands.CompoundCommand;
import org.eclipse.gef.requests.CreateRequest;
import org.eclipse.gef.requests.CreationFactory;
import org.eclipse.gef.ui.actions.Clipboard;
import org.eclipse.gef.ui.actions.EditorPartAction;
import org.eclipse.jface.viewers.ISelection;
import org.eclipse.jface.viewers.ISelectionChangedListener;
import org.eclipse.jface.viewers.IStructuredSelection;
import org.eclipse.jface.viewers.SelectionChangedEvent;
import org.eclipse.ui.IEditorPart;
import org.eclipse.ui.ISharedImages;
import org.eclipse.ui.actions.ActionFactory;
import org.eclipse.ui.internal.WorkbenchImages;

import com.primeton.studio.gef.core.Connection;
import com.primeton.studio.gef.core.Diagram;
import com.primeton.studio.gef.core.NodeElement;
import com.primeton.studio.gef.ui.commands.ConnectionCreateCommand;
import com.primeton.studio.gef.ui.commands.NodeCreateCommand;
import com.primeton.studio.gef.ui.editor.AbstractGraphicalEditor;
import com.primeton.studio.gef.ui.parts.DiagramEditPart;
import com.primeton.studio.gef.ui.parts.ExtenderReader;

/**
 * TODO此处填写 class 信息
 *
 * @author niegy (mailto:[e-mail])
 */
/*
 * 修改历史
 * $Log: PasteNodeAction.java,v $
 * Revision 1.7  2006/12/30 02:45:04  niegy
 * 重构代码
 *
 * Revision 1.6  2006/12/20 02:57:07  niegy
 * 重构代码
 *
 * Revision 1.5  2006/12/16 09:04:42  niegy
 * 重构代码
 *
 * Revision 1.4  2006/12/05 05:18:59  niegy
 * 修改模型，增加连线的扩展点
 *
 * Revision 1.3  2006/11/22 07:07:38  niegy
 * 增加table
 *
 * Revision 1.2  2006/11/18 12:13:54  niegy
 * 增加容器模型
 *
 * Revision 1.1  2006/11/17 03:15:13  niegy
 * create
 * 
 */
public class PasteNodeAction extends EditorPartAction implements
		ISelectionChangedListener {
    protected List addedObjects = new ArrayList();
	private Command pasteCommand = null;
	private AbstractGraphicalEditor graphicalEditor;

	/* Editor Model List */
	protected List selecteds;

	/* Selection Object */
	protected IStructuredSelection selection;

	/**
	 * @param editor
	 */
	public PasteNodeAction(IEditorPart editor) {
		super(editor);
		graphicalEditor = (AbstractGraphicalEditor)editor;
	}

	/*
	 * (non-Javadoc)
	 * 
	 * @see org.eclipse.gef.ui.actions.WorkbenchPartAction#init()
	 */
	protected void init() {
		setId(ActionFactory.PASTE.getId());
		setText("&Paste"); 
		setImageDescriptor(WorkbenchImages
				.getImageDescriptor(ISharedImages.IMG_TOOL_PASTE));
		setDisabledImageDescriptor(WorkbenchImages
				.getImageDescriptor(ISharedImages.IMG_TOOL_PASTE_DISABLED));
	}

	/*
	 * (non-Javadoc)
	 * 
	 * @see org.eclipse.gef.ui.actions.WorkbenchPartAction#calculateEnabled()
	 */
	protected boolean calculateEnabled() {
		Command command=createPasteCommand();
		if(command!=null)
			return command.canExecute();
		return false;
	}


	/*
	 * (non-Javadoc)
	 * 
	 * @see org.eclipse.gef.Disposable#dispose()
	 */
	public void dispose() {
		selecteds = null;
	}

	private void createModelChild(NodeElement oldNode,NodeElement parent,Hashtable newModels,CompoundCommand ret){
		if(oldNode.getNodes().size()<=0) return;
		for (Iterator iter = oldNode.getNodes().iterator(); iter.hasNext();) {
			NodeElement nodeElement = (NodeElement) iter.next();
			CreateRequest request = new CreateRequest(nodeElement);			 
            CreationFactory factory = ExtenderReader.getInstance().createModelCreationFactory(graphicalEditor.getPluginId(),nodeElement);
            request.setFactory(factory);
            Rectangle constraint = null;
            if(nodeElement.getLocation() != null && nodeElement.getSize() != null){
            	request.setLocation(new Point(nodeElement.getLocation().getX(), nodeElement.getLocation().getY()));
            	request.setSize(new Dimension(nodeElement.getSize().getWidth(), nodeElement.getSize().getHeight()));
            	constraint = new Rectangle(nodeElement.getLocation().getX()+10,nodeElement.getLocation().getY()+10,
                     nodeElement.getSize().getWidth(),nodeElement.getSize().getHeight());
            }
            NodeCreateCommand command = new NodeCreateCommand(parent, request, constraint,graphicalEditor.getPluginId());
            if(command != null){
                NodeElement newNode = command.getNode();
                newNode.setCollapsed(nodeElement.isCollapsed());
                newModels.put(nodeElement, newNode);
                addedObjects.add(newNode);
                ret.add(command);
            }	
		 
		}
	}
	
    protected Command createPasteCommand() {
        // TODO Auto-generated method stub
        if (getClipboardContents() == null)
            return null;
        if (!(getClipboardContents() instanceof List))
            return null;

        List models = (List) getClipboardContents();
        Hashtable newModels = new Hashtable();
        addedObjects.clear();
        CompoundCommand ret = new CompoundCommand();
        if(selection == null)return ret;
        Object obj = selection.getFirstElement();
        if (obj instanceof DiagramEditPart) {
            DiagramEditPart container = (DiagramEditPart) obj;
            for (int i = 0; i < models.size(); i++) {
                if (models.get(i) instanceof NodeElement) {
                    NodeElement nodeElement = (NodeElement) models.get(i);
                    CreateRequest request = new CreateRequest(nodeElement);
                    CreationFactory factory = ExtenderReader.getInstance().createModelCreationFactory(graphicalEditor.getPluginId(),nodeElement);
                    request.setFactory(factory);//new BizModelCreationFactory(node.getClass()));
                    if(nodeElement.getLocation() != null)
                    	request.setLocation(new Point(nodeElement.getLocation().getX(), nodeElement.getLocation().getY()));
                    if(nodeElement.getSize() != null)
                    	request.setSize(new Dimension(nodeElement.getSize().getWidth(), nodeElement.getSize().getHeight()));
                
                    Diagram automation = container.getCastedModel();
                    Rectangle constraint = new Rectangle(nodeElement.getLocation().getX()+10,nodeElement.getLocation().getY()+10,
                            nodeElement.getSize().getWidth(),nodeElement.getSize().getHeight());
                    NodeCreateCommand command = new NodeCreateCommand(automation, request, constraint,graphicalEditor.getPluginId());
                    if(command != null){
                        NodeElement newNode = command.getNode();
                        newNode.setCollapsed(nodeElement.isCollapsed());
                        newModels.put(nodeElement, newNode);
                        addedObjects.add(newNode);
                        ret.add(command);
                        createModelChild(nodeElement,newNode,newModels,ret);
                    }

                }
            }

            Enumeration keys = newModels.keys();
            while (keys.hasMoreElements()) {
                NodeElement nodeElement = (NodeElement) keys.nextElement();
                for (int j = 0; j < nodeElement.getSourceConnections().size(); j++) {
                    Connection conn = (Connection)nodeElement.getSourceConnections().get(j);
                    if (newModels.containsKey(conn.getTargetNode())) {      
                    	CreateRequest request = new CreateRequest(conn);
                        CreationFactory factory = ExtenderReader.getInstance().cloneModelCreationFactory(graphicalEditor.getPluginId(),conn);
                        request.setFactory(factory);
                    	ConnectionCreateCommand cmd = 
                            new ConnectionCreateCommand((NodeElement)newModels.get(nodeElement),request,graphicalEditor);                                        
                        cmd.setTarget((NodeElement) newModels.get(conn.getTargetNode()));                       
//                        Connection connection = ExtenderReader.getInstance().createConnection(graphicalEditor.getPluginId());
//                        cmd.setConnection(connection);
                        ret.add(cmd);
                    }
                }
            }
        }

        return ret;
    }

	/*
	 * (non-Javadoc)
	 * 
	 * @see org.eclipse.jface.viewers.ISelectionChangedListener#selectionChanged(org.eclipse.jface.viewers.SelectionChangedEvent)
	 */
	public void selectionChanged(SelectionChangedEvent event) {
		selecteds = getSelection(event);
		refresh();

	}

    public  List getSelection(SelectionChangedEvent event) {
        // TODO Auto-generated method stub
        List selecteds = null;
        ISelection s = event.getSelection();
        if (!(s instanceof IStructuredSelection))
            return null;
        
        selection = (IStructuredSelection) s;
        if (selection != null) {
            selecteds = new ArrayList();
            for (int i = 0; i < selection.size(); i++) {
                Object obj = selection.toList().get(i);
                
                Object model = ((EditPart) obj).getModel();
                if (!(model instanceof NodeElement)) {
                    selecteds = null;
                    break;
                }
                selecteds.add(model);   
            }
        }
        return selecteds;
    }
    

	public void run() {
		if (selection.size() > 0) {
			Object selectedObject = selection.getFirstElement();
			
			Command command=createPasteCommand();

			if (command == null)
				return;

			execute(command);
			pasteCommand = null;

			EditPartViewer viewer = getEditPartViewer();

			List tmp = new ArrayList();
			tmp.addAll(addedObjects);
			for (int i = 0; i < tmp.size(); i++) {
				Object editpart = viewer.getEditPartRegistry().get(tmp.get(i));
				if (editpart instanceof EditPart) {
					viewer.appendSelection((EditPart) editpart);
				}
			}
		}
	}

	private EditPartViewer getEditPartViewer() {
        AbstractGraphicalEditor editor = (AbstractGraphicalEditor) getWorkbenchPart();
		return editor.getGraphicalViewer();
	}

    protected Object getClipboardContents() {
		return Clipboard.getDefault().getContents();
	}

}