Application: gvSIG desktop » gvSIG geoprocess

package org.gvsig.sextante.app.extension.gui;

import java.awt.geom.Rectangle2D;

import java.util.ArrayList;

import java.util.List;

import org.gvsig.andami.PluginServices;

import org.gvsig.app.extension.ProjectExtension;

import org.gvsig.app.project.Project;

import org.gvsig.app.project.ProjectManager;

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.fmap.raster.layers.FLyrRasterSE;

import org.gvsig.sextante.app.extension.core.FileTools;

import org.gvsig.sextante.app.extension.core.gvRasterLayer;

import org.gvsig.sextante.app.extension.core.gvTable;

import org.gvsig.sextante.app.extension.core.gvVectorLayer;

import es.unex.sextante.core.AbstractInputFactory;

import es.unex.sextante.core.NamedExtent;

import es.unex.sextante.dataObjects.IDataObject;

public class gvInputFactory extends AbstractInputFactory {

        public void createDataObjects() {

                ArrayList list = new ArrayList();

//                Project project = ((ProjectExtension)PluginServices.getExtension(ProjectExtension.class)).getProject();

                final Project project = ProjectManager.getInstance().getCurrentProject();

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

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

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

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

                        LayersIterator iter = new LayersIterator(layers);

                        while (iter.hasNext()){

                                FLayer layer = iter.nextLayer();

                                if (layer instanceof FLyrRasterSE){

                                        gvRasterLayer rasterLayer = new gvRasterLayer();

                                        rasterLayer.create((FLyrRasterSE)layer);

                                        list.add(rasterLayer);

                                }

                                else if (layer instanceof FLyrVect){

                                        gvVectorLayer vectorLayer =new gvVectorLayer();

                                        vectorLayer.create((FLyrVect)layer);

                                        list.add(vectorLayer);

                                }

                        }

                }

                List<Document> tables = project.getDocuments(TableManager.TYPENAME);

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

                        gvTable table = new gvTable();

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

                        list.add(table);

                }

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

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

                        m_Objects[i] = (IDataObject)list.get(i);

                }

        }

        public NamedExtent[] getPredefinedExtents() {

                final Project project = ProjectManager.getInstance().getCurrentProject();

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

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

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

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

                        Rectangle2D extent = view.getMapContext().getViewPort().getExtent();

                        String sName = view.getName();

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

                }

                return ne;

        }

        public String[] getRasterLayerInputExtensions() {

                return FileTools.RASTER_EXT_IN;

        }

        public String[] getVectorLayerInputExtensions() {

                return FileTools.VECTOR_EXT_IN;

        }

        public String[] getTableInputExtensions() {

                return FileTools.TABLE_EXT;

        }

        public IDataObject openDataObjectFromFile(String sFilename) {

                Object object = FileTools.open(sFilename);

                if (object == null) {

                        return null;

                } else 

                        if (object instanceof FLyrRasterSE) {

                                gvRasterLayer layer = new gvRasterLayer();

                                layer.create(object);

                                return layer;

                } else 

                        if (object instanceof FLyrVect) {

                                gvVectorLayer layer = new gvVectorLayer();

                                layer.create(object);

                                return layer;

                } else 

                        return null;

        }

        public void close(String sName) {

                IDataObject dataObject = this.getInputFromName(sName);

                Object obj = dataObject.getBaseDataObject();

                if (obj instanceof FLayer){

                        removeLayer((FLayer)obj);

                } else {

                        Project project = ((ProjectExtension)PluginServices.getExtension(ProjectExtension.class)).getProject();

                        project.remove((Document) obj);

                }

        }

        private static void removeLayer(FLayer baseLayer) {

                Project project = ((ProjectExtension)PluginServices.getExtension(ProjectExtension.class)).getProject();

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

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

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

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

                        LayersIterator iter = new LayersIterator(layers);

                        while (iter.hasNext()){

                                FLayer layer = iter.nextLayer();

                                if (layer.equals(baseLayer)){

                                        layers.removeLayer(baseLayer);

                                        return;

                                }

                        }

                }

        }

}