Mercurial > repos > yufei-luo > s_mart

comparison SMART/Java/Sav/ProgramLauncher.java @ 31:0ab839023fe4

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Uploaded
author m-zytnicki
date Tue, 30 Apr 2013 14:33:21 -0400
parents 769e306b7933
children
comparison
equal deleted inserted replaced
30:5677346472b5 31:0ab839023fe4
1 /**
2 *
3 * Copyright INRA-URGI 2009-2010
4 *
5 * This software is governed by the CeCILL license under French law and
6 * abiding by the rules of distribution of free software. You can use,
7 * modify and/ or redistribute the software under the terms of the CeCILL
8 * license as circulated by CEA, CNRS and INRIA at the following URL
9 * "http://www.cecill.info".
10 *
11 * As a counterpart to the access to the source code and rights to copy,
12 * modify and redistribute granted by the license, users are provided only
13 * with a limited warranty and the software's author, the holder of the
14 * economic rights, and the successive licensors have only limited
15 * liability.
16 *
17 * In this respect, the user's attention is drawn to the risks associated
18 * with loading, using, modifying and/or developing or reproducing the
19 * software by the user in light of its specific status of free software,
20 * that may mean that it is complicated to manipulate, and that also
21 * therefore means that it is reserved for developers and experienced
22 * professionals having in-depth computer knowledge. Users are therefore
23 * encouraged to load and test the software's suitability as regards their
24 * requirements in conditions enabling the security of their systems and/or
25 * data to be ensured and, more generally, to use and operate it in the
26 * same conditions as regards security.
27 *
28 * The fact that you are presently reading this means that you have had
29 * knowledge of the CeCILL license and that you accept its terms.
30 *
31 */
32 import java.util.*;
33 import java.io.*;
34 import javax.swing.SwingUtilities;
35 import javax.swing.*;
36 import java.util.concurrent.CountDownLatch;
37
38 public class ProgramLauncher extends SwingWorker<Boolean, String> {
39
40 String[] command;
41 JTextArea logArea;
42 JLabel messageField;
43 JProgressBar progressBar;
44 JLabel etaField;
45 int exitValue;
46
47
48 public ProgramLauncher (LinkedList <String> c, JTextArea la, JLabel mf, JProgressBar pb, JLabel ef) {
49 command = new String[c.size()];
50 logArea = la;
51 messageField = mf;
52 progressBar = pb;
53 etaField = ef;
54 exitValue = -1;
55 c.toArray(command);
56 }
57
58
59 public ProgramLauncher (String[] c, JTextArea la, JLabel mf, JProgressBar pb, JLabel ef) {
60 command = c;
61 logArea = la;
62 messageField = mf;
63 progressBar = pb;
64 etaField = ef;
65 exitValue = -1;
66 }
67
68
69 @Override
70 public Boolean doInBackground() {
71 ProcessBuilder pb = new ProcessBuilder(command);
72 Process process = null;
73 BufferedReader outputReader = null;
74 pb = pb.redirectErrorStream(true);
75 Map<String, String> env = pb.environment();
76 env.put("PYTHONPATH", System.getProperty("user.dir") + java.io.File.separator + "Python");
77 env.put("SMARTPATH", System.getProperty("user.dir") + java.io.File.separator + "Python");
78 env.put("SMARTMYSQLPATH", Global.mysqlCommand);
79 env.put("SMARTRPATH", Global.rCommand);
80 String commandJoined = Arrays.toString(command);
81
82 try {
83 publish("=== Starting command '" + commandJoined.trim() + "' ===\n");
84 process = pb.start();
85
86 BufferedInputStream outputStream = new BufferedInputStream(process.getInputStream());
87 InputStream is = process.getInputStream();
88 InputStreamReader isr = new InputStreamReader(is);
89 outputReader = new BufferedReader(isr);
90 }
91 catch (Exception exception) {
92 publish("!Process cannot be started (command is '" + commandJoined + "')!\n");
93 exception.printStackTrace();
94 return Boolean.FALSE;
95 }
96 if (outputReader == null) {
97 publish("!Problem in the output of the command!\n");
98 return Boolean.FALSE;
99 }
100 else {
101 try {
102 String line;
103 while ((line = outputReader.readLine()) != null) {
104 publish(line + "\n");
105 }
106 }
107 catch (IOException e) {
108 e.printStackTrace();
109 publish("!Cannot get the output of the command!\n");
110 return Boolean.FALSE;
111 }
112 }
113 try {
114 process.waitFor();
115 }
116 catch (InterruptedException e) {
117 e.printStackTrace();
118 publish("!Cannot wait for the end of the command!\n");
119 return Boolean.FALSE;
120 }
121 try {
122 exitValue = process.exitValue();
123 System.out.println(exitValue);
124 }
125 catch (IllegalThreadStateException e) {
126 e.printStackTrace();
127 publish("!Cannot get the exit value of the command!\n");
128 return Boolean.FALSE;
129 }
130 if (exitValue != 0) {
131 publish("!Problem during the execution of the command '" + commandJoined + "'!\n");
132 return Boolean.FALSE;
133 }
134 publish("=== Ending command '" + commandJoined.trim() + "' ===\n");
135 return Boolean.TRUE;
136 }
137
138
139 @Override
140 protected void process(List<String> chunks) {
141 String message = "";
142 String text = logArea.getText();
143 for (String chunk: chunks) {
144 text += chunk;
145 }
146 for (String lineSeparatedByCarriageReturn: text.split("\n")) {
147 for (String line: lineSeparatedByCarriageReturn.split("\r")) {
148 boolean progressLine = false;
149 if (line.matches(".*\\[=*\\s*\\]\\s*\\d*/\\d*\\s*")) {
150 String[] ratioElements = line.split("\\]")[1].trim().split("/");
151 int current = Integer.parseInt(ratioElements[0].trim());
152 int aim = Integer.parseInt(ratioElements[1].trim());
153 messageField.setText(line.split("\\[")[0].trim());
154 progressBar.setValue(current * 100 / aim);
155 etaField.setText("");
156 progressLine = true;
157 }
158 else if (line.matches(".*\\[=*\\s*\\]\\s*\\d*/\\d*\\s*ETA:\\s*.*")) {
159 String[] ratioElements = line.split("\\]")[1].split("E")[0].trim().split("/");
160 int current = Integer.parseInt(ratioElements[0].trim());
161 int aim = Integer.parseInt(ratioElements[1].trim());
162 String eta = line.split("ETA:")[1].trim();
163 messageField.setText(line.split("\\[")[0].trim());
164 progressBar.setValue(current * 100 / aim);
165 etaField.setText("ETA: " + eta);
166 progressLine = true;
167 }
168 else if (line.matches(".*\\[=*\\s*\\]\\s*\\d*\\s*completed in.*")) {
169 String nbElements = line.split("\\]")[1].split("completed")[0].trim();
170 String timeSpent = line.split("completed in")[1].trim();
171 message += line.split("\\[")[0].trim() + ": " + nbElements + " elements completed in " + timeSpent + "\n";
172 messageField.setText(line.split("\\[")[0].trim());
173 progressLine = true;
174 }
175 if (! progressLine) {
176 message += line + "\n";
177 }
178 }
179 }
180 String lines[] = message.split("\n");
181 String toBeWritten = "";
182 for (int i = Math.max(0, lines.length - Global.logAreaSize); i < lines.length; i++) {
183 toBeWritten += lines[i] + "\n";
184 }
185 logArea.setText(toBeWritten);
186 }
187
188 public int getExitValue() {
189 return exitValue;
190 }
191 }