1 files changed, 184 insertions, 0 deletions
diff --git a/src/uk/org/ury/library/viewer/LibraryTableModel.java b/src/uk/org/ury/library/viewer/LibraryTableModel.java
new file mode 100644
index 0000000..1ce5709
--- /dev/null
+++ b/src/uk/org/ury/library/viewer/LibraryTableModel.java
@@ -0,0 +1,184 @@
+/**
+ * 
+ */
+package uk.org.ury.library.viewer;
+
+import java.util.List;
+
+import javax.swing.table.AbstractTableModel;
+
+import uk.org.ury.database.exceptions.MissingPropertyException;
+import uk.org.ury.library.item.LibraryItem;
+import uk.org.ury.library.item.LibraryItemProperty;
+
+
+/**
+ * A table model for the library viewer.
+ * 
+ * @author Matt Windsor
+ */
+
+public class LibraryTableModel extends AbstractTableModel
+{
+
+  /**
+   * 
+   */
+  
+  private static final long serialVersionUID = -1744980619128903509L;
+
+  private List<LibraryItem> data;
+  
+  
+  /**
+   * Construct a new table model.
+   * 
+   * @param data  The list of data on which the model will be based.
+   */
+  
+  public
+  LibraryTableModel (List<LibraryItem> data)
+  {
+    this.data = data;
+  }
+  
+  
+  /* (non-Javadoc)
+   * @see javax.swing.table.TableModel#getColumnCount()
+   */
+  
+  @Override
+  public int
+  getColumnCount ()
+  {
+    return 6;
+  }
+
+  
+  /* (non-Javadoc)
+   * @see javax.swing.table.TableModel#getRowCount()
+   */
+  
+  @Override
+  public int
+  getRowCount ()
+  {
+    return data.size ();
+  }
+
+  
+  /**
+   * @param c  The column whose class should be investigated.
+   * 
+   * @return   the column class of column c.
+   */
+  
+  @Override
+  public Class<?>
+  getColumnClass (int c) 
+  {
+    return getValueAt (0, c).getClass ();
+  }
+  
+  
+  /* (non-Javadoc)
+   * @see javax.swing.table.TableModel#getValueAt(int, int)
+   */
+  
+  @Override
+  public Object
+  getValueAt (int rowIndex, int columnIndex)
+  {
+    LibraryItem li = data.get (rowIndex);
+    
+    try
+      {
+    	String[] columnData = {li.get (LibraryItemProperty.TITLE), 
+                               li.get (LibraryItemProperty.ARTIST),
+                               li.get (LibraryItemProperty.ALBUM)};
+
+        switch (columnIndex)
+          {
+          default: // Title, artist, album, unknown
+            
+        	if (columnIndex >= columnData.length)
+        	  return "Unknown";
+        	else
+        	  return columnData[columnIndex];
+
+          case 3: // Medium
+            
+            // TODO: Make this less kludge-y
+            
+            String mediumString = li.get (LibraryItemProperty.MEDIUM);
+            
+            if (mediumString.equals ("c"))
+              return "Compact Disc";
+            else if (mediumString.equals ("7"))
+              return "7\" Vinyl";
+            else if (mediumString.equals ("2"))
+              return "12\" Vinyl";
+            else
+              return "Unrecognised";
+            
+          case 4: // Clean?
+    
+            // Return true if marked true, false if marked false or unknown etc.
+            
+            String cleanString = li.get (LibraryItemProperty.IS_CLEAN); 
+    
+            // TODO: Nicer way of showing this
+            
+            if (cleanString.equals ("y"))
+              return "Yes";
+            else if (cleanString.equals ("n"))
+              return "No";
+            else
+              return "???";
+            
+          case 5: // isDigitised
+            
+            // Return true if marked true, false if marked false or unknown etc.
+            
+            String digitisedString = li.get (LibraryItemProperty.IS_DIGITISED); 
+            
+            if (digitisedString.equals ("t"))
+              return true;
+            else
+              return false;
+          }
+      }
+    catch (MissingPropertyException e)
+      {
+        return "Unknown";
+      }
+  }
+  
+  
+  /* (non-Javadoc)
+   * @see javax.swing.table.TableModel#getColumnName(int, int)
+   */
+  
+  @Override
+  public String
+  getColumnName (int index)
+  {
+    switch (index)
+      {
+      case 0:
+        return "Title";
+      case 1:
+        return "Artist";
+      case 2:
+        return "Album";
+      case 3:
+        return "Medium";
+      case 4:
+        return "Clean?";
+      case 5:
+        return "On system?";
+      default:
+        return "ERROR";
+      }
+  }
+}