--- a/flys-client/src/main/java/de/intevation/flys/client/client/ui/map/WMSLayersTree.java	Mon Nov 21 21:00:03 2011 +0000
+++ b/flys-client/src/main/java/de/intevation/flys/client/client/ui/map/WMSLayersTree.java	Tue Nov 22 13:04:46 2011 +0000
@@ -14,6 +14,28 @@
 
 public class WMSLayersTree extends TreeGrid {
 
+    /**
+     * An internal TreeNode that stores besides some string attribute a WMSLayer
+     * object.
+     */
+    public static class WMSLayerNode extends TreeNode {
+
+        protected WMSLayer wms;
+
+        public WMSLayerNode(WMSLayer wms) {
+            super();
+            this.wms = wms;
+
+            setAttribute("name", wms.getName());
+            setAttribute("title", wms.getTitle());
+        }
+
+        public WMSLayer getWMSLayer() {
+            return wms;
+        }
+    } // end of class WMSLayerNode
+
+
     protected Capabilities capabilites;
 
     public WMSLayersTree(Capabilities capabilites) {
@@ -57,11 +79,8 @@
         List<TreeNode> layerNodes = new ArrayList<TreeNode>();
 
         for (WMSLayer layer: layers) {
-            TreeNode tn  = new TreeNode();
-            tn.setAttribute("name", layer.getName());
-            tn.setAttribute("title", layer.getTitle());
-
-            TreeNode[] tns = buildTree(layer.getLayers());
+            WMSLayerNode tn = buildTreeNode(layer);
+            TreeNode[] tns  = buildTree(layer.getLayers());
 
             if (tns != null && tns.length > 0) {
                 tn.setAttribute("children-nodes", tns);
@@ -72,4 +91,9 @@
 
         return (TreeNode[]) layerNodes.toArray(new TreeNode[layerNodes.size()]);
     }
+
+
+    protected static WMSLayerNode buildTreeNode(WMSLayer wms) {
+        return new WMSLayerNode(wms);
+    }
 }