Changeset 33671 in osm for applications/editors/josm/plugins/cadastre-fr/src

Timestamp:

2017-09-25T21:47:56+02:00 (7 years ago)

Author:

donvip

Message:

Edigeo: support reading tar.bz2 files directly

Location:

applications/editors/josm/plugins/cadastre-fr/src/org/openstreetmap/josm/plugins/fr/cadastre/edigeo/pci

Files:

• EdigeoPciImporter.java (modified) (4 diffs)
• EdigeoPciReader.java (modified) (4 diffs)

applications/editors/josm/plugins/cadastre-fr/src/org/openstreetmap/josm/plugins/fr/cadastre/edigeo/pci/EdigeoPciImporter.java

@@ -13 +13 @@
 import org.openstreetmap.josm.data.osm.DataSet;
 import org.openstreetmap.josm.gui.io.importexport.OsmImporter;
-import org.openstreetmap.josm.gui.layer.OsmDataLayer;
 import org.openstreetmap.josm.gui.progress.ProgressMonitor;
 import org.openstreetmap.josm.io.IllegalDataException;
@@ -24 +23 @@
 
     static final ExtensionFileFilter EDIGEO_FILE_FILTER = new ExtensionFileFilter(
-            "thf", "thf", tr("Cadastre Edigeo files") + " (*.thf)");
+            "thf,tar.bz2", "thf", tr("Cadastre Edigeo files") + " (*.thf, *.tar.bz2)");
 
     protected File file;
@@ -38 +37 @@
     public void importData(File file, ProgressMonitor progressMonitor)
             throws IOException, IllegalDataException {
-        if (file != null) {
-            this.file = file;
-        }
+        this.file = file;
         // Do not call super.importData because Compression.getUncompressedFileInputStream skips the first entry
         try (InputStream in = new FileInputStream(file)) {
@@ -58 +55 @@
         }
     }
-
-    @Override
-    protected OsmDataLayer createLayer(DataSet dataSet, File associatedFile, String layerName) {
-        // FIXME: mapping pci => osm
-        //DataSetUpdater.updateDataSet(dataSet, handler, associatedFile);
-        return new OsmDataLayer(dataSet, layerName, associatedFile);
-    }
 }

applications/editors/josm/plugins/cadastre-fr/src/org/openstreetmap/josm/plugins/fr/cadastre/edigeo/pci/EdigeoPciReader.java

@@ -2 +2 @@
 package org.openstreetmap.josm.plugins.fr.cadastre.edigeo.pci;
 
+import java.io.BufferedInputStream;
 import java.io.File;
+import java.io.FileOutputStream;
 import java.io.IOException;
 import java.io.InputStream;
+import java.nio.file.Files;
 import java.nio.file.Path;
 import java.util.Arrays;
@@ -12 +15 @@
 import java.util.Map.Entry;
 
+import org.apache.commons.compress.archivers.tar.TarArchiveEntry;
+import org.apache.commons.compress.archivers.tar.TarArchiveInputStream;
+import org.apache.commons.compress.compressors.bzip2.BZip2CompressorInputStream;
+import org.apache.commons.compress.utils.IOUtils;
 import org.openstreetmap.josm.data.osm.DataSet;
 import org.openstreetmap.josm.data.osm.DataSet.UploadPolicy;
@@ -20 +27 @@
 import org.openstreetmap.josm.plugins.fr.cadastre.edigeo.EdigeoFileTHF;
 import org.openstreetmap.josm.plugins.fr.cadastre.edigeo.EdigeoFileVEC;
+import org.openstreetmap.josm.tools.Utils;
 
 /**
@@ -180 +188 @@
 
     DataSet parse(Path path, ProgressMonitor instance) throws IOException, ReflectiveOperationException {
-        DataSet data = new DataSet();
-        data.setUploadPolicy(UploadPolicy.DISCOURAGED);
-        EdigeoFileTHF thf = new EdigeoFileTHF(path).read().fill(data);
-        data.setName(thf.getSupport().getBlockIdentifier());
-        return data;
+        Path tmpDir = null;
+        Path thfPath = path;
+        try {
+            if (thfPath.toString().endsWith(".tar.bz2")) {
+                try (InputStream fin = Files.newInputStream(path);
+                     BufferedInputStream in = new BufferedInputStream(fin);
+                     BZip2CompressorInputStream bzIn = new BZip2CompressorInputStream(in);
+                     TarArchiveInputStream tar = new TarArchiveInputStream(bzIn)
+                ) {
+                    TarArchiveEntry entry;
+                    tmpDir = Files.createTempDirectory(Utils.getJosmTempDir().toPath(), "cadastre");
+                    while ((entry = tar.getNextTarEntry()) != null) {
+                        File file = tmpDir.resolve(entry.getName()).toFile();
+                        try (FileOutputStream out = new FileOutputStream(file)) {
+                            if (IOUtils.copy(tar, out) < entry.getSize()) {
+                                throw new IOException(String.format("Unable to write ''{0}'' entirely", file));
+                            } else if (file.toString().endsWith(".THF")) {
+                                thfPath = file.toPath();
+                            }
+                        }
+                    }
+                }
+            }
+            DataSet data = new DataSet();
+            data.setUploadPolicy(UploadPolicy.DISCOURAGED);
+            EdigeoFileTHF thf = new EdigeoFileTHF(thfPath).read().fill(data);
+            data.setName(thf.getSupport().getBlockIdentifier());
+            return data;
+        } finally {
+            if (tmpDir != null) {
+                Utils.deleteDirectory(tmpDir.toFile());
+            }
+        }
     }