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.Track;
  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         Track track = parser.parse(new FileInputStream(file));
  58
  59         List<Pair<Double,Double>> elevationProfile = computeElevationProfile(track);
  60         printTrack(elevationProfile);
  61         computeTotalClimb(elevationProfile);
  62         plotElevationProfile(elevationProfile);
  63         List<Pair<Double,Double>> trackXy = computeTrackXY(track);
  64         List<Pair<Double,Double>> trackLatLon = computeTrackLatLon(track);
  65         plotTrack(trackLatLon);
  66     }
  67
  68     private static List<Pair<Double, Double>> computeElevationProfile(Track aTrack) {
  69         List<Pair<Double,Double>> results = new ArrayList<Pair<Double,Double>>();
  70         double distance = 0.0;
  71         for (int i = 0; i < aTrack.size(); i++) {
  72             Point point = aTrack.getPoint(i);
  73             results.add(new Pair<Double,Double>(distance, point.getCoordinates().getX3()));
  74             if ( i+1 < aTrack.size()) {
  75                 Point nextPoint = aTrack.getPoint(i+1);
  76                 distance += ReferenceCoordinateSystem.distance(point, nextPoint);
  77             }
  78         }
  79         return results;
  80     }
  81
  82     private static List<Pair<Double, Double>> computeTrackXY(Track aTrack) {
  83         Point reference = aTrack.getPoint(0);
  84         Plane plane = new Plane(reference, reference); // assume the earth is spherical.
  85         List<Pair<Double,Double>> results = new ArrayList<Pair<Double,Double>>();
  86         for (int i = 0; i < aTrack.size(); i++) {
  87             Point point = aTrack.getPoint(i);
  88             Pair<Double,Double> projection = plane.normalizedProjection(point);
  89             results.add(projection);
  90             System.out.println(point);
  91         }
  92         return results;
  93     }
  94
  95     private static List<Pair<Double, Double>> computeTrackLatLon(Track aTrack) {
  96         List<Pair<Double,Double>> results = new ArrayList<Pair<Double,Double>>();
  97         for (int i = 0; i < aTrack.size(); i++) {
  98             Point point = aTrack.getPoint(i);
  99             results.add(new Pair<Double,Double>(point.getCoordinates().getX1(), point.getCoordinates().getX2()));
 100         }
 101         return results;
 102     }
 103
 104
 105
 106     private static void printTrack(List<Pair<Double,Double>> aHeightProfile) {
 107        for (Pair<Double,Double> point: aHeightProfile) {
 108            System.out.println(point.getFirst() + " " + point.getSecond());
 109        }
 110     }
 111
 112     private static void computeTotalClimb(List<Pair<Double,Double>> aHeightProfile) {
 113         double result = 0.0;
 114
 115         double lastHeight = aHeightProfile.get(0).getSecond();
 116         for ( int i = 1; i < aHeightProfile.size(); i++) {
 117             double height = aHeightProfile.get(i).getSecond();
 118             if ( height > lastHeight) {
 119                 result += (height-lastHeight);
 120             }
 121             lastHeight = height;
 122         }
 123         System.out.println("Total climb: " + result);
 124     }
 125
 126     private static void plotElevationProfile(List<Pair<Double,Double>> aHeightProfile) throws IOException {
 127         XYSeriesCollection dataset = createDataset(aHeightProfile, "height");
 128         JFreeChart chart = ChartFactory.createXYLineChart(
 129                 "Height Profile",
 130                 "Distance(m)",
 131                 "Height(m)",
 132                 dataset,
 133                 PlotOrientation.VERTICAL,
 134                 true,
 135                 true,
 136                 false);
 137         ChartUtilities.writeChartAsPNG(new FileOutputStream("height.png"), chart, 600, 300);
 138         ChartFrame frame = new ChartFrame("test", chart);
 139         frame.pack();
 140         frame.setVisible(true);
 141     }
 142
 143     private static void plotTrack(List<Pair<Double,Double>> aPoints) throws IOException, InterruptedException {
 144         XYSeriesCollection dataset = createDataset(aPoints, "track");
 145         JFreeChart chart = createLineChart(dataset);
 146
 147         Pair<Pair<Double,Double>,Pair<Double,Double>> bounds = getBounds(aPoints);
 148
 149         chart.getXYPlot().getDomainAxis().setLowerBound(bounds.getFirst().getFirst());
 150         chart.getXYPlot().getDomainAxis().setUpperBound(bounds.getFirst().getSecond());
 151
 152         chart.getXYPlot().getRangeAxis().setLowerBound(bounds.getSecond().getFirst());
 153         chart.getXYPlot().getRangeAxis().setUpperBound(bounds.getSecond().getSecond());
 154
 155         Image background = JpegUtils.loadJpegImage(new FileInputStream("/home/erik/vakantie.jpg"));
 156         chart.getPlot().setBackgroundImage(background);
 157
 158         XYLineAndShapeRenderer renderer = (XYLineAndShapeRenderer)chart.getXYPlot().getRenderer();
 159         renderer.setShapesVisible(true);
 160         renderer.setShapesFilled(true);
 161         renderer.setPaint(Color.BLACK);
 162
 163         ChartUtilities.writeChartAsPNG(new FileOutputStream("test.png"), chart, 1280, 800);
 164         ChartFrame frame = new ChartFrame("test", chart);
 165         frame.pack();
 166         frame.setVisible(true);
 167     }
 168
 169     /**
 170      * @param dataset
 171      * @return
 172      */
 173     private static JFreeChart createLineChart(XYSeriesCollection dataset) {
 174         NumberAxis xAxis = new NumberAxis("S->N");
 175         xAxis.setAutoRangeIncludesZero(false);
 176
 177         NumberAxis yAxis = new NumberAxis("W->E");
 178         yAxis.setAutoRangeIncludesZero(false);
 179
 180         XYItemRenderer renderer = new XYLineAndShapeRenderer(true, false);
 181         XYPlot plot = new ZoomableBackgroundXYPlot(dataset, xAxis, yAxis, renderer);
 182         plot.setOrientation(PlotOrientation.HORIZONTAL);
 183
 184         JFreeChart chart = new JFreeChart(
 185             "Track", JFreeChart.DEFAULT_TITLE_FONT, plot, true
 186         );
 187
 188         return chart;
 189         /*
 190         JFreeChart chart = ChartFactory.createXYLineChart(
 191                 "Track",
 192                 "S->N",
 193                 "W->E",
 194                 dataset,
 195                 PlotOrientation.HORIZONTAL,
 196                 true,
 197                 true,
 198                 false);
 199         return chart;
 200         */
 201     }
 202
 203     /**
 204      * @param aHeightProfile
 205      * @return
 206      */
 207     private static XYSeriesCollection createDataset(List<Pair<Double, Double>> aHeightProfile, String aName) {
 208         XYSeries series = new XYSeries(aName, false);
 209         for (Pair<Double,Double> point: aHeightProfile) {
 210             series.add(point.getFirst(), point.getSecond());
 211         }
 212         XYSeriesCollection dataset = new XYSeriesCollection(series);
 213         return dataset;
 214     }
 215
 216     private static Pair<Pair<Double,Double>,Pair<Double,Double>> getBounds(List<Pair<Double,Double>> aList) {
 217         Pair<Double,Double> first = aList.get(0);
 218         double minx= first.getFirst();
 219         double maxx = minx;
 220         double miny = first.getSecond();
 221         double maxy = miny;
 222
 223         for (int i = 0; i < aList.size(); i++) {
 224             Pair<Double,Double> value = aList.get(i);
 225             minx = Math.min(minx, value.getFirst());
 226             maxx = Math.max(maxx, value.getFirst());
 227             miny = Math.min(miny, value.getSecond());
 228             maxy = Math.max(maxy, value.getSecond());
 229         }
 230         if ( maxx == minx ) {
 231             maxx += 1.0; // to avoid problems.
 232         }
 233         if ( maxy == miny ) {
 234             maxy += 1.0; // to avoid problems.
 235         }
 236         final double paddingFactor = 0.3; // allow some space around min and max
 237         return new Pair<Pair<Double,Double>,Pair<Double,Double>>(
 238                 new Pair<Double,Double>( minx - paddingFactor*(maxx-minx),
 239                                          maxx + paddingFactor*(maxx-minx)),
 240                 new Pair<Double,Double>( miny - paddingFactor*(maxy-miny),
 241                                          maxy + paddingFactor*(maxy-miny))
 242                 );
 243     }
 244 }
 245