From 77e569aacb74c8fda3630ee56c20d5d4bde0eb8f Mon Sep 17 00:00:00 2001
From: Tor-Einar Skog <tor-einar.skog@bioforsk.no>
Date: Sat, 15 Oct 2016 09:04:58 -0700
Subject: [PATCH] Enabling reading/writing of 3D coordinates

---
 src/main/java/org/wololo/jts2geojson/GeoJSONReader.java | 7 ++++++-
 src/main/java/org/wololo/jts2geojson/GeoJSONWriter.java | 7 ++++++-
 2 files changed, 12 insertions(+), 2 deletions(-)

diff --git a/src/main/java/org/wololo/jts2geojson/GeoJSONReader.java b/src/main/java/org/wololo/jts2geojson/GeoJSONReader.java
index a700a14..7bc9889 100644
--- a/src/main/java/org/wololo/jts2geojson/GeoJSONReader.java
+++ b/src/main/java/org/wololo/jts2geojson/GeoJSONReader.java
@@ -96,7 +96,12 @@ Geometry convert(GeometryCollection gc) {
     }
 
     Coordinate convert(double[] c) {
-        return new Coordinate(c[0], c[1]);
+        if(c.length == 2){
+            return new Coordinate(c[0], c[1]);
+        }
+        else{
+            return new Coordinate(c[0], c[1], c[2]);
+        }
     }
 
     Coordinate[] convert(double[][] ca) {
diff --git a/src/main/java/org/wololo/jts2geojson/GeoJSONWriter.java b/src/main/java/org/wololo/jts2geojson/GeoJSONWriter.java
index 2375323..5bf1522 100644
--- a/src/main/java/org/wololo/jts2geojson/GeoJSONWriter.java
+++ b/src/main/java/org/wololo/jts2geojson/GeoJSONWriter.java
@@ -102,7 +102,12 @@ org.wololo.geojson.GeometryCollection convert(GeometryCollection gc) {
     }
 
     double[] convert(Coordinate coordinate) {
-        return new double[] { coordinate.x, coordinate.y };
+        if(Double.isNaN( coordinate.z )) {
+            return new double[] { coordinate.x, coordinate.y };
+        }
+        else {
+            return new double[] { coordinate.x, coordinate.y, coordinate.z };
+        }
     }
 
     double[][] convert(Coordinate[] coordinates) {