Application: gvSIG desktop » gvSIG raster

/* gvSIG. Desktop Geographic Information System.

 *

 * Copyright ? 2007-2018 gvSIG Association

 *

 * 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.

 *

 * For any additional information, do not hesitate to contact us

 * at info AT gvsig.com, or visit our website www.gvsig.com.

 */

package org.gvsig.raster.wms.provider;

import java.awt.Component;

import java.util.ArrayList;

import java.util.Collections;

import java.util.Enumeration;

import java.util.Iterator;

import java.util.List;

import javax.swing.tree.TreeNode;

import org.apache.commons.lang3.StringUtils;

import org.gvsig.remoteclient.wms.WMSLayer;

import org.gvsig.remoteclient.wms.WMSStyle;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.i18n.I18nManager;

/**

 * @author fdiaz

 *

 */

public class LayerNode implements TreeNode {

    private TreeNode parent;

    private List<LayerNode> children;

    private List<Style> styles;

    private int selectedStyleIndex = -1;

    private String name;

    private String title;

    /**

     * @param layer

     * @param parent

     *

     */

    public LayerNode(WMSLayer layer, TreeNode parent) {

        this.name = layer.getName();

        this.title = layer.getTitle();

        this.parent = parent;

        setSelectedStyleIndex(-1);

        ArrayList<WMSStyle> layerStyles = layer.getStyles();

        styles = new ArrayList<>(layerStyles.size());

        for (Iterator<WMSStyle> iterator = layerStyles.iterator(); iterator.hasNext();) {

            WMSStyle wmsStyle = iterator.next();

            styles.add(new Style(wmsStyle));

        }

        if(layerStyles.size()>0){

            setSelectedStyleIndex(0);

        }

        ArrayList<WMSLayer> layerChildren = layer.getChildren();

        children = new ArrayList<>(layerChildren.size());

        for (Iterator<WMSLayer> iterator = layerChildren.iterator(); iterator.hasNext();) {

            WMSLayer wmsLayer = iterator.next();

            children.add(new LayerNode(wmsLayer, this));

        }

    }

    /* (non-Javadoc)

     * @see javax.swing.tree.TreeNode#getChildAt(int)

     */

    @Override

    public TreeNode getChildAt(int childIndex) {

        return children.get(childIndex);

    }

    /* (non-Javadoc)

     * @see javax.swing.tree.TreeNode#getChildCount()

     */

    @Override

    public int getChildCount() {

        return children.size();

    }

    /* (non-Javadoc)

     * @see javax.swing.tree.TreeNode#getParent()

     */

    @Override

    public TreeNode getParent() {

        return parent;

    }

    /* (non-Javadoc)

     * @see javax.swing.tree.TreeNode#getIndex(javax.swing.tree.TreeNode)

     */

    @Override

    public int getIndex(TreeNode node) {

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

            if(children.get(i).equals(node)){

                return i;

            }

        }

        return -1;

    }

    /* (non-Javadoc)

     * @see javax.swing.tree.TreeNode#getAllowsChildren()

     */

    @Override

    public boolean getAllowsChildren() {

        return (children.size() > 0);

    }

    /* (non-Javadoc)

     * @see javax.swing.tree.TreeNode#isLeaf()

     */

    @Override

    public boolean isLeaf() {

        return (children.size() == 0);

    }

    /* (non-Javadoc)

     * @see javax.swing.tree.TreeNode#children()

     */

    @Override

    public Enumeration children() {

        return Collections.enumeration(children);

    }

    /**

     * @return the selectedStyle

     */

    public int getSelectedStyleIndex() {

        return selectedStyleIndex;

    }

    /**

     * @param selectedStyle the selectedStyle to set

     */

    public void setSelectedStyleIndex(int selectedStyle) {

        this.selectedStyleIndex = selectedStyle;

    }

    /**

     * @return

     */

    public Style getSelectedStyle() {

        if(selectedStyleIndex>=0 && selectedStyleIndex<styles.size()){

            return styles.get(selectedStyleIndex);

        }

        return null;

    }

    /**

     * @param selectedStyleIndex the selectedStyle to set

     */

    public void setSelectedStyle(Style style) {

        this.selectedStyleIndex = styles.indexOf(style);

    }

    public List<Style> getStyles() {

        return styles;

    }

    public String toString() {

        String str;

            if (StringUtils.isEmpty(name)) {

                str = title;

            } else {

                str = "[" + name + "] " + title;

            }

        return str;

    }

    /**

     * @return

     */

    public String getName() {

        return name;

    }

    /**

     * @return

     */

    public String getTitle() {

        return title;

    }

    /**

     * @author fdiaz

     *

     */

    public class Style {

        private String name;

        private String title;

        /**

         * @param source

         */

        public Style(WMSStyle source){

            setName(source.getName());

            setTitle(source.getTitle());

        }

        /**

         * @return the name

         */

        public String getName() {

            return name;

        }

        /**

         * @param name the name to set

         */

        public void setName(String name) {

            this.name = name;

        }

        /**

         * @return the title

         */

        public String getTitle() {

            return title;

        }

        /**

         * @param title the title to set

         */

        public void setTitle(String title) {

            this.title = title;

        }

    }

}