LinksLFNodeFigure.java example

Explorer
gmf-tooling-master
package org.eclipse.gmf.tooling.runtime.linklf;

import org.eclipse.draw2d.AbstractPointListShape;
import org.eclipse.draw2d.ConnectionAnchor;
import org.eclipse.draw2d.IFigure;
import org.eclipse.draw2d.PositionConstants;
import org.eclipse.draw2d.ScalablePolygonShape;
import org.eclipse.draw2d.geometry.Point;
import org.eclipse.draw2d.geometry.PointList;
import org.eclipse.draw2d.geometry.PrecisionPoint;
import org.eclipse.gmf.runtime.diagram.ui.editparts.GraphicalEditPart;
import org.eclipse.gmf.runtime.diagram.ui.editparts.IBorderItemEditPart;
import org.eclipse.gmf.runtime.draw2d.ui.figures.BaseSlidableAnchor;
import org.eclipse.gmf.runtime.draw2d.ui.figures.IBorderItemLocator;
import org.eclipse.gmf.runtime.gef.ui.figures.DefaultSizeNodeFigure;

public class LinksLFNodeFigure extends DefaultSizeNodeFigure {

	private final GraphicalEditPart myHost;

	public LinksLFNodeFigure(GraphicalEditPart hostEP, int width, int height) {
		super(width, height);
		myHost = hostEP;
	}

	@Override
	protected ConnectionAnchor createAnchor(PrecisionPoint p) {
		if (p == null) {
			// If the old terminal for the connection anchor cannot be resolved (by SlidableAnchor) a null
			// PrecisionPoint will passed in - this is handled here
			return createDefaultAnchor();
		}
		SlidableSnapToGridAnchor result = new SlidableSnapToGridAnchor(this, p);
		result.setEditPartViewer(myHost.getViewer());
		return result;
	}

	@Override
	public ConnectionAnchor getSourceConnectionAnchorAt(Point p) {
		return super.getSourceConnectionAnchorAt(p);
	}

	@Override
	public ConnectionAnchor getTargetConnectionAnchorAt(Point p) {
		return super.getTargetConnectionAnchorAt(p);
	}

	@Override
	protected double getSlidableAnchorArea() {
		return 0.9;
	}

	protected ConnectionAnchor createConnectionAnchor(Point p) {
		if (p == null) {
			return getConnectionAnchor(szAnchor);
		} else {
			Point temp = p.getCopy();
			translateToRelative(temp);
			PrecisionPoint pt = BaseSlidableAnchor.getAnchorRelativeLocation(temp, getBounds());
			if (isDefaultAnchorArea(pt)) {
				return getConnectionAnchor(szAnchor);
			}

			if (myHost instanceof IBorderItemEditPart) {
				IBorderItemLocator locator = ((IBorderItemEditPart) myHost).getBorderItemLocator();
				switch (locator.getCurrentSideOfParent()) {
				case PositionConstants.WEST:
				case PositionConstants.EAST:
					pt.setPreciseX(pt.preciseX() > 0.5 ? 1.0 : 0.0);
					break;
				case PositionConstants.SOUTH:
				case PositionConstants.NORTH:
					pt.setPreciseY(pt.preciseY() > 0.5 ? 1.0 : 0.0);
					break;
				}
			}

			return createAnchor(pt);
		}
	}

	@Override
	public PointList getPolygonPoints() {
		if (getChildren().size() == 1) {
			IFigure primaryShape = (IFigure) getChildren().get(0);
			if (primaryShape instanceof ScalablePolygonShape) {
				ScalablePolygonShape scalable = (ScalablePolygonShape) primaryShape;
				PointList result = scalable.getScaledPoints().getCopy();
				result.translate(scalable.getLocation());
				return result;
			}
			if (primaryShape instanceof AbstractPointListShape) {
				return ((AbstractPointListShape) primaryShape).getPoints().getCopy();
			}
		}
		return super.getPolygonPoints();
	}

}