resultsetexamplesource.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.tools.DatabaseHandler;
import edu.udo.cs.yale.example.ExampleSet;
import edu.udo.cs.yale.example.ExampleTable;
import edu.udo.cs.yale.example.MemoryExampleTable;
import edu.udo.cs.yale.example.DataRowReader;
import edu.udo.cs.yale.example.ResultSetDataRowReader;
import edu.udo.cs.yale.example.DataRowFactory;
import edu.udo.cs.yale.example.Attribute;

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

/** Abstract superclass for operators that provide access to an {@link ExampleSet} via a {@link ResultSet}.
 *  @version $Id: ResultSetExampleSource.java,v 2.4 2003/06/18 17:11:17 fischer Exp $
 */
public abstract class ResultSetExampleSource extends Operator {

    private static final Class[] INPUT_CLASSES = {};
    private static final Class[] OUTPUT_CLASSES = { ExampleSet.class };

    /** Returns a {@link ResultSet}. */
    public abstract ResultSet getResultSet() throws OperatorException;

    /** Since the {@link ResultSet} does not provide information about possible values
     *  of nominal attributes, subclasses must set these by implementing this method. 
     *  @param attributeList List of {@link Attribute} */
    public abstract void setNominalValues(List attributeList, 
					  ResultSet resultSet,
					  Attribute label) throws OperatorException;

    public IOObject[] apply() throws OperatorException {
	int dataRowType   = getParameterAsInt("datamanagement");
	ResultSet resultSet = getResultSet();
	List attributeList  = null;
	try {
	    attributeList  = DatabaseHandler.createAttributes(resultSet);
	} catch (SQLException e) {
	    throw new UserError(this, e, 304, e.getMessage());
	}
	setNominalValues(attributeList, resultSet, find(attributeList, getParameterAsString("label_attribute")));
	DataRowReader reader = new ResultSetDataRowReader(new DataRowFactory(dataRowType),
							  attributeList,
							  resultSet);
	ExampleTable table = new MemoryExampleTable(attributeList, reader);
	return new IOObject[] { createExampleSet(table) };
    }

    private static Attribute find(List attributeList, String name) throws OperatorException {
	if (name == null) return null;
	Iterator i = attributeList.iterator();
	while (i.hasNext()) {
	    Attribute attribute = (Attribute)i.next();
	    if (attribute.getName().equals(name))
		return attribute;
	}
	throw new UserError(null, 111, name);
    }

    protected ExampleSet createExampleSet(ExampleTable table) throws OperatorException {
	String labelName  = getParameterAsString("label_attribute");
	String weightName = getParameterAsString("weight_attribute");
	String idName     = getParameterAsString("id_attribute");
	Attribute label  = table.findAttribute(labelName); 
	Attribute weight = table.findAttribute(weightName);
	Attribute id     = table.findAttribute(idName);
	return table.createCompleteExampleSet(label, null, weight, id);
    }

    public List getParameterTypes() {
	List types = super.getParameterTypes();

	types.add(new ParameterTypeString("label_attribute", "The (case sensitive) name of the label attribute"));
	types.add(new ParameterTypeString("weight_attribute", "The (case sensitive) name of the weight attribute"));
	types.add(new ParameterTypeString("id_attribute", "The (case sensitive) name of the id attribute"));
	types.add(new ParameterTypeCategory("datamanagement", "Determines, how the data is represented internally.",
					    DataRowFactory.TYPE_NAMES, DataRowFactory.TYPE_DOUBLE_ARRAY));
	return types;
    }

    public Class[] getInputClasses() {
	return INPUT_CLASSES;
    }

    public Class[] getOutputClasses() {
	return OUTPUT_CLASSES;
    }

}