From 5f2967e9cc0fb3db958ca08b4c9169b60e5197a1 Mon Sep 17 00:00:00 2001
From: YaFedImYaEatIm <gondra2000@hotmail.com>
Date: Sun, 2 Oct 2016 01:37:42 +1300
Subject: [PATCH] Simplified get Route By Airport

---
 .../group9/GUI/AirportRouteMapController.java | 20 ++++++++-----------
 1 file changed, 8 insertions(+), 12 deletions(-)

diff --git a/src/main/java/seng202/group9/GUI/AirportRouteMapController.java b/src/main/java/seng202/group9/GUI/AirportRouteMapController.java
index 4fa8071..cafd34f 100644
--- a/src/main/java/seng202/group9/GUI/AirportRouteMapController.java
+++ b/src/main/java/seng202/group9/GUI/AirportRouteMapController.java
@@ -53,19 +53,15 @@ public class AirportRouteMapController extends Controller{
         airportsTable.getSelectionModel().selectedItemProperty().addListener(new ChangeListener<Airport>() {
             public void changed(ObservableValue<? extends Airport> observable, Airport oldValue, Airport newValue) {
                 Airport selectedAirport= (Airport) airportsTable.getSelectionModel().getSelectedItems().get(0);
-                for (int i = 0 ; i < currentDataset.getAirports().size(); i ++){
-                    if (currentDataset.getAirports().get(i).equals(selectedAirport)){
-                        ArrayList<RoutePath> routePaths = new ArrayList<RoutePath>();
-                        for (Route route:currentDataset.getAirports().get(i).getArrivalRoutes()){
-                            routePaths.add(route.getRoutePath());
-                        }
-                        for (Route route:currentDataset.getAirports().get(i).getDepartureRoutes()){
-                            routePaths.add(route.getRoutePath());
-                        }
-                        map.displayRoutes(routePaths);
-                        break;
-                    }
+
+                ArrayList<RoutePath> routePaths = new ArrayList<RoutePath>();
+                for (Route route:selectedAirport.getArrivalRoutes()){
+                    routePaths.add(route.getRoutePath());
+                }
+                for (Route route:selectedAirport.getDepartureRoutes()){
+                    routePaths.add(route.getRoutePath());
                 }
+                map.displayRoutes(routePaths);
             }
         });
     }