/*
* 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.
*/
/*
* DatasetListPanel.java
* Copyright (C) 1999 University of Waikato, Hamilton, New Zealand
*
*/
package weka.gui.experiment;
import weka.core.ClassDiscovery.StringCompare;
import weka.core.converters.ConverterUtils;
import weka.core.converters.Saver;
import weka.core.converters.ConverterUtils.DataSource;
import weka.core.Utils;
import weka.experiment.Experiment;
import weka.gui.ConverterFileChooser;
import weka.gui.JListHelper;
import weka.gui.ViewerDialog;
import java.awt.BorderLayout;
import java.awt.GridBagConstraints;
import java.awt.GridBagLayout;
import java.awt.Insets;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.awt.event.MouseAdapter;
import java.awt.event.MouseEvent;
import java.awt.event.MouseListener;
import java.awt.event.WindowAdapter;
import java.awt.event.WindowEvent;
import java.io.File;
import java.util.Collections;
import java.util.Vector;
import javax.swing.BorderFactory;
import javax.swing.JButton;
import javax.swing.JCheckBox;
import javax.swing.JFileChooser;
import javax.swing.JFrame;
import javax.swing.JList;
import javax.swing.JOptionPane;
import javax.swing.JPanel;
import javax.swing.JScrollPane;
import javax.swing.event.ListSelectionEvent;
import javax.swing.event.ListSelectionListener;
/**
* This panel controls setting a list of datasets for an experiment to
* iterate over.
*
* @author Len Trigg (trigg@cs.waikato.ac.nz)
* @version $Revision: 1.27 $
*/
public class DatasetListPanel
extends JPanel
implements ActionListener {
/** for serialization. */
private static final long serialVersionUID = 7068857852794405769L;
/** The experiment to set the dataset list of. */
protected Experiment m_Exp;
/** The component displaying the dataset list. */
protected JList m_List;
/** Click to add a dataset. */
protected JButton m_AddBut = new JButton("Add new...");
/** Click to edit the selected algorithm. */
protected JButton m_EditBut = new JButton("Edit selected...");
/** Click to remove the selected dataset from the list. */
protected JButton m_DeleteBut = new JButton("Delete selected");
/** Click to move the selected dataset(s) one up. */
protected JButton m_UpBut = new JButton("Up");
/** Click to move the selected dataset(s) one down. */
protected JButton m_DownBut = new JButton("Down");
/** Make file paths relative to the user (start) directory. */
protected JCheckBox m_relativeCheck = new JCheckBox("Use relative paths");
/** The user (start) directory. */
// protected File m_UserDir = new File(System.getProperty("user.dir"));
/** The file chooser component. */
protected ConverterFileChooser m_FileChooser =
new ConverterFileChooser(ExperimenterDefaults.getInitialDatasetsDirectory());
/**
* Creates the dataset list panel with the given experiment.
*
* @param exp a value of type 'Experiment'
*/
public DatasetListPanel(Experiment exp) {
this();
setExperiment(exp);
}
/**
* Create the dataset list panel initially disabled.
*/
public DatasetListPanel() {
m_List = new JList();
m_List.addListSelectionListener(new ListSelectionListener() {
public void valueChanged(ListSelectionEvent e) {
setButtons(e);
}
});
MouseListener mouseListener = new MouseAdapter() {
public void mouseClicked(MouseEvent e) {
if (e.getClickCount() == 2) {
// unfortunately, locationToIndex only returns the nearest entry
// and not the exact one, i.e. if there's one item in the list and
// one doublelclicks somewhere in the list, this index will be
// returned
int index = m_List.locationToIndex(e.getPoint());
if (index > -1)
actionPerformed(new ActionEvent(m_EditBut, 0, ""));
}
}
};
m_List.addMouseListener(mouseListener);
//m_FileChooser.setFileSelectionMode(JFileChooser.FILES_AND_DIRECTORIES);
m_FileChooser.setCoreConvertersOnly(true);
m_FileChooser.setMultiSelectionEnabled(true);
m_FileChooser.setFileSelectionMode(ConverterFileChooser.FILES_AND_DIRECTORIES);
m_FileChooser.setAcceptAllFileFilterUsed(false);
m_DeleteBut.setEnabled(false);
m_DeleteBut.addActionListener(this);
m_AddBut.setEnabled(false);
m_AddBut.addActionListener(this);
m_EditBut.setEnabled(false);
m_EditBut.addActionListener(this);
m_UpBut.setEnabled(false);
m_UpBut.addActionListener(this);
m_DownBut.setEnabled(false);
m_DownBut.addActionListener(this);
m_relativeCheck.setSelected(ExperimenterDefaults.getUseRelativePaths());
m_relativeCheck.setToolTipText("Store file paths relative to "
+"the start directory");
setLayout(new BorderLayout());
setBorder(BorderFactory.createTitledBorder("Datasets"));
JPanel topLab = new JPanel();
GridBagLayout gb = new GridBagLayout();
GridBagConstraints constraints = new GridBagConstraints();
topLab.setBorder(BorderFactory.createEmptyBorder(10, 5, 10, 5));
// topLab.setLayout(new GridLayout(1,2,5,5));
topLab.setLayout(gb);
constraints.gridx=0;constraints.gridy=0;constraints.weightx=5;
constraints.fill = GridBagConstraints.HORIZONTAL;
constraints.gridwidth=1;constraints.gridheight=1;
constraints.insets = new Insets(0,2,0,2);
topLab.add(m_AddBut,constraints);
constraints.gridx=1;constraints.gridy=0;constraints.weightx=5;
constraints.gridwidth=1;constraints.gridheight=1;
topLab.add(m_EditBut,constraints);
constraints.gridx=2;constraints.gridy=0;constraints.weightx=5;
constraints.gridwidth=1;constraints.gridheight=1;
topLab.add(m_DeleteBut,constraints);
constraints.gridx=0;constraints.gridy=1;constraints.weightx=5;
constraints.fill = GridBagConstraints.HORIZONTAL;
constraints.gridwidth=1;constraints.gridheight=1;
constraints.insets = new Insets(0,2,0,2);
topLab.add(m_relativeCheck,constraints);
JPanel bottomLab = new JPanel();
gb = new GridBagLayout();
constraints = new GridBagConstraints();
bottomLab.setBorder(BorderFactory.createEmptyBorder(10, 5, 10, 5));
bottomLab.setLayout(gb);
constraints.gridx=0;constraints.gridy=0;constraints.weightx=5;
constraints.fill = GridBagConstraints.HORIZONTAL;
constraints.gridwidth=1;constraints.gridheight=1;
constraints.insets = new Insets(0,2,0,2);
bottomLab.add(m_UpBut,constraints);
constraints.gridx=1;constraints.gridy=0;constraints.weightx=5;
constraints.gridwidth=1;constraints.gridheight=1;
bottomLab.add(m_DownBut,constraints);
add(topLab, BorderLayout.NORTH);
add(new JScrollPane(m_List), BorderLayout.CENTER);
add(bottomLab, BorderLayout.SOUTH);
}
/**
* sets the state of the buttons according to the selection state of the
* JList.
*
* @param e the event
*/
private void setButtons(ListSelectionEvent e) {
if ( (e == null) || (e.getSource() == m_List) ) {
m_DeleteBut.setEnabled(m_List.getSelectedIndex() > -1);
m_EditBut.setEnabled(m_List.getSelectedIndices().length == 1);
m_UpBut.setEnabled(JListHelper.canMoveUp(m_List));
m_DownBut.setEnabled(JListHelper.canMoveDown(m_List));
}
}
/**
* Tells the panel to act on a new experiment.
*
* @param exp a value of type 'Experiment'
*/
public void setExperiment(Experiment exp) {
m_Exp = exp;
m_List.setModel(m_Exp.getDatasets());
m_AddBut.setEnabled(true);
setButtons(null);
}
/**
* Gets all the files in the given directory
* that match the currently selected extension.
*
* @param directory the directory to get the files for
* @param files the list to add the files to
*/
protected void getFilesRecursively(File directory, Vector files) {
try {
String[] currentDirFiles = directory.list();
for (int i = 0; i < currentDirFiles.length; i++) {
currentDirFiles[i] = directory.getCanonicalPath() + File.separator +
currentDirFiles[i];
File current = new File(currentDirFiles[i]);
if (m_FileChooser.getFileFilter().accept(current)) {
if (current.isDirectory()) {
getFilesRecursively(current, files);
} else {
files.addElement(current);
}
}
}
} catch (Exception e) {
System.err.println("IOError occured when reading list of files");
}
}
/**
* Handle actions when buttons get pressed.
*
* @param e a value of type 'ActionEvent'
*/
public void actionPerformed(ActionEvent e) {
boolean useRelativePaths = m_relativeCheck.isSelected();
if (e.getSource() == m_AddBut) {
// Let the user select an arff file from a file chooser
int returnVal = m_FileChooser.showOpenDialog(this);
if(returnVal == JFileChooser.APPROVE_OPTION) {
if (m_FileChooser.isMultiSelectionEnabled()) {
File [] selected = m_FileChooser.getSelectedFiles();
for (int i = 0; i < selected.length; i++) {
if (selected[i].isDirectory()) {
Vector files = new Vector();
getFilesRecursively(selected[i], files);
// sort the result
Collections.sort(files, new StringCompare());
for (int j = 0; j < files.size(); j++) {
File temp = (File)files.elementAt(j);
if (useRelativePaths) {
try {
temp = Utils.convertToRelativePath(temp);
} catch (Exception ex) {
ex.printStackTrace();
}
}
m_Exp.getDatasets().addElement(temp);
}
} else {
File temp = selected[i];
if (useRelativePaths) {
try {
temp = Utils.convertToRelativePath(temp);
} catch (Exception ex) {
ex.printStackTrace();
}
}
m_Exp.getDatasets().addElement(temp);
}
}
setButtons(null);
} else {
if (m_FileChooser.getSelectedFile().isDirectory()) {
Vector files = new Vector();
getFilesRecursively(m_FileChooser.getSelectedFile(), files);
// sort the result
Collections.sort(files, new StringCompare());
for (int j = 0; j < files.size(); j++) {
File temp = (File)files.elementAt(j);
if (useRelativePaths) {
try {
temp = Utils.convertToRelativePath(temp);
} catch (Exception ex) {
ex.printStackTrace();
}
}
m_Exp.getDatasets().addElement(temp);
}
} else {
File temp = m_FileChooser.getSelectedFile();
if (useRelativePaths) {
try {
temp = Utils.convertToRelativePath(temp);
} catch (Exception ex) {
ex.printStackTrace();
}
}
m_Exp.getDatasets().addElement(temp);
}
setButtons(null);
}
}
} else if (e.getSource() == m_DeleteBut) {
// Delete the selected files
int [] selected = m_List.getSelectedIndices();
if (selected != null) {
for (int i = selected.length - 1; i >= 0; i--) {
int current = selected[i];
m_Exp.getDatasets().removeElementAt(current);
if (m_Exp.getDatasets().size() > current) {
m_List.setSelectedIndex(current);
} else {
m_List.setSelectedIndex(current - 1);
}
}
}
setButtons(null);
} else if (e.getSource() == m_EditBut) {
// Delete the selected files
int selected = m_List.getSelectedIndex();
if (selected != -1) {
ViewerDialog dialog = new ViewerDialog(null);
String filename = m_List.getSelectedValue().toString();
int result;
try {
DataSource source = new DataSource(filename);
result = dialog.showDialog(source.getDataSet());
// nasty workaround for Windows regarding locked files:
// if file Reader in Loader is not closed explicitly, we cannot
// overwrite the file.
source = null;
System.gc();
// workaround end
if ((result == ViewerDialog.APPROVE_OPTION) && (dialog.isChanged())) {
result = JOptionPane.showConfirmDialog(
this,
"File was modified - save changes?");
if (result == JOptionPane.YES_OPTION) {
Saver saver = ConverterUtils.getSaverForFile(filename);
saver.setFile(new File(filename));
saver.setInstances(dialog.getInstances());
saver.writeBatch();
}
}
}
catch (Exception ex) {
JOptionPane.showMessageDialog(
this,
"Error loading file '" + filename + "':\n" + ex.toString(),
"Error loading file",
JOptionPane.INFORMATION_MESSAGE);
}
}
setButtons(null);
} else if (e.getSource() == m_UpBut) {
JListHelper.moveUp(m_List);
} else if (e.getSource() == m_DownBut) {
JListHelper.moveDown(m_List);
}
}
/**
* Tests out the dataset list panel from the command line.
*
* @param args ignored
*/
public static void main(String [] args) {
try {
final JFrame jf = new JFrame("Dataset List Editor");
jf.getContentPane().setLayout(new BorderLayout());
DatasetListPanel dp = new DatasetListPanel();
jf.getContentPane().add(dp,
BorderLayout.CENTER);
jf.addWindowListener(new WindowAdapter() {
public void windowClosing(WindowEvent e) {
jf.dispose();
System.exit(0);
}
});
jf.pack();
jf.setVisible(true);
System.err.println("Short nap");
Thread.currentThread().sleep(3000);
System.err.println("Done");
dp.setExperiment(new Experiment());
} catch (Exception ex) {
ex.printStackTrace();
System.err.println(ex.getMessage());
}
}
}