/*

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

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

 *

 * This code is free software; you can redistribute it and/or modify it

 * under the terms of the GNU General Public License version 2 only, as

 * published by the Free Software Foundation.  Oracle designates this

 * particular file as subject to the "Classpath" exception as provided

 * by Oracle in the LICENSE file that accompanied this code.

 *

 * This code is distributed in the hope that it will be useful, but WITHOUT

 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

 * version 2 for more details (a copy is included in the LICENSE file that

 * accompanied this code).

 *

 * You should have received a copy of the GNU General Public License version

 * 2 along with this work; if not, write to the Free Software Foundation,

 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

 *

 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

 * or visit www.oracle.com if you need additional information or have any

 * questions.

 */

package sun.swing.table;

import sun.swing.DefaultLookup;

import java.awt.Component;

import java.awt.Color;

import java.awt.FontMetrics;

import java.awt.Graphics;

import java.awt.Insets;

import java.awt.Point;

import java.awt.Rectangle;

import java.io.Serializable;

import javax.swing.*;

import javax.swing.plaf.UIResource;

import javax.swing.border.Border;

import javax.swing.table.*;

public class DefaultTableCellHeaderRenderer extends DefaultTableCellRenderer

        implements UIResource {

    private boolean horizontalTextPositionSet;

    private Icon sortArrow;

    private EmptyIcon emptyIcon = new EmptyIcon();

    public DefaultTableCellHeaderRenderer() {

        setHorizontalAlignment(JLabel.CENTER);

    }

    public void setHorizontalTextPosition(int textPosition) {

        horizontalTextPositionSet = true;

        super.setHorizontalTextPosition(textPosition);

    }

    public Component getTableCellRendererComponent(JTable table, Object value,

            boolean isSelected, boolean hasFocus, int row, int column) {

        Icon sortIcon = null;

        boolean isPaintingForPrint = false;

        if (table != null) {

            JTableHeader header = table.getTableHeader();

            if (header != null) {

                Color fgColor = null;

                Color bgColor = null;

                if (hasFocus) {

                    fgColor = DefaultLookup.getColor(this, ui, "TableHeader.focusCellForeground");

                    bgColor = DefaultLookup.getColor(this, ui, "TableHeader.focusCellBackground");

                }

                if (fgColor == null) {

                    fgColor = header.getForeground();

                }

                if (bgColor == null) {

                    bgColor = header.getBackground();

                }

                setForeground(fgColor);

                setBackground(bgColor);

                setFont(header.getFont());

                isPaintingForPrint = header.isPaintingForPrint();

            }

            if (!isPaintingForPrint && table.getRowSorter() != null) {

                if (!horizontalTextPositionSet) {

                    // There is a row sorter, and the developer hasn't

                    // set a text position, change to leading.

                    setHorizontalTextPosition(JLabel.LEADING);

                }

                SortOrder sortOrder = getColumnSortOrder(table, column);

                if (sortOrder != null) {

                    switch(sortOrder) {

                    case ASCENDING:

                        sortIcon = DefaultLookup.getIcon(

                            this, ui, "Table.ascendingSortIcon");

                        break;

                    case DESCENDING:

                        sortIcon = DefaultLookup.getIcon(

                            this, ui, "Table.descendingSortIcon");

                        break;

                    case UNSORTED:

                        sortIcon = DefaultLookup.getIcon(

                            this, ui, "Table.naturalSortIcon");

                        break;

                    }

                }

            }

        }

        setText(value == null ? "" : value.toString());

        setIcon(sortIcon);

        sortArrow = sortIcon;

        Border border = null;

        if (hasFocus) {

            border = DefaultLookup.getBorder(this, ui, "TableHeader.focusCellBorder");

        }

        if (border == null) {

            border = DefaultLookup.getBorder(this, ui, "TableHeader.cellBorder");

        }

        setBorder(border);

        return this;

    }

    public static SortOrder getColumnSortOrder(JTable table, int column) {

        SortOrder rv = null;

        if (table == null || table.getRowSorter() == null) {

            return rv;

        }

        java.util.List<? extends RowSorter.SortKey> sortKeys =

            table.getRowSorter().getSortKeys();

        if (sortKeys.size() > 0 && sortKeys.get(0).getColumn() ==

            table.convertColumnIndexToModel(column)) {

            rv = sortKeys.get(0).getSortOrder();

        }

        return rv;

    }

    @Override

    public void paintComponent(Graphics g) {

        boolean b = DefaultLookup.getBoolean(this, ui,

                "TableHeader.rightAlignSortArrow", false);

        if (b && sortArrow != null) {

            //emptyIcon is used so that if the text in the header is right

            //aligned, or if the column is too narrow, then the text will

            //be sized appropriately to make room for the icon that is about

            //to be painted manually here.

            emptyIcon.width = sortArrow.getIconWidth();

            emptyIcon.height = sortArrow.getIconHeight();

            setIcon(emptyIcon);

            super.paintComponent(g);

            Point position = computeIconPosition(g);

            sortArrow.paintIcon(this, g, position.x, position.y);

        } else {

            super.paintComponent(g);

        }

    }

    private Point computeIconPosition(Graphics g) {

        FontMetrics fontMetrics = g.getFontMetrics();

        Rectangle viewR = new Rectangle();

        Rectangle textR = new Rectangle();

        Rectangle iconR = new Rectangle();

        Insets i = getInsets();

        viewR.x = i.left;

        viewR.y = i.top;

        viewR.width = getWidth() - (i.left + i.right);

        viewR.height = getHeight() - (i.top + i.bottom);

        SwingUtilities.layoutCompoundLabel(

            this,

            fontMetrics,

            getText(),

            sortArrow,

            getVerticalAlignment(),

            getHorizontalAlignment(),

            getVerticalTextPosition(),

            getHorizontalTextPosition(),

            viewR,

            iconR,

            textR,

            getIconTextGap());

        int x = getWidth() - i.right - sortArrow.getIconWidth();

        int y = iconR.y;

        return new Point(x, y);

    }

    private class EmptyIcon implements Icon, Serializable {

        int width = 0;

        int height = 0;

        public void paintIcon(Component c, Graphics g, int x, int y) {}

        public int getIconWidth() { return width; }

        public int getIconHeight() { return height; }

    }

}