Pentagon.java

package org.eclipse.uml2.diagram.sequence.figures;

import org.eclipse.draw2d.Graphics;
import org.eclipse.draw2d.IFigure;
import org.eclipse.draw2d.Label;
import org.eclipse.draw2d.MarginBorder;
import org.eclipse.draw2d.Shape;
import org.eclipse.draw2d.geometry.PointList;
import org.eclipse.draw2d.geometry.Rectangle;
import org.eclipse.swt.graphics.Image;
import org.eclipse.uml2.diagram.sequence.draw2d.layouts.FillingBorderLayout;
import org.eclipse.uml2.diagram.sequence.internal.layout.model.LMVisibleFrameWithPentagon;
import org.eclipse.uml2.diagram.sequence.part.UMLDiagramEditorPlugin;



public class Pentagon extends Shape  implements LMVisibleFrameWithPentagon.PentagonLayouter  {
    public static final int BIT_SIZE = 5;
    private boolean myEnablePentagonLayouter = true;

    public Pentagon () {
    	setLineWidth(1);
    	setFont(UMLDiagramEditorPlugin.getInstance().getDefaultFont());
        setLayoutManager(new FillingBorderLayout());
    }
    
	public void setEnablePentagonLayouter(boolean enablePentagonLayouter) {
		myEnablePentagonLayouter = enablePentagonLayouter;
	}
    
    protected void fillShape(Graphics graphics) {
        graphics.fillPolygon(getPoints(getOutlineBounds(this)));
    }

    protected void outlineShape(Graphics graphics) {
        graphics.drawPolygon(getPoints(getOutlineBounds(this)));
    }

    /**
     * Text label to be shown in the pentagon
     */
    public void setContent(IFigure content) {
        content.setBorder(new MarginBorder(0, BIT_SIZE, 2, BIT_SIZE));
        add(content, FillingBorderLayout.CENTER);
        updateToPreferredSize();
    }
    
    public void updateToPreferredSize(){
    	setSize(getPreferredSize());
    }
    
    protected PointList getPoints(Rectangle rect) {           
        PointList points = new PointList(6);
        
        points.addPoint(rect.x, rect.y);
        points.addPoint(rect.x + rect.width, rect.y);
        if (BIT_SIZE > rect.height || BIT_SIZE > rect.width) {
            points.addPoint(rect.x + rect.width, rect.y + rect.height);
            //assert false;
        } else {
            points.addPoint(rect.x + rect.width, rect.y + rect.height - BIT_SIZE);
            points.addPoint(rect.x + rect.width - BIT_SIZE, rect.y + rect.height);
        }
        points.addPoint(rect.x, rect.y + rect.height);
        points.addPoint(rect.x, rect.y);//XXX: if ShapedPolygon.outlineShape(Graphics graphics) called drawPolygon but not drawPolyline it would be useless point
        
        return points;
    }
    
    
    //LMVisibleFrameWithPentagon.PentagonLayouter methods
    public int getPreferredWidth() {
        return getPreferredSize().width;
    }

    public int getPreferredHeight() {
        return getPreferredSize().height;
    }

    public void setX(int x) {
    	if (myEnablePentagonLayouter){
	        Rectangle bounds = getBounds().getCopy();
	        bounds.x = x;
	        setBounds(bounds);
    	}
    }

    public void setY(int y) {
    	if (myEnablePentagonLayouter){
	        Rectangle bounds = getBounds().getCopy();
	        bounds.y = y;
	        setBounds(bounds);
    	}
    }
    
    public Label createPentagonLabel(){
    	return new Label(){
    		@Override
    		public void setText(String s) {
    			super.setText(s);
    			Pentagon.this.updateToPreferredSize();
    		}
    		
    		@Override
    		public void setIcon(Image image) {
    			super.setIcon(image);
    			Pentagon.this.updateToPreferredSize();
    		}
    	};
    }
    
	/**
	 * Calculates shape outline bounds.
	 * @see org.eclipse.draw2d.RectangleFigure#outlineShape(Graphics)
	 */
	public static Rectangle getOutlineBounds(Shape shape) {
		Rectangle b = new Rectangle(shape.getBounds());
		final int lineWidth = shape.getLineWidth();
		final int lineOffset = lineWidth / 2;
		b.x += lineOffset;
		b.y += lineOffset;
		b.width -= lineWidth;
		b.height -= lineWidth;
		return b;
	}

}