source: josm/trunk/src/org/openstreetmap/josm/io/GpxImporter.java @ 5241

// License: GPL. For details, see LICENSE file.
package org.openstreetmap.josm.io;

import static org.openstreetmap.josm.tools.I18n.tr;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.zip.GZIPInputStream;

import javax.swing.JOptionPane;

import org.openstreetmap.josm.Main;
import org.openstreetmap.josm.actions.ExtensionFileFilter;
import org.openstreetmap.josm.gui.layer.GpxLayer;
import org.openstreetmap.josm.gui.layer.markerlayer.MarkerLayer;
import org.openstreetmap.josm.gui.progress.ProgressMonitor;
import org.openstreetmap.josm.gui.util.GuiHelper;
import org.xml.sax.SAXException;

public class GpxImporter extends FileImporter {

    protected static class GpxImporterData {
        public GpxLayer gpxLayer;
        public MarkerLayer markerLayer;
        public Runnable postLayerTask;
    }

    public GpxImporter() {
        super(new ExtensionFileFilter("gpx,gpx.gz", "gpx", tr("GPX Files") + " (*.gpx *.gpx.gz)"));
    }

    @Override public void importData(File file, ProgressMonitor progressMonitor) throws IOException {
        InputStream is;
        if (file.getName().endsWith(".gpx.gz")) {
            is = new GZIPInputStream(new FileInputStream(file));
        } else {
            is = new FileInputStream(file);
        }
        String fileName = file.getName();
        final GpxImporterData data = loadLayers(is, file, fileName, tr("Markers from {0}", fileName), progressMonitor);

        // FIXME: remove UI stuff from the IO subsystem
        GuiHelper.runInEDT(new Runnable() {
            public void run() {
                if (data.markerLayer != null) {
                    Main.main.addLayer(data.markerLayer);
                }
                if (data.gpxLayer != null) {
                    Main.main.addLayer(data.gpxLayer);
                }
                data.postLayerTask.run();
            }
        });
    }

    public GpxImporterData loadLayers(InputStream is, final File associatedFile,
            final String gpxLayerName, String markerLayerName, ProgressMonitor progressMonitor) throws IOException {
        final GpxImporterData data = new GpxImporterData();
        try {
            final GpxReader r = new GpxReader(is);
            final boolean parsedProperly = r.parse(true);
            r.data.storageFile = associatedFile;
            if (r.data.hasRoutePoints() || r.data.hasTrackPoints()) {
                data.gpxLayer = new GpxLayer(r.data, gpxLayerName, associatedFile != null);
            }
            if (Main.pref.getBoolean("marker.makeautomarkers", true) && !r.data.waypoints.isEmpty()) {
                data.markerLayer = new MarkerLayer(r.data, markerLayerName, associatedFile, data.gpxLayer, false);
                if (data.markerLayer.data.size() == 0) {
                    data.markerLayer = null;
                }
            }
            data.postLayerTask = new Runnable() {
                @Override
                public void run() {
                    if (data.markerLayer != null) {
                        data.markerLayer.addMouseHandler();
                    }
                    if (!parsedProperly) {
                        String msg;
                        if (associatedFile == null) {
                            msg = tr("Error occurred while parsing gpx data for layer ''{0}''. Only a part of the file will be available.",
                                    gpxLayerName);
                        } else {
                            msg = tr("Error occurred while parsing gpx file ''{0}''. Only a part of the file will be available.",
                                    associatedFile.getPath());
                        }
                        JOptionPane.showMessageDialog(null, msg);
                    }
                }
            };
        } catch (SAXException e) {
            e.printStackTrace();
            throw new IOException(tr("Parsing data for layer ''{0}'' failed", gpxLayerName));
        }
        return data;
    }
}