com.barrybecker4.game.twoplayer.go.ui.gametree.GoTreeCellRenderer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of bb4-go Show documentation
bb4-games java code.
There is a newer version: 1.6
/** Copyright by Barry G. Becker, 2000-2011. Licensed under MIT License: http://www.opensource.org/licenses/MIT  */
package com.barrybecker4.game.twoplayer.go.ui.gametree;

import com.barrybecker4.ui.util.ColorMap;
import com.barrybecker4.common.format.FormatUtil;
import com.barrybecker4.game.twoplayer.common.TwoPlayerMove;
import com.barrybecker4.game.twoplayer.common.search.tree.SearchTreeNode;
import com.barrybecker4.game.twoplayer.common.ui.gametree.GameTreeCellRenderer;
import com.barrybecker4.game.twoplayer.go.ui.rendering.GoStoneRenderer;

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

/**
 *  This class defines how to draw entrees in the textual game tree ui.
 *
 *  @author Barry Becker
 */
public final class GoTreeCellRenderer extends GameTreeCellRenderer {

    private static final Color ROW_BG_COLOR = new Color( 220, 210, 240 );

    private static final int STONE_IMG_SIZE = 11;
    private static final int TEXT_MARGIN = 4;
    private static final Font FONT = new Font("Sans Serif", Font.PLAIN, 10);

    /** the node we are going to render. */
    private SearchTreeNode node_;

    private static final ColorMap COLORMAP = new GoTreeColorMap();

    /**
     *  Default Constructor.
     */
    public GoTreeCellRenderer() {
        setColorMap(COLORMAP);
        this.setPreferredSize(new Dimension(1000, 16));
    }

    @Override
    public Component getTreeCellRendererComponent(
            JTree tree, Object value,
            boolean sel, boolean expanded,
            boolean leaf, int row, boolean hasFocus1 ) {

        super.getTreeCellRendererComponent( tree, value, sel,
                expanded, leaf, row, hasFocus1 );

        node_ = (SearchTreeNode) value;

        Color bg = getBGColor( value );

        this.setBackgroundNonSelectionColor( bg );
        this.setBackgroundSelectionColor( Color.orange );

        return this;
    }

    @Override
    protected Color getBGColor( Object value ) {
        return ROW_BG_COLOR;
    }


    /**
      * Paints the value.  The background is filled based on what was selected.
      */
    @Override
    public void paint(Graphics g) {
        drawCellBackground(g);

        Graphics2D g2 = (Graphics2D) g;
        TwoPlayerMove move = (TwoPlayerMove) node_.getUserObject();
        if (move == null) {
            return;
        }

        drawStoneIcon(move.isPlayer1(), g2);
        drawMoveText(move, g2);
    }

    /**
     * Draw a nice icon to show who's move it corresponds to
     */
    private void drawStoneIcon(boolean isPlayer1, Graphics2D g2) {
        if (isPlayer1)  {
            g2.drawImage(GoStoneRenderer.BLACK_STONE_IMG.getImage(), 1, 0, STONE_IMG_SIZE, STONE_IMG_SIZE, null);
        } else {
            g2.drawImage(GoStoneRenderer.WHITE_STONE_IMG.getImage(), 1, 0, STONE_IMG_SIZE, STONE_IMG_SIZE, null);
        }
    }

    /**
     * Draw the inherited value, the base value, and the move(with its attributes) to the right of the icon.
     */
    private void drawMoveText(TwoPlayerMove move, Graphics2D g2) {

        int inheritedValue = move.getInheritedValue();
        int value = move.getValue();

        g2.setFont(FONT);

        int end = TEXT_MARGIN + STONE_IMG_SIZE;
        end = drawStringValue("inhrtd=", inheritedValue, end, g2);
        end = drawStringValue("val=", value, end, g2);

        StringBuilder bldr = new StringBuilder();
        if (node_.isPruned()) {
            bldr.append(" *PRUNED");
        }  else {
            int numKids = node_.getChildMoves()==null? 0 : node_.getChildMoves().length;
            bldr.append(" kids=").append(numKids);
            bldr.append(node_.toString());
        }

        g2.drawString(bldr.toString(), end, 8);
    }

    private int drawStringValue(String prefix, int value, int offset, Graphics2D g2) {

        Color c = getColorMap().getColorForValue(value);
        String text = prefix + FormatUtil.formatNumber(value);
        g2.setColor(c);
        int width = (int)g2.getFontMetrics().getStringBounds(text, g2).getWidth();
        g2.fillRect(offset, 1, width + 2 * TEXT_MARGIN, 10);
        g2.setColor(this.getForeground());
        g2.drawString(text, offset + TEXT_MARGIN, 9);
        return offset + width + 3 * TEXT_MARGIN;
    }

    private void drawCellBackground(Graphics g) {
        Color bColor;

        if(selected) {
            bColor = getBackgroundSelectionColor();
        } else {
            bColor = getBackgroundNonSelectionColor();
            if(bColor == null)
                bColor = getBackground();
        }

        int imageOffset;
        if (bColor != null) {
            //Icon currentI = getIcon();

            imageOffset = 0;
            g.setColor(bColor);
            // if I don't  add 500, then some of the text does not have a bg.
            if (getComponentOrientation().isLeftToRight()) {
                g.fillRect(imageOffset, 0, getWidth() - imageOffset, getHeight());
            } else {
                g.fillRect(0, 0, getWidth() - imageOffset, getHeight());
            }
        }
    }

}