kdbexamplesource.java

著名的开源仿真软件yale

/*
 *  YALE - Yet Another Learning Environment
 *  Copyright (C) 2002, 2003
 *      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.operator;

import edu.udo.cs.yale.operator.parameter.*;
import edu.udo.cs.yale.example.Attribute;
import edu.udo.cs.yale.tools.LogService;
import org.kobjects.jdbc.TableManager;
import org.kobjects.jdbc.Column;
import org.kobjects.jdbc.util.AbstractResultSet;

import java.util.List;
import java.util.Iterator;
import java.sql.ResultSet;

/** This operator can read arff, comma separated values (csv), dbase and bibtex files.
 *  It uses Stefan Haustein's kdb tools.
 *  
 *  @yale.index arff
 *  @yale.index dbase
 *  @yale.index csv
 *  @yale.index BibTeX
 *  @yale.xmlclass KDBExampleSource
 *  @version $Id: KDBExampleSource.java,v 2.8 2003/07/02 13:57:10 fischer Exp $
 */
public class KDBExampleSource extends ResultSetExampleSource {

    private static final String[] FORMAT_TYPES = { "arff", "csv", "dbase", "bibtex" };

    public ResultSet getResultSet() {
	String dataFile    = getParameterAsString("data_file");
	int format         = getParameterAsInt("format");

	String dataFileAbsolutePath = getExperiment().resolveFileName(dataFile).getAbsolutePath();
	return TableManager.getResultSet(FORMAT_TYPES[format]+":"+dataFileAbsolutePath,
					 TableManager.READ);
    }

    public void setNominalValues(List attributeList, ResultSet resultSet, Attribute label) {
	if (resultSet instanceof AbstractResultSet) {
	    AbstractResultSet ars = (AbstractResultSet)resultSet;
	    Iterator i = attributeList.iterator();
	    int j = 0;
	    while (i.hasNext()) {
		j++;
		Attribute attribute = (Attribute)i.next();
		Object[] values = ars.getColumnSet().getColumn(j).getValues();
		if (attribute.isNominal()) {
		    if (values == null) {
			LogService.logMessage("Information about class values is null!", LogService.WARNING);
		    } else {
			for (int k = 0; k < values.length; k++) {
			    attribute.mapString(values[k].toString());
			}
		    }
		}
	    }
	} else {
	    LogService.logMessage("Result set does not provide information about class values!", LogService.WARNING);
	}
    }

    public List getParameterTypes() {
	List types = super.getParameterTypes();
	types.add(new ParameterTypeFile("data_file", "The file containing the data", false));
	types.add(new ParameterTypeCategory("format", "The data file format.",
					    FORMAT_TYPES, 0));
	return types;
    }

}