TreeRenderer.java

/*
 * TreeRenderer.java
  *
 * This file is part of TDA - Thread Dump Analysis Tool.
 *
 * TDA is free software; you can redistribute it and/or modify
 * it under the terms of the Lesser GNU General Public License as published by
 * the Free Software Foundation; either version 2.1 of the License, or
 * (at your option) any later version.
 *
 * TDA 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
 * Lesser GNU General Public License for more details.
 *
 * You should have received a copy of the Lesser GNU General Public License
 * along with TDA; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 *
 * $Id: TreeRenderer.java,v 1.10 2008-03-13 21:16:08 irockel Exp $
 */

package com.pironet.tda.utils;

import com.pironet.tda.Category;
import com.pironet.tda.HistogramInfo;
import com.pironet.tda.LogFileContent;
import com.pironet.tda.Logfile;
import com.pironet.tda.TDA;
import com.pironet.tda.ThreadInfo;
import java.awt.Color;
import java.awt.Component;
import javax.swing.Icon;
import javax.swing.JTree;
import javax.swing.tree.DefaultMutableTreeNode;
import javax.swing.tree.DefaultTreeCellRenderer;

/**
 * adds icons to tda root tree
 * @author irockel
 */
public class TreeRenderer extends DefaultTreeCellRenderer {

    public TreeRenderer() {
       // empty constructor
    }

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

        super.getTreeCellRendererComponent(tree, value, sel, expanded, leaf, row, hasFocus);
        if (leaf && isCategory(value)) {
            setIcon(getIconFromCategory(value));
        } else if (leaf && isThreadInfo(value)) {
            setIcon(TDA.createImageIcon("Thread.gif"));
        } else if(leaf && isHistogramInfo(value)) {
            setIcon(TDA.createImageIcon("Histogram.gif"));
        } else if (leaf && isLogfile(value)) {
            setIcon(TDA.createImageIcon("Root.gif"));
        } else if (leaf && isLogFileContent(value)) {
            setIcon(TDA.createImageIcon("LogfileContent.gif"));
        } else if(!leaf) {
            if(((DefaultMutableTreeNode) value).isRoot() || isLogfile(value)) {
                setIcon(TDA.createImageIcon("Root.gif"));
            } else if(isThreadInfo(value)) {
                if(((ThreadInfo) ((DefaultMutableTreeNode) value).getUserObject()).areALotOfWaiting()) {
                    setIcon(TDA.createImageIcon("MonitorRed.gif"));
                } else {
                    setIcon(TDA.createImageIcon("Monitor.gif"));
                }
            } else {
                setIcon(TDA.createImageIcon("ThreadDump.gif"));
            }
        }
        this.setBackgroundNonSelectionColor(new Color(0,0,0,0));
        
        return this;
    }

    protected boolean isCategory(Object value) {
        DefaultMutableTreeNode node = (DefaultMutableTreeNode) value;
        return(node.getUserObject() instanceof Category);
    }
        
    protected Icon getIconFromCategory(Object value) {
        DefaultMutableTreeNode node = (DefaultMutableTreeNode) value;
        Category nodeInfo = (Category) node.getUserObject();
        
        return(nodeInfo.getIcon());
    }    

    private boolean isHistogramInfo(Object value) {
        DefaultMutableTreeNode node = (DefaultMutableTreeNode) value;
        return(node.getUserObject() instanceof HistogramInfo);
    }
    
    private boolean isThreadInfo(Object value) {
        DefaultMutableTreeNode node = (DefaultMutableTreeNode) value;
        return((node.getUserObject() instanceof ThreadInfo));
    }
    
    private boolean isLogfile(Object value) {
        DefaultMutableTreeNode node = (DefaultMutableTreeNode) value;
        return(node.getUserObject() instanceof Logfile);
    }

    private boolean isLogFileContent(Object value) {
        DefaultMutableTreeNode node = (DefaultMutableTreeNode) value;
        return(node.getUserObject() instanceof LogFileContent);
    }
}