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.wmts.provider.tree;

import java.net.URL;

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

import org.gvsig.raster.wmts.ogc.WMTSClient;

import org.gvsig.raster.wmts.ogc.struct.WMTSLayer;

import org.gvsig.raster.wmts.ogc.struct.WMTSTheme;

import org.gvsig.raster.wmts.ogc.struct.WMTSThemes;

import org.gvsig.tools.ToolsLocator;

/**

 * @author fdiaz

 *

 */

public class DefaultWMTSRootNode extends AbstractWMTSNode implements WMTSRootNode {

    private List<WMTSNode> children;

    private WMTSThemes source;

    private URL service;

    /**

     * @param source

     * @param service

     * @param parent

     */

    public DefaultWMTSRootNode(WMTSThemes source, URL service) {

        super(null);

        this.source = source;

        this.service = service;

        children = new ArrayList<WMTSNode>();

        List<WMTSLayer> wmtsLayers = new ArrayList<WMTSLayer>();

        for (int i = 0; i < source.getChildCount(); i++) {

            WMTSTheme theme = source.getChildren(i);

            wmtsLayers.addAll(getLayers(theme));

        }

        for (Iterator iterator = wmtsLayers.iterator(); iterator.hasNext();) {

            WMTSLayer wmtsLayer = (WMTSLayer) iterator.next();

            children.add(new DefaultWMTSLayerNode(wmtsLayer, this, service));

        }

        this.title = ToolsLocator.getI18nManager().getTranslation("_base");

    }

    private List<WMTSLayer> getLayers(WMTSTheme theme){

        List<WMTSLayer> wmtsLayers = new ArrayList<WMTSLayer>();

        if(theme.getLayer()!=null){

            wmtsLayers.add(theme.getLayer());

        }

        for (int i = 0; i < theme.getChildCount(); i++) {

            wmtsLayers.addAll(getLayers(theme.getChildren(i)));

        }

        return wmtsLayers;

    }

    @Override

    public TreeNode getChildAt(int childIndex) {

        return children.get(childIndex);

    }

    @Override

    public int getChildCount() {

        return children.size();

    }

    @Override

    public int getIndex(TreeNode node) {

        return this.children.indexOf(node);

    }

    @Override

    public boolean getAllowsChildren() {

        return true;

    }

    @Override

    public boolean isLeaf() {

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

    }

    @Override

    public Enumeration children() {

        return Collections.enumeration(getChildren());

    }

    /**

     * @return the children

     */

    public List<WMTSNode> getChildren() {

        if (this.children == null) {

            this.children = new ArrayList<WMTSNode>();

        }

        return this.children;

    }

}