Application: gvSIG desktop

package org.gvsig.fmap.mapcontrol;

import java.util.ArrayList;

import java.util.Iterator;

import java.util.List;

import javax.swing.event.TreeModelEvent;

import javax.swing.event.TreeModelListener;

import javax.swing.tree.TreeModel;

import javax.swing.tree.TreePath;

import org.gvsig.fmap.mapcontext.layers.FLayer;

import org.gvsig.fmap.mapcontext.layers.FLayers;

public class LayersTreeModel implements TreeModel {

        protected final FLayers layers;

        protected final List<TreeModelListener> listeners = new ArrayList();

        public LayersTreeModel(FLayers layers) {

            this.layers = layers;

        }

        @Override

        public Object getRoot() {

            return layers;

        }

        @Override

        public Object getChild(Object parent, int index) {

            if (!(parent instanceof FLayers)) {

                return null;

            }

            return ((FLayers) parent).getLayer(index);

        }

        @Override

        public int getChildCount(Object parent) {

            if (!(parent instanceof FLayers)) {

                return 0;

            }

            return ((FLayers) parent).getLayersCount();

        }

        @Override

        public boolean isLeaf(Object node) {

            return !(node instanceof FLayers);

        }

        @Override

        public int getIndexOfChild(Object parent, Object child) {

            for( int i=0; i<((FLayers)parent).getLayersCount() ; i++ ) {

                FLayer layer = ((FLayers)parent).getLayer(i);

                if( child == layer ) {

                    return i;

                }

            }

            return -1;

        }

        @Override

        public void valueForPathChanged(TreePath path, Object value) {

        }

        @Override

        public void addTreeModelListener(TreeModelListener listener) {

            listeners.add(listener);

        }

        @Override

        public void removeTreeModelListener(TreeModelListener listener) {

            listeners.remove(listener);

        }

        public void reload() {

            TreeModelEvent event = new TreeModelEvent(this, new TreePath(this.layers));

            Iterator<TreeModelListener> iterator = listeners.iterator();

            TreeModelListener listener;

            while (iterator.hasNext()) {

                listener = iterator.next();

                listener.treeStructureChanged(event);

            }

        }

}