X-Git-Url: http://wamblee.org/gitweb/?a=blobdiff_plain;f=gps%2Fsrc%2Forg%2Fwamblee%2Fgpx%2FGpxPlotter.java;h=f239cabf4a3b7aadc01807f000f0ea0caf45b3dd;hb=157ef3a8ca2bbb7b242d68b769777423c03105b2;hp=8cb03a191d24f8a8e545cbcb36d7fbfdc9949e25;hpb=5b16b58e1c14bfe168cd3e1a5e63ef5f58e525e6;p=utils diff --git a/gps/src/org/wamblee/gpx/GpxPlotter.java b/gps/src/org/wamblee/gpx/GpxPlotter.java index 8cb03a19..f239cabf 100644 --- a/gps/src/org/wamblee/gpx/GpxPlotter.java +++ b/gps/src/org/wamblee/gpx/GpxPlotter.java @@ -16,71 +16,51 @@ package org.wamblee.gpx; +import java.awt.Color; +import java.awt.Image; import java.io.File; import java.io.FileInputStream; -import java.io.FileNotFoundException; import java.io.FileOutputStream; import java.io.IOException; import java.util.ArrayList; -import java.util.Iterator; import java.util.List; -import javax.xml.parsers.ParserConfigurationException; - -import org.dom4j.Document; -import org.dom4j.Element; import org.jfree.chart.ChartFactory; import org.jfree.chart.ChartFrame; import org.jfree.chart.ChartUtilities; import org.jfree.chart.JFreeChart; +import org.jfree.chart.axis.NumberAxis; import org.jfree.chart.plot.PlotOrientation; +import org.jfree.chart.plot.XYPlot; +import org.jfree.chart.renderer.xy.XYItemRenderer; +import org.jfree.chart.renderer.xy.XYLineAndShapeRenderer; import org.jfree.data.xy.XYSeries; import org.jfree.data.xy.XYSeriesCollection; import org.wamblee.general.Pair; -import org.wamblee.xml.DomUtils; -import org.xml.sax.SAXException; +import org.wamblee.gps.geometry.Plane; +import org.wamblee.gps.geometry.Point; +import org.wamblee.gps.geometry.ReferenceCoordinateSystem; +import org.wamblee.gps.track.Track; +import org.wamblee.utils.JpegUtils; /** * Parses a GPX file and prints out a data file with each trackpoints distance from the start of the - * track and its elevation, separated by a space. + * track and its elevation, separated 0by a space. */ public class GpxPlotter { - public static void main(String[] aArgs) throws ParserConfigurationException, FileNotFoundException, SAXException, IOException { - File file = new File(aArgs[0]); - Document doc = DomUtils.convert(DomUtils.read(new FileInputStream(file))); - - Track track = parseTrack(doc); + public static void main(String[] aArgs) throws Exception { + File file = new File(aArgs[0]); + GpxParser parser = new GpxParser(); + Track track = parser.parse(new FileInputStream(file)); List> elevationProfile = computeElevationProfile(track); printTrack(elevationProfile); computeTotalClimb(elevationProfile); plotElevationProfile(elevationProfile); - } - - /** - * @param doc - */ - private static Track parseTrack(Document doc) { - Track track = new Track(); - Element root = doc.getRootElement().element("trk").element("trkseg"); - for ( Iterator i =root.elementIterator("trkpt"); i.hasNext(); ) { - Element trkpt = (Element)i.next(); - track.addPoint(parseTrackPoint(trkpt)); - } - return track; - } - - /** - * @param trkpt - */ - private static TrackPoint parseTrackPoint(Element trkpt) { - //System.out.println(trkpt.asXML() + "|\n"); - double latitude = new Double(trkpt.attributeValue("lat")); - double longitude = new Double(trkpt.attributeValue("lon")); - double elevation = new Double(trkpt.elementText("ele")); - //System.out.println(" lat = " + lat + " lon = " + lon + " ele = " + ele); - return new TrackPoint(latitude, longitude, elevation); + List> trackXy = computeTrackXY(track); + List> trackLatLon = computeTrackLatLon(track); + plotTrack(trackLatLon); } private static List> computeElevationProfile(Track aTrack) { @@ -97,6 +77,30 @@ public class GpxPlotter { return results; } + private static List> computeTrackXY(Track aTrack) { + Point reference = aTrack.getPoint(0); + Plane plane = new Plane(reference, reference); // assume the earth is spherical. + List> results = new ArrayList>(); + for (int i = 0; i < aTrack.size(); i++) { + Point point = aTrack.getPoint(i); + Pair projection = plane.normalizedProjection(point); + results.add(projection); + System.out.println(point); + } + return results; + } + + private static List> computeTrackLatLon(Track aTrack) { + List> results = new ArrayList>(); + for (int i = 0; i < aTrack.size(); i++) { + Point point = aTrack.getPoint(i); + results.add(new Pair(point.getCoordinates().getX1(), point.getCoordinates().getX2())); + } + return results; + } + + + private static void printTrack(List> aHeightProfile) { for (Pair point: aHeightProfile) { System.out.println(point.getFirst() + " " + point.getSecond()); @@ -118,11 +122,7 @@ public class GpxPlotter { } private static void plotElevationProfile(List> aHeightProfile) throws IOException { - XYSeries series = new XYSeries("height"); - for (Pair point: aHeightProfile) { - series.add(point.getFirst(), point.getSecond()); - } - XYSeriesCollection dataset = new XYSeriesCollection(series); + XYSeriesCollection dataset = createDataset(aHeightProfile, "height"); JFreeChart chart = ChartFactory.createXYLineChart( "Height Profile", "Distance(m)", @@ -132,11 +132,112 @@ public class GpxPlotter { true, true, false); - ChartUtilities.writeChartAsPNG(new FileOutputStream("test.png"), chart, 600, 300); + ChartUtilities.writeChartAsPNG(new FileOutputStream("height.png"), chart, 600, 300); + ChartFrame frame = new ChartFrame("test", chart); + frame.pack(); + frame.setVisible(true); + } + + private static void plotTrack(List> aPoints) throws IOException, InterruptedException { + XYSeriesCollection dataset = createDataset(aPoints, "track"); + JFreeChart chart = createLineChart(dataset); + + Pair,Pair> bounds = getBounds(aPoints); + + chart.getXYPlot().getDomainAxis().setLowerBound(bounds.getFirst().getFirst()); + chart.getXYPlot().getDomainAxis().setUpperBound(bounds.getFirst().getSecond()); + + chart.getXYPlot().getRangeAxis().setLowerBound(bounds.getSecond().getFirst()); + chart.getXYPlot().getRangeAxis().setUpperBound(bounds.getSecond().getSecond()); + + Image background = JpegUtils.loadJpegImage(new FileInputStream("/home/erik/vakantie.jpg")); + chart.getPlot().setBackgroundImage(background); + + XYLineAndShapeRenderer renderer = (XYLineAndShapeRenderer)chart.getXYPlot().getRenderer(); + renderer.setShapesVisible(true); + renderer.setShapesFilled(true); + renderer.setPaint(Color.BLACK); + + ChartUtilities.writeChartAsPNG(new FileOutputStream("test.png"), chart, 1280, 800); ChartFrame frame = new ChartFrame("test", chart); frame.pack(); frame.setVisible(true); } + /** + * @param dataset + * @return + */ + private static JFreeChart createLineChart(XYSeriesCollection dataset) { + NumberAxis xAxis = new NumberAxis("S->N"); + xAxis.setAutoRangeIncludesZero(false); + + NumberAxis yAxis = new NumberAxis("W->E"); + yAxis.setAutoRangeIncludesZero(false); + + XYItemRenderer renderer = new XYLineAndShapeRenderer(true, false); + XYPlot plot = new ZoomableBackgroundXYPlot(dataset, xAxis, yAxis, renderer); + plot.setOrientation(PlotOrientation.HORIZONTAL); + + JFreeChart chart = new JFreeChart( + "Track", JFreeChart.DEFAULT_TITLE_FONT, plot, true + ); + + return chart; + /* + JFreeChart chart = ChartFactory.createXYLineChart( + "Track", + "S->N", + "W->E", + dataset, + PlotOrientation.HORIZONTAL, + true, + true, + false); + return chart; + */ + } + + /** + * @param aHeightProfile + * @return + */ + private static XYSeriesCollection createDataset(List> aHeightProfile, String aName) { + XYSeries series = new XYSeries(aName, false); + for (Pair point: aHeightProfile) { + series.add(point.getFirst(), point.getSecond()); + } + XYSeriesCollection dataset = new XYSeriesCollection(series); + return dataset; + } + + private static Pair,Pair> getBounds(List> aList) { + Pair first = aList.get(0); + double minx= first.getFirst(); + double maxx = minx; + double miny = first.getSecond(); + double maxy = miny; + + for (int i = 0; i < aList.size(); i++) { + Pair value = aList.get(i); + minx = Math.min(minx, value.getFirst()); + maxx = Math.max(maxx, value.getFirst()); + miny = Math.min(miny, value.getSecond()); + maxy = Math.max(maxy, value.getSecond()); + } + if ( maxx == minx ) { + maxx += 1.0; // to avoid problems. + } + if ( maxy == miny ) { + maxy += 1.0; // to avoid problems. + } + final double paddingFactor = 0.3; // allow some space around min and max + return new Pair,Pair>( + new Pair( minx - paddingFactor*(maxx-minx), + maxx + paddingFactor*(maxx-minx)), + new Pair( miny - paddingFactor*(maxy-miny), + maxy + paddingFactor*(maxy-miny)) + ); + } }