Application: gvSIG desktop » gvSIG projections

package org.gvsig.proj.swing.impl.tree;

import java.util.ArrayList;

import java.util.Collections;

import java.util.Enumeration;

import javax.swing.Icon;

import javax.swing.tree.TreeNode;

import org.gvsig.tools.swing.api.ToolsSwingLocator;

import org.gvsig.tools.swing.icontheme.IconTheme;

public class BranchNode implements TreeNode {

        final protected TreeNode parent;

        protected ArrayList<TreeNode> children = new ArrayList<TreeNode>();

        final protected String message;

        protected Icon expandedIcon;

        protected Icon collapsedIcon;

        final protected TreeContainer container;

        final protected IconTheme iconTheme = ToolsSwingLocator.getIconThemeManager().getCurrent();

        public BranchNode(TreeNode parent,

                        TreeContainer container,

                        String message

                        ) {

                this(parent, container, message, null, null);

        }

        public BranchNode(TreeNode parent,

                        TreeContainer container,

                        String message,

                        Icon icon

                        ) {

                this(parent, container, message, icon, icon);

        }

        public BranchNode(TreeNode parent,

                        TreeContainer container,

                        String message,

                        Icon expandedIcon,

                        Icon collapsedIcon

                        ) {

                this.container = container;

                this.parent = parent;

                this.message = message;

                this.expandedIcon = expandedIcon;

                this.collapsedIcon = collapsedIcon;

        }

        public void add(TreeNode child) {

                children.add(child);

        }

        public void clear() {

                this.children.clear();

        }

        /**

         * Requests children to be loaded if they have not been already

         * loaded. Subclasses may load children in an asynchronous way

         */

        public void loadChildren() {

                System.out.println("load");

        }

        /**

         * Requests children to be loaded (forcing the load even if they

         * were already loaded). Subclasses may load children in an asynchronous way

         */

        public void reloadChildren() {

                for (TreeNode node: children) {

                        if (node instanceof BranchNode) {

                                ((BranchNode)node).reloadChildren();

                        }

                }

        }

        @Override

        public TreeNode getParent() {

                return parent;

        }

        @Override

        public boolean isLeaf() {

                return false;

        }

        @Override

        public boolean getAllowsChildren() {

                return true;

        }

        @Override

        public TreeNode getChildAt(int childIndex) {

                return children.get(childIndex);

        }

        @Override

        public int getChildCount() {

                return children.size();

        }

        @Override

        public int getIndex(TreeNode node) {

        if (!isNodeChild(node)) {

            return -1;

        }

        return children.indexOf(node);

        }

    /**

     * Returns true if <code>aNode</code> is a child of this node.  If

     * <code>aNode</code> is null, this method returns false.

     *

     * @return  true if <code>aNode</code> is a child of this node; false if

     *                  <code>aNode</code> is null

     */

    protected boolean isNodeChild(TreeNode aNode) {

        if (aNode == null) {

            return false;

        }

        if (!(aNode instanceof CrsTreeNode)) {

                return false;

        }

        if (getChildCount() == 0) {

            return false;

        } else {

            return (aNode.getParent() == this);

        }

    }

        @SuppressWarnings("rawtypes")

        @Override

        public Enumeration children() {

                return Collections.enumeration(children);

        }

        public Icon getIcon(boolean expanded) {

                if (expanded) {

                        return expandedIcon;

                }

                return collapsedIcon;

        }

        @Override

        public String toString() {

                return this.message;

        }

}