weka.gui.sql.ResultSetTableModel Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of weka-stable Show documentation
Show all versions of weka-stable Show documentation
The Waikato Environment for Knowledge Analysis (WEKA), a machine
learning workbench. This is the stable version. Apart from bugfixes, this version
does not receive any other updates.
/*
* 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 .
*/
/*
* ResultSetTableModel.java
* Copyright (C) 2005-2012 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: 10222 $
*/
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
*/
@Override
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
*/
@Override
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
*/
@Override
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
*/
@Override
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
*/
@Override
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
*/
@Override
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
*/
@Override
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
*/
@Override
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
*/
@Override
public void setValueAt(Object aValue, int rowIndex, int columnIndex) {
// ignore
}
/**
* frees up the memory.
*
* @throws Throwable if something goes wrong
*/
@Override
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();
}
}