1 files changed, 56 insertions, 0 deletions
diff --git a/src/main/java/com/benburwell/planes/gui/navigationaids/AirportsTableModel.java b/src/main/java/com/benburwell/planes/gui/navigationaids/AirportsTableModel.java
new file mode 100644
index 0000000..2850196
--- /dev/null
+++ b/src/main/java/com/benburwell/planes/gui/navigationaids/AirportsTableModel.java
@@ -0,0 +1,56 @@
+package com.benburwell.planes.gui.navigationaids;
+
+import com.benburwell.planes.data.Airport;
+
+import javax.swing.table.AbstractTableModel;
+import java.util.List;
+
+/**
+ * Created by ben on 11/19/16.
+ */
+public class AirportsTableModel extends AbstractTableModel {
+    public final String[] COLUMN_HEADERS = {"Identifier", "Name", "Country", "Municipality", "Scheduled Service", "IATA Code", "Local Code"};
+
+    public List<Airport> airports;
+
+    public AirportsTableModel(List<Airport> airports) {
+        this.airports = airports;
+    }
+
+    @Override
+    public String getColumnName(int columnIndex) {
+        return COLUMN_HEADERS[columnIndex];
+    }
+
+    @Override
+    public int getRowCount() {
+        return airports.size();
+    }
+
+    @Override
+    public int getColumnCount() {
+        return COLUMN_HEADERS.length;
+    }
+
+    @Override
+    public Object getValueAt(int rowIndex, int columnIndex) {
+        Airport airport = this.airports.get(rowIndex);
+        switch (columnIndex) {
+            case 0:
+                return airport.getIdent();
+            case 1:
+                return airport.getName();
+            case 2:
+                return airport.getIsoCountry();
+            case 3:
+                return airport.getMunicipality();
+            case 4:
+                return airport.isScheduledService();
+            case 5:
+                return airport.getIataCode();
+            case 6:
+                return airport.getLocalCode();
+        }
+        return null;
+    }
+}