/*
* DoTask.java
* Copyright (C) 2007 University of Waikato, Hamilton, New Zealand
* @author Richard Kirkby (rkirkby@cs.waikato.ac.nz)
*
* 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.
*/
package tr.gov.ulakbim.jDenetX;
import tr.gov.ulakbim.jDenetX.core.Globals;
import tr.gov.ulakbim.jDenetX.core.Measurement;
import tr.gov.ulakbim.jDenetX.core.StringUtils;
import tr.gov.ulakbim.jDenetX.core.TimingUtils;
import tr.gov.ulakbim.jDenetX.options.ClassOption;
import tr.gov.ulakbim.jDenetX.options.FlagOption;
import tr.gov.ulakbim.jDenetX.options.IntOption;
import tr.gov.ulakbim.jDenetX.options.Option;
import tr.gov.ulakbim.jDenetX.tasks.FailedTaskReport;
import tr.gov.ulakbim.jDenetX.tasks.Task;
import tr.gov.ulakbim.jDenetX.tasks.TaskThread;
public class DoTask {
public static final char[] progressAnimSequence = new char[]{'-', '\\',
'|', '/'};
public static final int MAX_STATUS_STRING_LENGTH = 79;
public static void main(String[] args) {
try {
if (args.length < 1) {
System.err.println();
System.err.println(Globals.getWorkbenchInfoString());
System.err.println();
System.err.println("No task specified.");
} else {
// create standard options
FlagOption suppressStatusOutputOption = new FlagOption(
"suppressStatusOutput", 'S',
"Suppress the task status output that is normally send to stderr.");
FlagOption suppressResultOutputOption = new FlagOption(
"suppressResultOutput", 'R',
"Suppress the task result output that is normally send to stdout.");
IntOption statusUpdateFrequencyOption = new IntOption(
"statusUpdateFrequency",
'F',
"How many milliseconds to wait between status updates.",
1000, 0, Integer.MAX_VALUE);
Option[] extraOptions = new Option[]{
suppressStatusOutputOption, suppressResultOutputOption,
statusUpdateFrequencyOption};
// build a single string by concatenating cli options
StringBuilder cliString = new StringBuilder();
for (int i = 0; i < args.length; i++) {
cliString.append(" " + args[i]);
}
// parse options
Task task = (Task) ClassOption.cliStringToObject(cliString
.toString(), Task.class, extraOptions);
Object result = null;
if (suppressStatusOutputOption.isSet()) {
result = task.doTask();
} else {
System.err.println();
System.err.println(Globals.getWorkbenchInfoString());
System.err.println();
boolean preciseTiming = TimingUtils.enablePreciseTiming();
// start the task thread
TaskThread taskThread = new TaskThread(task);
taskThread.start();
int progressAnimIndex = 0;
// inform user of progress
while (!taskThread.isComplete()) {
StringBuilder progressLine = new StringBuilder();
progressLine
.append(progressAnimSequence[progressAnimIndex]);
progressLine.append(' ');
progressLine.append(StringUtils
.secondsToDHMSString(taskThread
.getCPUSecondsElapsed()));
progressLine.append(" [");
progressLine
.append(taskThread.getCurrentStatusString());
progressLine.append("] ");
double fracComplete = taskThread
.getCurrentActivityFracComplete();
if (fracComplete >= 0.0) {
progressLine.append(StringUtils.doubleToString(
fracComplete * 100.0, 2, 2));
progressLine.append("% ");
}
progressLine.append(taskThread
.getCurrentActivityString());
while (progressLine.length() < MAX_STATUS_STRING_LENGTH) {
progressLine.append(" ");
}
if (progressLine.length() > MAX_STATUS_STRING_LENGTH) {
progressLine.setLength(MAX_STATUS_STRING_LENGTH);
progressLine.setCharAt(
MAX_STATUS_STRING_LENGTH - 1, '~');
}
System.err.print(progressLine.toString());
System.err.print('\r');
if (++progressAnimIndex >= progressAnimSequence.length) {
progressAnimIndex = 0;
}
try {
Thread
.sleep(statusUpdateFrequencyOption
.getValue());
} catch (InterruptedException ignored) {
// wake up
}
}
StringBuilder cleanupString = new StringBuilder();
for (int i = 0; i < MAX_STATUS_STRING_LENGTH; i++) {
cleanupString.append(' ');
}
System.err.println(cleanupString);
result = taskThread.getFinalResult();
if (!(result instanceof FailedTaskReport)) {
System.err.print("Task completed in "
+ StringUtils.secondsToDHMSString(taskThread
.getCPUSecondsElapsed()));
if (preciseTiming) {
System.err.print(" (CPU time)");
}
System.err.println();
System.err.println();
}
}
if (result instanceof FailedTaskReport) {
System.err.println("Task failed. Reason: ");
((FailedTaskReport) result).getFailureReason()
.printStackTrace();
} else {
if (!suppressResultOutputOption.isSet()) {
if (result instanceof Measurement[]) {
StringBuilder sb = new StringBuilder();
Measurement.getMeasurementsDescription(
(Measurement[]) result, sb, 0);
System.out.println(sb.toString());
} else {
System.out.println(result);
}
System.out.flush();
}
}
}
} catch (Exception ex) {
ex.printStackTrace();
}
}
}