Application: gvSIG desktop

/* gvSIG. Sistema de Informaci?n Geogr?fica de la Generalitat Valenciana

 *

 * Copyright (C) 2005 IVER T.I. and Generalitat Valenciana.

 *

 * 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., 59 Temple Place - Suite 330, Boston, MA  02111-1307,USA.

 *

 * For more information, contact:

 *

 *  Generalitat Valenciana

 *   Conselleria d'Infraestructures i Transport

 *   Av. Blasco Ib??ez, 50

 *   46010 VALENCIA

 *   SPAIN

 *

 *      +34 963862235

 *   gvsig@gva.es

 *      www.gvsig.gva.es

 *

 *    or

 *

 *   IVER T.I. S.A

 *   Salamanca 50

 *   46005 Valencia

 *   Spain

 *

 *   +34 963163400

 *   dac@iver.es

 */

/* CVS MESSAGES:

*

* $Id: StyleTree.java 3537 2006-01-05 23:15:59Z jaume $

* $Log$

* Revision 1.1.2.3  2006-01-05 23:15:53  jaume

* *** empty log message ***

*

* Revision 1.1.2.2  2006/01/04 18:09:02  jaume

* Time dimension

*

* Revision 1.1.2.1  2006/01/03 18:08:40  jaume

* *** empty log message ***

*

* Revision 1.1.2.1  2006/01/02 18:08:01  jaume

* Tree de estilos

*

*

*/

/**

 * 

 */

package com.iver.cit.gvsig.gui.panels;

import java.awt.Color;

import java.awt.Component;

import java.util.ArrayList;

import java.util.Hashtable;

import java.util.Iterator;

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.TreeCellRenderer;

import javax.swing.tree.TreeModel;

import javax.swing.tree.TreePath;

import com.iver.cit.gvsig.fmap.layers.WMSLayerNode;

import com.iver.cit.gvsig.fmap.layers.WMSLayerNode.FMapWMSStyle;

/**

 * 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 is 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 no 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

 *

 */

public class StyleTree extends JTree {

    public StyleTree(){

        super();

        initialize();

    }

    private void initialize(){

        setToggleClickCount(1);

        setRowHeight(22);

        setCellRenderer(new TreeCellRenderer() {

            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("", ((StylesTreeModel) getModel()).isSelected((FMapWMSStyle) value));//selected);

                    leafRadioButton.setBackground(Color.WHITE);

                    leafComponent.add(leafRadioButton);

                    leafComponent.add(new JLabel(((FMapWMSStyle) value).title));

                    return leafComponent;

                } else {

                        JTree t = new JTree();

                            Component branchComponent = t.getCellRenderer().getTreeCellRendererComponent(tree, value, selected, expanded, leaf, row, hasFocus); 

                            return branchComponent;

                        /*

                    JPanel branchComponent = new JPanel();

                    branchComponent.setBackground(Color.WHITE);

                    if (value instanceof WMSLayerNode)

                        branchComponent.add(new JLabel(((WMSLayerNode) value).getTitle()));

                    return branchComponent;

                    */

                }

            }});

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

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

                ((StylesTreeModel)getModel()).setSelectedLeaf(getSelectionPath());

                clearSelection();

                repaint();

            }

        });

    }

    public Vector getStylesSelection(){

        return ((StylesTreeModel)getModel()).getStylesSelection();

    }

    /**

     * 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){

        StylesTreeModel model = (StylesTreeModel) getModel();

        if (model.getChildCount(model.getRoot())!=styleNames.size()){

            throw new RuntimeException("Bad arguments: styleNames length is "

                    +styleNames.size()+", should be "

                    +model.getChildCount(model.getRoot())+".");

        }

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

            if (styleNames.get(i)==null || ((String)styleNames.get(i)).equals(""))

                continue;

            Object node = model.getChild(model.getRoot(), i);

            for (int j = 0; j < model.getChildCount(node); j++) {

                if (((String)styleNames.get(i)).equals(((FMapWMSStyle)model.getChild(node, j)).name))

                    model.selectStyle(model.getChild(node, j));

            }

        }

    }

}

class StylesTreeModel implements TreeModel {

    WMSLayerNode root;

    Hashtable selections = new Hashtable();

    public StylesTreeModel(WMSLayerNode root){

        this.root = root;

    }

    /**

     * Marks this style as selected in the StylesTreeModel.

     * @param style

     */

    public void selectStyle(Object style) {

        selections.put(((FMapWMSStyle)style).parent, style);

    }

    /**

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

     * as they was represented in the StylesTree.

     * @return

     */

    protected Vector getStylesSelection() {

        Vector values = new Vector();

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

            values.add(i, "");

        }

        Iterator it = selections.keySet().iterator();

        int i = 0;

        while (it.hasNext()){

            Object key = it.next();

            values.set(i, ((FMapWMSStyle)selections.get(key)).name);

            i++;

        }

        return values;

    }

    /**

     * 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)){

                selections.put(((FMapWMSStyle) item).parent, item);

            }

        }

    }

    /**

     * Will return true if this style is selected.

     * @param style

     * @return

     */

    protected boolean isSelected(FMapWMSStyle style){

        return selections.get(style.parent) == style;

    }

    public int getChildCount(Object parent) {

        int count=0;

        if (parent == root)

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

        else

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

        return count;

    }

    public boolean isLeaf(Object node) {

        return (node instanceof FMapWMSStyle);

    }

    public void addTreeModelListener(TreeModelListener l) {

    }

    public void removeTreeModelListener(TreeModelListener l) {

    }

    public Object getChild(Object parent, int index) {

        if (parent instanceof FMapWMSStyle)

            return null;

        if (parent == root)

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

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

    }

    public int getIndexOfChild(Object parent, Object child) {

        if (parent instanceof FMapWMSStyle){ 

            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;

    }

    public void valueForPathChanged(TreePath path, Object newValue) {

    }

    /**

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

     * @param node

     */

    public void addLayerBranch(WMSLayerNode node){

        WMSLayerNode myNode = (WMSLayerNode) node.clone();

        myNode.setParent((WMSLayerNode) getRoot());

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

    }

    public Object getRoot() {

        if (root == null) {

            root = new WMSLayerNode();

            root.setParent(null);

        }

        return root;

    }

    /**

     * @param node, the tree's desired node.

     * @return Returns the title.

     */

    public String getTitle(Object node) {

        if (node instanceof WMSLayerNode)

            return ((WMSLayerNode) node).getTitle();

        return ((FMapWMSStyle) node).toString();

    }

    public String toString(){

        return getTitle(this);

    }

}