weka.gui.streams.InstanceSavePanel 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 .
*/
/*
* InstanceSavePanel.java
* Copyright (C) 1999-2012 University of Waikato, Hamilton, New Zealand
*
*/
package weka.gui.streams;
import java.awt.BorderLayout;
import java.awt.Color;
import java.awt.Label;
import java.awt.Panel;
import java.awt.TextField;
import java.io.FileOutputStream;
import java.io.PrintWriter;
import weka.core.Instance;
import weka.core.Instances;
/**
* A bean that saves a stream of instances to a file.
*
* @author Len Trigg ([email protected])
* @version $Revision: 8034 $
*/
public class InstanceSavePanel
extends Panel
implements InstanceListener {
/** for serialization */
private static final long serialVersionUID = -6061005366989295026L;
private Label count_Lab;
private int m_Count;
private TextField arffFile_Tex;
private boolean b_Debug;
private PrintWriter outputWriter;
public void input(Instance instance) throws Exception {
if (b_Debug)
System.err.println("InstanceSavePanel::input(" + instance +")");
m_Count++;
count_Lab.setText(""+m_Count+" instances");
if (outputWriter != null)
outputWriter.println(instance.toString());
}
public void inputFormat(Instances instanceInfo) {
if (b_Debug)
System.err.println("InstanceSavePanel::inputFormat()\n"
+instanceInfo.toString());
m_Count = 0;
count_Lab.setText(""+m_Count+" instances");
try {
outputWriter = new PrintWriter(new FileOutputStream(arffFile_Tex.getText()));
outputWriter.println(instanceInfo.toString());
if (b_Debug)
System.err.println("InstanceSavePanel::inputFormat() - written header");
} catch (Exception ex) {
outputWriter = null;
System.err.println("InstanceSavePanel::inputFormat(): "+ex.getMessage());
}
}
public void batchFinished() {
if (b_Debug)
System.err.println("InstanceSavePanel::batchFinished()");
if (outputWriter != null)
outputWriter.close();
}
public InstanceSavePanel() {
setLayout(new BorderLayout());
arffFile_Tex = new TextField("arffoutput.arff");
add("Center", arffFile_Tex);
count_Lab = new Label("0 instances");
add("East", count_Lab);
// setSize(60,40);
setBackground(Color.lightGray);
}
public void setDebug(boolean debug) {
b_Debug = debug;
}
public boolean getDebug() {
return b_Debug;
}
public void setArffFile(String newArffFile) {
arffFile_Tex.setText(newArffFile);
}
public String getArffFile() {
return arffFile_Tex.getText();
}
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("InstanceSavePanel::instanceProduced() - unknown event type");
break;
}
} catch (Exception ex) {
System.err.println(ex.getMessage());
}
} else {
System.err.println("InstanceSavePanel::instanceProduced() - Unknown source object type");
}
}
}