resultsettablemodel.java

代码是一个分类器的实现,其中使用了部分weka的源代码。可以将项目导入eclipse运行

/*
 *    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.sql.ResultSetMetaData;
import java.sql.Types;
import java.util.Iterator;
import java.util.HashSet;

import javax.swing.table.TableModel;
import javax.swing.event.TableModelListener;
import javax.swing.event.TableModelEvent;

/**
* The model for an SQL ResultSet.
*
*
* @author     FracPete (fracpete at waikato dot ac dot nz)
* @version    $Revision: 1.2 $
*/

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
   */
  public void addTableModelListener(TableModelListener l) {
    m_Listeners.add(l);
  }

  /**
   * returns the most specific superclass for all the cell values in the 
   * column (always String)
   */
  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
   */
  public int getColumnCount() {
    return m_Helper.getColumnCount() + 1;
  }

  /**
   * returns the name of the column at columnIndex
   */
  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
   */
  public int getRowCount() {
    return m_Helper.getRowCount();
  }

  /**
   * returns the value for the cell at columnindex and rowIndex
   */
  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
   */
  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
   */
  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
   */
  public void removeTableModelListener(TableModelListener l) {
    m_Listeners.remove(l);
  }

  /**
   * sets the value in the cell at columnIndex and rowIndex to aValue.
   * but only the value and the value can be changed
   */
  public void setValueAt(Object aValue, int rowIndex, int columnIndex) {
    // ignore
  }

  /**
   * notfies all listener of the change in the specified cell
   */
  private void notifyListener(int rowIndex, int columnIndex) {
    Iterator                iter;
    TableModelListener      l;

    if (m_Listeners.size() > 0) {
      iter = m_Listeners.iterator();
      while (iter.hasNext()) {
        l = (TableModelListener) iter.next();
        l.tableChanged(new TableModelEvent(this, rowIndex, columnIndex));
      }
    }
  }

  /**
   * frees up the memory
   */
  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();
  }
}