/*
* 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 2 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, write to the Free Software
* Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
*/
/*
* ResultSetTableModel.java
* Copyright (C) 2005 University of Waikato, Hamilton, New Zealand
*
*/
package weka.gui.sql;
import java.sql.ResultSet;
import java.util.HashSet;
import javax.swing.event.TableModelListener;
import javax.swing.table.TableModel;
/**
* The model for an SQL ResultSet.
*
* @author FracPete (fracpete at waikato dot ac dot nz)
* @version $Revision: 1.3 $
*/
public class ResultSetTableModel implements TableModel {
/** the listeners. */
protected HashSet m_Listeners;
/** the data. */
protected Object[][] m_Data;
/** for retrieving the data etc. */
protected ResultSetHelper m_Helper;
/**
* initializes the model, retrieves all rows.
*
* @param rs the ResultSet to get the data from
*/
public ResultSetTableModel(ResultSet rs) {
this(rs, 0);
}
/**
* initializes the model, retrieves only the given amount of rows (0 means
* all).
*
* @param rs the ResultSet to get the data from
* @param rows the maximum number of rows to retrieve, 0 retrieves all
*/
public ResultSetTableModel(ResultSet rs, int rows) {
super();
m_Listeners = new HashSet();
m_Helper = new ResultSetHelper(rs, rows);
m_Data = m_Helper.getCells();
}
/**
* adds a listener to the list that is notified each time a change to data
* model occurs.
*
* @param l the listener to add
*/
public void addTableModelListener(TableModelListener l) {
m_Listeners.add(l);
}
/**
* returns the most specific superclass for all the cell values in the
* column (always String).
*
* @param columnIndex the index of the column
* @return the class
*/
public Class getColumnClass(int columnIndex) {
Class result;
result = null;
if ( (m_Helper.getColumnClasses() != null)
&& (columnIndex >= 0)
&& (columnIndex < getColumnCount()) ) {
if (columnIndex == 0)
result = Integer.class;
else
result = m_Helper.getColumnClasses()[columnIndex - 1];
}
return result;
}
/**
* returns the number of columns in the model.
*
* @return the number of columns
*/
public int getColumnCount() {
return m_Helper.getColumnCount() + 1;
}
/**
* returns the name of the column at columnIndex.
*
* @param columnIndex the index of the column
* @return the name
*/
public String getColumnName(int columnIndex) {
String result;
result = "";
if ( (m_Helper.getColumnNames() != null)
&& (columnIndex >= 0)
&& (columnIndex < getColumnCount()) ) {
if (columnIndex == 0)
result = "Row";
else
result = m_Helper.getColumnNames()[columnIndex - 1];
}
return result;
}
/**
* returns the number of rows in the model.
*
* @return the number of data rows
*/
public int getRowCount() {
return m_Data.length;
}
/**
* returns the value for the cell at columnindex and rowIndex.
*
* @param rowIndex the row of the cell
* @param columnIndex the column of the cell
* @return the data value
*/
public Object getValueAt(int rowIndex, int columnIndex) {
Object result;
result = null;
if ( (rowIndex >= 0) && (rowIndex < getRowCount())
&& (columnIndex >= 0) && (columnIndex < getColumnCount()) ) {
if (columnIndex == 0)
result = new Integer(rowIndex + 1);
else
result = m_Data[rowIndex][columnIndex - 1];
}
return result;
}
/**
* checks whether the value of the cell is NULL.
*
* @param rowIndex the row of the cell
* @param columnIndex the column of the cell
* @return true if the cell value is NULL
*/
public boolean isNullAt(int rowIndex, int columnIndex) {
return (getValueAt(rowIndex, columnIndex) == null);
}
/**
* returns whether the column at the given index is numeric.
*
* @param columnIndex the column to check
* @return whether the column is numeric
*/
public boolean isNumericAt(int columnIndex) {
boolean result;
result = false;
if ( (columnIndex >= 0) && (columnIndex < getColumnCount()) ) {
if (columnIndex == 0) {
result = true;
}
else {
if (m_Helper.getNumericColumns() == null)
result = false;
else
result = m_Helper.getNumericColumns()[columnIndex - 1];
}
}
return result;
}
/**
* returns true if the cell at rowindex and columnindexis editable.
*
* @param rowIndex the row of the cell
* @param columnIndex the column of the cell
* @return always false
*/
public boolean isCellEditable(int rowIndex, int columnIndex) {
return false;
}
/**
* removes a listener from the list that is notified each time a change to
* the data model occurs.
*
* @param l the listener to remove
*/
public void removeTableModelListener(TableModelListener l) {
m_Listeners.remove(l);
}
/**
* sets the value in the cell at columnIndex and rowIndex to aValue.
* Ignored.
*
* @param aValue the value to set - ignored
* @param rowIndex the row of the cell
* @param columnIndex the column of the cell
*/
public void setValueAt(Object aValue, int rowIndex, int columnIndex) {
// ignore
}
/**
* frees up the memory.
*
* @throws Throwable if something goes wrong
*/
public void finalize() throws Throwable {
try {
m_Helper.getResultSet().close();
m_Helper.getResultSet().getStatement().close();
m_Helper = null;
}
catch (Exception e) {
// ignored
}
m_Data = null;
super.finalize();
}
}