fixes #1085

svn changeset:3668/svn branch:trunk

diff --git a/src/com/itmill/toolkit/terminal/gwt/client/ui/ITree.java b/src/com/itmill/toolkit/terminal/gwt/client/ui/ITree.java
index 4568339c72273c5b4096e5d4430b1871598a7460..64693b5b938caee54bb500826d783816589f7471 100644 (file)
--- a/src/com/itmill/toolkit/terminal/gwt/client/ui/ITree.java
+++ b/src/com/itmill/toolkit/terminal/gwt/client/ui/ITree.java
@@ -179,6 +179,8 @@ public class ITree extends FlowPanel implements Paintable {
 
         private boolean open;
 
+        private Icon icon;
+
         public TreeNode() {
             constructDom();
             sinkEvents(Event.ONCLICK);
@@ -264,6 +266,20 @@ public class ITree extends FlowPanel implements Paintable {
             if (uidl.getBooleanAttribute("selected")) {
                 setSelected(true);
             }
+
+            if (uidl.hasAttribute("icon")) {
+                if (icon == null) {
+                    icon = new Icon(client);
+                    DOM.insertBefore(nodeCaptionDiv, icon.getElement(),
+                            nodeCaptionSpan);
+                }
+                icon.setUri(uidl.getStringAttribute("icon"));
+            } else {
+                if (icon != null) {
+                    DOM.removeChild(nodeCaptionDiv, icon.getElement());
+                    icon = null;
+                }
+            }
         }
 
         private void setState(boolean state, boolean notifyServer) {