/*
 * Copyright (c) 2014 tabletoptool.com team.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the GNU Public License v3.0
 * which accompanies this distribution, and is available at
 * http://www.gnu.org/licenses/gpl.html
 * 
 * Contributors:
 *     rptools.com team - initial implementation
 *     tabletoptool.com team - further development
 */
package com.t3.client.ui.tokenpanel;

import java.awt.Color;
import java.awt.Component;
import java.awt.Dimension;
import java.awt.Graphics;
import java.awt.Transparency;
import java.awt.image.BufferedImage;

import javax.swing.Icon;
import javax.swing.ImageIcon;
import javax.swing.JTree;
import javax.swing.SwingUtilities;
import javax.swing.tree.DefaultTreeCellRenderer;

import com.t3.client.TabletopTool;
import com.t3.image.ImageUtil;
import com.t3.model.Token;
import com.t3.swing.SwingUtil;
import com.t3.util.ImageManager;

public class TokenPanelTreeCellRenderer extends DefaultTreeCellRenderer {

    private BufferedImage image;
    private int row;
    private int rowWidth;
    
    
    @Override
	public Component getTreeCellRendererComponent(JTree tree, Object value, boolean sel, boolean expanded, boolean leaf, int row,
            boolean hasFocus) {

        setBorder(null);

        String text = "";
        this.row = row;

        setBackgroundNonSelectionColor(Color.white);
        if (value instanceof Token) {
            Token token = (Token) value;
            
            int height = getPreferredSize().height;
            if (height < 1) {
            	height = 15;
            }
            if (image == null || image.getHeight() != height) {
            	image = new BufferedImage(height, height, Transparency.BITMASK);
            } else {
            	ImageUtil.clearImage(image);
            }
            
            // Make a thumbnail of the image
            // TODO: This could be cached somehow, right now it's quick enough though
            BufferedImage tokenImage = ImageManager.getImage(token.getImageAssetId(), this);
            Dimension dim = new Dimension(tokenImage.getWidth(), tokenImage.getHeight());
            SwingUtil.constrainTo(dim, height);
            
            Graphics g = image.getGraphics();
            // TODO: Center the image
            g.drawImage(tokenImage, 0, 0, dim.width, dim.height, this);
            g.dispose();
            
            text = token.getName();
            if (TabletopTool.getPlayer().isGM() && token.getGMName() != null && token.getGMName().length() > 0) {
            	text += " (" + token.getGMName() + ")";
            }

            if (!token.isVisible()) {
            	setBackgroundNonSelectionColor(Color.lightGray);
            }
        }        
        if (value instanceof TokenPanelTreeModel.View) {
        	TokenPanelTreeModel.View view = (TokenPanelTreeModel.View)value ;
        	
        	text = view.getDisplayName();
        }
        
        super.getTreeCellRendererComponent(tree, text, sel, expanded, leaf, row, hasFocus);

        Icon icon = getIcon();
        rowWidth = (icon != null ? icon.getIconWidth() + 2 : 0 ) + SwingUtilities.computeStringWidth(getFontMetrics(getFont()), text);
        
        return this;
    }
    
    @Override
    public Dimension getPreferredSize() {
    	int height = row > 0 ? getFontMetrics(getFont()).getHeight() + 4 : 0;
    	return new Dimension(super.getPreferredSize().width, height);
    }
    
    @Override
    public Icon getLeafIcon() {
    	return new ImageIcon(image);
    }
}