gvSIG projects pool

/**

 * gvSIG. Desktop Geographic Information System.

 *

 * Copyright (C) 2007-2013 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 3

 * 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.topology.app.mainplugin;

import java.util.Map;

import javax.swing.ImageIcon;

import javax.swing.event.TreeModelListener;

import javax.swing.tree.TreeModel;

import javax.swing.tree.TreePath;

import org.gvsig.andami.IconThemeHelper;

import org.gvsig.app.project.documents.view.ViewDocument;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.DataStore;

import org.gvsig.fmap.mapcontext.MapContextLocator;

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

import org.gvsig.fmap.mapcontext.layers.operations.SingleLayer;

import org.gvsig.fmap.mapcontrol.CompoundLayersTreeModel;

import org.gvsig.fmap.mapcontrol.MapControlLocator;

import org.gvsig.tools.swing.api.ListElement;

import org.gvsig.topology.lib.api.TopologyDataSet;

import org.gvsig.topology.lib.api.TopologyLocator;

import org.gvsig.topology.swing.api.TopologySwingServices;

/**

 *

 * @author jjdelcerro

 */

public class DataSetsTreeModel implements TreeModel {

    private final ViewDocument view;

    private final CompoundLayersTreeModel deletaged;

    private final TopologySwingServices services;

    private final ListElement<Object> root;

    public DataSetsTreeModel(TopologySwingServices services, ViewDocument view) {

        this.services = services;

        this.view = view;

        CompoundLayersTreeModel model = (CompoundLayersTreeModel) MapControlLocator.getMapControlManager().createCompoundLayersTreeModel();

        model.addLayers(view.getMapContext().getLayers());

        this.deletaged = model;

        this.root = new ListElement<>("Project", this.deletaged.getRoot());

    }

    private Object getNode(Object value) {

        if( value instanceof ListElement ) {

            return ((ListElement) value).getValue();

        }

        return value;

    }

    @Override

    public Object getRoot() {

        return this.root;

    }

    @Override

    public Object getChild(Object parent, int index) {

        parent = this.getNode(parent);

        Object x = this.deletaged.getChild(parent, index);

        if( x instanceof SingleLayer ) {

            SingleLayer layer = (SingleLayer)x;

            DataStore store = layer.getDataStore();

            x = TopologyLocator.getTopologyManager().createDataSet(

                    layer.getName(),

                    store

            );

        }

        return x;

    }

    @Override

    public int getChildCount(Object parent) {

        parent = this.getNode(parent);

        return this.deletaged.getChildCount(parent);

    }

    @Override

    public boolean isLeaf(Object node) {

        node = this.getNode(node);

        return this.deletaged.isLeaf(node);

    }

    @Override

    public void valueForPathChanged(TreePath path, Object newValue) {

        this.deletaged.valueForPathChanged(path, newValue);

    }

    @Override

    public int getIndexOfChild(Object parent, Object child) {

        parent = this.getNode(parent);

        return this.deletaged.getIndexOfChild(parent, child);

    }

    @Override

    public void addTreeModelListener(TreeModelListener l) {

        this.deletaged.addTreeModelListener(l);

    }

    @Override

    public void removeTreeModelListener(TreeModelListener l) {

        this.deletaged.removeTreeModelListener(l);

    }

    public ImageIcon getTreeIcon(Object item) {

        if( this.getRoot()==item ) {

            return IconThemeHelper.getImageIcon("topology-tree-project");

        }

        if (item instanceof ListElement) {

            item = ((ListElement) item).getValue();

        }

        if( this.view.getMapContext().getLayers()==item ) {

            return IconThemeHelper.getImageIcon("document-view-tree-icon");

        }

        if (item instanceof TopologyDataSet) {

            DataStore store = ((TopologyDataSet)item).getStore();

            String iconName = MapContextLocator.getMapContextManager().getIconLayer(store);

            return IconThemeHelper.getImageIcon(iconName);

        }

        if (item instanceof FLayer) {

            String iconName = ((FLayer) item).getTocImageIcon();

            return IconThemeHelper.getImageIcon(iconName);

        }

        return IconThemeHelper.getImageIcon("topology-tree-folder");

    }

}