ConnectionPointReferenceFigure.java

/*******************************************************************************
 * Copyright (c) 2006 CEA List.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *     CEA List - initial API and implementation
 *******************************************************************************/
package org.eclipse.papyrus.uml.diagram.statemachine.custom.figures;

import java.util.Iterator;

import org.eclipse.draw2d.ColorConstants;
import org.eclipse.draw2d.Graphics;
import org.eclipse.draw2d.IFigure;
import org.eclipse.draw2d.geometry.Dimension;
import org.eclipse.draw2d.geometry.Rectangle;
import org.eclipse.papyrus.uml.diagram.common.figure.node.AffixedNamedElementFigure;

/**
 * ConnectionPointReferenceFigure.
 * 
 */
public class ConnectionPointReferenceFigure extends AffixedNamedElementFigure {

	/**
	 * kind
	 */
	protected int kind = 0;


	/**
	 * Default Constructor.
	 */
	public ConnectionPointReferenceFigure() {
		super();
		Dimension dim1 = new Dimension(22, 22);
		this.setPreferredSize(new Dimension(dim1));
		this.setMaximumSize(new Dimension(dim1));
		this.setMinimumSize(new Dimension(dim1));
		this.setSize(dim1);

	}

	@Override
	public void setName(String name) {
		// TODO Auto-generated method stub
		super.setName(name);
	}

	@Override
	public void paint(Graphics graphics) {
		graphics.pushState();

		IFigure ancestor = getParent();
		while((ancestor != null) && !(ancestor instanceof RegionFigure))
			ancestor = ancestor.getParent();
		if(ancestor != null) {
			if(((RegionFigure)ancestor).getBorderColor() != null)
				graphics.setForegroundColor(((RegionFigure)ancestor).getBorderColor());
			if(((RegionFigure)ancestor).getBackgroundColor() != null)
				graphics.setBackgroundColor(((RegionFigure)ancestor).getBackgroundColor());
		} else {
			ancestor = getParent();
			Iterator it = ancestor.getChildren().iterator();
			boolean found = false;
			while(it.hasNext() && !found) {
				ancestor = (IFigure)it.next();
				if(ancestor instanceof StateMachineFigure)
					found = true;
			}
			if(found) {
				if(((StateMachineFigure)ancestor).getBorderColor() != null)
					graphics.setForegroundColor(((StateMachineFigure)ancestor).getBorderColor());
				if(((StateMachineFigure)ancestor).getBackgroundColor() != null)
					graphics.setBackgroundColor(((StateMachineFigure)ancestor).getBackgroundColor());
			}
		}
		graphics.setLineWidth(1);
		Rectangle r = Rectangle.SINGLETON;
		r.setBounds(getBounds());
		r.width -= 2;
		r.height -= 2;
		graphics.drawOval(r);
		r.shrink(1, 1);
		graphics.fillOval(r);

		if(kind == 0) {

			r.x += r.width / 4;
			r.y += r.height / 7;
			r.width = r.width / 2;
			r.height = r.height / 2;
			graphics.setForegroundColor(ColorConstants.black);
			graphics.setBackgroundColor(ColorConstants.black);
			graphics.setLineWidth(2);
			graphics.drawArc(r, 0, 180);
			r.x += r.width / 2;
			r.y += r.height / 2;
			graphics.drawArc(r, 90, 90);


			r.x -= r.width / 4 - 1;
			r.y += 0.75 * r.height;
			r.width = Math.max((int)r.width / 2, 2);
			r.height = Math.max((int)r.height / 2, 2);
			graphics.fillOval(r);
		} else {
			if(kind == 2) {
				// intersection coordinates.
				int x = (int)(bounds.width / (2 * Math.sqrt(2)));
				int y = (int)(bounds.height / (2 * Math.sqrt(2)));

				// cross.
				graphics.drawLine(bounds.getCenter().translate(x, -y), bounds.getCenter().translate(-x, y));
				graphics.drawLine(bounds.getCenter().translate(-x, -y), bounds.getCenter().translate(x, y));
			}
		}


		graphics.popState();
	}

	public void setKind(int kind) {
		this.kind = kind;
	}
}