1 files changed, 103 insertions, 0 deletions
diff --git a/src/com/itmill/toolkit/demo/sampler/features/trees/TreeMultiSelectExample.java b/src/com/itmill/toolkit/demo/sampler/features/trees/TreeMultiSelectExample.java
new file mode 100644
index 0000000000..c012b81d56
--- /dev/null
+++ b/src/com/itmill/toolkit/demo/sampler/features/trees/TreeMultiSelectExample.java
@@ -0,0 +1,103 @@
+package com.itmill.toolkit.demo.sampler.features.trees;
+
+import java.util.Set;
+
+import com.itmill.toolkit.data.Item;
+import com.itmill.toolkit.demo.sampler.ExampleUtil;
+import com.itmill.toolkit.event.Action;
+import com.itmill.toolkit.ui.AbstractSelect;
+import com.itmill.toolkit.ui.Button;
+import com.itmill.toolkit.ui.Tree;
+import com.itmill.toolkit.ui.VerticalLayout;
+import com.itmill.toolkit.ui.Button.ClickEvent;
+
+public class TreeMultiSelectExample extends VerticalLayout implements
+        Button.ClickListener, Action.Handler {
+
+    private static final Action ACTION_ADD = new Action("Add child item");
+    private static final Action ACTION_DELETE = new Action("Delete");
+    private static final Action[] ACTIONS = new Action[] { ACTION_ADD,
+            ACTION_DELETE };
+
+    private Tree t;
+    private Button delete;
+    private int itemId;
+
+    public TreeMultiSelectExample() {
+        setSpacing(true);
+
+        // Create new Tree object using a hierarchical container from
+        // ExampleUtil class
+        t = new Tree("Hardware Inventory", ExampleUtil.getHardwareContainer());
+
+        // Add Actionhandler
+
+        t.addActionHandler(this);
+
+        t.setImmediate(true);
+
+        // Set tree to show the 'name' property as caption for items
+        t.setItemCaptionPropertyId(ExampleUtil.hw_PROPERTY_NAME);
+        t.setItemCaptionMode(AbstractSelect.ITEM_CAPTION_MODE_PROPERTY);
+
+        // Starting itemId # for new items
+        itemId = t.getContainerDataSource().size();
+
+        // Expand whole tree
+        for (int i = 0; i < itemId; i++) {
+            t.expandItemsRecursively(i);
+        }
+
+        // Create the 'delete button'
+        delete = new Button("Delete", this, "buttonClick");
+
+        addComponent(delete);
+        addComponent(t);
+
+        // Set multiselect mode
+        t.setMultiSelect(true);
+    }
+
+    public void buttonClick(ClickEvent event) {
+        // Delete all the selected objects
+        Object[] toDelete = ((Set<Object>) t.getValue()).toArray();
+        for (int i = 0; i < toDelete.length; i++) {
+            handleAction(ACTION_DELETE, t, toDelete[i]);
+        }
+    }
+
+    /*
+     * Returns the set of available actions
+     */
+    public Action[] getActions(Object target, Object sender) {
+        return ACTIONS;
+    }
+
+    /*
+     * Handle actions
+     */
+    public void handleAction(Action action, Object sender, Object target) {
+        if (action == ACTION_ADD) {
+            // Allow children for the target item
+            t.setChildrenAllowed(target, true);
+
+            // Create new item, disallow children, add name, set parent
+            Item i = t.addItem(itemId);
+            t.setChildrenAllowed(itemId, false);
+            String newItemName = "New Item # " + itemId;
+            i.getItemProperty(ExampleUtil.hw_PROPERTY_NAME).setValue(
+                    newItemName);
+            t.setParent(itemId, target);
+            t.expandItem(target);
+            itemId++;
+        } else if (action == ACTION_DELETE) {
+            Object parent = t.getParent(target);
+            t.removeItem(target);
+            // If the deleted object's parent has no more children, set it's
+            // childrenallowed property to false
+            if (parent != null && t.getChildren(parent).size() == 0) {
+                t.setChildrenAllowed(parent, false);
+            }
+        }
+    }
+}