diff --git a/src/jnet/client/gui/MapTreeCellRenderer.java b/src/jnet/client/gui/MapTreeCellRenderer.java index be55750..c5045f9 100644 --- a/src/jnet/client/gui/MapTreeCellRenderer.java +++ b/src/jnet/client/gui/MapTreeCellRenderer.java @@ -1,10 +1,8 @@ package jnet.client.gui; -import jnet.client.Client; import java.awt.Component; import java.awt.image.BufferedImage; import java.io.File; -import java.util.Iterator; import javax.imageio.ImageIO; import javax.swing.ImageIcon; import javax.swing.JLabel; @@ -14,18 +12,17 @@ import javax.swing.tree.DefaultMutableTreeNode; import javax.swing.tree.DefaultTreeCellRenderer; import jnet.lib.Status; import jnet.lib.object.Map; -import jnet.lib.object.MapObject; public class MapTreeCellRenderer extends DefaultTreeCellRenderer { - + private JLabel label; private BufferedImage image; - + public MapTreeCellRenderer() { label = new JLabel(); - + } - + @Override public Component getTreeCellRendererComponent(JTree tree, Object value, boolean selected, boolean expanded, boolean leaf, int row, boolean hasFocus) { label = (JLabel) super.getTreeCellRendererComponent(tree, value, selected, expanded, leaf, row, hasFocus); @@ -33,19 +30,22 @@ public class MapTreeCellRenderer extends DefaultTreeCellRenderer { if (o instanceof Map) { try { Map m = (Map) o; + String imagePath = ""; switch (m.getStatus()) { case Status.OK: // online - image = ImageIO.read(new File("img/flag_green.png")); + imagePath = "/img/flag_green.png"; break; case Status.WARNING: // warning - image = ImageIO.read(new File("img/flag_orange.png")); + imagePath = "/img/flag_orange.png"; break; case Status.OFFLINE: // offline - image = ImageIO.read(new File("img/flag_red.png")); + imagePath = "/img/flag_red.png"; break; default: // other - image = ImageIO.read(new File("img/flag_gray.png")); + imagePath = "/img/flag_gray.png"; } + image = ImageIO.read(getClass().getResourceAsStream(imagePath)); + if (image != null) { label.setIcon(new ImageIcon(image)); } @@ -59,5 +59,5 @@ public class MapTreeCellRenderer extends DefaultTreeCellRenderer { label.setBorder(border); return label; } - + }