deleteshapeshellaction.java

一个可视化编辑器的基础结构

package org.goshawk.workflow.GUI.Action;

import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.util.Collection;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;

import org.goshawk.workflow.GUI.Display.EditorCanvas;
import org.goshawk.workflow.GUI.Display.EditorFrame;
import org.goshawk.workflow.GUI.Display.ShapeMap;
import org.goshawk.workflow.GUI.MouseCommand.MouseCommand;
import org.goshawk.workflow.GUI.ShapeShell.ArrowShell;
import org.goshawk.workflow.GUI.ShapeShell.ShapeShell;

public class DeleteShapeShellAction implements ActionListener
{
	private EditorCanvas canvas = EditorFrame.getCanvas();
	private ShapeMap shapeMap = canvas.getShapeMap();
	private ShapeShell shape = null;
	
	public DeleteShapeShellAction()
	{
		
	}
	
	public DeleteShapeShellAction(ShapeShell shape)
	{
		this.shape = shape;
	}
	public void actionPerformed(ActionEvent e)
	{
		Map holdShape = new HashMap();
		if(this.shape!=null)
		{
			shapeMap.remove(this.shape.getId());
		}
		else
		{
			Collection shapeSet = shapeMap.values();
			Iterator iter = shapeSet.iterator();
			while(iter.hasNext())
			{
				ShapeShell shape = (ShapeShell) iter.next();
				
				if(shape!=null&&!shape.getId().endsWith(MouseCommand.RECT_AREA)&&!shape.isSelected())
				{
					holdShape.put(shape.getId(),shape);
				}
			}
			shapeMap.clear();
			shapeMap.putAll(holdShape);
		}
		holdShape = null;
		holdShape = new HashMap();
		Collection shapeSet = shapeMap.values();
		Iterator iter = shapeSet.iterator();
		while(iter.hasNext())
		{
			ShapeShell shape = (ShapeShell) iter.next();
			
			if(shape!=null&&shape instanceof ArrowShell)
			{
				if(shapeMap.get(((ArrowShell)shape).getStartNode().getId())==null|shapeMap.get(((ArrowShell)shape).getEndNode().getId())==null)
					holdShape.put(shape.getId(),shape);
			}
		}
		shapeSet = holdShape.values();
		iter = shapeSet.iterator();
		while(iter.hasNext())
		{
			ShapeShell shape = (ShapeShell) iter.next();
			if(shape!=null)
				shapeMap.remove(shape.getId());
		}
		canvas.repaint();
	}

}