+package tim.prune.save;
+
+import javax.swing.table.AbstractTableModel;
+
+import tim.prune.I18nManager;
+
+/**
+ * Class to hold table model information for save dialog
+ */
+public class FieldSelectionTableModel extends AbstractTableModel
+{
+ private FieldInfo[] _info = null;
+
+
+ /**
+ * Constructor giving list size
+ * @param inSize number of fields
+ */
+ public FieldSelectionTableModel(int inSize)
+ {
+ _info = new FieldInfo[inSize];
+ }
+
+
+ /**
+ * Set the given FieldInfo object in the array
+ * @param inInfo FieldInfo object describing the field
+ * @param inIndex index to place in array
+ */
+ public void addFieldInfo(FieldInfo inInfo, int inIndex)
+ {
+ _info[inIndex] = inInfo;
+ }
+
+
+ /**
+ * @see javax.swing.table.TableModel#getColumnCount()
+ */
+ public int getColumnCount()
+ {
+ return 3;
+ }
+
+
+ /**
+ * @see javax.swing.table.TableModel#getRowCount()
+ */
+ public int getRowCount()
+ {
+ return _info.length;
+ }
+
+
+ /**
+ * @see javax.swing.table.TableModel#getValueAt(int, int)
+ */
+ public Object getValueAt(int inRowIndex, int inColumnIndex)
+ {
+ if (inColumnIndex == 0)
+ {
+ return _info[inRowIndex].getField().getName();
+ }
+ else if (inColumnIndex == 1)
+ {
+ return Boolean.valueOf(_info[inRowIndex].hasData());
+ }
+ return Boolean.valueOf(_info[inRowIndex].isSelected());
+ }
+
+
+ /**
+ * @return true if cell is editable
+ */
+ public boolean isCellEditable(int inRowIndex, int inColumnIndex)
+ {
+ // only the select column is editable
+ return inColumnIndex == 2;
+ }
+
+
+ /**
+ * Set the given cell value
+ * @see javax.swing.table.TableModel#setValueAt(java.lang.Object, int, int)
+ */
+ public void setValueAt(Object inValue, int inRowIndex, int inColumnIndex)
+ {
+ // ignore edits to other columns
+ if (inColumnIndex == 2)
+ _info[inRowIndex].setSelected(((Boolean) inValue).booleanValue());
+ }
+
+
+ /**
+ * Swap the specified items in the array
+ * @param inIndex1 first index
+ * @param inIndex2 second index
+ */
+ public void swapItems(int inIndex1, int inIndex2)
+ {
+ if (inIndex1 >= 0 && inIndex1 < _info.length && inIndex2 >= 0 && inIndex2 < _info.length)
+ {
+ FieldInfo temp = _info[inIndex1];
+ _info[inIndex1] = _info[inIndex2];
+ _info[inIndex2] = temp;
+ }
+ }
+
+
+ /**
+ * @return Class of cell data
+ */
+ public Class<?> getColumnClass(int inColumnIndex)
+ {
+ if (inColumnIndex==0) return String.class;
+ return Boolean.class;
+ }
+
+
+ /**
+ * Get the name of the column
+ */
+ public String getColumnName(int inColNum)
+ {
+ if (inColNum == 0) return I18nManager.getText("dialog.save.table.field");
+ else if (inColNum == 1) return I18nManager.getText("dialog.save.table.hasdata");
+ return I18nManager.getText("dialog.save.table.save");
+ }
+
+
+ /**
+ * Retrieve the FieldInfo object at the given index
+ * @param inIndex index, starting at 0
+ * @return FieldInfo object at this position
+ */
+ public FieldInfo getFieldInfo(int inIndex)
+ {
+ if (inIndex < 0 || inIndex >= _info.length)
+ {
+ return null;
+ }
+ return _info[inIndex];
+ }
+}