remoteexperimentsubtask.java

来自「wekaUT是 university texas austin 开发的基于wek」· Java 代码 · 共 97 行

/*
 *    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.
 */

/*
 *    RemoteExperimentSubTask.java
 *    Copyright (C) 2000 Mark Hall
 *
 */

package weka.experiment;

import java.io.File;

/**
 * Class to encapsulate an experiment as a task that can be executed on
 * a remote host.
 *
 * @author Mark Hall ([e-mail])
 * @version $Revision: 1.1.1.1 $
 */
public class RemoteExperimentSubTask implements Task {

    /* The (sub) experiment to execute */
    private Experiment m_experiment;

    /**
     * Set the experiment for this sub task
     * @param task the experiment
     */
    public void setExperiment(Experiment task) {
      m_experiment = task;
    }

    /**
     * Get the experiment for this sub task
     * @return this sub task's experiment
     */
    public Experiment getExperiment() {
      return m_experiment;
    }
    
    /**
     * Run the experiment
     */
    public TaskStatusInfo execute() {
      //      FastVector result = new FastVector();
      TaskStatusInfo result = new TaskStatusInfo();
      result.setStatusMessage("Running...");
      String goodResult = "(sub)experiment completed successfully";
      String subTaskType;
      if (m_experiment.getRunLower() != m_experiment.getRunUpper()) {
	subTaskType = "(datataset "
	  + ((File)m_experiment.getDatasets().elementAt(0)).getName();
      } else {
	subTaskType = "(exp run # "+
	  m_experiment.getRunLower();
      }
      try {	
	System.err.println("Initializing " + subTaskType + ")...");
	m_experiment.initialize();
	System.err.println("Iterating " + subTaskType + ")...");
	m_experiment.runExperiment();
	System.err.println("Postprocessing " + subTaskType + ")...");
	m_experiment.postProcess();
      } catch (Exception ex) {
	ex.printStackTrace();
	String badResult =  "(sub)experiment " + subTaskType 
	  + ") failed : "+ex.toString();
	result.setExecutionStatus(TaskStatusInfo.FAILED);
	//	result.addElement(new Integer(RemoteExperiment.FAILED));
	//	result.addElement(badResult);
	result.setStatusMessage(badResult);
	result.setTaskResult("Failed");
	return result;
      }            
      //      result.addElement(new Integer(RemoteExperiment.FINISHED));
      //      result.addElement(goodResult);
      result.setExecutionStatus(TaskStatusInfo.FINISHED);
      result.setStatusMessage(goodResult+" "+subTaskType+").");
      result.setTaskResult("No errors");
      return result;
    }
}