No commit message

No commit message

src/main/java/fr/soleil/comete/swt/Tree.java

@@ -23,9 +23,13 @@ package fr.soleil.comete.swt;
 import java.io.File;
 import java.lang.ref.WeakReference;
 import java.util.ArrayList;
+import java.util.LinkedList;
 import java.util.List;
 
 import org.eclipse.swt.SWT;
+import org.eclipse.swt.events.SelectionEvent;
+import org.eclipse.swt.events.SelectionListener;
+import org.eclipse.swt.graphics.Point;
 import org.eclipse.swt.graphics.Rectangle;
 import org.eclipse.swt.layout.FillLayout;
 import org.eclipse.swt.widgets.Composite;
@@ -33,6 +37,7 @@ import org.eclipse.swt.widgets.Display;
 import org.eclipse.swt.widgets.Shell;
 import org.eclipse.swt.widgets.TreeItem;
 
+import fr.soleil.comete.definition.event.TreeNodeSelectionEvent;
 import fr.soleil.comete.definition.listener.ITreeNodeSelectionListener;
 import fr.soleil.comete.definition.widget.ITree;
 import fr.soleil.comete.definition.widget.util.BasicTreeNode;
@@ -44,10 +49,10 @@ import fr.soleil.comete.swt.util.TreeNodeTool;
  *
  * @author saintin
  */
-public class Tree extends CometeComposite<org.eclipse.swt.widgets.Tree> implements ITree {
+public class Tree extends CometeComposite<org.eclipse.swt.widgets.Tree> implements ITree, SelectionListener {
 
-    protected static final Rectangle SHELL_BOUNDS = new Rectangle(0, 0, 800, 800);
     protected final List<WeakReference<ITreeNodeSelectionListener>> selectionListeners;
+
     protected ITreeNode rootNode;
 
     /**
@@ -56,24 +61,13 @@ public class Tree extends CometeComposite<org.eclipse.swt.widgets.Tree> implemen
     public Tree(Composite parent, int style) {
         super(parent, style);
         selectionListeners = new ArrayList<WeakReference<ITreeNodeSelectionListener>>();
-        setLayout(new FillLayout());
     }
 
     @Override
     protected org.eclipse.swt.widgets.Tree initControl() {
-        return new org.eclipse.swt.widgets.Tree(this, SWT.NONE);
-    }
-
-    @Override
-    public void setHorizontalAlignment(int halign) {
-        // TODO Auto-generated method stub
-
-    }
-
-    @Override
-    public int getHorizontalAlignment() {
-        // TODO Auto-generated method stub
-        return 0;
+        org.eclipse.swt.widgets.Tree tree = new org.eclipse.swt.widgets.Tree(this, SWT.MULTI);
+        tree.addSelectionListener(this);
+        return tree;
     }
 
     @Override
@@ -84,72 +78,56 @@ public class Tree extends CometeComposite<org.eclipse.swt.widgets.Tree> implemen
     @Override
     public void setRootNode(ITreeNode rootNode) {
         if (getControl() != null) {
+            // remove all nodes to keep a single root, as swt allows multiple roots
+            getControl().removeAll();
+
             TreeNodeTool.createRootTreeNode(getControl(), rootNode);
             this.rootNode = rootNode;
+            //TODO fire
         }
     }
 
     @Override
     public ITreeNode getNodeAt(int x, int y) {
-        return null;
+        TreeItem item = getControl().getItem(new Point(x, y));
+        return TreeNodeTool.findTreeNodeForTreeItem(item, rootNode);
     }
 
     @Override
     public void selectNodes(boolean keepPreviousSelection, ITreeNode... nodes) {
-    }
-
-    @Override
-    public void deselectNodes(ITreeNode... nodes) {
 
+        TreeItem rootItem = getControl().getItem(0);
 
+        // find new items to select
+        List<TreeItem> itemsToSelect = new LinkedList<TreeItem>();
+        for (ITreeNode node : nodes) {
+            TreeItem item = TreeNodeTool.findTreeItemFromTreeNode(node, rootItem);
+            if (item != null && !itemsToSelect.contains(item)) {
+                itemsToSelect.add(item);
             }
-
-    @Override
-    public void clearSelection() {
-       TreeItem[] treeItem =  getControl().getSelection();
-
         }
 
-    @Override
-    public void removeAllTreeNodeSelectionListeners() {
-        synchronized (selectionListeners) {
-            selectionListeners.clear();
+        // get previously selected items
+        if (keepPreviousSelection) {
+            TreeItem[] currentSelection = getControl().getSelection();
+            for (TreeItem selectedItem : currentSelection) {
+                itemsToSelect.add(selectedItem);
             }
         }
 
-    public static void main(final String[] args) {
-        final Display display = new Display();
-        final Shell shell = new Shell(display);
-
-        shell.setBounds(SHELL_BOUNDS);
-        shell.setLayout(new FillLayout(SWT.VERTICAL));
-
-        Tree tree = new Tree(shell, SWT.NONE);
-        ITreeNode root = new BasicTreeNode();
-        root.setName("ROOT");
-        File[] roots = File.listRoots();
-        System.out.println("root length " + roots.length);
-        for (int i = 0; i < roots.length; i++) {
-            ITreeNode child = new BasicTreeNode();
-            child.setName(roots[i].toString());
-            child.setData(roots[i]);
-            root.addNodes(child);
+        getControl().setSelection((TreeItem[]) itemsToSelect.toArray());
     }
 
-        tree.setRootNode(root);
-
-        shell.open();
-
-        while (!shell.isDisposed()) {
+    @Override
+    public void deselectNodes(ITreeNode... nodes) {
 
-            if (!display.readAndDispatch()) {
 
-                display.sleep();
-            }
     }
 
-        display.dispose();
-        System.exit(0);
+    @Override
+    public void clearSelection() {
+        getControl().deselectAll();
+        // TODO fireSelectionChanged
     }
 
     @Override
@@ -222,4 +200,115 @@ public class Tree extends CometeComposite<org.eclipse.swt.widgets.Tree> implemen
 
     }
 
+    @Override
+    public void removeAllTreeNodeSelectionListeners() {
+        synchronized (selectionListeners) {
+            selectionListeners.clear();
+        }
+    }
+
+    @Override
+    public void widgetDefaultSelected(SelectionEvent e) {
+        // warnSelectionListeners(new EventObject(this));
+    }
+
+    @Override
+    public void widgetSelected(SelectionEvent e) {
+        // warnSelectionListeners(new EventObject(this));
+    }
+
+    // if (e != null) {
+    // TreePath[] paths = e.getPaths();
+    // if (paths != null) {
+    // List<TreePath> added = new ArrayList<TreePath>();
+    // List<TreePath> removed = new ArrayList<TreePath>();
+    // for (int i = 0; i < paths.length; i++) {
+    // if (e.isAddedPath(i)) {
+    // added.add(paths[i]);
+    // }
+    // else {
+    // removed.add(paths[i]);
+    // }
+    // }
+    // List<ITreeNode> nodes = extractNodes(removed);
+    // if (!nodes.isEmpty()) {
+    // warnSelectionListeners(new TreeNodeSelectionEvent(this, false,
+    // nodes.toArray(new ITreeNode[nodes.size()])));
+    // nodes.clear();
+    // }
+    // nodes = extractNodes(added);
+    // if (!nodes.isEmpty()) {
+    // warnSelectionListeners(new TreeNodeSelectionEvent(this, true,
+    // nodes.toArray(new ITreeNode[nodes.size()])));
+    // nodes.clear();
+    // }
+    // }
+    // }
+
+    protected void warnSelectionListeners(TreeNodeSelectionEvent event) {
+        if (event != null) {
+            synchronized (selectionListeners) {
+                List<WeakReference<ITreeNodeSelectionListener>> toRemove = new ArrayList<WeakReference<ITreeNodeSelectionListener>>();
+                for (WeakReference<ITreeNodeSelectionListener> ref : selectionListeners) {
+                    ITreeNodeSelectionListener selectionListener = ref.get();
+                    if (selectionListener == null) {
+                        toRemove.add(ref);
+                    }
+                    else {
+                        selectionListener.selectionChanged(event);
+                    }
+                }
+                selectionListeners.removeAll(toRemove);
+                toRemove.clear();
+            }
+        }
+    }
+
+    @Override
+    public void setHorizontalAlignment(int halign) {
+        // TODO Auto-generated method stub
+
+    }
+
+    @Override
+    public int getHorizontalAlignment() {
+        // TODO Auto-generated method stub
+        return 0;
+    }
+
+    public static void main(final String[] args) {
+        final Display display = new Display();
+        final Shell shell = new Shell(display);
+
+        shell.setBounds(new Rectangle(0, 0, 800, 800));
+        shell.setLayout(new FillLayout(SWT.VERTICAL));
+
+        Tree tree = new Tree(shell, SWT.NONE);
+        ITreeNode root = new BasicTreeNode();
+        root.setName("ROOT");
+        File[] roots = File.listRoots();
+        System.out.println("root length " + roots.length);
+        for (int i = 0; i < roots.length; i++) {
+            ITreeNode child = new BasicTreeNode();
+            child.setName(roots[i].toString());
+            child.setData(roots[i]);
+            root.addNodes(child);
+        }
+
+        tree.setRootNode(root);
+
+        shell.open();
+
+        while (!shell.isDisposed()) {
+
+            if (!display.readAndDispatch()) {
+
+                display.sleep();
+            }
+        }
+
+        display.dispose();
+        System.exit(0);
+    }
+
 }