Merge pull request #687 from mapzen/686-limit-station-icons

limit station icons
mapzen · Jul 19, 2016 · 6800b7e · 6800b7e
2 parents 04f6c72 + be85467
commit 6800b7e
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/app/src/main/kotlin/com/mapzen/erasermap/view/RouteModeView.kt b/app/src/main/kotlin/com/mapzen/erasermap/view/RouteModeView.kt
@@ -489,10 +489,10 @@ class RouteModeView : LinearLayout, RouteViewController, ViewPager.OnPageChangeL
                         val stationPoints = ArrayList<LngLat>()
                         if (instruction.getTransitInfo() != null) {
                             val stops = instruction.getTransitInfo()!!.getTransitStops()
-                            for (transitStop in stops) {
-                                stationPoints.add(LngLat(transitStop.getLon(),
-                                    transitStop.getLat()))
-                            }
+                            val firstStop = stops[0]
+                            stationPoints.add(LngLat(firstStop.getLon(), firstStop.getLat()))
+                            val lastStop = stops[stops.size-1]
+                            stationPoints.add(LngLat(lastStop.getLon(), lastStop.getLat()))
                         }
                         mapzenMap?.drawTransitRouteLine(points, stationPoints,
                             instruction.getTransitInfoColorHex() as String)
@@ -504,6 +504,7 @@ class RouteModeView : LinearLayout, RouteViewController, ViewPager.OnPageChangeL
 
     override fun hideRouteLine() {
         mapzenMap?.clearRouteLine()
+        mapzenMap?.clearTransitRouteLine()
     }
 
     fun clearRoute() {