org.eclipse.amp.agf/plugins/org.eclipse.amp.agf.core/src/org/eclipse/amp/agf/gef/Shapes.java

/**
 * <copyright>
 *
 * Copyright (c) 2010 Metascape, LLC.
 * 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:
 *   Metascape - Initial API and Implementation
 *
 * </copyright>
 *
 */

package org.eclipse.amp.agf.gef;

import org.eclipse.core.runtime.Assert;
import org.eclipse.draw2d.Ellipse;
import org.eclipse.draw2d.Figure;
import org.eclipse.draw2d.IFigure;
import org.eclipse.draw2d.RectangleFigure;
import org.eclipse.draw2d.RoundedRectangle;
import org.eclipse.draw2d.Shape;
import org.eclipse.draw2d.geometry.Dimension;
import org.eclipse.draw2d.geometry.Point;

/**
 * 
 * @author mparker
 * 
 */
public class Shapes {

    public static final Dimension SIZE_FULL = new Dimension(AgentEditPart.SCALE, AgentEditPart.SCALE);
    public static final Dimension SIZE_INSET_1 = new Dimension(AgentEditPart.SCALE - 1, AgentEditPart.SCALE - 1);

    public static IFigureProvider EMPTY_PROVIDER = new DefaultFigureProvider() {
        public IFigure getFigure(Object object) {
            return new Figure();
        };
    };

    public static IFigureProvider OVAL_SHAPE_PROVIDER = new DefaultFigureProvider() {
        public IFigure getFigure(Object object) {
            return createOval();
        };
    };

    public static IFigureProvider RECTANGLE_SHAPE_PROVIDER = new DefaultFigureProvider() {
        public IFigure getFigure(Object object) {
            return createRectangle();
        };
    };

    public static IFigureProvider ROUNDED_RECTANGLE_SHAPE_PROVIDER = new DefaultFigureProvider() {
        public IFigure getFigure(Object object) {
            return createRoundedRectangle();
        };
    };

    private static void setupShared(Shape figure) {
        figure.setFill(true);
        figure.setOutline(true);
        figure.setLineWidth(1);
    }

    public static Shape createOval() {
        Shape figure = new Ellipse() {
            protected boolean useLocalCoordinates() {
                return true;
            }
        };
        setupShared(figure);
        figure.setSize(SIZE_INSET_1);
        return figure;
    }

    public static Shape createRoundedRectangle() {
        Shape figure = new RoundedRectangle() {
            protected boolean useLocalCoordinates() {
                return true;
            }
        };
        setupShared(figure);
        figure.setSize(SIZE_FULL);
        return figure;
    }

    public static Shape createRectangle() {
        Shape figure = new RectangleFigure() {
            protected boolean useLocalCoordinates() {
                return true;
            }
        };
        setupShared(figure);
        figure.setSize(SIZE_FULL);
        return figure;
    }

    public static Point UPPER_LEFT = new Point(AgentEditPart.SCALE / 10, AgentEditPart.SCALE / 10);

    public static Point UPPER_RIGHT = new Point(AgentEditPart.SCALE / 10 + AgentEditPart.SCALE / 2,
                                                AgentEditPart.SCALE / 10);

    public static Point LOWER_LEFT = new Point(AgentEditPart.SCALE / 10 ,
                                               AgentEditPart.SCALE / 10+ AgentEditPart.SCALE / 2);

    public static Point LOWER_RIGHT = new Point(AgentEditPart.SCALE / 10 + AgentEditPart.SCALE / 2,
                                                AgentEditPart.SCALE / 10 + AgentEditPart.SCALE / 2);

    public static Point getLocationFor(int i) {
        switch (i) {
            case 0:
                return UPPER_LEFT;
            case 1:
                return UPPER_RIGHT;
            case 2:
                return LOWER_LEFT;
            case 3:
                return LOWER_RIGHT;
        }
        // let's not fail and simply allow a default case instead for overrun.
        return UPPER_LEFT;
    }

    public static Shape getMarkerShape(IFigure parent, int position) {
        Shape markerFigure;
        Assert.isNotNull(parent, "Marker parent cannot be null.");
        for (int i = parent.getChildren().size(); i <= position; i++) {
            markerFigure = new Ellipse();
            markerFigure.setFill(false);
            markerFigure.setOutline(false);
            parent.add(markerFigure);
            // markerFigure.setLocation(getLocationFor(i));
            parent.setConstraint(markerFigure, getLocationFor(i));
            markerFigure.setSize(new Dimension(AgentEditPart.SCALE / 3, AgentEditPart.SCALE / 3));
            setupShared(markerFigure);
        }
        markerFigure = (Shape) parent.getChildren().get(position);
        Assert.isNotNull(markerFigure);
        return markerFigure;
    }

}