Application: gvSIG desktop » gvSIG raster

/* gvSIG. Geographic Information System of the Valencian Government

*

* Copyright (C) 2007-2008 Infrastructures and Transports Department

* of the Valencian Government (CIT)

* 

* This program is free software; you can redistribute it and/or

* modify it under the terms of the GNU General Public License

* as published by the Free Software Foundation; either version 2

* of the License, or (at your option) any later version.

* 

* This program is distributed in the hope that it will be useful,

* but WITHOUT ANY WARRANTY; without even the implied warranty of

* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

* GNU General Public License for more details.

* 

* You should have received a copy of the GNU General Public License

* along with this program; if not, write to the Free Software

* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, 

* MA  02110-1301, USA.

* 

*/

package org.gvsig.raster.wms.app.wmsclient.gui.panel;

import java.awt.Color;

import java.awt.Component;

import java.util.ArrayList;

import java.util.Vector;

import javax.swing.JLabel;

import javax.swing.JPanel;

import javax.swing.JRadioButton;

import javax.swing.JTree;

import javax.swing.event.TreeModelListener;

import javax.swing.tree.DefaultTreeCellRenderer;

import javax.swing.tree.TreeModel;

import javax.swing.tree.TreePath;

import org.gvsig.fmap.dal.coverage.RasterLocator;

import org.gvsig.raster.wms.io.RemoteWMSStyle;

import org.gvsig.raster.wms.io.WMSLayerNode;

/**

 * This class holds a visual tree that handles the selection of the styles in the

 * selected layers. It encapsulates any gui interface handling the mouse selection.

 * <p>

 * It has a method getStylesSelection that returns the current selection in a 

 * bi-dimensional string array containing the layer name and the selected style name.

 * </p>

 * <p>

 * Keep in mind that although it's similar to LayersTree it is not the same.

 * It does not use the same instances of the WMSLayerNode that LayerTreeModel

 * because it allows to repeat layers. So, each layer is a completely new 

 * WMSLayerNode containing same properties but not the same parent (which is always

 * the tree root) and each style is a completely new FMapStyle containing same

 * properties but its parent is one of the layers of the StylesTree and not one

 * of those of the Layers tree.

 * </p>

 * 

 * @author jaume dominguez faus

 *

 */

public class StyleTree extends JTree {

        public boolean showLayerNames = false;

    public StyleTree(){

        super();

        initialize();

    }

    private void initialize(){

        setToggleClickCount(1);

        setRowHeight(22);

        setCellRenderer(new DefaultTreeCellRenderer() {

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

                if (leaf) {

                    JPanel leafComponent = new JPanel();

                    leafComponent.setBackground(Color.WHITE);

                    JRadioButton leafRadioButton = new JRadioButton("", ((StyleTreeModel) getModel()).isSelected((RemoteWMSStyle) value));//selected);

                    leafRadioButton.setBackground(Color.WHITE);

                    leafComponent.add(leafRadioButton);

                    leafComponent.add(new JLabel(((RemoteWMSStyle) value).getTitle()));

                    return leafComponent;

                } else {

                        super.getTreeCellRendererComponent(tree, value, selected, expanded, leaf, row, hasFocus);

                        if (value instanceof WMSLayerNode) {

                                WMSLayerNode layer = (WMSLayerNode) value;

                                if (!showLayerNames) {

                                        if (layer.getName() != null || layer.getName() == "") {

                                                String text = layer.toString();

                                                text = text.substring(text.indexOf(']')+2, text.length());

                                                setText(text);

                                        }

                                }

                        }

                            return this;

                }

            }});

        addMouseListener(new java.awt.event.MouseAdapter() { 

            public void mouseClicked(java.awt.event.MouseEvent e) {

                ((StyleTreeModel)getModel()).setSelectedLeaf(getSelectionPath());

                clearSelection();

                repaint();

            }

        });

    }

    /**

     * Expands this tree.

     *

     */

    public void expandAll(){

        int row = 0; 

        while (row < getRowCount()) {

          expandRow(row);

          row++;

          }

    }

    /**

     * Returns a Vector of Strings containing the style titles.

     */

    public Vector getStyleSelectionTitles(){

        if (getModel() instanceof StyleTreeModel)

            return ((StyleTreeModel)getModel()).getStyleSelectionTitles();

        else

            return null;

    }

    /**

     * Sets the selected styles in the StylesTree. The argument styleNames is

     * a Vector with exactly the same amount of strings than the amount of

     * themes (layers) contained by the tree. A blank or null string will

     * leave the default style for that layer, but this element <b>must exist</b>

     * in the array in any case.

     * 

     * @param styleNames, Vector containing the style names. 

     * The styles order <b>must match</b> with the layer order. 

     */

    public void initSelections(Vector styleNames){

            if (styleNames!=null) {

                    StyleTreeModel model = (StyleTreeModel) getModel();

                    model.setStylesSelection(styleNames);

            }

    }

}

class StyleTreeModel implements TreeModel {

        private WMSLayerNode root;

        private ArrayList<WMSLayerNode> layers = new ArrayList<WMSLayerNode>();

        public StyleTreeModel(WMSLayerNode root){

        this.root = root;

    }

        /**

         * Will return true if this style is selected.

         * @param style

         * @return

         */

        public boolean isSelected(RemoteWMSStyle style) {

                return style.getParent().getSelectedStyle().equals(style);

        }

        /**

         * Gets the names of the selected styles into a Vector using the same order

         * as they was represented in the StylesTree.

         * @return

         */

        public Vector<String> getStyleSelectionTitles() {

                Vector<String> v = new Vector<String>();

                for (int i = 0; i < layers.size(); i++) {

                        WMSLayerNode layer = (WMSLayerNode) layers.get(i);

                        RemoteWMSStyle sty = layer.getSelectedStyle();

                        if (sty == null) 

                                v.add("");

                        else 

                                v.add(sty.getTitle());

                }

                return v;

        }

        /**

         * Sets the selected styles in the StylesTree. The argument styleNames is

         * a Vector with exactly the same amount of strings than the amount of

         * themes (layers) contained by the tree. A blank or null string will

         * leave the default style for that layer, but this element <b>must exist</b>

         * in the array in any case.

         * 

         * @param styleNames, Vector containing the style names. 

         * The styles order <b>must match</b> with the layer order. 

         */

        public void setStylesSelection(Vector styleNames) {

                for (int i = 0; i < layers.size(); i++) {

                        WMSLayerNode layer = (WMSLayerNode) layers.get(i);

                        for (int j = 0; j < layer.getStyles().size(); j++) {

                                RemoteWMSStyle sty = (RemoteWMSStyle) layer.getStyles().get(j);

                                if ( sty.getName().equals((String) styleNames.get(i))) {

                                        layer.setSelectedStyleByIndex(j);

                                }

                        }

                }

        }

        /**

         * Adds a new branch to this tree. It must be a layer node.

         * @param node

         * @return <b>True</b>, if the added branch actually defines any style.

         *         <b>False</b>, if the layer has no styles.

         */

        @SuppressWarnings("unchecked")

        public boolean addLayerBranch(WMSLayerNode node) {

                layers.add(node);

                if (node.getStyles()==null || node.getStyles().size()==0){

                        return false;

                }

                for (int i = 0; i < node.getStyles().size(); i++) {

                        ((RemoteWMSStyle) node.getStyles().get(i)).setParent(node);

                }

                ((WMSLayerNode)getRoot()).getChildren().add(node);

                return true;

        }

        /**

     * Sets a leaf (an style) selected.

     * @param selectionPath to this leaf.

     */

        protected void setSelectedLeaf(TreePath selectionPath) {

                if (selectionPath!=null){

                        Object[] objects = selectionPath.getPath();

                        Object item = objects[objects.length-1];

                        if (isLeaf(item)){

                                RemoteWMSStyle style = (RemoteWMSStyle) item;

                                WMSLayerNode layer = style.getParent() ;

                                for (int i = 0; i < layer.getStyles().size(); i++) {

                                        RemoteWMSStyle sty = (RemoteWMSStyle) layer.getStyles().get(i);

                                        if (sty.getName().equals(style.getName())) {

                                                layer.setSelectedStyleByIndex(i);

                                                return;

                                        }

                                }

                        }

                }

        }

        /*

         *  (non-Javadoc)

         * @see javax.swing.tree.TreeModel#getRoot()

         */

        public Object getRoot() {

                if (root == null) {

            root = new WMSLayerNode();

            root.setParent(null);

        }

        return root;

        }

        /*

         *  (non-Javadoc)

         * @see javax.swing.tree.TreeModel#getChildCount(java.lang.Object)

         */

        public int getChildCount(Object parent) {

        int count=0;

        if (parent == root)

            count = ((WMSLayerNode) parent).getChildren().size();

        else

            count = ((WMSLayerNode) parent).getStyles().size();

        return count;

    }

        /*

         *  (non-Javadoc)

         * @see javax.swing.tree.TreeModel#isLeaf(java.lang.Object)

         */

        public boolean isLeaf(Object node) {

                return (node instanceof RemoteWMSStyle);

        }

        /*

         *  (non-Javadoc)

         * @see javax.swing.tree.TreeModel#addTreeModelListener(javax.swing.event.TreeModelListener)

         */

        public void addTreeModelListener(TreeModelListener l) {

        }

        /*

         *  (non-Javadoc)

         * @see javax.swing.tree.TreeModel#removeTreeModelListener(javax.swing.event.TreeModelListener)

         */

        public void removeTreeModelListener(TreeModelListener l) {

        }

        /*

         *  (non-Javadoc)

         * @see javax.swing.tree.TreeModel#getChild(java.lang.Object, int)

         */

        public Object getChild(Object parent, int index) {

                if (parent instanceof RemoteWMSStyle)

            return null;

        if (parent == root)

            return ((WMSLayerNode) parent).getChildren().get(index);

        return ((WMSLayerNode) parent).getStyles().get(index);

        }

        /*

         *  (non-Javadoc)

         * @see javax.swing.tree.TreeModel#getIndexOfChild(java.lang.Object, java.lang.Object)

         */

        public int getIndexOfChild(Object parent, Object child) {

                if (parent instanceof RemoteWMSStyle) { 

            return -1;   

        }

                WMSLayerNode l = (WMSLayerNode)parent;

        if (l.getParent() == null){

            for (int i = 0; i < l.getChildren().size(); i++) {

                if (l.getChildren().get(i).equals(child)){

                   return i;

                }

            }

        } else {

            for (int i = 0; i < l.getStyles().size(); i++) {

                if (l.getChildren().get(i).equals(child)){

                    return i;

                }

            }

        }

        return -1;

        }

        /*

         *  (non-Javadoc)

         * @see javax.swing.tree.TreeModel#valueForPathChanged(javax.swing.tree.TreePath, java.lang.Object)

         */

        public void valueForPathChanged(TreePath path, Object newValue) {

        }

}