Application: gvSIG desktop » gvSIG geoprocess

package org.gvsig.geoprocess.lib.sextante.core;

import java.awt.geom.Rectangle2D;

import java.util.ArrayList;

import java.util.List;

import es.unex.sextante.core.AbstractInputFactory;

import es.unex.sextante.core.NamedExtent;

import es.unex.sextante.dataObjects.I3DRasterLayer;

import es.unex.sextante.dataObjects.IDataObject;

import org.gvsig.andami.PluginServices;

import org.gvsig.app.extension.ProjectExtension;

import org.gvsig.app.project.Project;

import org.gvsig.app.project.documents.Document;

import org.gvsig.app.project.documents.table.TableDocument;

import org.gvsig.app.project.documents.table.TableManager;

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

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

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

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

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

import org.gvsig.fmap.mapcontext.layers.vectorial.FLyrVect;

import org.gvsig.geoprocess.lib.sextante.dataObjects.FileTools;

import org.gvsig.geoprocess.lib.sextante.dataObjects.gvRasterLayerRead;

import org.gvsig.geoprocess.lib.sextante.dataObjects.gvTable;

import org.gvsig.geoprocess.lib.sextante.dataObjects.gvVectorLayer;

import org.gvsig.raster.fmap.layers.FLyrRaster;

public class gvInputFactory extends AbstractInputFactory {

    public void createDataObjects() {

        final List<IDataObject> list = new ArrayList<IDataObject>();

        final Project project =

            ((ProjectExtension) PluginServices

                .getExtension(ProjectExtension.class)).getProject();

        final List<Document> views = project.getDocuments(ViewManager.TYPENAME);

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

            final DefaultViewDocument view = (DefaultViewDocument) views.get(i);

            final FLayers layers = view.getMapContext().getLayers();

            final LayersIterator iter = new LayersIterator(layers);

            while (iter.hasNext()) {

                final FLayer layer = iter.nextLayer();

                if (layer instanceof FLyrRaster) {

                    final gvRasterLayerRead rasterLayer =

                        new gvRasterLayerRead();

                    rasterLayer.create((FLyrRaster) layer);

                    list.add(rasterLayer);

                } else

                    if (layer instanceof FLyrVect) {

                        final gvVectorLayer vectorLayer = new gvVectorLayer();

                        vectorLayer.create((FLyrVect) layer);

                        list.add(vectorLayer);

                    }

            }

        }

        final List<Document> tables =

            project.getDocuments(TableManager.TYPENAME);

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

            final gvTable table = new gvTable();

            table.create((TableDocument) tables.get(i));

            list.add(table);

        }

        m_Objects = list.toArray(new IDataObject[list.size()]);

    }

    public NamedExtent[] getPredefinedExtents() {

        final Project project =

            ((ProjectExtension) PluginServices

                .getExtension(ProjectExtension.class)).getProject();

        final List<Document> views = project.getDocuments(ViewManager.TYPENAME);

        final NamedExtent ne[] = new NamedExtent[views.size()];

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

            final DefaultViewDocument view = (DefaultViewDocument) views.get(i);

            final Rectangle2D extent =

                view.getMapContext().getViewPort().getExtent();

            final String sName = view.getName();

            ne[i] = new NamedExtent(sName, extent);

        }

        return ne;

    }

    public String[] getRasterLayerInputExtensions() {

        return FileTools.RASTER_EXT_IN;

    }

    public String[] get3DRasterLayerInputExtensions() {

        return FileTools.RASTER3D_EXT_IN;

    }

    public String[] getVectorLayerInputExtensions() {

        return FileTools.VECTOR_EXT_IN;

    }

    public String[] getTableInputExtensions() {

        return FileTools.TABLE_EXT;

    }

    public IDataObject openDataObjectFromFile(final String sFilename) {

        final Object object = FileTools.open(sFilename);

        if (object == null) {

            return null;

        } else

            if (object instanceof FLyrRaster) {

                final gvRasterLayerRead layer = new gvRasterLayerRead();

                layer.create((FLyrRaster) object);

                return layer;

            } else

                if (object instanceof FLyrVect) {

                    final gvVectorLayer layer = new gvVectorLayer();

                    layer.create((FLyrVect) object);

                    return layer;

                } else

                    if (object instanceof I3DRasterLayer) {

                        return (I3DRasterLayer) object;

                    } else {

                        return null;

                    }

    }

    public void close(final String sName) {

        final IDataObject dataObject = this.getInputFromName(sName);

        final Object obj = dataObject.getBaseDataObject();

        if (obj instanceof FLayer) {

            removeLayer((FLayer) obj);

        } else {

            final Project project =

                ((ProjectExtension) PluginServices

                    .getExtension(ProjectExtension.class)).getProject();

            project.remove((Document) obj);

        }

    }

    private static void removeLayer(final FLayer baseLayer) {

        final Project project =

            ((ProjectExtension) PluginServices

                .getExtension(ProjectExtension.class)).getProject();

        final List<Document> views = project.getDocuments(ViewManager.TYPENAME);

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

            final DefaultViewDocument view = (DefaultViewDocument) views.get(i);

            final FLayers layers = view.getMapContext().getLayers();

            final LayersIterator iter = new LayersIterator(layers);

            while (iter.hasNext()) {

                final FLayer layer = iter.nextLayer();

                if (layer.equals(baseLayer)) {

                    layers.removeLayer(baseLayer);

                    return;

                }

            }

        }

    }

    @Override

    public I3DRasterLayer[] get3DRasterLayers() {

        return new I3DRasterLayer[0];

    }

}