TreeItem old format

nifty-gui · Jan 25, 2014 · 94b9236 · 94b9236
1 parent e3487c1
commit 94b9236
Showing 1 changed file with 15 additions and 3 deletions.
diff --git a/nifty-controls/src/main/java/de/lessvoid/nifty/controls/TreeItem.java b/nifty-controls/src/main/java/de/lessvoid/nifty/controls/TreeItem.java
@@ -59,13 +59,25 @@ public TreeItem(@Nullable final T itemValue) {
     this();
     value = itemValue;
   }
-
+  
   public void addTreeItem(final TreeItem<T> item) {
-    children.add(item);
+      children.add(item);
+      item.setParentItem(this);
   }
 
   public void addTreeItems(@Nonnull final Collection<TreeItem<T>> items) {
-    children.addAll(items);
+      for (TreeItem<T> t : items)
+          addTreeItem(t);
+  }
+
+  public void removeTreeItem(final TreeItem<T> item) {
+      children.remove(item);
+      item.setParentItem(null);
+  }
+
+  public void removeTreeItems(@Nonnull final Collection<TreeItem<T>> items) {
+      for (TreeItem<T> t : items)
+          removeTreeItem(t);
   }
 
   /**