com.barrybecker4.game.twoplayer.common.ui.gametree.GameTreeCellRenderer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of bb4-twoplayer 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.common.ui.gametree;

import com.barrybecker4.ui.util.ColorMap;
import com.barrybecker4.game.twoplayer.common.TwoPlayerMove;
import com.barrybecker4.game.twoplayer.common.search.strategy.SearchStrategy;
import com.barrybecker4.game.twoplayer.common.search.tree.SearchTreeNode;
import com.barrybecker4.game.twoplayer.common.ui.TwoPlayerPieceRenderer;

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

/**
 *  this class defines how to draw entrees in the game tree ui.
 *
 *  @author Barry Becker
 */
public class GameTreeCellRenderer extends DefaultTreeCellRenderer {

    private Color p1Color_ = Color.green;
    private ColorMap colormap_ = null;

    private static final Color SELECTED_COLOR = new Color( 200, 20, 0 );

    private final JLabel pieceLabel_ = new JLabel();
    private final JPanel cellPanel_ = new JPanel();

    protected  GameTreeCellRenderer() {
        commonInit();
    }

    /**
     * Default Constructor
     */
    public GameTreeCellRenderer(TwoPlayerPieceRenderer pieceRenderer) {
        setColorMap(createColormap(pieceRenderer));
        commonInit();
    }

    /**
     * initialize the colormap used to color the game tree rows, nodes, and arcs.
     * We will use this colormap for both the text tree and the graphical
     * tree viewers so they have consistent coloring.
     * @return the colormap
     */
    private static synchronized ColorMap createColormap(TwoPlayerPieceRenderer renderer) {

        final double[] values = {-SearchStrategy.WINNING_VALUE,
                                 -SearchStrategy.WINNING_VALUE/20.0,
                                 0.0,
                                 SearchStrategy.WINNING_VALUE/20.0,
                                 SearchStrategy.WINNING_VALUE};
        final Color[] colors = {renderer.getPlayer2Color().darker(),
                                renderer.getPlayer2Color(),
                                new Color( 160, 160, 160),
                                renderer.getPlayer1Color(),
                                renderer.getPlayer1Color().darker()};
        return new ColorMap( values, colors);
    }

    private void commonInit()  {

        cellPanel_.setBackground( UIManager.getColor( "Tree.textBackground" ) );
        setOpaque( false );
        pieceLabel_.setOpaque( true );
        pieceLabel_.setText(" ");
        pieceLabel_.setBackground( p1Color_ );

        cellPanel_.setLayout( new FlowLayout() );
        cellPanel_.add( pieceLabel_ );
        cellPanel_.add( this );
        cellPanel_.setMinimumSize( new Dimension( 300, 30 ) );
        cellPanel_.setVisible( true );
    }

    public ColorMap getColorMap() {
        return colormap_;
    }

    protected void setColorMap(ColorMap cmap) {
        colormap_ = cmap;
    }

    @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 );

        Color bg = getBGColor( value );
        Color fg = getFGColor( value );

        setForeground( fg );
        //setBackground( Color.WHITE );
        //setBackgroundNonSelectionColor( bg );
        setBackgroundSelectionColor( Color.orange );

        return this;
    }

    private static Color getFGColor( Object value ) {

        SearchTreeNode node = (SearchTreeNode) value;
        TwoPlayerMove m = (TwoPlayerMove) node.getUserObject();
        if ( m == null )
            return Color.gray;

        if ( m.isSelected() )
            return SELECTED_COLOR;

        return Color.black;
    }

    protected Color getBGColor( Object value )  {

        Color c;
        SearchTreeNode node = (SearchTreeNode) value;
        int numChildren = node.getChildCount();
        setText( getText() + " kids=" + numChildren );
        TwoPlayerMove m = (TwoPlayerMove) node.getUserObject();
        if ( m == null ) return Color.gray;    // passing move?

        if (colormap_ != null) {
            c = colormap_.getColorForValue(m.getInheritedValue());
        }
        else {
            int val = (int) (2.0 * Math.sqrt( Math.abs( m.getInheritedValue() ) ));
            if ( m.getInheritedValue() < 0 )
                val = -val;
            int v1 = 255 - Math.min( Math.max( val, 0 ), 255 );
            int v2 = 255 - Math.min( Math.max( -val, 0 ), 255 );
            int v3 = 255;
            c = new Color( v1, v2, v3 );
        }
        return c;
    }

}