Application: gvSIG desktop » gvSIG projections

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

import java.awt.Image;

import java.awt.Rectangle;

import java.awt.image.ImageObserver;

import java.util.Objects;

import javax.swing.ImageIcon;

import javax.swing.JTree;

import javax.swing.tree.TreeNode;

import javax.swing.tree.TreePath;

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

public class LoadingNode extends LeafNode {

        private String message;

        private ImageIcon icon;

        final protected TreePath path;

        final protected JTree tree;

        public LoadingNode(BranchNode parent, TreeContainer container, String message) {

                super(parent);

                // we don't want to reuse the icon for other nodes, since we'll add an observer to the icon 

                this.icon = new ImageIcon(ToolsSwingLocator.getIconThemeManager().getCurrent().get("spinner").getImage());

                this.message = message;

                this.tree = container.getTree();

                TreeNode[] nodeList = container.getModel().getPathToRoot(this);

                if (nodeList!=null) {

                        path = new TreePath(nodeList);

                }

                else {

                        path = null;

                }

                // required for animaged gifs in treenodes

            this.icon.setImageObserver(new ImageObserver() {

                @Override

                public boolean imageUpdate(

                    Image img, int infoflags, int x, int y, int w, int h) {

                        if (path == null || !LoadingNode.this.tree.isShowing()) {

                                return false;        

                        }

                        Rectangle cellRect = LoadingNode.this.tree.getPathBounds(path);

                  if ((infoflags & (FRAMEBITS | ALLBITS)) != 0 && Objects.nonNull(cellRect)) {

                          LoadingNode.this.tree.repaint(cellRect);

                  }

                  return (infoflags & (ALLBITS | ABORT)) == 0;

                }

          });

        }

        public void setMessage(String message) {

                this.message = message;

        }

        @Override

        public String toString() {

                return message;

        }

        @Override

        public ImageIcon getIcon() {

                return icon;

        }

}