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-dev Show documentation
The Waikato Environment for Knowledge Analysis (WEKA), a machine learning workbench. This version represents the developer version, the "bleeding edge" of development, you could say. New functionality gets added to this version.
There is a newer version: 3.9.6
Show newest version
/*
 *   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();
  }
}