systemeditpart.java

来自「mywork是rcp开发的很好的例子」· Java 代码 · 共 117 行

package net.sf.freenote.uml;

import java.beans.PropertyChangeEvent;
import java.util.List;

import net.sf.freenote.FreeNoteConstants;
import net.sf.freenote.directedit.ShapeDirectEditManager;
import net.sf.freenote.directedit.TitleCellEditorLocator;
import net.sf.freenote.parts.ShapesContainerEditPolicy;
import net.sf.freenote.parts.ShapesXYLayoutEditPolicy;
import net.sf.freenote.uml.figure.SystemFigure;
import net.sf.freenote.uml.model.SystemShape;

import org.eclipse.draw2d.IFigure;
import org.eclipse.gef.EditPart;
import org.eclipse.gef.EditPolicy;
import org.eclipse.gef.GraphicalEditPart;
import org.eclipse.gef.Request;
import org.eclipse.gef.RequestConstants;
import org.eclipse.gef.editpolicies.GraphicalEditPolicy;
import org.eclipse.swt.graphics.Color;

/**
 * System的控制器
 * @author levin
 * @since 2008-2-3 下午10:26:21
 */
public class SystemEditPart extends UmlShapeEditPart {
	private final static Color SelectedBlue = new Color(null, 200, 200, 240);

	@Override
	protected IFigure createFigureForModel() {
		return new SystemFigure();
	}

	@Override
	public void propertyChange(PropertyChangeEvent evt) {
		String prop = evt.getPropertyName();
		if (FreeNoteConstants.CHILD_ADDED_PROP.equals(prop)|| FreeNoteConstants.CHILD_REMOVED_PROP.equals(prop)) {
			refreshChildren();
		}
		super.propertyChange(evt);
	}

	@Override
	public IFigure getContentPane() {
		return ((SystemFigure)getFigure()).getContentsPane();
	}
	
	protected List getModelChildren() {
		return ((SystemShape) getModel()).getChildren();
	}	

	@Override
	protected void createEditPolicies() {
		super.createEditPolicies();
		installEditPolicy(EditPolicy.LAYOUT_ROLE, new ShapesXYLayoutEditPolicy());
		installEditPolicy(EditPolicy.CONTAINER_ROLE, new ShapesContainerEditPolicy());
		installEditPolicy(EditPolicy.SELECTION_FEEDBACK_ROLE, new ContainerHighlightEditPolicy());
	}

	
	@Override
	public void performRequest(Request req) {
		if (req.getType() == RequestConstants.REQ_DIRECT_EDIT){
			new ShapeDirectEditManager(this,new TitleCellEditorLocator(getFigure())).show();
		}
	}


	class ContainerHighlightEditPolicy extends GraphicalEditPolicy {
		private Color revertColor;

		public void eraseTargetFeedback(Request request) {
			if (revertColor != null) {
				setContainerBackground(revertColor);
				revertColor = null;
			}
		}

		private Color getContainerBackground() {
			return getContainerFigure().getBackgroundColor();
		}

		private IFigure getContainerFigure() {
			return ((GraphicalEditPart) getHost()).getFigure();
		}

		public EditPart getTargetEditPart(Request request) {
			return request.getType().equals(RequestConstants.REQ_SELECTION_HOVER) ? getHost() : null;
		}

		private void setContainerBackground(Color c) {
			getContainerFigure().setBackgroundColor(c);
		}

		protected void showHighlight() {
			if (revertColor == null) {
				revertColor = getContainerBackground();
				setContainerBackground(SelectedBlue);
			}
		}

		public void showTargetFeedback(Request request) {
			if (request.getType().equals(RequestConstants.REQ_MOVE)
					|| request.getType().equals(RequestConstants.REQ_ADD)
					|| request.getType().equals(RequestConstants.REQ_CLONE)
					|| request.getType().equals(RequestConstants.REQ_CONNECTION_START)
					|| request.getType().equals(RequestConstants.REQ_CONNECTION_END)
					|| request.getType().equals(RequestConstants.REQ_CREATE))
				showHighlight();
		}

	}
	
}