use of edu.rit.pj.WorkerIteration in project ffx by mjschnie.
the class Runner method main.
// Main program.
/**
* Main program.
*
* @param args an array of {@link java.lang.String} objects.
* @throws java.lang.Exception if any.
*/
public static void main(String[] args) throws Exception {
if (args.length != 1) {
usage();
}
// Initialize world communicator.
Comm.init(args);
world = Comm.world();
rank = world.rank();
// Set up worker team.
team = new WorkerTeam();
// Master process sets up job generator.
if (rank == team.masterRank()) {
omitted = new HashSet<Integer>();
// Assume argument is a checkpoint file name and try to read it.
Scanner scanner = null;
try {
scanner = new Scanner(new File(args[0]));
} catch (IOException exc) {
}
// Read checkpoint file.
if (scanner != null) {
while (scanner.hasNextLine()) {
Scanner linescanner = new Scanner(scanner.nextLine());
String word;
int jobnum;
if (!linescanner.hasNext()) {
continue;
}
word = linescanner.next();
if (!word.equals("***")) {
continue;
}
if (!linescanner.hasNext()) {
continue;
}
word = linescanner.next();
if (word.equals("Generator")) {
if (!linescanner.hasNext()) {
continue;
}
if (generatorExpression == null) {
generatorExpression = linescanner.next();
}
} else if (word.equals("Job")) {
if (!linescanner.hasNextInt()) {
continue;
}
jobnum = linescanner.nextInt();
if (!linescanner.hasNext()) {
continue;
}
word = linescanner.next();
if (word.equals("finished")) {
omitted.add(jobnum);
}
}
}
scanner.close();
} else // Assume argument is a job generator constructor expression.
{
generatorExpression = args[0];
}
// Create job generator.
if (generatorExpression == null) {
stderr.printf("Runner: No job generator in checkpoint file %s%n", args[0]);
} else {
try {
generator = (JobGenerator) Instance.newInstance(generatorExpression);
stdout.printf("*** Generator %s%n", generatorExpression);
generator.omit(omitted);
} catch (Throwable exc) {
stderr.printf("Runner: Could not create job generator %s%n", generatorExpression);
exc.printStackTrace(stderr);
}
}
}
// Abort every process if job generator was not created.
BooleanItemBuf buf = BooleanBuf.buffer(generator != null);
world.broadcast(team.masterRank(), buf);
if (!buf.item) {
System.exit(1);
}
// Generate and run jobs.
team.execute(new WorkerRegion() {
public void run() throws Exception {
execute(generator, new WorkerIteration<Job>() {
public void sendTaskInput(Job job, Comm comm, int wRank, int tag) {
stdout.printf("*** Job %d started %s %s%n", job.getJobNumber(), new Date(), job.getDescription());
}
public void run(Job job) {
job.run();
}
public void receiveTaskOutput(Job job, Comm comm, int wRank, int tag) {
stdout.printf("*** Job %d finished %s%n", job.getJobNumber(), new Date());
}
});
}
});
if (rank == team.masterRank()) {
stdout.printf("*** All jobs finished%n");
}
}
Aggregations