Re: turn off the visibility of a JTree NODE

Trufel wrote:


It sounds like you want to implement your own TreeModel, which will
allow you to do all sorts of things that are not easy otherwise.

Start with <http://www.chka.de/swing/tree/TreeModel-dynamic.html> and
then use <http://www.chka.de/swing/tree/AbstractTreeModel.java> as the
basis for your own model.

In essence, what you need to do is alter the internal state of your
TreeModel so that the key methods (see the JavaDoc for TreeModel)
reflect how you want your tree to appear at any one time. When you
change the state, you will need to fire appropriate events so that the
listening JTree will know to update its view of your Model.

The minimal methods that you will need to implement for your own tree are:

     public Object getRoot();
     public Object getChild(Object parent, int index);
     public int getChildCount(Object parent);
     public boolean isLeaf(Object node);
     public int getIndexOfChild(Object parent, Object child);

You will only need to implement

     public void valueForPathChanged(TreePath path, Object newValue);

if you want your users to be able to edit the tree by double clicking on
nodes.

Attached is an example of a dynamic TreeModel, which may be useful. It
implements a tree view of a (number of) web site(s). Note the add(URI)
and remove(URI) methods, which dynamically modify the state of the tree,
and fire the necessary event methods to notify the JTree.

Hope this helps.

Rogan

P.S. You may notice that I am actually using an AbstractTreeModel from
the SpringFramework RichClient. It is essentially the same as the one I
pointed you to above, and it SHOULD be a drop in replacement if you
don't want the whole Spring Framework as a dependency.

Oh, and the method from UrlUtils is just:

     public static URI getParent(URI uri) {
         if (!uri.getScheme().startsWith("http"))
             return null;
         try {
             boolean parent = false;

             String s = uri.toString();
             int q = s.indexOf('?');
             if (q>-1) {
                 s = s.substring(0, q);
                 parent = true;
             }
             int f = s.indexOf(';');
             if (f>-1) {
                 s = s.substring(0, f);
                 parent = true;
             }
             if (parent)
                 return new URI(s);
             int sl = s.lastIndexOf('/');
             // if the url ends in /, cut of the last component
             if (sl == s.length()-1) {
                 sl = s.lastIndexOf('/', s.length()-2);
             }
             s = s.substring(0, sl+1);
             sl = s.lastIndexOf('/');
             // if the last slash is part of the "://", there is no parent
             if (sl==6 || sl==7)
                 return null;
             return new URI(s);
         } catch (URISyntaxException use) {
          use.printStackTrace();
             return null;
         }
     }

===============8x=====8x=======8x=======

/**
  *
  */
package org.owasp.webscarab.util.swing;

import javax.swing.JFrame;
import javax.swing.JScrollPane;
import javax.swing.JTree;
import javax.swing.SwingUtilities;
import javax.swing.tree.TreePath;

import org.owasp.webscarab.util.UrlUtils;
import org.springframework.richclient.tree.AbstractTreeModel;

import java.net.URI;
import java.util.Comparator;
import java.util.HashSet;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.TreeMap;

/**
  * @author rdawes
  *
  */
public class UriTreeModel extends AbstractTreeModel {

    private static final URI[] NO_CHILDREN = new URI[0];

    private static final Comparator<URI> comp = new UriComparator();

    private Map<URI, URI[]> nodes = new TreeMap<URI, URI[]>(comp);

    private Set<URI> implied = new HashSet<URI>();

    public UriTreeModel() {
        super("Root");
    }

    public boolean add(URI uri) {
        return add(uri, false);
    }

    private boolean add(URI uri, boolean implicit) {
        URI parent = UrlUtils.getParent(uri);
        if (implied.contains(uri) && !implicit) {
            implied.remove(uri);
            fireTreeNodeChanged(path(parent), getIndexOfChild(parent, uri), uri);
        }
        if (nodes.containsKey(uri))
            return false;
        if (parent != null)
            add(parent, true);
        int position = insertChild(parent, uri);
        if (implicit)
            implied.add(uri);
        if (position < 0) {
            Object[] path = path(parent);
            fireTreeNodeInserted(path, -position - 1, uri);
        }
        return true;
    }

    private Object[] path(URI node) {
        List<Object> path = new LinkedList<Object>();
        if (node != null) {
            path.add(0, node);
            URI parent = node;
            while ((parent = UrlUtils.getParent(parent)) != null) {
                path.add(0, parent);
            }
        }
        path.add(0, getRoot());
        return path.toArray();
    }

    private int insertChild(URI parent, URI child) {
        URI[] children = getChildren(parent);
        int position = getIndexOfChild(parent, child);
        if (position < 0) {
            URI[] newChildren = new URI[children.length + 1];
            System.arraycopy(children, 0, newChildren, 0, -position - 1);
            System.arraycopy(children, -position - 1, newChildren, -position,
                    children.length + position + 1);
            newChildren[-position - 1] = child;
            nodes.put(parent, newChildren);
            nodes.put(child, NO_CHILDREN);
        }
        return position;
    }

    public boolean remove(URI uri) {
        URI parent = UrlUtils.getParent(uri);
        URI[] children = getChildren(uri);
        if (children.length > 0) {
            implied.add(uri);
            fireTreeNodeChanged(path(parent), getIndexOfChild(parent, uri), uri);
            return false;
        }
        int position = removeChild(parent, uri);
        if (position < 0) {
            return false;
        }
        fireTreeNodeRemoved(path(parent), position, uri);
        if (isImplied(parent) && getChildCount(parent) == 0)
            remove(parent);
        return true;

    }

    private int removeChild(URI parent, URI child) {
        URI[] children = getChildren(parent);
        int position = getIndexOfChild(parent, child);
        if (position > -1) {
            URI[] newChildren;
            if (children.length == 1) {
                newChildren = NO_CHILDREN;
            } else {
                newChildren = new URI[children.length - 1];
                System.arraycopy(children, 0, newChildren, 0, position);
                System.arraycopy(children, position + 1, newChildren, position,
                        children.length - position - 1);
            }
            nodes.put(parent, newChildren);
            nodes.remove(child);
        }
        return position;
    }

    public boolean isImplied(URI uri) {
        return implied.contains(uri);
    }

    private URI[] getChildren(Object parent) {
        URI[] children;
        if (parent == getRoot()) {
            children = nodes.get(null);
        } else {
            children = nodes.get((URI) parent);
        }
        if (children == null)
            children = NO_CHILDREN;
        return children;
    }

    public Object getChild(Object parent, int index) {
        return getChildren(parent)[index];
    }

    public int getChildCount(Object parent) {
        return getChildren(parent).length;
    }

    /*
     * This method implements TreeModel.getIndexOfChild(Object, Object),
with an additional twist.
     * It can also be used to tell callers the position where the child
node would be inserted
     * amongst its siblings. Nodes are sorted according to the
<code>UriComparator<code>
     * (non-Javadoc)
     * @see javax.swing.tree.TreeModel#getIndexOfChild(java.lang.Object,
java.lang.Object)
     */
    public int getIndexOfChild(Object parent, Object child) {
        URI[] children = getChildren(parent);
        for (int i = 0; i < children.length; i++) {
            int result = comp.compare((URI) child, children[i]);
            if (result == 0) {
                return i;
            } else if (result < 0) {
                return -i - 1;
            }
        }
        return -children.length - 1;
    }

    public boolean isLeaf(Object node) {
        if (node == getRoot()) return false;
        if (node.toString().endsWith("/"))
            return false;
        URI[] children = getChildren(node);
        return (children == null || children.length == 0);
    }

    public void valueForPathChanged(TreePath path, Object newValue) {
        // we don't support editing
    }

    private static class UriComparator implements Comparator<URI> {

        public UriComparator() {
        }

        public int compare(URI u1, URI u2) {
            if (u1 == null && u2 == null)
                return 0;
            if (u1 == null)
                return -1;
            if (u2 == null)
                return 1;
            int result = u1.getHost().toLowerCase().compareTo(
                    u2.getHost().toLowerCase());
            if (result != 0)
                return result;
            result = u1.getScheme().compareTo(u2.getScheme());
            if (result != 0)
                return result;
            result = u2.getPort() - u1.getPort();
            if (result != 0)
                return result;
            result = u1.getPath().compareTo(u2.getPath());
            if (result != 0)
                return result;
            String q1 = u1.getQuery() == null ? "" : u1.getQuery();
            String q2 = u2.getQuery() == null ? "" : u2.getQuery();
            result = q1.compareTo(q2);
            return result;
        }

    }

    public static void addUri(final UriTreeModel model, final URI uri)
            throws Exception {
        SwingUtilities.invokeAndWait(new Runnable() {
            public void run() {
                model.add(uri);
            }
        });
    }

    public static void removeUri(final UriTreeModel model, final URI uri)
            throws Exception {
        SwingUtilities.invokeAndWait(new Runnable() {
            public void run() {
                model.remove(uri);
            }
        });
    }

    // If expand is true, expands all nodes in the tree.
    // Otherwise, collapses all nodes in the tree.
    public static void expandAll(JTree tree, boolean expand) {
        Object root = tree.getModel().getRoot();

        // Traverse tree from root
        expandAll(tree, new TreePath(root), expand);
    }
    private static void expandAll(JTree tree, TreePath parent, boolean
expand) {
        // Traverse children
        Object node = parent.getLastPathComponent();
        int count = tree.getModel().getChildCount(node);
        if (count >= 0) {
            for (int i=0; i<count; i++) {
                Object n = tree.getModel().getChild(node, i);
                TreePath path = parent.pathByAddingChild(n);
                expandAll(tree, path, expand);
            }
        }

        // Expansion or collapse must be done bottom-up
        if (expand) {
            tree.expandPath(parent);
        } else {
            tree.collapsePath(parent);
        }
    }

    public static void main(String[] args) throws Exception {
        UriTreeModel model = new UriTreeModel();
        JFrame frame = new JFrame();
        frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
        JTree tree = new JTree(model);
        tree.setShowsRootHandles(true);
        tree.setRootVisible(false);
        frame.getContentPane().add(new JScrollPane(tree));
        frame.setBounds(200, 200, 400, 400);
        frame.setVisible(true);
        URI u1 = new URI("http://efgh/");
        URI u2 = new URI("http://abcd/");
        URI u3 = new URI("http://abbb/");
        URI u4 = new URI("http://bcde/");
        addUri(model, u1);
        Thread.sleep(1000);
        addUri(model, u2);
        addUri(model, u3);
        addUri(model, u4);
        expandAll(tree, true);
        Thread.sleep(1000);
// removeUri(model, new URI("http://abcd/"));
// removeUri(model, new URI("http://abcd/efgh/ijkl/"));
// removeUri(model, new URI("http://abcd/efgh/"));
    }

}