wamblee.org Git - utils/blob - gps/src/main/java/org/wamblee/gpx/GpxPlotter.java

   1 /*
   2  * Copyright 2006 the original author or authors.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  *      http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16
  17 package org.wamblee.gpx;
  18
  19 import java.awt.Color;
  20 import java.awt.Image;
  21 import java.io.File;
  22 import java.io.FileInputStream;
  23 import java.io.FileOutputStream;
  24 import java.io.IOException;
  25 import java.util.ArrayList;
  26 import java.util.List;
  27
  28 import org.jfree.chart.ChartFactory;
  29 import org.jfree.chart.ChartFrame;
  30 import org.jfree.chart.ChartUtilities;
  31 import org.jfree.chart.JFreeChart;
  32 import org.jfree.chart.axis.NumberAxis;
  33 import org.jfree.chart.plot.PlotOrientation;
  34 import org.jfree.chart.plot.XYPlot;
  35 import org.jfree.chart.renderer.xy.XYItemRenderer;
  36 import org.jfree.chart.renderer.xy.XYLineAndShapeRenderer;
  37 import org.jfree.data.xy.XYSeries;
  38 import org.jfree.data.xy.XYSeriesCollection;
  39 import org.wamblee.general.Pair;
  40 import org.wamblee.gps.geometry.Plane;
  41 import org.wamblee.gps.geometry.Point;
  42 import org.wamblee.gps.geometry.ReferenceCoordinateSystem;
  43 import org.wamblee.gps.track.TrackSegment;
  44 import org.wamblee.utils.JpegUtils;
  45
  46 /**
  47  * Parses a GPX file and prints out a data file with each trackpoints distance from the start of the
  48  * track and its elevation, separated 0by a space.
  49  *
  50  * @author Erik Brakkee
  51  */
  52 public class GpxPlotter {
  53
  54     public static void main(String[] aArgs) throws Exception {
  55         File file = new File(aArgs[0]);
  56         GpxParser parser = new GpxParser();
  57         TrackSegment track = parser.parse(file.getName(), new FileInputStream(file));
  58
  59         TrackStatistics profile = new TrackStatistics(track);
  60         profile.writeHeightProfilePng(new FileOutputStream("x.png"), 600, 300);
  61
  62         List<Pair<Double,Double>> elevationProfile = computeElevationProfile(track);
  63         printTrack(elevationProfile);
  64         computeTotalClimb(elevationProfile);
  65         plotElevationProfile(elevationProfile);
  66         List<Pair<Double,Double>> trackXy = computeTrackXY(track);
  67         List<Pair<Double,Double>> trackLatLon = computeTrackLatLon(track);
  68         plotTrack(trackLatLon);
  69     }
  70
  71     private static List<Pair<Double, Double>> computeElevationProfile(TrackSegment aTrack) {
  72         List<Pair<Double,Double>> results = new ArrayList<Pair<Double,Double>>();
  73         double distance = 0.0;
  74         for (int i = 0; i < aTrack.size(); i++) {
  75             Point point = aTrack.getPoint(i);
  76             results.add(new Pair<Double,Double>(distance, point.getCoordinates().getX3()));
  77             if ( i+1 < aTrack.size()) {
  78                 Point nextPoint = aTrack.getPoint(i+1);
  79                 distance += ReferenceCoordinateSystem.distance(point, nextPoint);
  80             }
  81         }
  82         return results;
  83     }
  84
  85     private static List<Pair<Double, Double>> computeTrackXY(TrackSegment aTrack) {
  86         Point reference = aTrack.getPoint(0);
  87         Plane plane = new Plane(reference, reference); // assume the earth is spherical.
  88         List<Pair<Double,Double>> results = new ArrayList<Pair<Double,Double>>();
  89         for (int i = 0; i < aTrack.size(); i++) {
  90             Point point = aTrack.getPoint(i);
  91             Pair<Double,Double> projection = plane.normalizedProjection(point);
  92             results.add(projection);
  93             System.out.println(point);
  94         }
  95         return results;
  96     }
  97
  98     private static List<Pair<Double, Double>> computeTrackLatLon(TrackSegment aTrack) {
  99         List<Pair<Double,Double>> results = new ArrayList<Pair<Double,Double>>();
 100         for (int i = 0; i < aTrack.size(); i++) {
 101             Point point = aTrack.getPoint(i);
 102             results.add(new Pair<Double,Double>(point.getCoordinates().getX1(), point.getCoordinates().getX2()));
 103         }
 104         return results;
 105     }
 106
 107
 108
 109     private static void printTrack(List<Pair<Double,Double>> aHeightProfile) {
 110        for (Pair<Double,Double> point: aHeightProfile) {
 111            System.out.println(point.getFirst() + " " + point.getSecond());
 112        }
 113     }
 114
 115     private static void computeTotalClimb(List<Pair<Double,Double>> aHeightProfile) {
 116         double result = 0.0;
 117
 118         double lastHeight = aHeightProfile.get(0).getSecond();
 119         for ( int i = 1; i < aHeightProfile.size(); i++) {
 120             double height = aHeightProfile.get(i).getSecond();
 121             if ( height > lastHeight) {
 122                 result += (height-lastHeight);
 123             }
 124             lastHeight = height;
 125         }
 126         System.out.println("Total climb: " + result);
 127     }
 128
 129     private static void plotElevationProfile(List<Pair<Double,Double>> aHeightProfile) throws IOException {
 130         XYSeriesCollection dataset = createDataset(aHeightProfile, "height");
 131         JFreeChart chart = ChartFactory.createXYLineChart(
 132                 "Height Profile",
 133                 "Distance(m)",
 134                 "Height(m)",
 135                 dataset,
 136                 PlotOrientation.VERTICAL,
 137                 true,
 138                 true,
 139                 false);
 140         ChartUtilities.writeChartAsPNG(new FileOutputStream("height.png"), chart, 600, 300);
 141         ChartFrame frame = new ChartFrame("test", chart);
 142         frame.pack();
 143         frame.setVisible(true);
 144     }
 145
 146     private static void plotTrack(List<Pair<Double,Double>> aPoints) throws IOException, InterruptedException {
 147         XYSeriesCollection dataset = createDataset(aPoints, "track");
 148         JFreeChart chart = createLineChart(dataset);
 149
 150         Pair<Pair<Double,Double>,Pair<Double,Double>> bounds = getBounds(aPoints);
 151
 152         chart.getXYPlot().getDomainAxis().setLowerBound(bounds.getFirst().getFirst());
 153         chart.getXYPlot().getDomainAxis().setUpperBound(bounds.getFirst().getSecond());
 154
 155         chart.getXYPlot().getRangeAxis().setLowerBound(bounds.getSecond().getFirst());
 156         chart.getXYPlot().getRangeAxis().setUpperBound(bounds.getSecond().getSecond());
 157
 158         Image background = JpegUtils.loadJpegImage(new FileInputStream("/home/erik/vakantie.jpg"));
 159         chart.getPlot().setBackgroundImage(background);
 160
 161         XYLineAndShapeRenderer renderer = (XYLineAndShapeRenderer)chart.getXYPlot().getRenderer();
 162         renderer.setShapesVisible(true);
 163         renderer.setShapesFilled(true);
 164         renderer.setPaint(Color.BLACK);
 165
 166         ChartUtilities.writeChartAsPNG(new FileOutputStream("test.png"), chart, 1280, 800);
 167         ChartFrame frame = new ChartFrame("test", chart);
 168         frame.pack();
 169         frame.setVisible(true);
 170     }
 171
 172     /**
 173      * @param dataset
 174      * @return
 175      */
 176     private static JFreeChart createLineChart(XYSeriesCollection dataset) {
 177         NumberAxis xAxis = new NumberAxis("S->N");
 178         xAxis.setAutoRangeIncludesZero(false);
 179
 180         NumberAxis yAxis = new NumberAxis("W->E");
 181         yAxis.setAutoRangeIncludesZero(false);
 182
 183         XYItemRenderer renderer = new XYLineAndShapeRenderer(true, false);
 184         XYPlot plot = new ZoomableBackgroundXYPlot(dataset, xAxis, yAxis, renderer);
 185         plot.setOrientation(PlotOrientation.HORIZONTAL);
 186
 187         JFreeChart chart = new JFreeChart(
 188             "Track", JFreeChart.DEFAULT_TITLE_FONT, plot, true
 189         );
 190
 191         return chart;
 192         /*
 193         JFreeChart chart = ChartFactory.createXYLineChart(
 194                 "Track",
 195                 "S->N",
 196                 "W->E",
 197                 dataset,
 198                 PlotOrientation.HORIZONTAL,
 199                 true,
 200                 true,
 201                 false);
 202         return chart;
 203         */
 204     }
 205
 206     /**
 207      * @param aHeightProfile
 208      * @return
 209      */
 210     private static XYSeriesCollection createDataset(List<Pair<Double, Double>> aHeightProfile, String aName) {
 211         XYSeries series = new XYSeries(aName, false);
 212         for (Pair<Double,Double> point: aHeightProfile) {
 213             series.add(point.getFirst(), point.getSecond());
 214         }
 215         XYSeriesCollection dataset = new XYSeriesCollection(series);
 216         return dataset;
 217     }
 218
 219     private static Pair<Pair<Double,Double>,Pair<Double,Double>> getBounds(List<Pair<Double,Double>> aList) {
 220         Pair<Double,Double> first = aList.get(0);
 221         double minx= first.getFirst();
 222         double maxx = minx;
 223         double miny = first.getSecond();
 224         double maxy = miny;
 225
 226         for (int i = 0; i < aList.size(); i++) {
 227             Pair<Double,Double> value = aList.get(i);
 228             minx = Math.min(minx, value.getFirst());
 229             maxx = Math.max(maxx, value.getFirst());
 230             miny = Math.min(miny, value.getSecond());
 231             maxy = Math.max(maxy, value.getSecond());
 232         }
 233         if ( maxx == minx ) {
 234             maxx += 1.0; // to avoid problems.
 235         }
 236         if ( maxy == miny ) {
 237             maxy += 1.0; // to avoid problems.
 238         }
 239         final double paddingFactor = 0.3; // allow some space around min and max
 240         return new Pair<Pair<Double,Double>,Pair<Double,Double>>(
 241                 new Pair<Double,Double>( minx - paddingFactor*(maxx-minx),
 242                                          maxx + paddingFactor*(maxx-minx)),
 243                 new Pair<Double,Double>( miny - paddingFactor*(maxy-miny),
 244                                          maxy + paddingFactor*(maxy-miny))
 245                 );
 246     }
 247 }
 248