/* * Copyright (C) 2012 Dr. John Lindsay <jlindsay@uoguelph.ca> * * This program is free software: you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation, either version 3 of the License, or * (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program. If not, see <http://www.gnu.org/licenses/>. */ package whitebox.ui.plugin_dialog; import java.util.Vector; import javax.swing.table.AbstractTableModel; /** * * @author Dr. John Lindsay <jlindsay@uoguelph.ca> */ public class ReclassTableModel extends AbstractTableModel { public static final int NEW_INDEX = 0; public static final int FROM_INDEX = 1; public static final int TO_INDEX = 2; public static final int HIDDEN_INDEX = 3; protected String[] columnNames; protected Vector dataVector; public ReclassTableModel(String[] columnNames) { this.columnNames = columnNames; dataVector = new Vector(); } @Override public String getColumnName(int column) { return columnNames[column]; } @Override public boolean isCellEditable(int row, int column) { if (column == HIDDEN_INDEX) { return false; } else { return true; } } @Override public Class getColumnClass(int column) { switch (column) { case NEW_INDEX: case FROM_INDEX: case TO_INDEX: return String.class; default: return Object.class; } } @Override public Object getValueAt(int row, int column) { ReclassEntry record = (ReclassEntry) dataVector.get(row); switch (column) { case NEW_INDEX: return record.getNewValue(); case FROM_INDEX: return record.getFromValue(); case TO_INDEX: return record.getToValue(); default: return new Object(); } } @Override public void setValueAt(Object value, int row, int column) { ReclassEntry record = (ReclassEntry) dataVector.get(row); switch (column) { case NEW_INDEX: record.setNewValue((String) value); break; case FROM_INDEX: record.setFromValue((String) value); break; case TO_INDEX: record.setToValue((String) value); break; default: System.out.println("invalid index"); } fireTableCellUpdated(row, column); } @Override public int getRowCount() { return dataVector.size(); } @Override public int getColumnCount() { return columnNames.length; } public boolean hasEmptyRow() { if (dataVector.isEmpty()) { return false; } ReclassEntry record = (ReclassEntry) dataVector.get(dataVector.size() - 1); if (record.getNewValue().trim().equals("") && record.getFromValue().trim().equals("") && record.getToValue().trim().equals("")) { return true; } else { return false; } } public void addEmptyRow() { dataVector.add(new ReclassEntry()); fireTableRowsInserted( dataVector.size() - 1, dataVector.size() - 1); } }