Application: gvSIG desktop » Library: org.gvsig.tools

package org.gvsig.tools.swing.impl;

import java.util.ArrayList;

import java.util.List;

import javax.swing.AbstractListModel;

import javax.swing.ComboBoxModel;

import javax.swing.JComboBox;

import javax.swing.event.TreeModelEvent;

import javax.swing.event.TreeModelListener;

import javax.swing.tree.DefaultMutableTreeNode;

import javax.swing.tree.TreeModel;

public class TreeComboUtils {

    public static void setTreeModel(JComboBox comboBox, TreeModel aTreeModel) {

        ComboBoxModel model = createComboBoxModel(aTreeModel);

        comboBox.setModel(model);

        comboBox.setRenderer(new TreeComboCellRenderer(model));

    }

    public static ComboBoxModel createComboBoxModel(TreeModel aTreeModel) {

        return new TreeModelToComboBoxModelAdapter(aTreeModel);

    }

    public static class TreeModelToComboBoxModelAdapter extends AbstractListModel implements ComboBoxModel, TreeModelListener {

        private static final long serialVersionUID = 922222366298284635L;

        TreeModel source;

        boolean invalid = true;

        DefaultMutableTreeNode currentValue;

        List<DefaultMutableTreeNode> cache = new ArrayList<>();

        @SuppressWarnings("LeakingThisInConstructor")

        public TreeModelToComboBoxModelAdapter(TreeModel aTreeModel) {

            source = aTreeModel;

            aTreeModel.addTreeModelListener(this);

        }

        @Override

        public void setSelectedItem(Object anObject) {

            if ((currentValue != null && !currentValue.equals(anObject))

                    || currentValue == null && anObject != null) {

                currentValue = this.getNode(anObject);

            }

            fireContentsChanged(this, -1, -1);

        }

        @Override

        public Object getSelectedItem() {

            if (currentValue == null) {

                return null;

            }

            return currentValue.getUserObject();

        }

        @Override

        public int getSize() {

            validate();

            return cache.size();

        }

        @Override

        public Object getElementAt(int index) {

            return cache.get(index).getUserObject();

        }

        public DefaultMutableTreeNode getNode(int index) {

            return cache.get(index);

        }

        public DefaultMutableTreeNode getNode(Object anObject) {

            for (DefaultMutableTreeNode node : cache) {

                if (node.getUserObject().equals(anObject)) {

                    return node;

                }

            }

            return null;

        }

        @Override

        public void treeNodesChanged(TreeModelEvent e) {

            invalid = true;

        }

        @Override

        public void treeNodesInserted(TreeModelEvent e) {

            invalid = true;

        }

        @Override

        public void treeNodesRemoved(TreeModelEvent e) {

            invalid = true;

        }

        @Override

        public void treeStructureChanged(TreeModelEvent e) {

            invalid = true;

        }

        void validate() {

            if (invalid) {

                cache = new ArrayList<>();

                cacheTree(source.getRoot(), 0);

                if (cache.size() > 0) {

                    currentValue = cache.get(0);

                }

                invalid = false;

                fireContentsChanged(this, 0, 0);

            }

        }

        void cacheTree(Object anObject, int level) {

            if (source.isLeaf(anObject)) {

                addNode(anObject, level, false);

            } else {

                int c = source.getChildCount(anObject);

                int i;

                Object child;

                addNode(anObject, level, true);

                level++;

                for (i = 0; i < c; i++) {

                    child = source.getChild(anObject, i);

                    cacheTree(child, level);

                }

                level--;

            }

        }

        void addNode(Object anObject, int level, boolean isNode) {

            cache.add(new ListEntry(anObject, level, isNode));

        }

        private static class ListEntry extends DefaultMutableTreeNode {

            private static final long serialVersionUID = 304579876188493086L;

            int level;

            public ListEntry(Object anObject, int aLevel, boolean isNode) {

                super(anObject, isNode);

                this.setUserObject(anObject);

                level = aLevel;

            }

            @Override

            public int getLevel() {

                return level;

            }

        }

    }

}