Statistics
| Revision:

svn-gvsig-desktop / branches / v10 / extensions / extCAD / src / com / iver / cit / gvsig / ExportTo.java @ 11867

History | View | Annotate | Download (25.2 KB)

1
package com.iver.cit.gvsig;
2

    
3
import java.awt.Component;
4
import java.awt.Cursor;
5
import java.awt.geom.Point2D;
6
import java.io.File;
7
import java.io.IOException;
8
import java.nio.charset.Charset;
9
import java.sql.Connection;
10
import java.sql.DriverManager;
11
import java.sql.SQLException;
12
import java.sql.Types;
13
import java.util.Vector;
14

    
15
import javax.swing.JComponent;
16
import javax.swing.JFileChooser;
17
import javax.swing.JOptionPane;
18
import javax.swing.ProgressMonitor;
19

    
20
import org.cresques.cts.ICoordTrans;
21

    
22
import com.hardcode.driverManager.Driver;
23
import com.hardcode.driverManager.DriverLoadException;
24
import com.hardcode.gdbms.engine.values.Value;
25
import com.iver.andami.PluginServices;
26
import com.iver.andami.messages.NotificationManager;
27
import com.iver.andami.plugins.Extension;
28
import com.iver.cit.gvsig.fmap.DriverException;
29
import com.iver.cit.gvsig.fmap.MapContext;
30
import com.iver.cit.gvsig.fmap.core.DefaultFeature;
31
import com.iver.cit.gvsig.fmap.core.FShape;
32
import com.iver.cit.gvsig.fmap.core.IFeature;
33
import com.iver.cit.gvsig.fmap.core.IGeometry;
34
import com.iver.cit.gvsig.fmap.core.ShapeFactory;
35
import com.iver.cit.gvsig.fmap.core.v02.FLabel;
36
import com.iver.cit.gvsig.fmap.drivers.DBLayerDefinition;
37
import com.iver.cit.gvsig.fmap.drivers.DriverAttributes;
38
import com.iver.cit.gvsig.fmap.drivers.DriverIOException;
39
import com.iver.cit.gvsig.fmap.drivers.FieldDescription;
40
import com.iver.cit.gvsig.fmap.drivers.ILayerDefinition;
41
import com.iver.cit.gvsig.fmap.drivers.SHPLayerDefinition;
42
import com.iver.cit.gvsig.fmap.drivers.VectorialDriver;
43
import com.iver.cit.gvsig.fmap.drivers.dxf.DXFMemoryDriver;
44
import com.iver.cit.gvsig.fmap.drivers.gml.GMLDriver;
45
import com.iver.cit.gvsig.fmap.drivers.jdbc.postgis.PostGISWriter;
46
import com.iver.cit.gvsig.fmap.drivers.jdbc.postgis.PostGisDriver;
47
import com.iver.cit.gvsig.fmap.drivers.shp.IndexedShpDriver;
48
import com.iver.cit.gvsig.fmap.edition.DefaultRowEdited;
49
import com.iver.cit.gvsig.fmap.edition.EditionException;
50
import com.iver.cit.gvsig.fmap.edition.IWriter;
51
import com.iver.cit.gvsig.fmap.edition.writers.dxf.DxfFieldsMapping;
52
import com.iver.cit.gvsig.fmap.edition.writers.dxf.DxfWriter;
53
import com.iver.cit.gvsig.fmap.edition.writers.gml.GMLWriter;
54
import com.iver.cit.gvsig.fmap.edition.writers.shp.ShpWriter;
55
import com.iver.cit.gvsig.fmap.layers.FBitSet;
56
import com.iver.cit.gvsig.fmap.layers.FLayer;
57
import com.iver.cit.gvsig.fmap.layers.FLayers;
58
import com.iver.cit.gvsig.fmap.layers.FLyrAnnotation;
59
import com.iver.cit.gvsig.fmap.layers.FLyrVect;
60
import com.iver.cit.gvsig.fmap.layers.LayerFactory;
61
import com.iver.cit.gvsig.fmap.layers.ReadableVectorial;
62
import com.iver.cit.gvsig.fmap.layers.SelectableDataSource;
63
import com.iver.cit.gvsig.jdbc_spatial.DlgConnection;
64
import com.iver.cit.gvsig.jdbc_spatial.gui.jdbcwizard.ConnectionSettings;
65
import com.iver.cit.gvsig.project.documents.view.IProjectView;
66
import com.iver.cit.gvsig.project.documents.view.gui.View;
67
import com.iver.utiles.PostProcessSupport;
68
import com.iver.utiles.SimpleFileFilter;
69
import com.iver.utiles.swing.threads.AbstractMonitorableTask;
70

    
71
public class ExportTo extends Extension {
72
        private String lastPath = null;
73
        private class WriterTask extends AbstractMonitorableTask
74
        {
75
                FLyrVect lyrVect;
76
                IWriter writer;
77
                int rowCount;
78
                ReadableVectorial va;
79
                SelectableDataSource sds;
80
                FBitSet bitSet;
81
                MapContext mapContext;
82
                VectorialDriver reader;
83

    
84
                public WriterTask(MapContext mapContext, FLyrVect lyr, IWriter writer, Driver reader) throws DriverException, DriverIOException
85
                {
86
                        this.mapContext = mapContext;
87
                        this.lyrVect = lyr;
88
                        this.writer = writer;
89
                        this.reader = (VectorialDriver) reader;
90

    
91
                        setInitialStep(0);
92
                        setDeterminatedProcess(true);
93
                        setStatusMessage(PluginServices.getText(this, "exportando_features"));
94

    
95
                        va = lyrVect.getSource();
96
                        sds = lyrVect.getRecordset();
97

    
98
                        bitSet = sds.getSelection();
99

    
100
                        if (bitSet.cardinality() == 0)
101
                                rowCount = va.getShapeCount();
102
                        else
103
                                rowCount = bitSet.cardinality();
104

    
105
                        setFinalStep(rowCount);
106

    
107
                }
108
                public void run() throws Exception {
109
                        va.start();
110
                        ICoordTrans ct = lyrVect.getCoordTrans();
111
                        DriverAttributes attr = va.getDriverAttributes();
112
                        boolean bMustClone = false;
113
                        if (attr != null) {
114
                                if (attr.isLoadedInMemory()) {
115
                                        bMustClone = attr.isLoadedInMemory();
116
                                }
117
                        }
118
                        if (lyrVect instanceof FLyrAnnotation && lyrVect.getShapeType()!=FShape.POINT) {
119
                                SHPLayerDefinition lyrDef=(SHPLayerDefinition)writer.getTableDefinition();
120
                                lyrDef.setShapeType(FShape.POINT);
121
                                writer.initialize(lyrDef);
122
                        }
123

    
124
                        // Creamos la tabla.
125
                        writer.preProcess();
126

    
127
                        if (bitSet.cardinality() == 0) {
128
                                rowCount = va.getShapeCount();
129
                                for (int i = 0; i < rowCount; i++) {
130
                                        IGeometry geom = va.getShape(i);
131
                                        if (lyrVect instanceof FLyrAnnotation && geom.getGeometryType()!=FShape.POINT) {
132
                                                Point2D p=FLabel.createLabelPoint((FShape)geom.getInternalShape());
133
                                                geom=ShapeFactory.createPoint2D(p.getX(),p.getY());
134
                                        }
135
                                        if (ct != null) {
136
                                                if (bMustClone)
137
                                                        geom = geom.cloneGeometry();
138
                                                geom.reProject(ct);
139
                                        }
140
                                        reportStep();
141
                                        setNote(PluginServices.getText(this, "exporting_") + i);
142
                                        if (isCanceled())
143
                                                break;
144

    
145
                                        if (geom != null) {
146
                                                Value[] values = sds.getRow(i);
147
                                                IFeature feat = new DefaultFeature(geom, values, "" + i);
148
                                                DefaultRowEdited edRow = new DefaultRowEdited(feat,
149
                                                                DefaultRowEdited.STATUS_ADDED, i);
150
                                                writer.process(edRow);
151
                                        }
152
                                }
153
                        } else {
154
                                int counter = 0;
155
                                for (int i = bitSet.nextSetBit(0); i >= 0; i = bitSet
156
                                                .nextSetBit(i + 1)) {
157
                                        IGeometry geom = va.getShape(i);
158
                                        if (lyrVect instanceof FLyrAnnotation && geom.getGeometryType()!=FShape.POINT) {
159
                                                Point2D p=FLabel.createLabelPoint((FShape)geom.getInternalShape());
160
                                                geom=ShapeFactory.createPoint2D(p.getX(),p.getY());
161
                                        }
162
                                        if (ct != null) {
163
                                                if (bMustClone)
164
                                                        geom = geom.cloneGeometry();
165
                                                geom.reProject(ct);
166
                                        }
167
                                        reportStep();
168
                                        setNote(PluginServices.getText(this, "exporting_") + counter);
169
                                        if (isCanceled())
170
                                                break;
171

    
172
                                        if (geom != null) {
173
                                                Value[] values = sds.getRow(i);
174
                                                IFeature feat = new DefaultFeature(geom, values, "" + i);
175
                                                DefaultRowEdited edRow = new DefaultRowEdited(feat,
176
                                                                DefaultRowEdited.STATUS_ADDED, i);
177

    
178
                                                writer.process(edRow);
179
                                        }
180
                                }
181

    
182
                        }
183

    
184
                        writer.postProcess();
185
                        va.stop();
186
                        if (reader != null){
187
                                int res = JOptionPane.showConfirmDialog(
188
                                        (JComponent) PluginServices.getMDIManager().getActiveWindow()
189
                                        , PluginServices.getText(this, "insertar_en_la_vista_la_capa_creada"),
190
                                        PluginServices.getText(this,"insertar_capa"),
191
                                        JOptionPane.YES_NO_OPTION);
192

    
193
                                if (res == JOptionPane.YES_OPTION)
194
                                {
195
                                        PostProcessSupport.executeCalls();
196
                                        ILayerDefinition lyrDef = (ILayerDefinition) writer.getTableDefinition();
197
                                        FLayer newLayer = LayerFactory.createLayer(
198
                                                        lyrDef.getName(), reader, mapContext.getProjection());
199
                                        mapContext.getLayers().addLayer(newLayer);
200
                                }
201
                        }
202

    
203
                }
204

    
205
        }
206
        private class MultiWriterTask extends AbstractMonitorableTask{
207
                Vector tasks=new Vector();
208
                public void addTask(WriterTask wt) {
209
                        tasks.add(wt);
210
                }
211
                public void run() throws Exception {
212
                        for (int i = 0; i < tasks.size(); i++) {
213
                                ((WriterTask)tasks.get(i)).run();
214
                        }
215
                        tasks.clear();
216
                }
217

    
218

    
219
        }
220
        /**
221
         * @see com.iver.andami.plugins.IExtension#initialize()
222
         */
223
        public void initialize() {
224
        }
225

    
226
        /**
227
         * @see com.iver.andami.plugins.IExtension#execute(java.lang.String)
228
         */
229
        public void execute(String actionCommand) {
230
                com.iver.andami.ui.mdiManager.IWindow f = PluginServices.getMDIManager()
231
                                .getActiveWindow();
232

    
233
                if (f instanceof View) {
234
                        View vista = (View) f;
235
                        IProjectView model = vista.getModel();
236
                        MapContext mapa = model.getMapContext();
237
                        FLayers layers = mapa.getLayers();
238
                        FLayer[] actives = layers.getActives();
239
                        try {
240
                                // NOTA: SI HAY UNA SELECCI?N, SOLO SE SALVAN LOS SELECCIONADOS
241
                                for (int i = 0; i < actives.length; i++) {
242
                                        if (actives[i] instanceof FLyrVect) {
243
                                                FLyrVect lv = (FLyrVect) actives[i];
244
                                                int numSelec = lv.getRecordset().getSelection()
245
                                                                .cardinality();
246
                                                if (numSelec > 0) {
247
                                                        int resp = JOptionPane.showConfirmDialog(
248
                                                                        (Component) PluginServices.getMainFrame(),
249
                                                                        PluginServices.getText(this,"se_van_a_guardar_") + numSelec
250
                                                                                        + PluginServices.getText(this,"features_desea_continuar"),
251
                                                                        PluginServices.getText(this,"export_to"), JOptionPane.YES_NO_OPTION);
252
                                                        if (resp != JOptionPane.YES_OPTION) {
253
                                                                continue;
254
                                                        }
255
                                                } // if numSelec > 0
256
                                                if (actionCommand.equals("SHP")) {
257
                                                        saveToShp(mapa, lv);
258
                                                }
259
                                                if (actionCommand.equals("DXF")) {
260
                                                        saveToDxf(mapa, lv);
261
                                                }
262
                                                if (actionCommand.equals("POSTGIS")) {
263
                                                        saveToPostGIS(mapa, lv);
264
                                                }
265
                                                if (actionCommand.equals("GML")) {
266
                                                        saveToGml(mapa, lv);
267
                                                }
268
                                        } // actives[i]
269
                                } // for
270
                        } catch (EditionException e) {
271
                                e.printStackTrace();
272
                                NotificationManager.addError(e.getMessage(), e);
273
                        } catch (DriverException e) {
274
                                e.printStackTrace();
275
                                NotificationManager.addError(e.getMessage(), e);
276
                        } catch (DriverIOException e) {
277
                                e.printStackTrace();
278
                                NotificationManager.addError(e.getMessage(), e);
279
                        }
280

    
281
                }
282
        }
283

    
284
        public void saveToPostGIS(MapContext mapContext, FLyrVect layer) throws EditionException, DriverIOException {
285
                try {
286
                        String tableName = JOptionPane.showInputDialog(PluginServices
287
                                        .getText(this, "intro_tablename"));
288
                        if (tableName == null)
289
                                return;
290
                        tableName = tableName.toLowerCase();
291
                        DlgConnection dlg = new DlgConnection();
292
                        dlg.setModal(true);
293
                        dlg.setVisible(true);
294
                        ConnectionSettings cs = dlg.getConnSettings();
295
                        if (cs == null)
296
                                return;
297
                        Connection conex = DriverManager.getConnection(cs
298
                                        .getConnectionString(), cs.getUser(), cs.getPassw());
299

    
300
                        DBLayerDefinition dbLayerDef = new DBLayerDefinition();
301
                        // Fjp:
302
                        // Cambio: En Postgis, el nombre de cat?logo est? siempre vac?o. Es algo heredado de Oracle, que no se usa.
303
                        // dbLayerDef.setCatalogName(cs.getDb());
304
                        dbLayerDef.setCatalogName("");
305
                        
306
                        // A?adimos el schema dentro del layer definition para poder tenerlo en cuenta.
307
                        dbLayerDef.setSchema(cs.getSchema());
308
                        
309
                        dbLayerDef.setTableName(tableName);
310
                        dbLayerDef.setName(tableName);
311
                        dbLayerDef.setShapeType(layer.getShapeType());
312
                        SelectableDataSource sds = layer.getRecordset();
313
                        FieldDescription[] fieldsDescrip = sds.getFieldsDescription();
314
                        dbLayerDef.setFieldsDesc(fieldsDescrip);
315
                // Creamos el driver. OJO: Hay que a?adir el campo ID a la
316
                // definici?n de campos.
317

    
318
                boolean bFound = false;
319
                for (int i=0; i < fieldsDescrip.length; i++)
320
                {
321
                        FieldDescription f = fieldsDescrip[i];
322
                        if (f.getFieldName().equalsIgnoreCase("gid"))
323
                        {
324
                                bFound = true;
325
                                break;
326
                        }
327
                }
328
                // Si no est?, lo a?adimos
329
                if (!bFound)
330
                {
331
                        int numFieldsAnt = fieldsDescrip.length;
332
                        FieldDescription[] newFields = new FieldDescription[dbLayerDef.getFieldsDesc().length + 1];
333
                    for (int i=0; i < numFieldsAnt; i++)
334
                    {
335
                            newFields[i] = fieldsDescrip[i];
336
                    }
337
                    newFields[numFieldsAnt] = new FieldDescription();
338
                    newFields[numFieldsAnt].setFieldDecimalCount(0);
339
                    newFields[numFieldsAnt].setFieldType(Types.INTEGER);
340
                    newFields[numFieldsAnt].setFieldLength(7);
341
                    newFields[numFieldsAnt].setFieldName("gid");
342
                    dbLayerDef.setFieldsDesc(newFields);
343

    
344
                }
345

    
346

    
347

    
348
                        dbLayerDef.setFieldGeometry("the_geom");
349
                        dbLayerDef.setFieldID("gid");
350

    
351
                        dbLayerDef.setWhereClause("");
352
                        String strSRID = layer.getProjection().getAbrev().substring(5);
353
                        dbLayerDef.setSRID_EPSG(strSRID);
354
                        dbLayerDef.setConnection(conex);
355

    
356
                        PostGISWriter writer=(PostGISWriter)LayerFactory.getWM().getWriter("PostGIS Writer");
357
                        writer.setWriteAll(true);
358
                        writer.setCreateTable(true);
359
                        writer.initialize(dbLayerDef);
360
                        PostGisDriver postGISDriver=new PostGisDriver();
361
                        postGISDriver.setLyrDef(dbLayerDef);
362
                        postGISDriver.open();
363
                        PostProcessSupport.clearList();
364
                        Object[] params = new Object[2];
365
                        params[0] = conex;
366
                        params[1] = dbLayerDef;
367
                        PostProcessSupport.addToPostProcess(postGISDriver, "setData",
368
                                        params, 1);
369

    
370
                        writeFeatures(mapContext, layer, writer, postGISDriver);
371

    
372
                } catch (DriverException e) {
373
                        e.printStackTrace();
374
                        throw new EditionException(e);
375
                } catch (DriverLoadException e) {
376
                        throw new EditionException(e);
377
                } catch (SQLException e) {
378
                        throw new EditionException(e);
379
                } catch (com.hardcode.gdbms.engine.data.driver.DriverException e) {
380
                        e.printStackTrace();
381
                        throw new EditionException(e);
382
                }
383

    
384
        }
385

    
386
        /**
387
         * Lanza un thread en background que escribe las features. Cuando termina, pregunta al usuario si quiere
388
         * a?adir la nueva capa a la vista. Para eso necesita un driver de lectura ya configurado.
389
         * @param mapContext
390
         * @param layer
391
         * @param writer
392
         * @param reader
393
         * @throws DriverException
394
         * @throws DriverIOException
395
         */
396
        private void writeFeatures(MapContext mapContext, FLyrVect layer, IWriter writer, Driver reader) throws DriverException, DriverIOException
397
        {
398
                PluginServices.cancelableBackgroundExecution(new WriterTask(mapContext, layer, writer, reader));
399
        }
400
        private void writeMultiFeatures(MapContext mapContext, FLyrVect layers, IWriter[] writers, Driver[] readers) throws DriverException, DriverIOException{
401
                MultiWriterTask mwt=new MultiWriterTask();
402
                for (int i=0;i<writers.length;i++) {
403
                        mwt.addTask(new WriterTask(mapContext, layers, writers[i], readers[i]));
404
                }
405
                PluginServices.cancelableBackgroundExecution(mwt);
406
        }
407
        /**
408
         * @param layer
409
         *            FLyrVect to obtain features. If selection, only selected
410
         *            features will be precessed.
411
         * @param writer
412
         *            (Must be already initialized)
413
         * @throws EditionException
414
         * @throws DriverException
415
         * @throws DriverIOException
416
         * @throws com.hardcode.gdbms.engine.data.driver.DriverException
417
         */
418
        public void writeFeaturesNoThread(FLyrVect layer, IWriter writer)
419
                        throws EditionException, DriverException, DriverIOException,
420
                        com.hardcode.gdbms.engine.data.driver.DriverException {
421
                ReadableVectorial va = layer.getSource();
422
                SelectableDataSource sds = layer.getRecordset();
423

    
424
                // Creamos la tabla.
425
                writer.preProcess();
426

    
427
                int rowCount;
428
                FBitSet bitSet = layer.getRecordset().getSelection();
429

    
430
                if (bitSet.cardinality() == 0)
431
                        rowCount = va.getShapeCount();
432
                else
433
                        rowCount = bitSet.cardinality();
434

    
435
                ProgressMonitor progress = new ProgressMonitor(
436
                                (JComponent) PluginServices.getMDIManager().getActiveWindow(),
437
                                PluginServices.getText(this, "exportando_features"),
438
                                PluginServices.getText(this, "exportando_features"), 0,
439
                                rowCount);
440

    
441
                progress.setMillisToDecideToPopup(200);
442
                progress.setMillisToPopup(500);
443

    
444
                if (bitSet.cardinality() == 0) {
445
                        rowCount = va.getShapeCount();
446
                        for (int i = 0; i < rowCount; i++) {
447
                                IGeometry geom = va.getShape(i);
448

    
449
                                progress.setProgress(i);
450
                                if (progress.isCanceled())
451
                                        break;
452

    
453
                                if (geom != null) {
454
                                        Value[] values = sds.getRow(i);
455
                                        IFeature feat = new DefaultFeature(geom, values, "" + i);
456
                                        DefaultRowEdited edRow = new DefaultRowEdited(feat,
457
                                                        DefaultRowEdited.STATUS_ADDED, i);
458
                                        writer.process(edRow);
459
                                }
460
                        }
461
                } else {
462
                        int counter = 0;
463
                        for (int i = bitSet.nextSetBit(0); i >= 0; i = bitSet
464
                                        .nextSetBit(i + 1)) {
465
                                IGeometry geom = va.getShape(i);
466

    
467
                                progress.setProgress(counter++);
468
                                if (progress.isCanceled())
469
                                        break;
470

    
471
                                if (geom != null) {
472
                                        Value[] values = sds.getRow(i);
473
                                        IFeature feat = new DefaultFeature(geom, values, "" + i);
474
                                        DefaultRowEdited edRow = new DefaultRowEdited(feat,
475
                                                        DefaultRowEdited.STATUS_ADDED, i);
476

    
477
                                        writer.process(edRow);
478
                                }
479
                        }
480

    
481
                }
482

    
483
                writer.postProcess();
484
                progress.close();
485
        }
486

    
487
        public void saveToDxf(MapContext mapContext, FLyrVect layer) throws EditionException, DriverIOException {
488
                try {
489
                        JFileChooser jfc = new JFileChooser(lastPath);
490
                        SimpleFileFilter filterShp = new SimpleFileFilter("dxf",
491
                                        PluginServices.getText(this, "dxf_files"));
492
                        jfc.setFileFilter(filterShp);
493
                        if (jfc.showSaveDialog((Component) PluginServices.getMainFrame()) == JFileChooser.APPROVE_OPTION) {
494
                                File newFile = jfc.getSelectedFile();
495
                                String path = newFile.getAbsolutePath();
496
                                if (!(path.toLowerCase().endsWith(".dxf"))) {
497
                                        path = path + ".dxf";
498
                                }
499
                                newFile = new File(path);
500

    
501
                                DxfWriter writer = (DxfWriter) LayerFactory.getWM().getWriter(
502
                                                "DXF Writer");
503
                                SHPLayerDefinition lyrDef = new SHPLayerDefinition();
504
                                SelectableDataSource sds = layer.getRecordset();
505
                                FieldDescription[] fieldsDescrip = sds.getFieldsDescription();
506
                                lyrDef.setFieldsDesc(fieldsDescrip);
507
                                lyrDef.setFile(newFile);
508
                                lyrDef.setName(newFile.getName());
509
                                lyrDef.setShapeType(layer.getShapeType());
510
                                writer.setFile(newFile);
511
                                writer.initialize(lyrDef);
512
                                writer.setProjection(layer.getProjection());
513
                                DxfFieldsMapping fieldsMapping = new DxfFieldsMapping();
514
                                // TODO: Recuperar aqu? los campos del cuadro de di?logo.
515
                                writer.setFieldMapping(fieldsMapping);
516
                                DXFMemoryDriver dxfDriver=new DXFMemoryDriver();
517
                                dxfDriver.open(newFile);
518
                                writeFeatures(mapContext, layer, writer, dxfDriver);
519
                                String fileName = newFile.getAbsolutePath();
520
                                lastPath  = fileName.substring(0, fileName.lastIndexOf(File.separatorChar));
521
                        }
522

    
523
                } catch (DriverException e) {
524
                        e.printStackTrace();
525
                        throw new EditionException(e);
526
                } catch (com.hardcode.gdbms.engine.data.driver.DriverException e) {
527
                        e.printStackTrace();
528
                        throw new EditionException(e);
529
                } catch (IOException e) {
530
                        // TODO Auto-generated catch block
531
                        e.printStackTrace();
532
                }
533

    
534
        }
535

    
536
        public void saveToShp(MapContext mapContext, FLyrVect layer) throws EditionException, DriverIOException {
537
                try {
538
                        JFileChooser jfc = new JFileChooser();
539
                        SimpleFileFilter filterShp = new SimpleFileFilter("shp",
540
                                        PluginServices.getText(this, "shp_files"));
541
                        jfc.setFileFilter(filterShp);
542
                        if (jfc.showSaveDialog((Component) PluginServices.getMainFrame()) == JFileChooser.APPROVE_OPTION) {
543
                                File newFile = jfc.getSelectedFile();
544
                                String path = newFile.getAbsolutePath();
545
                                if( newFile.exists()){
546
                                        int resp = JOptionPane.showConfirmDialog(
547
                                                        (Component) PluginServices.getMainFrame(),PluginServices.getText(this,"fichero_ya_existe_seguro_desea_guardarlo"),
548
                                                        PluginServices.getText(this,"guardar"), JOptionPane.YES_NO_OPTION);
549
                                        if (resp != JOptionPane.YES_OPTION) {
550
                                                return;
551
                                        }
552
                                }
553
                                if (!(path.toLowerCase().endsWith(".shp"))) {
554
                                        path = path + ".shp";
555
                                }
556
                                newFile = new File(path);
557

    
558

    
559

    
560
                                SelectableDataSource sds = layer.getRecordset();
561
                                FieldDescription[] fieldsDescrip = sds.getFieldsDescription();
562

    
563
                                if (layer.getShapeType() == FShape.MULTI) // Exportamos a 3
564
                                // ficheros
565
                                {
566
                                        ShpWriter writer1 = (ShpWriter) LayerFactory.getWM().getWriter(
567
                                        "Shape Writer");
568
                                        Driver[] drivers=new Driver[3];
569
                                        ShpWriter[] writers=new ShpWriter[3];
570

    
571
                                        // puntos
572
                                        String auxPoint = path.replaceFirst(".shp", "_points.shp");
573

    
574
                                        SHPLayerDefinition lyrDefPoint = new SHPLayerDefinition();
575
                                        lyrDefPoint.setFieldsDesc(fieldsDescrip);
576
                                        File filePoints = new File(auxPoint);
577
                                        lyrDefPoint.setFile(filePoints);
578
                                        lyrDefPoint.setName(filePoints.getName());
579
                                        lyrDefPoint.setShapeType(FShape.POINT);
580
                                        writer1.setFile(filePoints);
581
                                        writer1.initialize(lyrDefPoint);
582
                                        writers[0]=writer1;
583
                                        drivers[0]=getOpenShpDriver(filePoints);
584
                                        //drivers[0]=null;
585

    
586

    
587

    
588
                                        ShpWriter writer2 = (ShpWriter) LayerFactory.getWM().getWriter(
589
                                        "Shape Writer");
590
                                        // Lineas
591
                                        String auxLine = path.replaceFirst(".shp", "_line.shp");
592
                                        SHPLayerDefinition lyrDefLine = new SHPLayerDefinition();
593
                                        lyrDefLine.setFieldsDesc(fieldsDescrip);
594

    
595
                                        File fileLines = new File(auxLine);
596
                                        lyrDefLine.setFile(fileLines);
597
                                        lyrDefLine.setName(fileLines.getName());
598
                                        lyrDefLine.setShapeType(FShape.LINE);
599
                                        writer2.setFile(fileLines);
600
                                        writer2.initialize(lyrDefLine);
601
                                        writers[1]=writer2;
602
                                        drivers[1]=getOpenShpDriver(fileLines);
603
                                        //drivers[1]=null;
604

    
605
                                        ShpWriter writer3 = (ShpWriter) LayerFactory.getWM().getWriter(
606
                                        "Shape Writer");
607
                                        // Pol?gonos
608
                                        String auxPolygon = path.replaceFirst(".shp", "_polygons.shp");
609
                                        SHPLayerDefinition lyrDefPolygon = new SHPLayerDefinition();
610
                                        lyrDefPolygon.setFieldsDesc(fieldsDescrip);
611
                                        File filePolygons = new File(auxPolygon);
612
                                        lyrDefPolygon.setFile(filePolygons);
613
                                        lyrDefPolygon.setName(filePolygons.getName());
614
                                        lyrDefPolygon.setShapeType(FShape.POLYGON);
615
                                        writer3.setFile(filePolygons);
616
                                        writer3.initialize(lyrDefPolygon);
617
                                        writers[2]=writer3;
618
                                        drivers[2]=getOpenShpDriver(filePolygons);
619
                                        //drivers[2]=null;
620

    
621
                                        writeMultiFeatures(mapContext,layer, writers, drivers);
622
                                } else {
623
                                        ShpWriter writer = (ShpWriter) LayerFactory.getWM().getWriter(
624
                                                "Shape Writer");
625
                                        IndexedShpDriver drv = getOpenShpDriver(newFile);
626
                                        SHPLayerDefinition lyrDef = new SHPLayerDefinition();
627
                                        lyrDef.setFieldsDesc(fieldsDescrip);
628
                                        lyrDef.setFile(newFile);
629
                                        lyrDef.setName(newFile.getName());
630
                                        lyrDef.setShapeType(layer.getShapeType());
631
                                        writer.setFile(newFile);
632
                                        writer.initialize(lyrDef);
633
                                        // CODIGO PARA EXPORTAR UN SHP A UN CHARSET DETERMINADO
634
                                        // ES UTIL PARA QUE UN DBF SE VEA CORRECTAMENTE EN EXCEL, POR EJEMPLO
635
//                                        Charset resul = (Charset) JOptionPane.showInputDialog((Component)PluginServices.getMDIManager().getActiveWindow(),
636
//                                                                PluginServices.getText(ExportTo.class, "select_charset_for_writing"),
637
//                                                                "Charset", JOptionPane.QUESTION_MESSAGE, null,
638
//                                                                Charset.availableCharsets().values().toArray(),
639
//                                                                writer.getCharsetForWriting().displayName());
640
//                                        if (resul == null)
641
//                                                return;
642
//                                        Charset charset = resul;
643
//                                        writer.setCharsetForWriting(charset);
644
                                        writeFeatures(mapContext, layer, writer, drv);
645

    
646
                                }
647
                        }
648
                } catch (DriverException e) {
649
                        e.printStackTrace();
650
                        throw new EditionException(e);
651
                } catch (com.hardcode.gdbms.engine.data.driver.DriverException e) {
652
                        e.printStackTrace();
653
                        throw new EditionException(e);
654
                }
655
//                catch (IOException e) {
656
//                        e.printStackTrace();
657
//                        throw new EditionException(e);
658
//                }
659
                catch (IOException e) {
660
                        // TODO Auto-generated catch block
661
                        e.printStackTrace();
662
                }
663

    
664
        }
665
        /**
666
         * This method saves a layer to GML
667
         * @param mapContext
668
         * @param layer
669
         * @throws EditionException
670
         * @throws DriverIOException
671
         */
672
        public void saveToGml(MapContext mapContext, FLyrVect layer) throws EditionException, DriverIOException {
673
                try {
674
                        JFileChooser jfc = new JFileChooser();
675
                        SimpleFileFilter filterShp = new SimpleFileFilter("gml",
676
                                        PluginServices.getText(this, "gml_files"));
677
                        jfc.setFileFilter(filterShp);
678
                        if (jfc.showSaveDialog((Component) PluginServices.getMainFrame()) == JFileChooser.APPROVE_OPTION) {
679
                                File newFile = jfc.getSelectedFile();
680
                                String path = newFile.getAbsolutePath();
681
                                if (!(path.toLowerCase().endsWith(".gml"))) {
682
                                        path = path + ".gml";
683
                                }
684
                                newFile = new File(path);
685

    
686
                                GMLWriter writer = (GMLWriter)LayerFactory.getWM().getWriter("GML Writer");
687

    
688
                                SHPLayerDefinition lyrDef = new SHPLayerDefinition();
689
                                SelectableDataSource sds = layer.getRecordset();
690
                                FieldDescription[] fieldsDescrip = sds.getFieldsDescription();
691
                                lyrDef.setFieldsDesc(fieldsDescrip);
692
                                lyrDef.setName(newFile.getName());
693
                                lyrDef.setShapeType(layer.getShapeType());
694

    
695
                                writer.setFile(newFile);
696
                                writer.setSchema(lyrDef);
697
                                writer.setBoundedBy(layer.getFullExtent(),layer.getProjection());
698
                                writer.initialize(lyrDef);
699
                                GMLDriver gmlDriver=new GMLDriver();
700
                                gmlDriver.open(newFile);
701
                                writeFeatures(mapContext, layer, writer, gmlDriver);
702
                        }
703

    
704
                } catch (DriverException e) {
705
                        e.printStackTrace();
706
                        throw new EditionException(e);
707
                } catch (com.hardcode.gdbms.engine.data.driver.DriverException e) {
708
                        e.printStackTrace();
709
                        throw new EditionException(e);
710
                }
711
//                catch (IOException e) {
712
//                        e.printStackTrace();
713
//                        throw new EditionException(e);
714
//                }
715
                catch (IOException e) {
716
                        e.printStackTrace();
717
                }
718

    
719
        }
720
        private IndexedShpDriver getOpenShpDriver(File fileShp) throws IOException {
721
                IndexedShpDriver drv = new IndexedShpDriver();
722
                if (!fileShp.exists()) {
723
                        fileShp.createNewFile();
724
                        File newFileSHX=new File(fileShp.getAbsolutePath().replaceAll("[.]shp",".shx"));
725
                        newFileSHX.createNewFile();
726
                        File newFileDBF=new File(fileShp.getAbsolutePath().replaceAll("[.]shp",".dbf"));
727
                        newFileDBF.createNewFile();
728
                }
729
                drv.open(fileShp);
730
                return drv;
731
        }
732
        /**
733
         * @see com.iver.andami.plugins.IExtension#isEnabled()
734
         */
735
        public boolean isEnabled() {
736
                int status = EditionUtilities.getEditionStatus();
737
                if (( status == EditionUtilities.EDITION_STATUS_ONE_VECTORIAL_LAYER_ACTIVE || status == EditionUtilities.EDITION_STATUS_ONE_VECTORIAL_LAYER_ACTIVE_AND_EDITABLE)
738
                                || (status == EditionUtilities.EDITION_STATUS_MULTIPLE_VECTORIAL_LAYER_ACTIVE)|| (status == EditionUtilities.EDITION_STATUS_MULTIPLE_VECTORIAL_LAYER_ACTIVE_AND_EDITABLE))
739
                {
740
                        return true;
741
                }
742
                return false;
743
        }
744

    
745
        /**
746
         * @see com.iver.andami.plugins.IExtension#isVisible()
747
         */
748
        public boolean isVisible() {
749
                com.iver.andami.ui.mdiManager.IWindow f = PluginServices.getMDIManager()
750
                                .getActiveWindow();
751

    
752
                if (f == null) {
753
                        return false;
754
                }
755

    
756
                if (f instanceof View)
757
                        return true;
758
                return false;
759
        }
760

    
761
}