/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License (the "License").
 * You may not use this file except in compliance with the License.
 *
 * You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
 * or http://www.opensolaris.org/os/licensing.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL HEADER in each
 * file and include the License file at usr/src/OPENSOLARIS.LICENSE.
 * If applicable, add the following below this CDDL HEADER, with the
 * fields enclosed by brackets "[]" replaced with your own identifying
 * information: Portions Copyright [yyyy] [name of copyright owner]
 *
 * CDDL HEADER END
 */

/*
 * Copyright (c) 2009, 2012, Oracle and/or its affiliates. All rights reserved.
 */

package com.oracle.solaris.vp.util.swing;

import java.awt.*;
import javax.swing.*;

public class ArrowIcon implements Icon {
    //
    // Enums
    //

    public enum Direction {
	UP, DOWN, LEFT, RIGHT
    };

    //
    // Static data
    //

    public static final ArrowIcon UP = new ArrowIcon(Direction.UP);
    public static final ArrowIcon DOWN = new ArrowIcon(Direction.DOWN);
    public static final ArrowIcon LEFT = new ArrowIcon(Direction.LEFT);
    public static final ArrowIcon RIGHT = new ArrowIcon(Direction.RIGHT);

    public static final int _SIZE = 7;

    private static final Color COLOR_DISABLED =
	UIManager.getColor("controlShadow");

    private static final Color COLOR_ENABLED =
	UIManager.getColor("controlDkShadow");

    //
    // Instance data
    //

    private Direction direction;
    private int size;
    private Color color;

    //
    // Constructors
    //

    public ArrowIcon(Direction direction, int size, Color color) {
	this.direction = direction;
	this.size = size;
	this.color = color;
    }

    public ArrowIcon(Direction direction, int size) {
	this(direction, size, null);
    }

    public ArrowIcon(Direction direction) {
	this(direction, _SIZE);
    }

    //
    // Icon methods
    //

    public void paintIcon(Component c, Graphics g, int xTran, int yTran) {
	g.translate(xTran, yTran);

	Color color = getColor();
	g.setColor(color != null ? color :
	    (c.isEnabled() ? COLOR_ENABLED : COLOR_DISABLED));

	int size = getSize();
	for (int x = 0; x < (size + 1) / 2; x++) {
	    drawLine(g, x, x, x, size - x - 1);
	}

	g.translate(-xTran, -yTran);
    }

    public int getIconWidth() {
	switch (getDirection()) {
	    case LEFT:
	    case RIGHT:
	    return (size + 1) / 2;
	}
	return size;
    }

    public int getIconHeight() {
	switch (getDirection()) {
	    case LEFT:
	    case RIGHT:
	    return size;
	}
	return (size + 1) / 2;
    }

    //
    // ArrowIcon methods
    //

    public Color getColor() {
	return color;
    }

    public Direction getDirection() {
	return direction;
    }

    public int getSize() {
	return size;
    }

    //
    // Private methods
    //

    private void drawLine(Graphics g, int x1, int y1, int x2, int y2) {
	int size = (getSize() + 1) / 2;

	// Rotate coordinates as appropriate
	switch (direction) {
	    case LEFT:
		x1 = size - x1 - 1;
		x2 = size - x2 - 1;
	    break;

	    case DOWN:
		int tmp1 = y1;
		int tmp2 = y2;
		y1 = x1;
		y2 = x2;
		x1 = tmp1;
		x2 = tmp2;
	    break;

	    case UP:
		tmp1 = y1;
		tmp2 = y2;
		y1 = size - x1 - 1;
		y2 = size - x2 - 1;
		x1 = tmp1;
		x2 = tmp2;
	    break;
	}

	g.drawLine(x1, y1, x2, y2);
    }

    //
    // Static methods
    //

    /**
     * Unit test.
     */
    public static void main(String args[]) {
	JFrame frame = new JFrame();

	boolean raised[] = {true, false};
	boolean active[] = {true, false};

	int rows = Direction.values().length;
	int columns = 1;
	JPanel p = new JPanel(new FlowLayout(15));
	p.setBorder(BorderFactory.createEmptyBorder(15, 15, 15, 15));

	for (Direction d : Direction.values()) {
	    ArrowIcon icon = new ArrowIcon(d, 41);
	    System.out.printf("%s: %s, %s\n", d, icon.getIconWidth(),
		icon.getIconHeight());
	    JLabel l = new JLabel(icon);
	    l.setDisabledIcon(icon);
	    l.setText(d.toString());
	    l.setEnabled(true);
	    l.setBackground(Color.green);
	    l.setOpaque(true);
	    p.add(l);
	}

	Container c = frame.getContentPane();
	c.setLayout(new BorderLayout());
	c.add(p, BorderLayout.CENTER);
	frame.pack();
	frame.setVisible(true);
    }
}