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.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.fmap.dal.OpenDataStoreParameters;

import org.gvsig.remoteclient.wms.WMSLayer;

import org.gvsig.remoteclient.wms.WMSStyle;

/**

 * @author fdiaz

 *

 */

public class DefaultWMSLayerNode implements WMSLayerNode {

    private TreeNode parent;

    private List<WMSLayerNode> children;

    private List<WMSStyleData> styles;

    private int selectedStyleIndex = -1;

    private String name;

    private String title;

    /**

     * @param layer

     * @param parent

     *

     */

    public DefaultWMSLayerNode(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 DefaultWMSStyle(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 DefaultWMSLayerNode(wmsLayer, this));

        }

    }

    @Override

    public TreeNode getChildAt(int childIndex) {

        return children.get(childIndex);

    }

    @Override

    public int getChildCount() {

        return children.size();

    }

    @Override

    public TreeNode getParent() {

        return parent;

    }

    @Override

    public int getIndex(TreeNode node) {

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

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

                return i;

            }

        }

        return -1;

    }

    @Override

    public boolean getAllowsChildren() {

        return (children.size() > 0);

    }

    @Override

    public boolean isLeaf() {

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

    }

    @Override

    public Enumeration children() {

        return Collections.enumeration(children);

    }

    @Override

    public int getSelectedStyleIndex() {

        return selectedStyleIndex;

    }

    @Override

    public void setSelectedStyleIndex(int selectedStyle) {

        this.selectedStyleIndex = selectedStyle;

    }

    @Override

    public WMSStyleData getSelectedStyle() {

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

            return styles.get(selectedStyleIndex);

        }

        return null;

    }

    @Override

    public void setSelectedStyle(WMSStyleData defaultWMSStyle) {

        this.selectedStyleIndex = styles.indexOf(defaultWMSStyle);

    }

    @Override

    public List<WMSStyleData> getStyles() {

        return styles;

    }

    @Override

    public String toString() {

        String str;

            if (StringUtils.isEmpty(name)) {

                str = title;

            } else {

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

            }

        return str;

    }

    @Override

    public String getName() {

        return name;

    }

    @Override

    public String getTitle() {

        return title;

    }

    @Override

    public OpenDataStoreParameters getParameters() {

        // TODO Auto-generated method stub

        return null;

    }

}