| 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.io.File;

import javax.swing.ImageIcon;

import javax.swing.JOptionPane;

import com.iver.andami.PluginServices;

import com.iver.andami.messages.NotificationManager;

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.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.SingleLayerIterator;

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.wizard.NetWizard;

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

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 GenerateNetworkExtension extends Extension implements IPreferenceExtension {

        private static final IPreference thePreferencePage = new RoutePage();

        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);

                                                        return;

                                                }

                                        }

                                } catch (DriverException e) {

                                        e.printStackTrace();

                                        NotificationManager.addError(e);

                                }

                        }

                }

        }

        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 = NetworkUtils.getNetworkFile(lyr);

//                        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){

                                        FLyrVect lyrVect = (FLyrVect) activeLayers[0];

                                        int shapeType ;

                                        try {

                                                shapeType = lyrVect.getShapeType();

                                                if (shapeType == FShape.LINE)

                                                        return true;

                                        } catch (DriverException e) {

                                                // TODO Auto-generated catch block

                                                e.printStackTrace();

                                        }

                                }        

                }

                return false;

        }

        public IPreference getPreferencesPage() {

                return thePreferencePage;

        }

}