weka.gui.streams.InstanceTable Maven / Gradle / Ivy

/*
 *   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 .
 */

/*
 *    InstanceTable.java
 *    Copyright (C) 1999-2012 University of Waikato, Hamilton, New Zealand
 *
 */

package weka.gui.streams;

import java.awt.BorderLayout;

import javax.swing.JPanel;
import javax.swing.JScrollPane;
import javax.swing.JTable;
import javax.swing.table.AbstractTableModel;
import javax.swing.table.TableModel;

import weka.core.Instance;
import weka.core.Instances;

/**
 * A bean that takes a stream of instances and displays in a table.
 * 
 * @author Len Trigg ([email protected])
 * @version $Revision: 10222 $
 */
public class InstanceTable extends JPanel implements InstanceListener {

  /** for serialization */
  private static final long serialVersionUID = -2462533698100834803L;

  private final JTable m_InstanceTable;
  private boolean m_Debug;
  // private boolean m_Clear; NOT USED
  // private String m_UpdateString; NOT USED
  private Instances m_Instances;

  public void inputFormat(Instances instanceInfo) {

    if (m_Debug) {
      System.err.println("InstanceTable::inputFormat()\n"
        + instanceInfo.toString());
    }
    m_Instances = instanceInfo;
  }

  public void input(Instance instance) throws Exception {

    if (m_Debug) {
      System.err.println("InstanceTable::input(" + instance + ")");
    }
    m_Instances.add(instance);
  }

  public void batchFinished() {

    TableModel newModel = new AbstractTableModel() {
      private static final long serialVersionUID = 5447106383000555291L;

      @Override
      public String getColumnName(int col) {
        return m_Instances.attribute(col).name();
      }

      @Override
      public Class getColumnClass(int col) {
        return "".getClass();
      }

      @Override
      public int getColumnCount() {
        return m_Instances.numAttributes();
      }

      @Override
      public int getRowCount() {
        return m_Instances.numInstances();
      }

      @Override
      public Object getValueAt(int row, int col) {
        return new String(m_Instances.instance(row).toString(col));
      }
    };
    m_InstanceTable.setModel(newModel);
    if (m_Debug) {
      System.err.println("InstanceTable::batchFinished()");
    }
  }

  public InstanceTable() {

    setLayout(new BorderLayout());
    m_InstanceTable = new JTable();
    add("Center", new JScrollPane(m_InstanceTable));
  }

  public void setDebug(boolean debug) {

    m_Debug = debug;
  }

  public boolean getDebug() {

    return m_Debug;
  }

  @Override
  public void instanceProduced(InstanceEvent e) {

    Object source = e.getSource();
    if (source instanceof InstanceProducer) {
      try {
        InstanceProducer a = (InstanceProducer) source;
        switch (e.getID()) {
        case InstanceEvent.FORMAT_AVAILABLE:
          inputFormat(a.outputFormat());
          break;
        case InstanceEvent.INSTANCE_AVAILABLE:
          input(a.outputPeek());
          break;
        case InstanceEvent.BATCH_FINISHED:
          batchFinished();
          break;
        default:
          System.err.println("InstanceTable::instanceProduced()"
            + " - unknown event type");
          break;
        }
      } catch (Exception ex) {
        System.err.println(ex.getMessage());
      }
    } else {
      System.err.println("InstanceTable::instanceProduced()"
        + " - Unknown source object type");
    }
  }
}