/*
* This file is part of JGrasstools (http://www.jgrasstools.org)
* (C) HydroloGIS - www.hydrologis.com
*
* JGrasstools 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 3 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, see <http://www.gnu.org/licenses/>.
*/
package org.jgrasstools.grass.utils;
import java.io.BufferedReader;
import java.io.File;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.PrintStream;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
/**
* Runner for GRASS commands.
*
* @author Andrea Antonello (www.hydrologis.com)
*/
public class GrassModuleRunnerWithScript {
private List<GrassRunnerListener> listeners = new ArrayList<>();
private final PrintStream outputStream;
private final PrintStream errorStream;
private StringBuilder outSb = new StringBuilder();
private StringBuilder errSb = new StringBuilder();
public GrassModuleRunnerWithScript( final PrintStream outputStream, final PrintStream errorStream ) {
this.outputStream = outputStream;
this.errorStream = errorStream;
}
public String runModule( String[] cmdArgs, final String mapset ) throws Exception {
String gisbaseProperty = System.getProperty(GrassUtils.GRASS_ENVIRONMENT_GISBASE_KEY);
if (gisbaseProperty == null || !new File(gisbaseProperty).exists()) {
throw new IllegalArgumentException("GISBASE environment not set.");
}
String shellProperty = System.getProperty(GrassUtils.GRASS_ENVIRONMENT_SHELL_KEY);
String shell = shellProperty;
if (shell == null && (GrassUtils.isUnix() || GrassUtils.isMacOSX())) {
shell = "/bin/sh";
}
if (shell == null) {
throw new IllegalArgumentException("SHELL not set.");
}
StringBuilder grassCommandSb = new StringBuilder();
for( String cmdArg : cmdArgs ) {
grassCommandSb.append(" ").append(cmdArg);
}
String grassCommand = grassCommandSb.toString().trim();
File runScript = GrassUtilsSextante.createGrassModuleRunScript(grassCommand, gisbaseProperty, mapset, shell);
if (GrassUtils.isWindows()) {
cmdArgs = new String[]{runScript.getAbsolutePath()};
} else {
cmdArgs = new String[]{shell, runScript.getAbsolutePath()};
}
final boolean[] outputDone = {false};
final boolean[] errorDone = {false};
ProcessBuilder processBuilder = new ProcessBuilder(cmdArgs);
String homeDir = System.getProperty("java.home");
processBuilder.directory(new File(homeDir));
final Process process = processBuilder.start();
Thread outputThread = new Thread(){
public void run() {
try {
InputStream is = process.getInputStream();
InputStreamReader isr = new InputStreamReader(is);
BufferedReader br = new BufferedReader(isr);
String line;
while( (line = br.readLine()) != null ) {
print(line);
}
} catch (Exception e) {
e.printStackTrace();
printErr(e.getLocalizedMessage());
} finally {
outputDone[0] = true;
for( GrassRunnerListener listener : listeners ) {
listener.processfinished(mapset);
}
}
}
};
Thread errorThread = new Thread(){
public void run() {
try {
InputStream is = process.getErrorStream();
InputStreamReader isr = new InputStreamReader(is);
BufferedReader br = new BufferedReader(isr);
String line;
while( (line = br.readLine()) != null ) {
print(line);
}
} catch (Exception e) {
e.printStackTrace();
printErr(e.getLocalizedMessage());
} finally {
errorDone[0] = true;
}
}
};
outputThread.start();
errorThread.start();
/*
* if streams were provided, then the user doesn't expect
* a sync result and the text of it.
*/
if (outputStream == null) {
while( !outputDone[0] && !errorDone[0] ) {
Thread.sleep(100);
}
if (errSb.length() != 0) {
return errSb.toString();
} else {
return outSb.toString();
}
}
return null;
}
public void addListener( GrassRunnerListener listener ) {
if (!listeners.contains(listener)) {
listeners.add(listener);
}
}
public void removeListener( GrassRunnerListener listener ) {
listeners.remove(listener);
}
private void print( String line ) {
if (line.contains("\u0008")) {
line = line.replaceAll("\u0008+", "\n");
}
if (outputStream != null) {
outputStream.append(line).append("\n");
} else {
outSb.append(line).append("\n");
}
}
private void printErr( String line ) {
if (errorStream != null) {
errorStream.append(line).append("\n");
} else {
errSb.append(line).append("\n");
}
}
public static void main( String[] args ) throws Exception {
System.setProperty("PATH", "/home/moovida/TMP/test/");
ProcessBuilder builder = new ProcessBuilder("list.sh");
Map<String, String> environment = builder.environment();
environment.put("PATH", "/home/moovida/TMP/test/");
Process process = builder.start();
InputStream is = process.getInputStream();
InputStreamReader isr = new InputStreamReader(is);
BufferedReader br = new BufferedReader(isr);
String line;
while( (line = br.readLine()) != null ) {
System.out.println(line);
}
}
}