| gvSIG desktop 1

/* gvSIG. Sistema de Informaci?n Geogr?fica de la Generalitat Valenciana

 *

 * Copyright (C) 2004 IVER T.I. and Generalitat Valenciana.

 *

 * 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 2

 * 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., 59 Temple Place - Suite 330, Boston, MA  02111-1307,USA.

 *

 * For more information, contact:

 *

 *  Generalitat Valenciana

 *   Conselleria d'Infraestructures i Transport

 *   Av. Blasco Ib??ez, 50

 *   46010 VALENCIA

 *   SPAIN

 *

 *      +34 963862235

 *   gvsig@gva.es

 *      www.gvsig.gva.es

 *

 *    or

 *

 *   IVER T.I. S.A

 *   Salamanca 50

 *   46005 Valencia

 *   Spain

 *

 *   +34 963163400

 *   dac@iver.es

 */

package com.iver.cit.gvsig.graph;

import java.awt.Component;

import java.io.File;

import javax.swing.ImageIcon;

import javax.swing.JOptionPane;

import com.iver.andami.PluginServices;

import com.iver.andami.plugins.Extension;

import com.iver.andami.preferences.IPreference;

import com.iver.andami.preferences.IPreferenceExtension;

import com.iver.andami.ui.mdiManager.IWindow;

import com.iver.cit.gvsig.ProjectExtension;

import com.iver.cit.gvsig.fmap.DriverException;

import com.iver.cit.gvsig.fmap.MapContext;

import com.iver.cit.gvsig.fmap.MapControl;

import com.iver.cit.gvsig.fmap.core.FShape;

import com.iver.cit.gvsig.fmap.drivers.DriverIOException;

import com.iver.cit.gvsig.fmap.edition.EditionException;

import com.iver.cit.gvsig.fmap.edition.writers.dbf.DbfWriter;

import com.iver.cit.gvsig.fmap.layers.FLayer;

import com.iver.cit.gvsig.fmap.layers.FLyrVect;

import com.iver.cit.gvsig.fmap.layers.SelectableDataSource;

import com.iver.cit.gvsig.fmap.layers.SingleLayerIterator;

import com.iver.cit.gvsig.fmap.operations.strategies.Strategy;

import com.iver.cit.gvsig.fmap.operations.strategies.StrategyManager;

import com.iver.cit.gvsig.geoprocess.core.fmap.FeaturePersisterProcessor2;

import com.iver.cit.gvsig.graph.core.GvFlag;

import com.iver.cit.gvsig.graph.core.IGraph;

import com.iver.cit.gvsig.graph.core.Network;

import com.iver.cit.gvsig.graph.core.loaders.NetworkLoader;

import com.iver.cit.gvsig.graph.core.loaders.NetworkRedLoader;

import com.iver.cit.gvsig.graph.core.writers.NetworkFileRedWriter;

import com.iver.cit.gvsig.graph.core.writers.NetworkGvTableWriter;

import com.iver.cit.gvsig.graph.gui.RouteControlPanel;

import com.iver.cit.gvsig.graph.gui.wizard.NetWizard;

import com.iver.cit.gvsig.graph.preferences.RoutePage;

import com.iver.cit.gvsig.project.documents.table.ProjectTable;

import com.iver.cit.gvsig.project.documents.view.IProjectView;

import com.iver.cit.gvsig.project.documents.view.gui.IView;

import com.iver.cit.gvsig.project.documents.view.gui.View;

import com.iver.utiles.swing.threads.AbstractMonitorableTask;

public class NetworkExtension extends Extension implements IPreferenceExtension {

        private static final IPreference thePreferencePage = new RoutePage();

//        private String fieldType;

//        private String fieldDist;

//        private String fieldSense;

        public void initialize() {

        }

        public void execute(String actionCommand) {

                IView view = (View) PluginServices.getMDIManager().getActiveWindow();

                MapControl mapControl = view.getMapControl();

                MapContext map = mapControl.getMapContext();

                SingleLayerIterator lyrIterator = new SingleLayerIterator(map

                                .getLayers());

                while (lyrIterator.hasNext()) {

                        FLayer lyr = lyrIterator.next();

                        if ((lyr.isActive()) && (lyr instanceof FLyrVect))

                        {

                                FLyrVect lyrVect = (FLyrVect) lyr;

                                int shapeType;

                                try {

                                        shapeType = lyrVect.getShapeType();

                                        if (shapeType == FShape.LINE)

                                        {

                                                if (actionCommand.equalsIgnoreCase("GENERATE_RED")) {

                                                        generateRedNetwork(lyrVect);

                                                }

                                                if (actionCommand.equalsIgnoreCase("LOAD_RED")) {

                                                        loadNetwork(lyrVect);

                                                }

                                                if (actionCommand.equalsIgnoreCase("MANAGE_FLAGS")) {

                                                        manageFlags(lyrVect);

                                                }

                                        }

                                } catch (DriverException e) {

                                        // TODO Auto-generated catch block

                                        e.printStackTrace();

                                }

                        }

                }

        }

        private void manageFlags(FLyrVect lyrVect) {

                Network net = (Network) lyrVect.getProperty("network");

                if ( net != null)

                {

                        GvFlag[] flags = net.getFlags();

                        if(flags.length > 0)

                                PluginServices.getMDIManager().addWindow(new RouteControlPanel(flags));

                        else

                                PluginServices.getMDIManager().addWindow(new RouteControlPanel());

                        return;

                }

        }

        /**

         * Suponemos que en el proyecto hay 2 tablas, una con los nodos

         * y otro con los edges.

         * Cargamos la red a partir de esas tablas y se la

         * asociamos a la capa. A partir de ah?, nuestras

         * herramientas pueden ver si la capa activa tiene

         * asociada o no una red y ponerse visibles / invisibles

         * Otra posible soluci?n es llevar nuestra propia lista de capas

         * con red (que ser? peque?ita), y as?, en lugar de recorrer

         * el MapContext, recorremos nuestra lista para ver la

         * capa que est? activa y con red. Me empieza a preocupar

         * que todas las herramientas iteren por la colecci?n de

         * capas para habilitarse/deshabilitarse:

         * 100 herramientas * 100 capas = 10.000 comprobaciones

         * Si comprobar algo cuesta 1 mseg => 10 segundos!!!

         * @param lyrVect

         */

        private void loadNetworkFromTables(FLyrVect lyrVect) {

                // Aqu? mostrar un di?lgo para seleccionar las tablas

                // de nodos y edges

                // y hacer un mapping (si es necesario) entre los

                // nombres de campos

                String tableNodes = "Nodes";

                String tableEdges = "Edges";

                ProjectExtension projectExt = (ProjectExtension) PluginServices.getExtension(ProjectExtension.class);

                ProjectTable ptNodes = projectExt.getProject().getTable(tableNodes);

                ProjectTable ptEdges = projectExt.getProject().getTable(tableEdges);

                SelectableDataSource sdsNodes = ptNodes.getModelo().getRecordset();

                SelectableDataSource sdsEdges = ptEdges.getModelo().getRecordset();

                NetworkLoader netLoader = new NetworkLoader(true);

                netLoader.setNodeReader(sdsNodes);

                netLoader.setEdgeReader(sdsEdges);

                IGraph g = netLoader.loadNetwork();

                System.out.println("Num nodos=" + g.numVertices() + " numEdges = " + g.numEdges());

                lyrVect.setProperty("network", g);

        }

        private void loadNetwork(FLyrVect lyrVect) {

                // Aqu? mostrar un di?lgo para seleccionar las tablas

                // de nodos y edges

                // y hacer un mapping (si es necesario) entre los

                // nombres de campos

                NetworkRedLoader netLoader = new NetworkRedLoader();

                File redFile = new File("c:/" + lyrVect.getName().replaceAll("\\Q.shp\\E", ".red"));

                netLoader.setNetFile(redFile);

                IGraph g = netLoader.loadNetwork();

                System.out.println("Num nodos=" + g.numVertices() + " numEdges = " + g.numEdges());

                // TODO: MOSTRAR UN CUADRO DE DI?LOGO CON UN COMBOBOX PARA QUE ESCOJA EL CAMPO DE NOMBRE DE CALLE.

                String fieldStreetName = JOptionPane.showInputDialog((Component) PluginServices.getMainFrame(),

                                "Street-Route field name?", "NOMBRE");

                Network net = new Network();

                // lyrVect.createSpatialIndex();

                net.setGraph(g);

                net.setLayer(lyrVect);

                ShortestPathExtension.solver.setNetwork(net);

                ShortestPathExtension.solver.setFielStreetName(fieldStreetName);

                lyrVect.setProperty("network", net);

        }

        private void generateNetwork(FLyrVect lyr) {

                NetworkGvTableWriter netBuilder = new NetworkGvTableWriter();

                // Por ahora, a pelo, pero hay que sacar un cuadro

                // de di?logo para hecer el mapping.

                // Tambi?n un cuadro de di?logo para seleccionar

                // en qu? tablas quiere escribir, y su formato

                // (dbf, postgres, etc)

                String fieldType = "tipored"; String fieldDist = "length"; String fieldSense = "sen";

                String fieldCost = "cost";

                try {

                        netBuilder.setLayer(lyr);

                        netBuilder.setFieldType(fieldType);

                        netBuilder.setFieldDist(fieldDist);

                        netBuilder.setFieldSense(fieldSense);

                        netBuilder.setFieldCost(fieldCost);

                        DbfWriter nodeWriter = new DbfWriter();

                        nodeWriter.setFile(new File("c:/nodes.dbf"));

                        DbfWriter edgeWriter = new DbfWriter();

                        edgeWriter.setFile(new File("c:/edges.dbf"));

                        netBuilder.setEdgeWriter(edgeWriter);

                        netBuilder.setNodeWriter(nodeWriter);

                        netBuilder.writeNetwork();

                } catch (DriverException e1) {

                        // TODO Auto-generated catch block

                        e1.printStackTrace();

                } catch (EditionException e1) {

                        // TODO Auto-generated catch block

                        e1.printStackTrace();

                }

                JOptionPane.showMessageDialog(null, PluginServices.getText(this, "done"));

        }

        class GenerateRedNetworkTask extends AbstractMonitorableTask{

                FLyrVect layer;

                File redFile;

                NetworkFileRedWriter netBuilder;

                /**

                 * Constructor

                 * */

                GenerateRedNetworkTask(FLyrVect layer, File redFile, NetworkFileRedWriter netBuilder){

                        this.layer = layer;

                        this.redFile = redFile;

                        this.netBuilder = netBuilder;

                        setInitialStep(0);

                        int numShapes;

                        try {

                                numShapes = layer.getSource().getShapeCount();

                                setFinalStep(numShapes);

                                setDeterminatedProcess(true);

                                setStatusMessage(PluginServices.

                                                getText(this, "Generando_red_a_partir_de_capa_lineal"));

                        } catch (DriverIOException e) {

                                // TODO Auto-generated catch block

                                e.printStackTrace();

                        }

                }

                public void run() throws Exception {

                        netBuilder.setLayer(layer);

                        netBuilder.setCancellableMonitorable(this);

                        netBuilder.setRedFile(redFile);

                        netBuilder.writeNetwork();

                        JOptionPane.showMessageDialog(null, PluginServices.getText(this, "done"));

                }        

                public String getNote() {

                        return "Procesando linea..."  + 

                        " " +

                        getCurrentStep()+ 

                        " "+

                        "de"+

                        " "+

                        getFinishStep();

                }

                public void cancel() {

                        setCanceled(true);

                }

        }

        private void generateRedNetwork(FLyrVect lyr) {

                lyr.createSpatialIndex();

                NetworkFileRedWriter netBuilder = new NetworkFileRedWriter();

                // Por ahora, a pelo, pero hay que sacar un cuadro

                // de di?logo para hecer el mapping.

                // Tambi?n un cuadro de di?logo para seleccionar

                // en qu? tablas quiere escribir, y su formato

                // (dbf, postgres, etc)

                ImageIcon icon = new ImageIcon(this.

                                getClass().

                                getClassLoader().

                                getResource("images/net-wizard-logo.jpg"));

                NetWizard wiz = new NetWizard(icon, lyr);

                PluginServices.getMDIManager().addWindow(wiz);

                if(!wiz.wasFinishPressed())

                        return;

//                try {

                        String fieldType = wiz.getFieldType();

                        String fieldLength = wiz.getFieldLength();

                        String fieldCost = wiz.getFieldCost();

                        String fieldSense = wiz.getFieldSense();

                        netBuilder.setLayer(lyr);

                        netBuilder.setFieldType(fieldType);

                        netBuilder.setFieldDist(fieldLength);

                        netBuilder.setFieldSense(fieldSense);

                        netBuilder.setFieldCost(fieldCost);

                        File redFile = new File("c:/" + lyr.getName().replaceAll("\\Q.shp\\E", ".red"));

//                        netBuilder.createRedFile(lyr, redFile);

                        GenerateRedNetworkTask task = new GenerateRedNetworkTask(lyr, redFile, netBuilder);

                        PluginServices.cancelableBackgroundExecution(task);

//                } catch (DriverException e1) {

//                        // TODO Auto-generated catch block

//                        e1.printStackTrace();

//

//                }

//                JOptionPane.showMessageDialog(null, PluginServices.getText(this, "done"));

        }

        public boolean isEnabled() {

                return true;

        }

        public boolean isVisible() {

                IWindow f = PluginServices.getMDIManager().getActiveWindow();

                if (f == null) {

                        return false;

                }

                if (f instanceof View) {

                        View vista = (View) f;

                        IProjectView model = vista.getModel();

                        MapContext mapa = model.getMapContext();

                        FLayer[] activeLayers = mapa.getLayers().getActives();

                        if (activeLayers.length > 0)

                                if (activeLayers[0] instanceof FLyrVect)

                                        return true;

                }

                return false;

        }

        public IPreference getPreferencesPage() {

                return thePreferencePage;

        }

}