package com.vci.client.uif.engine.client.TreeTable; import java.awt.Component; import javax.swing.JTree; import javax.swing.ToolTipManager; import javax.swing.tree.DefaultTreeCellRenderer; import com.vci.client.ui.swing.VCISwingUtil; import com.vci.client.ui.treeTable.TreeTableModel; public class TreeTableCellRenderer extends DefaultTreeCellRenderer { /** * */ private static final long serialVersionUID = -9164244995222072279L; 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); TreeTableModel.TreeTableNode node = null; if(value instanceof TreeTableModel.TreeTableNode){ node = (TreeTableModel.TreeTableNode)value; } boolean isRoot = node.getParent() == null; String image = "cog.png"; if(isRoot){ image = "house.png"; } else if(leaf){ image = "cog.png"; } else { image = "book.png"; } ToolTipManager.sharedInstance().registerComponent(tree); setIcon(VCISwingUtil.createImageIcon(image)); return this; } }