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("crs-loadingnode-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;

        }

}