Author: knopp
Date: Wed Apr 23 11:18:47 2008
New Revision: 650977

URL: http://svn.apache.org/viewvc?rev=650977&view=rev
Log:
added getItemClass

Modified:
    
wicket/trunk/wicket/src/main/java/org/apache/wicket/markup/html/tree/BaseTree.java

Modified: 
wicket/trunk/wicket/src/main/java/org/apache/wicket/markup/html/tree/BaseTree.java
URL: 
http://svn.apache.org/viewvc/wicket/trunk/wicket/src/main/java/org/apache/wicket/markup/html/tree/BaseTree.java?rev=650977&r1=650976&r2=650977&view=diff
==============================================================================
--- 
wicket/trunk/wicket/src/main/java/org/apache/wicket/markup/html/tree/BaseTree.java
 (original)
+++ 
wicket/trunk/wicket/src/main/java/org/apache/wicket/markup/html/tree/BaseTree.java
 Wed Apr 23 11:18:47 2008
@@ -118,20 +118,33 @@
                        public void onComponentTag(Component component, 
ComponentTag tag)
                        {
                                Object node = component.getModelObject();
-                               if (getTreeState().isNodeSelected(node))
+                               String klass = getItemClass(node);
+                               if (!Strings.isEmpty(klass))
                                {
                                        CharSequence oldClass = 
tag.getString("class");
                                        if (Strings.isEmpty(oldClass))
                                        {
-                                               tag.put("class", 
getSelectedClass());
+                                               tag.put("class", klass);
                                        }
                                        else
                                        {
-                                               tag.put("class", oldClass + " " 
+ getSelectedClass());
+                                               tag.put("class", oldClass + " " 
+ klass);
                                        }
                                }
                        }
                });
+       }
+
+       protected String getItemClass(Object node)
+       {
+               if (getTreeState().isNodeSelected(node))
+               {
+                       return getSelectedClass();
+               }
+               else
+               {
+                       return null;
+               }
        }
 
        /**


Reply via email to