saveastemplatedialog.java

一个很好的LIBSVM的JAVA源码。对于要研究和改进SVM算法的学者。可以参考。来自数据挖掘工具YALE工具包。

/*
 *  YALE - Yet Another Learning Environment
 *  Copyright (C) 2001-2004
 *      Simon Fischer, Ralf Klinkenberg, Ingo Mierswa, 
 *          Katharina Morik, Oliver Ritthoff
 *      Artificial Intelligence Unit
 *      Computer Science Department
 *      University of Dortmund
 *      44221 Dortmund,  Germany
 *  email: [e-mail]
 *  web:   http://yale.cs.uni-dortmund.de/
 *
 *  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., 59 Temple Place, Suite 330, Boston, MA 02111-1307
 *  USA.
 */
package edu.udo.cs.yale.gui;

import edu.udo.cs.yale.Yale;
import edu.udo.cs.yale.Experiment;
import edu.udo.cs.yale.operator.Operator;
import edu.udo.cs.yale.operator.OperatorChain;
import edu.udo.cs.yale.operator.parameter.ParameterType;
import edu.udo.cs.yale.tools.ParameterService;
import edu.udo.cs.yale.tools.Tools;

import javax.swing.*;
import javax.swing.table.*;
import java.awt.Component;
import java.awt.BorderLayout;
import java.awt.FlowLayout;
import java.awt.GridBagLayout;
import java.awt.GridBagConstraints;
import java.awt.event.*;
import java.io.File;
import java.io.IOException;
import java.io.FileFilter;
import java.util.List;
import java.util.LinkedList;
import java.util.Collections;
import java.util.Iterator;
import java.util.Set;

/** The save as template dialog assists the user in creating a new experiment template. Template experiments 
 *  are saved in the local .yale directory of the user. The name, description and additional parameters to set 
 *  can be specified by the user. */
public class SaveAsTemplateDialog extends JDialog {

    private class OperatorParameterPair implements Comparable {
	private String[] pair;
	
	public OperatorParameterPair(String[] pair) {
	    this.pair = pair;
	}

	public int compareTo(Object o) {
	    OperatorParameterPair opp = (OperatorParameterPair)o;
	    return (this.pair[0] + "." + this.pair[1]).compareTo(opp.pair[0] + "." + opp.pair[1]);
	}

	public String toString() {
	    return (this.pair[0] + "." + this.pair[1]);
	}

	public String[] getStringPair() { return pair; }
    }

    private class TemplateParameterTableModel extends AbstractTableModel {

	/** 0: operator, 1: parameter key */
	private OperatorParameterPair[] parameters;
	private boolean[] selection;

	public TemplateParameterTableModel(OperatorParameterPair[] parameters) {
	    this.parameters = parameters;
	    this.selection = new boolean[parameters.length];
	}

	public String getColumnName(int column) {
	    if (column == 0) return "Parameter name";
	    else return "Selected for Template";
	}

	public Object getValueAt(int row, int column) {
	    if (column == 0) return parameters[row].toString();
	    else return new Boolean(selection[row]);
	}

 	public void setValueAt(Object o, int row, int column) {
	    if (column == 1) {
		selection[row] = ((Boolean)o).booleanValue();
		fireTableCellUpdated(row, column);
	    }
 	}

	public boolean isCellEditable(int row, int column) {
	    return column == 1;
	}

	public int getRowCount() {
	    return parameters.length;
	}
	
	public int getColumnCount() {
	    return 2;
	}

	/** Makes sure that a checkbox is used for the boolean values. */
	public Class getColumnClass(int column) {
	    return getValueAt(0, column).getClass();
	}

	public List getSelectedParameters() {
	    List selected = new LinkedList();
	    for (int i = 0; i < parameters.length; i++) {
		if (selection[i])
		    selected.add(parameters[i].getStringPair());
	    }
	    return selected;
	}
    }

    private boolean ok = false;
    private JTextField nameField  = new JTextField();
    private JTextField descriptionField = new JTextField();
    private TemplateParameterTableModel tableModel = null;


    public SaveAsTemplateDialog(MainFrame mainFrame, Operator operator) {
	super(mainFrame, "Save as Template", true);

	JPanel rootPanel = new JPanel(new BorderLayout());
	rootPanel.setBorder(BorderFactory.createEmptyBorder(10, 10, 10, 10));
	GridBagLayout layout = new GridBagLayout();
	JPanel mainPanel = new JPanel(layout);
	GridBagConstraints c = new GridBagConstraints();
	c.fill = GridBagConstraints.BOTH;
	c.weightx = 1.0d;

	JPanel textPanel = SwingTools.createTextPanel("Save this experiment as Template...",
						      "Please specify a name and short description of this template. " +
						      "The selected parameters can be determined by the user in addition " +
						      "to the mandatory parameters.");
	c.gridwidth = GridBagConstraints.REMAINDER;
	layout.setConstraints(textPanel, c);
	mainPanel.add(textPanel);

	Component sep = Box.createVerticalStrut(10);
	c.gridwidth = GridBagConstraints.REMAINDER;
	layout.setConstraints(sep, c);
	mainPanel.add(sep);

	// add components to main panel
	JLabel label = new JLabel("Name");
	label.setToolTipText("The name of the template.");
	c.gridwidth = GridBagConstraints.RELATIVE;
	layout.setConstraints(label, c);
	mainPanel.add(label);


	nameField.setToolTipText("The name of the template.");
	c.gridwidth = GridBagConstraints.REMAINDER;
	layout.setConstraints(nameField, c);
	mainPanel.add(nameField);


	label = new JLabel("Description");
	label.setToolTipText("A short description of this template.");
	c.gridwidth = GridBagConstraints.RELATIVE;
	layout.setConstraints(label, c);
	mainPanel.add(label);

	descriptionField.setToolTipText("A short description of this template.");
	c.gridwidth = GridBagConstraints.REMAINDER;
	layout.setConstraints(descriptionField, c);
	mainPanel.add(descriptionField);

	sep = Box.createVerticalStrut(10);
	c.gridwidth = GridBagConstraints.REMAINDER;
	layout.setConstraints(sep, c);
	mainPanel.add(sep);

	tableModel = new TemplateParameterTableModel(getParameters(operator));
	JTable table = new JTable(tableModel);
	JScrollPane tablePane = new JScrollPane(table);
	table.setPreferredScrollableViewportSize(new java.awt.Dimension(600, 200));
	table.getTableHeader().setToolTipText("Select the parameters which could be defined by the user.");
	table.setRowHeight(table.getRowHeight()+SwingTools.TABLE_ROW_EXTRA_HEIGHT);
	table.getTableHeader().setReorderingAllowed(false);

	c.weighty = 1.0d;
	c.gridwidth = GridBagConstraints.REMAINDER;
	layout.setConstraints(tablePane, c);
	mainPanel.add(tablePane);

	// buttons
	JPanel buttonPanel = new JPanel(new FlowLayout(FlowLayout.CENTER));
	JButton okButton = new JButton("Ok");
	okButton.addActionListener(new ActionListener() {
		public void actionPerformed(ActionEvent e) { ok(); }
	    });
	buttonPanel.add(okButton);
	JButton cancelButton = new JButton("Cancel");
	cancelButton.addActionListener(new ActionListener() {
		public void actionPerformed(ActionEvent e) { cancel(); }
	    });
	buttonPanel.add(cancelButton);


	rootPanel.add(mainPanel, BorderLayout.CENTER);
	rootPanel.add(buttonPanel, BorderLayout.SOUTH);
	getContentPane().add(rootPanel);
	pack();
	setSize(700,500);
	setLocationRelativeTo(mainFrame);
    }

    private OperatorParameterPair[] getParameters(Operator operator) {
	List parameters = new LinkedList();
	addParameters(operator, parameters);
	Collections.sort(parameters);
	OperatorParameterPair[] result = new OperatorParameterPair[parameters.size()];
	parameters.toArray(result);
	return result;
    }

    private void addParameters(Operator operator, List parameters) {
	Iterator i = operator.getParameterTypes().iterator();
	while (i.hasNext()) {
	    ParameterType type = (ParameterType)i.next();
	    Object value = operator.getParameters().getParameter(type.getKey());
	    if (type.isOptional())
		parameters.add(new OperatorParameterPair(new String[] { operator.getName(), type.getKey() }));
	}
	if (operator instanceof OperatorChain) {
	    OperatorChain chain = (OperatorChain)operator;
	    for (int n = 0; n < chain.getNumberOfOperators(); n++) {
		addParameters(chain.getOperator(n), parameters);
	    }
	}
    }

    private void addMandatoryParameters(Operator operator, List parameters) {
	Iterator i = operator.getParameterTypes().iterator();
	while (i.hasNext()) {
	    ParameterType type = (ParameterType)i.next();
	    Object value = operator.getParameters().getParameter(type.getKey());
	    if (!type.isOptional())
		parameters.add(new String[] { operator.getName(), type.getKey() });
	}
	if (operator instanceof OperatorChain) {
	    OperatorChain chain = (OperatorChain)operator;
	    for (int n = 0; n < chain.getNumberOfOperators(); n++) {
		addMandatoryParameters(chain.getOperator(n), parameters);
	    }
	}
    }

    public boolean isOk() { return ok; }

    public Template getTemplate(Operator operator) { 
	String name = nameField.getText();
	List selectedOptional = tableModel.getSelectedParameters();
	addMandatoryParameters(operator, selectedOptional);
	return new Template(name, descriptionField.getText(), name + ".xml", selectedOptional);
    }

    private boolean checkIfNameOk() {
	String name = nameField.getText();
	if ((name == null) || (name.length() == 0)) {
	    SwingTools.showVerySimpleErrorMessage("Please specify a name for this template!");
	    return false;
	}

	File[] preDefinedTemplateFiles = ParameterService.getConfigFile("templates").listFiles(new FileFilter() {
		public boolean accept(File file) {
		    return file.getName().endsWith(".template");
		}
	    });
	File[] userDefinedTemplateFiles = ParameterService.getUserYaleDir().listFiles(new FileFilter() {
		public boolean accept(File file) {
		    return file.getName().endsWith(".template");
		}
	    });

	File[] templateFiles = new File[preDefinedTemplateFiles.length + userDefinedTemplateFiles.length];
	System.arraycopy(preDefinedTemplateFiles, 0, templateFiles, 0, preDefinedTemplateFiles.length);
	System.arraycopy(userDefinedTemplateFiles, 0, templateFiles, preDefinedTemplateFiles.length, userDefinedTemplateFiles.length);

	for (int i = 0; i < templateFiles.length; i++) {
	    String tempName = templateFiles[i].getName().substring(0, templateFiles[i].getName().lastIndexOf("."));
	    if (tempName.equals(name)) {
		SwingTools.showVerySimpleErrorMessage("Name '" + name + "' is already used.\nPlease change name or delete the old template!");
		return false;
	    }
	}
	return true;
    }

    private void ok() {
	if (checkIfNameOk()) {
	    ok = true;
	    dispose();
	}
    }

    private void cancel() {
	ok = false;
	dispose();
    }
}