diff --git a/src/main/java/seng202/group9/Controller/AirlineFilter.java b/src/main/java/seng202/group9/Controller/AirlineFilter.java index dc1e149..a13bbad 100644 --- a/src/main/java/seng202/group9/Controller/AirlineFilter.java +++ b/src/main/java/seng202/group9/Controller/AirlineFilter.java @@ -155,6 +155,10 @@ public class AirlineFilter extends Filter{ return filteredList; } +// public ArrayList getBaseList(){ +// return baseArray; +// } + /** * sets a new base list of the filter. * @param arrayList diff --git a/src/main/java/seng202/group9/Controller/Session.java b/src/main/java/seng202/group9/Controller/Session.java index 1f769d2..09da4c1 100644 --- a/src/main/java/seng202/group9/Controller/Session.java +++ b/src/main/java/seng202/group9/Controller/Session.java @@ -12,7 +12,7 @@ import java.util.HashMap; */ public class Session implements Serializable { private SceneCode sceneDisplayed; - private HashMap filteredAirlines; + private HashMap filteredAirlines; /** * Constructor for a new session @@ -50,7 +50,7 @@ public class Session implements Serializable { this.filteredAirlines = airlines; } - public HashMap getFilteredAirlines() { + public HashMap getFilteredAirlines() { return filteredAirlines; } } diff --git a/src/main/java/seng202/group9/GUI/AirlineFilterController.java b/src/main/java/seng202/group9/GUI/AirlineFilterController.java index 7c516b7..7a6b88c 100644 --- a/src/main/java/seng202/group9/GUI/AirlineFilterController.java +++ b/src/main/java/seng202/group9/GUI/AirlineFilterController.java @@ -81,16 +81,16 @@ public class AirlineFilterController extends Controller { //currentSession.setFilteredAirlines(FXCollections.observableArrayList(filter.getFilteredData())); - HashMap airlinesHM = new HashMap(); + HashMap airlinesHM = new HashMap(); ArrayList airlines = filter.getFilteredData(); - for (Airline airline: airlines) { - airlinesHM.put(airline.getName(), airline); + //for (Airline airline: airlines) { + for (int index = 0; index < airlines.size(); index++) { + airlinesHM.put(index, airlines.get(index).getName()); } currentSession.setFilteredAirlines(airlinesHM); Stage stage = (Stage) applyButton.getScene().getWindow(); stage.close(); - } public void load() { diff --git a/src/main/java/seng202/group9/GUI/AirlineRDController.java b/src/main/java/seng202/group9/GUI/AirlineRDController.java index 06036e9..2422ffa 100644 --- a/src/main/java/seng202/group9/GUI/AirlineRDController.java +++ b/src/main/java/seng202/group9/GUI/AirlineRDController.java @@ -17,6 +17,7 @@ import seng202.group9.Core.Airline; import javax.swing.*; import java.io.IOException; +import java.util.ArrayList; //make a class for the scenes to get the data I guess /** * The GUI controller class for airline_raw_data.fxml. @@ -84,7 +85,14 @@ public class AirlineRDController extends Controller { */ public void openFilter() { createPopUpStage(SceneCode.AIRLINE_FILTER, 600, 370); - tableViewAirlineRD.setItems(FXCollections.observableArrayList(currentSession.getFilteredAirlines().values())); + ArrayList d = new ArrayList(); + for(int i = 0; i < theDataSet.getAirlines().size(); i++) { + if (currentSession.getFilteredAirlines().containsValue(theDataSet.getAirlines().get(i).getName()) + && currentSession.getFilteredAirlines().containsKey(i)) { + d.add(theDataSet.getAirlines().get(i)); + } + } + tableViewAirlineRD.setItems(FXCollections.observableArrayList(d)); }