filtering N number of leaf nodes per parent node

[mark goking]

hi. i have a filteredtreemodel that extends DefaultTreeModel

what i want to do is at the end of each parent node, i place a Jbutton as a leaf node that says show/hide.
when this button is clicked, it will only show like the 1st 5 leaf nodes if button labeled, hide. if it is labeled show, it will display all leaf nodes

my problem is that whenever i iterate from the parentnode using either getChildAt() or children() from the DefaultMutableTreeNode and i loop the nodes one by one, it doesn not seem to be in order

if it was displayed in order, this would be easy. however it is not and i cant understand why it's outputted in random and not in sequence as when you added them to the treemodel.

anyone got ideas? thanks

[Miklos Szeles]

Hi Mark,
Please post a Short, Self Contained, Correct (Compilable), Example. Without that I can say only that there must be some problem in your code. The DefaultMutableTreeNode's documentation says the following:

Removes newChild from its parent and makes it a child of this node by adding it to the end of this node's child array.

So it puts the new nodes into the end of the list.
The getChildAt says:

Returns the child at the specified index in this node's child array.

I'm sure this class behaving according to it's documentation, so it will give back the nodes in the right order.

[mark goking]

hi miklos

this is my working code

public class QueueTreeFilter extends DefaultTreeModel {

    public static final int MAX_DISPLAY = 5;

    public QueueTreeFilter(DefaultMutableTreeNode root) {
        super(root);
    }

    @Override
    public Object getChild(Object parent, int index) {
        return getFilteredChildren(parent).get(index);
    }

    @Override
    public int getChildCount(Object parent) {
        return getFilteredChildCount(parent);
    }

    @Override
    public int getIndexOfChild(Object parent, Object child) {
        return getFilteredChildren(parent).indexOf(child);
    }

    private int getFilteredChildCount(Object parent) {
        return getFilteredChildren(parent).size();
    }

    private List<TreeNode> getFilteredChildren(Object parent) {
        List<TreeNode> filteredChildren = new ArrayList<TreeNode>();
        DefaultMutableTreeNode parentNode = (DefaultMutableTreeNode) parent;
        DefaultMutableTreeNode nextNode = null;
        CheckBoxNode cbn = null;
        int ctr = 0;

        for (Enumeration e=parentNode.children(); e.hasMoreElements();) {
            nextNode = (DefaultMutableTreeNode) e.nextElement();
            if (nextNode.children().hasMoreElements()) {
                filteredChildren.add(nextNode);
                ctr = 0;
            } else {
                // if last leaf node is a button
                if (((DefaultMutableTreeNode) nextNode.getParent()).getLastLeaf() == nextNode) filteredChildren.add(nextNode);
                else {
                    cbn = (CheckBoxNode) ((DefaultMutableTreeNode) nextNode.getParent()).getLastLeaf().getUserObject();

                    if (cbn.getType() == CheckBoxNode.BUTTON) {
                        if (cbn.getText().equals("Show")) {
                            if (ctr < MAX_DISPLAY) {
                                filteredChildren.add(nextNode);
                                ctr++;
                            }
                        } else filteredChildren.add(nextNode);
                    } else filteredChildren.add(nextNode);
                }
            }
        }
        return filteredChildren;
    }
    
}

im not sure if my filter tree model is the problem because when i run my app, it works fine either in show all or hide mode. the problem occurs when i click the button to either show or hide, it gives me an array index out of bounds exception. and i am not sure where in my filtertree class the problem lies. please advise. thank you

by the way, my jtree has checkbox leaf nodes in it, and the last leaf node for every parent node is a jbutton where the show/hide functionality happens. coz my leaf nodes may be lots of them so i want to be able to hide so only the 1st 5 leaf nodes of every parent node appears.

[Miklos Szeles]

Please reread SSCCE and post according to that, so we can experiment with your code.

[mark goking]

sorry what does SSCCE mean? i searched for it here but it didnt give any results. thank you

[Miklos Szeles]

Short, Self Contained, Correct (Compilable), Example

[mark goking]

just a quick quest. im wondering why defaultmutabletreenode's children() or getChildrenAt() do not return nodes according to the sequence when they were added.

if it was sequential it would be easy to locate which nodes should appear under which parent.

[Miklos Szeles]

They do return in the sequence they were added. Please try it without your FilteredTree code and you will see that it works fine. I'm almost sure that your code makes it invalid, but I won't spend my time debugging your code, till you won't post a runnable example.

[mark goking]

hi miklos. yes. i will post it. hopefully tomorrow night gmt+8. i did not bring my source with me right now. there are actually 6 files in total because my tree nodes are either jcheckbox and a jbutton as the last leaf node.

is it ok to post them all here? i might just rar them and host it some place else

[Miklos Szeles]

I think it's okay to paste that 6 file between code tags here.

[mark goking]

hi miklos. here are the codes. 7 files in all.

   package test;

import java.util.*;
import javax.swing.*;
import javax.swing.tree.*;

public class MyFrame extends javax.swing.JFrame {

    /** Creates new form MyFrame */
    public MyFrame() {
        initComponents();
        

        CheckBoxNode[] cpOptions = new CheckBoxNode[6];
        cpOptions[0] = new CheckBoxNode("Location 1", true);
        cpOptions[1] = new CheckBoxNode("Location 2", true);
        cpOptions[2] = new CheckBoxNode("Location 3", true);
        cpOptions[3] = new CheckBoxNode("Location 4", true);
        cpOptions[4] = new CheckBoxNode("Location 5", true);
        cpOptions[5] = new CheckBoxNode("Show", CheckBoxNode.BUTTON, false);
        Vector cpVector = new NamedVector("Location", cpOptions);

        // service options

        CheckBoxNode[] svcOptions = new CheckBoxNode[2];
        svcOptions[0] = new CheckBoxNode("Service 1", true);
        svcOptions[1] = new CheckBoxNode("Service 2", true);
        Vector svcVector = new NamedVector("Services", svcOptions);

        // status options
        CheckBoxNode[] statusOptions = new CheckBoxNode[6];
        statusOptions[0] = new CheckBoxNode("Status 1", true);
        statusOptions[1] = new CheckBoxNode("Status 2", true);
        statusOptions[2] = new CheckBoxNode("Status 3", true);
        statusOptions[3] = new CheckBoxNode("Status 4", true);
        statusOptions[4] = new CheckBoxNode("Status 5", true);
        statusOptions[5] = new CheckBoxNode("Show", CheckBoxNode.BUTTON, false);
        Vector statusVector = new NamedVector("Status", statusOptions);

        Object rootNodes[] = { cpVector, svcVector, statusVector };
        Vector rootVector = new NamedVector("Root", rootNodes);

        CheckBoxNodeRenderer renderer = new CheckBoxNodeRenderer();
        jTree1 = new JTree(rootVector);
        QueueTreeFilter qtf = new QueueTreeFilter((DefaultMutableTreeNode) jTree1.getModel().getRoot());
        jTree1.setCellRenderer(renderer);
        jTree1.setModel(qtf);
        jTree1.setCellEditor(new QueueCheckBoxNodeEditor(jTree1));
        jTree1.setEditable(true);
        jTree1.getSelectionModel().setSelectionMode(TreeSelectionModel.SINGLE_TREE_SELECTION);
        jScrollPane1.setViewportView(jTree1);
        expandAll(jTree1, true);
    }

     @SuppressWarnings("unchecked")
    // <editor-fold defaultstate="collapsed" desc="Generated Code">
    private void initComponents() {

        jScrollPane1 = new javax.swing.JScrollPane();
        jTree1 = new javax.swing.JTree();

        setDefaultCloseOperation(javax.swing.WindowConstants.EXIT_ON_CLOSE);

        jScrollPane1.setViewportView(jTree1);

        javax.swing.GroupLayout layout = new javax.swing.GroupLayout(getContentPane());
        getContentPane().setLayout(layout);
        layout.setHorizontalGroup(
            layout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING)
            .addGroup(layout.createSequentialGroup()
                .addContainerGap()
                .addComponent(jScrollPane1, javax.swing.GroupLayout.DEFAULT_SIZE, 378, Short.MAX_VALUE)
                .addContainerGap())
        );
        layout.setVerticalGroup(
            layout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING)
            .addGroup(layout.createSequentialGroup()
                .addContainerGap()
                .addComponent(jScrollPane1, javax.swing.GroupLayout.PREFERRED_SIZE, 352, javax.swing.GroupLayout.PREFERRED_SIZE)
                .addContainerGap(70, Short.MAX_VALUE))
        );

        pack();
    }// </editor-fold>

    /**
    * @param args the command line arguments
    */
    public static void main(String args[]) {
        java.awt.EventQueue.invokeLater(new Runnable() {
            public void run() {
                new MyFrame().setVisible(true);
            }
        });
    }

    // Variables declaration - do not modify
    private javax.swing.JScrollPane jScrollPane1;
    private javax.swing.JTree jTree1;
    // End of variables declaration

    public static void expandAll(JTree tree, boolean expand) {
        TreeNode root = (TreeNode) 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
        TreeNode node = (TreeNode) parent.getLastPathComponent();
        if (node.getChildCount() >= 0) {
            for (Enumeration e=node.children(); e.hasMoreElements();) {
                TreeNode n = (TreeNode) e.nextElement();
                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);
        }
    }

}
 

  package test;

import java.awt.Color;
import java.awt.Component;
import java.awt.Font;

import javax.swing.JButton;
import javax.swing.JCheckBox;
import javax.swing.UIManager;
import javax.swing.JTree;
import javax.swing.tree.DefaultMutableTreeNode;
import javax.swing.tree.DefaultTreeCellRenderer;
import javax.swing.tree.TreeCellRenderer;

public class CheckBoxNodeRenderer implements TreeCellRenderer {
    private Object leafRenderer = new Object();
    private JCheckBox cbRenderer = new JCheckBox();
    private JButton buttRenderer = new JButton();
    private DefaultTreeCellRenderer nonLeafRenderer = new DefaultTreeCellRenderer();
    private Color selectionBorderColor, selectionForeground, selectionBackground,textForeground, textBackground;

    protected Object getLeafRenderer() {
        return leafRenderer;
    }

    public CheckBoxNodeRenderer() {
        Font fontValue;
        fontValue = UIManager.getFont("Tree.font");
        if (fontValue != null) {
            cbRenderer.setFont(fontValue);
        }
        Boolean booleanValue = (Boolean) UIManager.get("Tree.drawsFocusBorderAroundIcon");
        cbRenderer.setFocusPainted((booleanValue != null)
        && (booleanValue.booleanValue()));

        selectionBorderColor = UIManager.getColor("Tree.selectionBorderColor");
        selectionForeground = UIManager.getColor("Tree.selectionForeground");
        selectionBackground = UIManager.getColor("Tree.selectionBackground");
        textForeground = UIManager.getColor("Tree.textForeground");
        textBackground = UIManager.getColor("Tree.textBackground");
    }

    @Override
    public Component getTreeCellRendererComponent(JTree tree, Object value,
        boolean selected, boolean expanded, boolean leaf, int row,
        boolean hasFocus) {

        Component returnValue;

        if (leaf) {
            String component_text = ((CheckBoxNode)((DefaultMutableTreeNode)value).getUserObject()).getText();
            if (component_text.indexOf("Show") > -1 || component_text.indexOf("Hide") > -1) {
                // if component is button node, basing it on text starting with show or hide
                buttRenderer.setText(component_text);
                leafRenderer = buttRenderer;
                returnValue = buttRenderer;
            } else {
                // component is checkbox node
                String stringValue = tree.convertValueToText(value, selected, expanded, leaf, row, false);
                cbRenderer.setText(stringValue);
                cbRenderer.setSelected(false);

                cbRenderer.setEnabled(tree.isEnabled());

                if (selected) {
                    cbRenderer.setForeground(selectionForeground);
                    cbRenderer.setBackground(selectionBackground);
                } else {
                    cbRenderer.setForeground(textForeground);
                    cbRenderer.setBackground(textBackground);
                }

                if ((value != null) && (value instanceof DefaultMutableTreeNode)) {
                    Object userObject = ((DefaultMutableTreeNode) value).getUserObject();
                    if (userObject instanceof CheckBoxNode) {
                        CheckBoxNode node = (CheckBoxNode) userObject;
                        cbRenderer.setText(node.getText());
                        cbRenderer.setSelected(node.isSelected());
                    }
                }
                leafRenderer = cbRenderer;
                returnValue = cbRenderer;
            }
        } else {
            returnValue = nonLeafRenderer.getTreeCellRendererComponent(tree,
            value, selected, expanded, leaf, row, hasFocus);
            nonLeafRenderer.setIcon(null);
        }

        return returnValue;
    }

}
  

    package test;

import java.awt.Component;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.awt.event.ItemEvent;
import java.awt.event.ItemListener;
import javax.swing.JButton;
import javax.swing.JCheckBox;
import javax.swing.JTree;

public class QueueCheckBoxNodeEditor extends CheckBoxNodeEditor {

  
    public QueueCheckBoxNodeEditor(JTree tree) {
        super(tree);
    }

    @Override
    public Component getTreeCellEditorComponent(JTree tree, Object value, boolean selected, boolean expanded, boolean leaf, int row) {

        Component editor = renderer.getTreeCellRendererComponent(tree, value, true, expanded, leaf, row, true);

        if (editor instanceof JCheckBox && ((JCheckBox) editor).getItemListeners().length == 0) {
            // editor always selected / focused
            ItemListener itemListener = new ItemListener() {
                @Override
                public void itemStateChanged(ItemEvent itemEvent) {
                    JCheckBox cb = (JCheckBox) itemEvent.getItem();
                    System.out.println("checkbox presseD");
                    if (stopCellEditing()) fireEditingStopped();
                }
            };
            ((JCheckBox) editor).addItemListener(itemListener);
        } else if (editor instanceof JButton && ((JButton) editor).getActionListeners().length == 0) {
            ActionListener actionListener = new ActionListener() {
                @Override
                public void actionPerformed(ActionEvent ae) {
                    JButton butt = (JButton) ae.getSource();
                    if (butt.getText().indexOf("Show") > -1) butt.setText("Hide");
                    else butt.setText("Show");
System.out.println("pressed");
                    if (stopCellEditing()) fireEditingStopped();
                }
            };
            ((JButton) editor).addActionListener(actionListener);
        }

        return editor;
    }

}

   package test;

import javax.swing.tree.DefaultMutableTreeNode;
import javax.swing.tree.DefaultTreeModel;
import javax.swing.tree.TreeNode;

import java.util.ArrayList;
import java.util.Enumeration;
import java.util.List;


public class QueueTreeFilter extends DefaultTreeModel {

    public static final int MAX_DISPLAY = 3;

    public QueueTreeFilter(DefaultMutableTreeNode root) {
        super(root);
    }

    @Override
    public Object getChild(Object parent, int index) {
        return getFilteredChildren(parent).get(index);
    }

    @Override
    public int getChildCount(Object parent) {
        return getFilteredChildCount(parent);
    }

    @Override
    public int getIndexOfChild(Object parent, Object child) {
        return getFilteredChildren(parent).indexOf(child);
    }

    private int getFilteredChildCount(Object parent) {
        return getFilteredChildren(parent).size();
    }

    private List<TreeNode> getFilteredChildren(Object parent) {
        List<TreeNode> filteredChildren = new ArrayList<TreeNode>();
        DefaultMutableTreeNode parentNode = (DefaultMutableTreeNode)parent;
        int ctr = 0;
        CheckBoxNode cbn = null;
        for (Enumeration e=parentNode.children(); e.hasMoreElements();) {
            DefaultMutableTreeNode nextNode = (DefaultMutableTreeNode) e.nextElement();
            if (nextNode.children().hasMoreElements()) {
                filteredChildren.add(nextNode);
                ctr = 0;
            } else {
                cbn = (CheckBoxNode) nextNode.getUserObject();
                if (cbn.getType() == CheckBoxNode.CHECKBOX) {
                    filteredChildren.add(nextNode);
                    ctr++;
                } else {
                    // if button
                    if (ctr > MAX_DISPLAY && !cbn.isSelected()) {

                        for (int i=filteredChildren.size()-1, ctrfor=0; ctrfor<MAX_DISPLAY; ctrfor++) {
                          //  System.out.println("filteredChildren.size="+filteredChildren.size());
                           // System.out.println("i="+i);
                            filteredChildren.remove(i);
                            i--;
                        }
                        ctr = 0;                        
                    }

                    if (cbn.getType() == CheckBoxNode.BUTTON) filteredChildren.add(nextNode);
                    // retract previous leaf node, and remove them, except 1st 5
                }
            }
        }

        return filteredChildren;
    }

}

 

   package test;

public class CheckBoxNode {
    private String text;
    private boolean selected;
    private int type;
    // added option type BUTTON, for show/hide purposes
    public static final int CHECKBOX = 0, BUTTON = 1;

    public CheckBoxNode(String text, boolean selected) {
        this.text = text;
        this.selected = selected;
        type = CHECKBOX;
    }

    public CheckBoxNode(String text, int type, boolean selected) {
        this.text = text;
        this.type = type;
        this.selected = selected;
    }

    public int getType() { return type; }
    public void setType(int type) { this.type = type; }

    public boolean isSelected() {
        return selected;
    }

    public void setSelected(boolean newValue) {
        selected = newValue;
    }

    public String getText() {
        return text;
    }

    public void setText(String newValue) {
        text = newValue;
    }

   /* @Override
    public String toString() {
        return getClass().getName() + "[" + text + "/" + selected + "]";
    }*/

}
 

 package test;

import java.awt.Component;
import java.awt.event.MouseEvent;
import java.util.EventObject;

import javax.swing.AbstractCellEditor;
import javax.swing.JButton;
import javax.swing.JCheckBox;
import javax.swing.JTree;
import javax.swing.event.ChangeEvent;
import javax.swing.tree.DefaultMutableTreeNode;
import javax.swing.tree.TreeCellEditor;
import javax.swing.tree.TreePath;

public class CheckBoxNodeEditor extends AbstractCellEditor implements TreeCellEditor {

    public CheckBoxNodeRenderer renderer = new CheckBoxNodeRenderer();
    ChangeEvent changeEvent = null;
    JTree tree;

    public CheckBoxNodeEditor(JTree tree) {
        this.tree = tree;
    }

    @Override
    public Object getCellEditorValue() {
        CheckBoxNode checkBoxNode = null;
        if (renderer.getLeafRenderer() instanceof JCheckBox) {
            JCheckBox checkbox = (JCheckBox) renderer.getLeafRenderer();
            checkBoxNode = new CheckBoxNode(checkbox.getText(), checkbox.isSelected());
        } else {
            JButton butt = (JButton) renderer.getLeafRenderer();
            checkBoxNode = new CheckBoxNode(butt.getText(), true);
        }
        return checkBoxNode;
    }

    @Override
    public boolean isCellEditable(EventObject event) {
        boolean returnValue = false;
        if (event instanceof MouseEvent) {
            MouseEvent mouseEvent = (MouseEvent) event;
            TreePath path = tree.getPathForLocation(mouseEvent.getX(), mouseEvent.getY());

            if (path != null) {
                Object node = path.getLastPathComponent();
                if ((node != null) && (node instanceof DefaultMutableTreeNode)) {
                    DefaultMutableTreeNode treeNode = (DefaultMutableTreeNode) node;
                    Object userObject = treeNode.getUserObject();
                    returnValue = ((treeNode.isLeaf()) && (userObject instanceof CheckBoxNode));
                }
            }
        }
        return returnValue;
    }

    @Override
    public Component getTreeCellEditorComponent(JTree tree, Object value, boolean selected, boolean expanded, boolean leaf, int row) {

        Component editor = renderer.getTreeCellRendererComponent(tree, value, true, expanded, leaf, row, true);

        /*
        // comment out. inherit from another class instead so this class can ba reused
        // editor always selected / focused
        ItemListener itemListener = new ItemListener() {
            public void itemStateChanged(ItemEvent itemEvent) {
                JCheckBox cb = (JCheckBox) itemEvent.getItem();
                // get jtable's row sorter's filter to gain access to filter variables
                if (trs != null) {
                    QueueFilter qf = (QueueFilter) trs.getRowFilter();
                    qf.setFilterStatus(cb.getText(), cb.isSelected());
                    trs.setRowFilter(qf);
                }

                if (stopCellEditing()) {
                    fireEditingStopped();
                }
            }
        };

        if (editor instanceof JCheckBox && ((JCheckBox) editor).getItemListeners().length == 0) {
            ((JCheckBox) editor).addItemListener(itemListener);
        }*/

        return editor;
    }

}
   

   package test;

import java.util.Vector;

public class NamedVector extends Vector {
    String name;

    public NamedVector(String name) {
        this.name = name;
    }

    public NamedVector(String name, Object elements[]) {
        this.name = name;
        for (int i = 0, n = elements.length; i < n; i++) {
            add(elements[i]);
        }
    }

    @Override
    public String toString() {
        return name;
    }

}
 

[mark goking]

what i want with my tree is that if i click the show button, it will change its status to HIDE, and will show all leaf nodes

in the CheckBoxNode class, i re-used the variable selected for the button so i could use it to base my show/hide state. but in the QueueCheckBoxNodeEditor where the button click action takes place, i dont think the "editor" object can be converted to class CheckBoxNode so im stuck with how to make this work.

and in the QueueTreeFilter class, if you system.out the nextNode object, it is not outputted in order. my logic was if it encounters a button, it will remove all leaf nodes before it except the 1st three.

[Miklos Szeles]

I think one bad thing in your filtered tree is that you change the model without notifying anybody about it. You change the tree's structure from a tree cell editor. I think that is a bad idea, maybe you should use SwingUtilities.InvokeLater. I'm just guessing, so I hope some JTree guru will give the correct answer to your question. Till then I suggest you to check these examples which shows how can you make a tree which allows hiding/showing nodes.
http://forums.sun.com/thread.jspa?threadID=563514&forumID=57
http://www.kidslovepc.com/javatable/java-jtree-show-hide.shtml

[mark goking]

hey thanks. i am using the 2nd link classes now. and i extended my CheckBoxNode class to inherit InvisibleNode

and in my main class, instead of the queuetreefilter i used the invisiblemodeltree. same sequence of code. but when i reload the tree model after setting the node invis and/or unsetting, the nodes are still there. they dont disappear.

i debugged the code, the visible attribute it set to invis when i i set it. the problem is when the filter is activated, it does not hide it

[mark goking]

hi. i actually made this work now using the invisibleNode link you gave to me. i jut got one tiny little problem,
you see, if Location is the root node, it works with no problems

after I added two sets Location and Location 2 under root, the error pops up. java.lang.NullPointerException. and the stacktrace doesnt give me the line number where the error may be but it's within the java class

at javax.swing.plaf.basic.BasicTreeUI.completeEditing(BasicTreeUI.java:2014)

any idea?

   package test;

import java.awt.Color;
import java.awt.Component;
import java.awt.Font;

import javax.swing.JButton;
import javax.swing.JCheckBox;
import javax.swing.UIManager;
import javax.swing.JTree;
import javax.swing.tree.DefaultMutableTreeNode;
import javax.swing.tree.DefaultTreeCellRenderer;
import javax.swing.tree.TreeCellRenderer;

public class CheckBoxNodeRenderer implements TreeCellRenderer {
    private Object leafRenderer = new Object();
    private JCheckBox cbRenderer = new JCheckBox();
    private JButton buttRenderer = new JButton();
    private DefaultTreeCellRenderer nonLeafRenderer = new DefaultTreeCellRenderer();
    private Color selectionBorderColor, selectionForeground, selectionBackground,textForeground, textBackground;

    protected Object getLeafRenderer() {
        return leafRenderer;
    }

    public CheckBoxNodeRenderer() {
        Font fontValue;
        fontValue = UIManager.getFont("Tree.font");
        if (fontValue != null) {
            cbRenderer.setFont(fontValue);
        }
        Boolean booleanValue = (Boolean) UIManager.get("Tree.drawsFocusBorderAroundIcon");
        cbRenderer.setFocusPainted((booleanValue != null)
        && (booleanValue.booleanValue()));

        selectionBorderColor = UIManager.getColor("Tree.selectionBorderColor");
        selectionForeground = UIManager.getColor("Tree.selectionForeground");
        selectionBackground = UIManager.getColor("Tree.selectionBackground");
        textForeground = UIManager.getColor("Tree.textForeground");
        textBackground = UIManager.getColor("Tree.textBackground");
    }

    @Override
    public Component getTreeCellRendererComponent(JTree tree, Object value,
        boolean selected, boolean expanded, boolean leaf, int row,
        boolean hasFocus) {

        Component returnValue;

        if (leaf) {
            String component_text = ((CheckBoxNode)value).getText();
            if (component_text.indexOf("Show") > -1 || component_text.indexOf("Hide") > -1) {
                // if component is button node, basing it on text starting with show or hide
                buttRenderer.setText(component_text);
                leafRenderer = buttRenderer;
                returnValue = buttRenderer;
            } else {
                // component is checkbox node
                String stringValue = tree.convertValueToText(value, selected, expanded, leaf, row, false);
                cbRenderer.setText(stringValue);
                cbRenderer.setSelected(false);

                cbRenderer.setEnabled(tree.isEnabled());

                if (selected) {
                    cbRenderer.setForeground(selectionForeground);
                    cbRenderer.setBackground(selectionBackground);
                } else {
                    cbRenderer.setForeground(textForeground);
                    cbRenderer.setBackground(textBackground);
                }

                // idont understand why this code has to run first in order
                // for the checkbox nodes to be initialized correctly.
                // first check instanceof defaultmutabletreenode, then do check
                // later if checkboxnode instance
                if ((value != null) && (value instanceof DefaultMutableTreeNode)) {
                    Object userObject = ((DefaultMutableTreeNode) value).getUserObject();
                    if (userObject instanceof CheckBoxNode) {
                        CheckBoxNode node = (CheckBoxNode) userObject;
                        cbRenderer.setText(node.getText());
                        cbRenderer.setSelected(node.isSelected());
                    } else {
                        if (value instanceof CheckBoxNode) {
                            CheckBoxNode node = (CheckBoxNode) value;
                            cbRenderer.setText(node.getText());
                            cbRenderer.setSelected(node.isSelected());
                        }
                    }
                }
                leafRenderer = cbRenderer;
                returnValue = cbRenderer;
            }
        } else {
            returnValue = nonLeafRenderer.getTreeCellRendererComponent(tree,
            value, selected, expanded, leaf, row, hasFocus);
            nonLeafRenderer.setIcon(null);
        }

        return returnValue;
    }

}
 

   package test;

import java.util.*;
import javax.swing.*;
import javax.swing.tree.*;
import test.invis.*;
public class MyFrame extends javax.swing.JFrame {

    /** Creates new form MyFrame */
    public MyFrame() {
        initComponents();
        
        jTree1 = new EasyJTree();
        InvisibleNode inroot = new InvisibleNode("Root");

        InvisibleNode in = new InvisibleNode("Location");
        in.add(new CheckBoxNode("Location 1", true));
        in.add(new CheckBoxNode("Location 2", true));
        in.add(new CheckBoxNode("Location 3", true));
        in.add(new CheckBoxNode("Location 4", true));
        in.add(new CheckBoxNode("Location 5", true));
        in.add(new CheckBoxNode("Location 6", true));
        in.add(new CheckBoxNode("Location 7", true));
        in.add(new CheckBoxNode("Location 8", true));
        in.add(new CheckBoxNode("Location 9", true));
        in.add(new CheckBoxNode("Location 10", true));
        in.add(new CheckBoxNode("Hide", CheckBoxNode.BUTTON));

        InvisibleNode in1 = new InvisibleNode("Location 2");
        in1.add(new CheckBoxNode("Hee-haw 1", true));
        in1.add(new CheckBoxNode("Hee-haw 2", true));
        in1.add(new CheckBoxNode("Hee-haw 3", true));
        in1.add(new CheckBoxNode("Hee-haw 4", true));
        in1.add(new CheckBoxNode("Hee-haw 5", true));
        in1.add(new CheckBoxNode("Hee-haw 6", true));
        in1.add(new CheckBoxNode("Hee-haw 7", true));
        in1.add(new CheckBoxNode("Hee-haw 8", true));
        in1.add(new CheckBoxNode("Hee-haw 9", true));
        in1.add(new CheckBoxNode("Hee-haw 10", true));
        in1.add(new CheckBoxNode("Hide", CheckBoxNode.BUTTON));

        inroot.add(in);
        inroot.add(in1);

        CheckBoxNodeRenderer renderer = new CheckBoxNodeRenderer();
        InvisibleTreeModel itm = new InvisibleTreeModel(inroot);
        jTree1.setModel(itm);
        jTree1.setCellRenderer(renderer);
       // jTree1.setModel(itm);
        jTree1.setCellEditor(new QueueCheckBoxNodeEditor(jTree1));
        jTree1.setEditable(true);
        jTree1.getSelectionModel().setSelectionMode(TreeSelectionModel.SINGLE_TREE_SELECTION);

        jScrollPane1.setViewportView(jTree1);
   
        expandAll(jTree1, true);
    }

     @SuppressWarnings("unchecked")
    // <editor-fold defaultstate="collapsed" desc="Generated Code">                          
    private void initComponents() {

        jButton1 = new javax.swing.JButton();
        jButton2 = new javax.swing.JButton();
        jButton3 = new javax.swing.JButton();
        jButton4 = new javax.swing.JButton();
        jScrollPane1 = new javax.swing.JScrollPane();
        jTree1 = new test.EasyJTree();

        setDefaultCloseOperation(javax.swing.WindowConstants.EXIT_ON_CLOSE);

        jButton1.setText("set invis");
        jButton1.addActionListener(new java.awt.event.ActionListener() {
            public void actionPerformed(java.awt.event.ActionEvent evt) {
                jButton1ActionPerformed(evt);
            }
        });

        jButton2.setText("unset invis");
        jButton2.addActionListener(new java.awt.event.ActionListener() {
            public void actionPerformed(java.awt.event.ActionEvent evt) {
                jButton2ActionPerformed(evt);
            }
        });

        jButton3.setText("hide all");
        jButton3.addActionListener(new java.awt.event.ActionListener() {
            public void actionPerformed(java.awt.event.ActionEvent evt) {
                jButton3ActionPerformed(evt);
            }
        });

        jButton4.setText("show all");
        jButton4.addActionListener(new java.awt.event.ActionListener() {
            public void actionPerformed(java.awt.event.ActionEvent evt) {
                jButton4ActionPerformed(evt);
            }
        });

        jScrollPane1.setViewportView(jTree1);

        javax.swing.GroupLayout layout = new javax.swing.GroupLayout(getContentPane());
        getContentPane().setLayout(layout);
        layout.setHorizontalGroup(
            layout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING)
            .addGroup(layout.createSequentialGroup()
                .addGap(53, 53, 53)
                .addComponent(jButton3)
                .addContainerGap(247, Short.MAX_VALUE))
            .addGroup(layout.createSequentialGroup()
                .addContainerGap()
                .addGroup(layout.createParallelGroup(javax.swing.GroupLayout.Alignment.TRAILING)
                    .addComponent(jButton4)
                    .addGroup(layout.createSequentialGroup()
                        .addComponent(jButton1)
                        .addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.RELATED, 114, Short.MAX_VALUE)
                        .addComponent(jButton2)))
                .addGap(60, 60, 60))
            .addGroup(layout.createSequentialGroup()
                .addContainerGap()
                .addComponent(jScrollPane1, javax.swing.GroupLayout.DEFAULT_SIZE, 378, Short.MAX_VALUE)
                .addContainerGap())
        );
        layout.setVerticalGroup(
            layout.createParallelGroup(javax.swing.GroupLayout.Alignment.LEADING)
            .addGroup(layout.createSequentialGroup()
                .addContainerGap()
                .addComponent(jScrollPane1, javax.swing.GroupLayout.PREFERRED_SIZE, 280, javax.swing.GroupLayout.PREFERRED_SIZE)
                .addGap(122, 122, 122)
                .addGroup(layout.createParallelGroup(javax.swing.GroupLayout.Alignment.BASELINE)
                    .addComponent(jButton1)
                    .addComponent(jButton2))
                .addPreferredGap(javax.swing.LayoutStyle.ComponentPlacement.RELATED)
                .addGroup(layout.createParallelGroup(javax.swing.GroupLayout.Alignment.BASELINE)
                    .addComponent(jButton3)
                    .addComponent(jButton4))
                .addContainerGap(javax.swing.GroupLayout.DEFAULT_SIZE, Short.MAX_VALUE))
        );

        pack();
    }// </editor-fold>                        

     private void jButton1ActionPerformed(java.awt.event.ActionEvent evt) {                                         
         // TODO add your handling code here:
         CheckBoxNode in = (CheckBoxNode) ((DefaultMutableTreeNode) jTree1.getSelectionPath().getLastPathComponent()).getUserObject();
         setNodeVisible(jTree1, in, false);
     }                                        

     private void jButton2ActionPerformed(java.awt.event.ActionEvent evt) {                                         
         // TODO add your handling code here:
         CheckBoxNode in = (CheckBoxNode) ((DefaultMutableTreeNode) jTree1.getSelectionPath().getLastPathComponent()).getUserObject();
         //in.setVisible(true);

         setNodeVisible(jTree1, in, true);
     }                                        

     private void jButton4ActionPerformed(java.awt.event.ActionEvent evt) {                                         
         // TODO add your handling code here:
         // show all
         InvisibleTreeModel itm = (InvisibleTreeModel) jTree1.getModel();
         if (itm.isActivatedFilter()) {
            itm.activateFilter(false);
            itm.reload();
        }
       //  expandAll(jTree1, true);
     }                                        

     private void jButton3ActionPerformed(java.awt.event.ActionEvent evt) {                                         
         // TODO add your handling code here:
         InvisibleTreeModel itm = (InvisibleTreeModel) jTree1.getModel();
         if (! itm.isActivatedFilter()) {
            itm.activateFilter(true);
            itm.reload();
        }
       //  expandAll(jTree1, true);
     }                                        


    public static void setNodeVisible(JTree tree, CheckBoxNode node, boolean isVisible) {
        System.out.println(node.getText());
        DefaultTreeModel model = (DefaultTreeModel)tree.getModel();
        TreePath[] path = tree.getSelectionPaths();
        node.setVisible(isVisible);
        model.nodeChanged(node);
        System.out.println(node.isVisible());
    }

    /**
    * @param args the command line arguments
    */
    public static void main(String args[]) {
        java.awt.EventQueue.invokeLater(new Runnable() {
            public void run() {
                new MyFrame().setVisible(true);
            }
        });
    }

    // Variables declaration - do not modify                     
    private javax.swing.JButton jButton1;
    private javax.swing.JButton jButton2;
    private javax.swing.JButton jButton3;
    private javax.swing.JButton jButton4;
    private javax.swing.JScrollPane jScrollPane1;
    private test.EasyJTree jTree1;
    // End of variables declaration                   

    public static void expandAll(JTree tree, boolean expand) {
        TreeNode root = (TreeNode) 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
        TreeNode node = (TreeNode) parent.getLastPathComponent();
        if (node.getChildCount() >= 0) {
            for (Enumeration e=node.children(); e.hasMoreElements();) {
                TreeNode n = (TreeNode) e.nextElement();
                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);
        }
    }

}
 

 package test;

import java.awt.Component;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.awt.event.ItemEvent;
import java.awt.event.ItemListener;
import javax.swing.JButton;
import javax.swing.JCheckBox;
import javax.swing.JTree;
import javax.swing.tree.*;
import test.invis.*;

public class QueueCheckBoxNodeEditor extends CheckBoxNodeEditor {

    public QueueCheckBoxNodeEditor(JTree tree) {
        super(tree);
    }

    @Override
    public Component getTreeCellEditorComponent(final JTree tree, Object value, boolean selected, boolean expanded, boolean leaf, int row) {

        Component editor = renderer.getTreeCellRendererComponent(tree, value, true, expanded, leaf, row, true);

        if (editor instanceof JCheckBox && ((JCheckBox) editor).getItemListeners().length == 0) {
            // editor always selected / focused
            ItemListener itemListener = new ItemListener() {
                @Override
                public void itemStateChanged(ItemEvent itemEvent) {
                    JCheckBox cb = (JCheckBox) itemEvent.getItem();
                   /* System.out.println(tree.getSelectionPath()+", "+cb.isSelected());
                    if (tree.getSelectionPath() != null) {
                    CheckBoxNode cbn = (CheckBoxNode) tree.getSelectionPath().getLastPathComponent();
                    cbn.setSelected(cb.isSelected());
                    }*/
                    if (stopCellEditing()) fireEditingStopped();
                }
            };
            ((JCheckBox) editor).addItemListener(itemListener);
        } else if (editor instanceof JButton && ((JButton) editor).getActionListeners().length == 0) {
            ActionListener actionListener = new ActionListener() {
                @Override
                public void actionPerformed(ActionEvent ae) {
                    JButton butt = (JButton) ae.getSource();

                    CheckBoxNode dmt = (CheckBoxNode) tree.getSelectionPath().getLastPathComponent();
                    InvisibleNode parent = (InvisibleNode) dmt.getParent();

                    if (butt.getText().indexOf("Show") > -1) {
                        //s et all nodes to visible
                        InvisibleNode in = null;
                        InvisibleTreeModel itm = (InvisibleTreeModel) tree.getModel();
                            System.out.println("totalchild="+parent.getChildCount());
                        for (int i=0; i<parent.getChildCount(); i++) {
                            in = (InvisibleNode) parent.getChildAt(i);
                            if (!in.isVisible()) {
                            in.setVisible(true);
                                System.out.println(in+","+in.isVisible());
                               // itm.nodeChanged(in);
                            }
                        }
                        itm.activateFilter(false);
                        itm.reload();
                    } else {
                        if (butt.getText().indexOf("Hide") > -1) {
                            InvisibleNode in = null;
                            InvisibleTreeModel itm = (InvisibleTreeModel) tree.getModel();
                            
                            for (int i=0; i<parent.getChildCount(); i++) {
                                in = (InvisibleNode) parent.getChildAt(i);
                                
                                if (i<=4) in.setVisible(true);
                                else if (i==(parent.getChildCount()-1)) in.setVisible(true);
                                else in.setVisible(false);

                                itm.nodeChanged(in);
System.out.println(in+","+in.isVisible());
                            }

                            itm.activateFilter(true);
                            itm.reload();
                        }
                    }


                    if (butt.getText().indexOf("Show") > -1) {
                        dmt.setText("Hide");
                    }
                    else {
                        dmt.setText("Show");
                    }


                    if (stopCellEditing()) fireEditingStopped();
                }
            };
            ((JButton) editor).addActionListener(actionListener);
        }

        return editor;
    }

}
   

   package test;

import java.awt.Component;
import java.awt.event.MouseEvent;
import java.util.EventObject;

import javax.swing.AbstractCellEditor;
import javax.swing.JButton;
import javax.swing.JCheckBox;
import javax.swing.JTree;
import javax.swing.event.ChangeEvent;
import javax.swing.tree.DefaultMutableTreeNode;
import javax.swing.tree.TreeCellEditor;
import javax.swing.tree.TreePath;

public class CheckBoxNodeEditor extends AbstractCellEditor implements TreeCellEditor {

    public CheckBoxNodeRenderer renderer = new CheckBoxNodeRenderer();
    ChangeEvent changeEvent = null;
    JTree tree;

    public CheckBoxNodeEditor(JTree tree) {
        this.tree = tree;
    }

    @Override
    public Object getCellEditorValue() {
        CheckBoxNode checkBoxNode = null;
        if (renderer.getLeafRenderer() instanceof JCheckBox) {
            JCheckBox checkbox = (JCheckBox) renderer.getLeafRenderer();
            checkBoxNode = new CheckBoxNode(checkbox.getText(), checkbox.isSelected());
        } else {
            
            JButton butt = (JButton) renderer.getLeafRenderer();
            checkBoxNode = new CheckBoxNode(butt.getText(), CheckBoxNode.BUTTON);
        }
        return checkBoxNode;
    }

    @Override
    public boolean isCellEditable(EventObject event) {
        boolean returnValue = false;
        if (event instanceof MouseEvent) {
            MouseEvent mouseEvent = (MouseEvent) event;
            TreePath path = tree.getPathForLocation(mouseEvent.getX(), mouseEvent.getY());

            if (path != null) {
                Object node = path.getLastPathComponent();
                if ((node != null) && (node instanceof CheckBoxNode)) {
                    CheckBoxNode treeNode = (CheckBoxNode) node;
                    //Object userObject = treeNode.getUserObject();
                    //System.out.println("returnvalue="+((treeNode.isLeaf()) && (node instanceof CheckBoxNode)));
                    returnValue = ((treeNode.isLeaf()) && (node instanceof CheckBoxNode));
                }
            }
        }
        return returnValue;
    }

    @Override
    public Component getTreeCellEditorComponent(JTree tree, Object value, boolean selected, boolean expanded, boolean leaf, int row) {

        Component editor = renderer.getTreeCellRendererComponent(tree, value, true, expanded, leaf, row, true);

        /*
        // comment out. inherit from another class instead so this class can ba reused
        // editor always selected / focused
        ItemListener itemListener = new ItemListener() {
            public void itemStateChanged(ItemEvent itemEvent) {
                JCheckBox cb = (JCheckBox) itemEvent.getItem();
                // get jtable's row sorter's filter to gain access to filter variables
                if (trs != null) {
                    QueueFilter qf = (QueueFilter) trs.getRowFilter();
                    qf.setFilterStatus(cb.getText(), cb.isSelected());
                    trs.setRowFilter(qf);
                }

                if (stopCellEditing()) {
                    fireEditingStopped();
                }
            }
        };

        if (editor instanceof JCheckBox && ((JCheckBox) editor).getItemListeners().length == 0) {
            ((JCheckBox) editor).addItemListener(itemListener);
        }*/

        return editor;
    }

}
 

   package test;

import test.invis.*;

public class CheckBoxNode extends InvisibleNode {
    private String text;
    private boolean selected;
    private int type;
    // added option type BUTTON, for show/hide purposes
    public static final int CHECKBOX = 0, BUTTON = 1;

    public CheckBoxNode(String text, boolean selected) {
        this.text = text;
        this.selected = selected;
        type = CHECKBOX;
    }

    public CheckBoxNode(String text, int type) {
        this.text = text;
        this.type = type;
    }

    public CheckBoxNode(String text, int type, boolean selected) {
        this.text = text;
        this.type = type;
        this.selected = selected;
    }

    public int getType() { return type; }
    public void setType(int type) { this.type = type; }

    public boolean isSelected() {
        return selected;
    }

    public void setSelected(boolean newValue) {
        selected = newValue;
    }

    public String getText() {
        return text;
    }

    public void setText(String newValue) {
        text = newValue;
    }

    public String toString() {
        return getText();
    }

   /* @Override
    public String toString() {
        return getClass().getName() + "[" + text + "/" + selected + "]";
    }*/

}
 

   
package test.invis;



import java.util.*;
import javax.swing.tree.*;

public class InvisibleNode extends DefaultMutableTreeNode {

    protected boolean isVisible;

    public InvisibleNode() {
        this(null);
    }

    public InvisibleNode(Object userObject) {
        this(userObject, true, true);
    }

    public InvisibleNode(Object userObject, boolean allowsChildren
    , boolean isVisible) {
        super(userObject, allowsChildren);
        this.isVisible = isVisible;
    }

    public TreeNode getChildAt(int index,boolean filterIsActive) {
        //System.out.println("InvisibleNode: filterActive="+filterIsActive);
        if (! filterIsActive) {
            return super.getChildAt(index);
        }
        if (children == null) {
            throw new ArrayIndexOutOfBoundsException("node has no children");
        }

       // System.out.println("getchild() invisiblenode set to invisible");

        int realIndex = -1;
        int visibleIndex = -1;
        Enumeration menum = children.elements();
        while (menum.hasMoreElements()) {
            InvisibleNode node = (InvisibleNode)menum.nextElement();
            if (node.isVisible()) {
                visibleIndex++;
            } else {
                //System.out.println(node + " set to invisible");
            }
            realIndex++;
            if (visibleIndex == index) {
                return (TreeNode)children.elementAt(realIndex);
            }
        }

        throw new ArrayIndexOutOfBoundsException("index unmatched");

    }

    public int getChildCount(boolean filterIsActive) {
        if (! filterIsActive) {
            return super.getChildCount();
        }
        if (children == null) {
            return 0;
        }

        int count = 0;
        Enumeration menum = children.elements();
        while (menum.hasMoreElements()) {
            InvisibleNode node = (InvisibleNode)menum.nextElement();
            if (node.isVisible()) {
                count++;
            }
        }

        return count;
    }

    public void setVisible(boolean visible) {
        this.isVisible = visible;
    }

    public boolean isVisible() {
        return isVisible;
    }

}

 

   
package test.invis;

import javax.swing.tree.*;

public class InvisibleTreeModel extends DefaultTreeModel {

    protected boolean filterIsActive;

    public InvisibleTreeModel(TreeNode root) {
        this(root, false);
    }

    public InvisibleTreeModel(TreeNode root, boolean asksAllowsChildren) {
        this(root, false, false);
    }

    public InvisibleTreeModel(TreeNode root, boolean asksAllowsChildren
    ,boolean filterIsActive) {
        super(root, asksAllowsChildren);
        this.filterIsActive = filterIsActive;
    }

    public void activateFilter(boolean newValue) {
        filterIsActive = newValue;
    }

    public boolean isActivatedFilter() {
        return filterIsActive;
    }

    public Object getChild(Object parent, int index) {
        if (filterIsActive) {
            //System.out.println("parent = " + parent + " instance of invisnode? " + (parent instanceof InvisibleNode));
            if (parent instanceof InvisibleNode) {
                return ((InvisibleNode)parent).getChildAt(index,filterIsActive);
            }
        }
        return ((TreeNode)parent).getChildAt(index);
    }

    public int getChildCount(Object parent) {
        if (filterIsActive) {
            if (parent instanceof InvisibleNode) {
                return ((InvisibleNode)parent).getChildCount(filterIsActive);
            }
        }
        return ((TreeNode)parent).getChildCount();
    }

}