Revision 727 org.gvsig.topology/trunk/org.gvsig.topology/org.gvsig.topology.app/org.gvsig.topology.app.mainplugin/src/main/java/org/gvsig/topology/app/mainplugin/DataSetsTreeModel.java

View differences:

DataSetsTreeModel.java
23 23
 */
24 24
package org.gvsig.topology.app.mainplugin;
25 25

  
26
import javax.swing.ImageIcon;
26 27
import javax.swing.event.TreeModelListener;
27 28
import javax.swing.tree.TreeModel;
28 29
import javax.swing.tree.TreePath;
30
import org.gvsig.andami.IconThemeHelper;
29 31
import org.gvsig.app.project.documents.view.ViewDocument;
30
import org.gvsig.fmap.dal.feature.FeatureStore;
31
import org.gvsig.fmap.mapcontext.layers.vectorial.FLyrVect;
32
import org.gvsig.fmap.dal.DataStore;
33
import org.gvsig.fmap.mapcontext.MapContextLocator;
34
import org.gvsig.fmap.mapcontext.layers.FLayer;
35
import org.gvsig.fmap.mapcontext.layers.operations.SingleLayer;
32 36
import org.gvsig.fmap.mapcontrol.CompoundLayersTreeModel;
33 37
import org.gvsig.fmap.mapcontrol.MapControlLocator;
34 38
import org.gvsig.tools.swing.api.ListElement;
39
import org.gvsig.topology.lib.api.TopologyDataSet;
35 40
import org.gvsig.topology.lib.api.TopologyLocator;
36 41
import org.gvsig.topology.swing.api.TopologySwingServices;
37 42

  
......
72 77
    public Object getChild(Object parent, int index) {
73 78
        parent = this.getNode(parent);
74 79
        Object x = this.deletaged.getChild(parent, index);
75
        if( x instanceof FLyrVect ) {
76
            FLyrVect layer = (FLyrVect)x;
77
            FeatureStore store = layer.getFeatureStore();
80
        if( x instanceof SingleLayer ) {
81
            SingleLayer layer = (SingleLayer)x;
82
            DataStore store = layer.getDataStore();
78 83
            x = TopologyLocator.getTopologyManager().createDataSet(
79 84
                    layer.getName(),
80 85
                    store
......
115 120
    public void removeTreeModelListener(TreeModelListener l) {
116 121
        this.deletaged.removeTreeModelListener(l);
117 122
    }
123

  
124
    public ImageIcon getTreeIcon(Object item) {
125
        if( this.getRoot()==item ) {
126
            return IconThemeHelper.getImageIcon("topology-tree-project");
127
        }
128
        if (item instanceof ListElement) {
129
            item = ((ListElement) item).getValue();
130
        }
131
        if( this.view.getMapContext().getLayers()==item ) {
132
            return IconThemeHelper.getImageIcon("document-view-tree-icon");
133
        }
134
        if (item instanceof TopologyDataSet) {
135
            DataStore store = ((TopologyDataSet)item).getStore();
136
            String iconName = MapContextLocator.getMapContextManager().getIconLayer(store);
137
            return IconThemeHelper.getImageIcon(iconName);
138
        }
139
        if (item instanceof FLayer) {
140
            String iconName = ((FLayer) item).getTocImageIcon();
141
            return IconThemeHelper.getImageIcon(iconName);
142
        }
143
        return IconThemeHelper.getImageIcon("topology-tree-folder");
144
    }
118 145
    
119 146
}

Also available in: Unified diff