diff options
Diffstat (limited to 'src/com/delwink/icebox/table/MainWindowTableModel.java')
-rw-r--r-- | src/com/delwink/icebox/table/MainWindowTableModel.java | 19 |
1 files changed, 14 insertions, 5 deletions
diff --git a/src/com/delwink/icebox/table/MainWindowTableModel.java b/src/com/delwink/icebox/table/MainWindowTableModel.java index 66a904e..187ee02 100644 --- a/src/com/delwink/icebox/table/MainWindowTableModel.java +++ b/src/com/delwink/icebox/table/MainWindowTableModel.java @@ -20,6 +20,8 @@ package com.delwink.icebox.table; import com.delwink.icebox.Inventory; import com.delwink.icebox.InventoryItem; import com.delwink.icebox.lang.Lang; +import java.util.ArrayList; +import java.util.List; import javax.swing.table.AbstractTableModel; /** @@ -27,15 +29,22 @@ import javax.swing.table.AbstractTableModel; * @author David McMackins II */ public class MainWindowTableModel extends AbstractTableModel { - private final Inventory INVENTORY; + private final List<InventoryItem> ITEMS; - public MainWindowTableModel(Inventory inventory) { - INVENTORY = inventory; + public MainWindowTableModel(Inventory inventory, boolean reorderOnly) { + if (reorderOnly) { + ITEMS = new ArrayList<>(); + for (InventoryItem item : inventory.getItems()) + if (item.getStock() <= item.getReorderAt()) + ITEMS.add(item); + } else { + ITEMS = new ArrayList<>(inventory.getItems()); + } } @Override public int getRowCount() { - return INVENTORY.getItems().size(); + return ITEMS.size(); } @Override @@ -56,7 +65,7 @@ public class MainWindowTableModel extends AbstractTableModel { @Override public Object getValueAt(int i, int j) { InventoryItem item = null; - for (InventoryItem temp : INVENTORY.getItems()) { + for (InventoryItem temp : ITEMS) { if (i-- == 0) { item = temp; break; |