Commit | Line | Data |
---|---|---|
c068a752 GB |
1 | /******************************************************************************* |
2 | * Copyright (c) 2013 École Polytechnique de Montréal | |
3 | * | |
4 | * All rights reserved. This program and the accompanying materials are | |
5 | * made available under the terms of the Eclipse Public License v1.0 which | |
6 | * accompanies this distribution, and is available at | |
7 | * http://www.eclipse.org/legal/epl-v10.html | |
8 | * | |
9 | * Contributors: | |
10 | * Geneviève Bastien - Initial API and implementation | |
11 | *******************************************************************************/ | |
12 | ||
13 | package org.eclipse.linuxtools.tmf.core.analysis; | |
14 | ||
15 | import java.util.ArrayList; | |
c068a752 GB |
16 | import java.util.HashMap; |
17 | import java.util.List; | |
18 | import java.util.Map; | |
19 | import java.util.concurrent.CountDownLatch; | |
20 | import java.util.concurrent.TimeUnit; | |
21 | ||
22 | import org.eclipse.core.runtime.IProgressMonitor; | |
23 | import org.eclipse.core.runtime.IStatus; | |
24 | import org.eclipse.core.runtime.Status; | |
25 | import org.eclipse.core.runtime.jobs.Job; | |
26 | import org.eclipse.linuxtools.internal.tmf.core.Activator; | |
27 | import org.eclipse.linuxtools.tmf.core.component.TmfComponent; | |
28 | import org.eclipse.linuxtools.tmf.core.exceptions.TmfAnalysisException; | |
29 | import org.eclipse.linuxtools.tmf.core.signal.TmfSignalHandler; | |
30 | import org.eclipse.linuxtools.tmf.core.signal.TmfStartAnalysisSignal; | |
31 | import org.eclipse.linuxtools.tmf.core.signal.TmfTraceClosedSignal; | |
32 | import org.eclipse.linuxtools.tmf.core.trace.ITmfTrace; | |
9de979b2 | 33 | import org.eclipse.linuxtools.tmf.core.trace.TmfTrace; |
c068a752 GB |
34 | import org.eclipse.osgi.util.NLS; |
35 | ||
36 | /** | |
37 | * Base class that analysis modules main class may extend. It provides default | |
38 | * behavior to some methods of the analysis module | |
39 | * | |
40 | * @author Geneviève Bastien | |
41 | * @since 3.0 | |
42 | */ | |
43 | public abstract class TmfAbstractAnalysisModule extends TmfComponent implements IAnalysisModule { | |
44 | ||
45 | private String fName, fId; | |
46 | private boolean fAutomatic = false, fStarted = false; | |
47 | private ITmfTrace fTrace; | |
a4524c1b AM |
48 | private final Map<String, Object> fParameters = new HashMap<>(); |
49 | private final List<String> fParameterNames = new ArrayList<>(); | |
50 | private final List<IAnalysisOutput> fOutputs = new ArrayList<>(); | |
51 | private List<IAnalysisParameterProvider> fParameterProviders = new ArrayList<>(); | |
c068a752 GB |
52 | private Job fJob = null; |
53 | ||
54 | private final Object syncObj = new Object(); | |
55 | ||
56 | /* Latch tracking if the analysis is completed or not */ | |
57 | private CountDownLatch fFinishedLatch = new CountDownLatch(0); | |
58 | ||
59 | private boolean fAnalysisCancelled = false; | |
60 | ||
61 | @Override | |
62 | public boolean isAutomatic() { | |
63 | return fAutomatic; | |
64 | } | |
65 | ||
66 | @Override | |
67 | public String getName() { | |
68 | return fName; | |
69 | } | |
70 | ||
71 | @Override | |
72 | public void setName(String name) { | |
73 | fName = name; | |
74 | } | |
75 | ||
76 | @Override | |
77 | public void setId(String id) { | |
78 | fId = id; | |
79 | } | |
80 | ||
81 | @Override | |
82 | public String getId() { | |
83 | return fId; | |
84 | } | |
85 | ||
86 | @Override | |
87 | public void setAutomatic(boolean auto) { | |
88 | fAutomatic = auto; | |
89 | } | |
90 | ||
91 | @Override | |
92 | public void setTrace(ITmfTrace trace) throws TmfAnalysisException { | |
93 | if (fTrace != null) { | |
94 | throw new TmfAnalysisException(NLS.bind(Messages.TmfAbstractAnalysisModule_TraceSetMoreThanOnce, getName())); | |
95 | } | |
96 | ||
97 | /* Check that analysis can be executed */ | |
98 | if (!canExecute(trace)) { | |
99 | throw new TmfAnalysisException(NLS.bind(Messages.TmfAbstractAnalysisModule_AnalysisCannotExecute, getName())); | |
100 | } | |
101 | fTrace = trace; | |
102 | /* Get the parameter providers for this trace */ | |
103 | fParameterProviders = TmfAnalysisManager.getParameterProviders(this, fTrace); | |
104 | for (IAnalysisParameterProvider provider : fParameterProviders) { | |
105 | provider.registerModule(this); | |
106 | } | |
107 | resetAnalysis(); | |
108 | fStarted = false; | |
109 | } | |
110 | ||
111 | /** | |
112 | * Gets the trace | |
113 | * | |
114 | * @return The trace | |
115 | */ | |
116 | protected ITmfTrace getTrace() { | |
117 | return fTrace; | |
118 | } | |
119 | ||
120 | @Override | |
121 | public void addParameter(String name) { | |
122 | fParameterNames.add(name); | |
123 | } | |
124 | ||
125 | @Override | |
126 | public synchronized void setParameter(String name, Object value) { | |
127 | if (!fParameterNames.contains(name)) { | |
128 | throw new RuntimeException(NLS.bind(Messages.TmfAbstractAnalysisModule_InvalidParameter, name, getName())); | |
129 | } | |
130 | Object oldValue = fParameters.get(name); | |
131 | fParameters.put(name, value); | |
132 | if ((value != null) && !(value.equals(oldValue))) { | |
133 | parameterChanged(name); | |
134 | } | |
135 | } | |
136 | ||
137 | @Override | |
138 | public synchronized void notifyParameterChanged(String name) { | |
139 | if (!fParameterNames.contains(name)) { | |
140 | throw new RuntimeException(NLS.bind(Messages.TmfAbstractAnalysisModule_InvalidParameter, name, getName())); | |
141 | } | |
142 | Object oldValue = fParameters.get(name); | |
143 | Object value = getParameter(name); | |
144 | if ((value != null) && !(value.equals(oldValue))) { | |
145 | parameterChanged(name); | |
146 | } | |
147 | } | |
148 | ||
149 | /** | |
150 | * Used to indicate that a parameter value has been changed | |
151 | * | |
152 | * @param name | |
153 | * The name of the modified parameter | |
154 | */ | |
155 | protected void parameterChanged(String name) { | |
156 | ||
157 | } | |
158 | ||
159 | @Override | |
160 | public Object getParameter(String name) { | |
161 | Object paramValue = fParameters.get(name); | |
162 | /* The parameter is not set, maybe it can be provided by someone else */ | |
163 | if ((paramValue == null) && (fTrace != null)) { | |
164 | for (IAnalysisParameterProvider provider : fParameterProviders) { | |
165 | paramValue = provider.getParameter(name); | |
166 | if (paramValue != null) { | |
167 | break; | |
168 | } | |
169 | } | |
170 | } | |
171 | return paramValue; | |
172 | } | |
173 | ||
174 | @Override | |
175 | public boolean canExecute(ITmfTrace trace) { | |
176 | return true; | |
177 | } | |
178 | ||
179 | /** | |
180 | * Set the countdown latch back to 1 so the analysis can be executed again | |
181 | */ | |
182 | protected void resetAnalysis() { | |
183 | fFinishedLatch = new CountDownLatch(1); | |
184 | } | |
185 | ||
186 | /** | |
187 | * Actually executes the analysis itself | |
188 | * | |
189 | * @param monitor | |
190 | * Progress monitor | |
191 | * @return Whether the analysis was completed successfully or not | |
192 | * @throws TmfAnalysisException | |
193 | * Method may throw an analysis exception | |
194 | */ | |
195 | protected abstract boolean executeAnalysis(final IProgressMonitor monitor) throws TmfAnalysisException; | |
196 | ||
197 | /** | |
198 | * Indicate the analysis has been canceled. It is abstract to force | |
199 | * implementing class to cleanup what they are running. This is called by | |
200 | * the job's canceling. It does not need to be called directly. | |
201 | */ | |
202 | protected abstract void canceling(); | |
203 | ||
204 | /** | |
205 | * To be called when the analysis is completed, whether normally or because | |
206 | * it was cancelled or for any other reason. | |
207 | * | |
208 | * It has to be called inside a synchronized block | |
209 | */ | |
210 | private void setAnalysisCompleted() { | |
211 | fStarted = false; | |
212 | fJob = null; | |
213 | fFinishedLatch.countDown(); | |
9de979b2 GB |
214 | if (fTrace instanceof TmfTrace) { |
215 | ((TmfTrace) fTrace).refreshSupplementaryFiles(); | |
216 | } | |
c068a752 GB |
217 | } |
218 | ||
219 | /** | |
220 | * Cancels the analysis if it is executing | |
221 | */ | |
222 | @Override | |
223 | public final void cancel() { | |
224 | synchronized (syncObj) { | |
225 | if (fJob != null) { | |
226 | if (fJob.cancel()) { | |
227 | fAnalysisCancelled = true; | |
228 | setAnalysisCompleted(); | |
229 | } | |
230 | } | |
231 | fStarted = false; | |
232 | } | |
233 | } | |
234 | ||
235 | private void execute() { | |
236 | ||
237 | /* | |
238 | * TODO: The analysis in a job should be done at the analysis manager | |
239 | * level instead of depending on this abstract class implementation, | |
240 | * otherwise another analysis implementation may block the main thread | |
241 | */ | |
242 | ||
243 | /* Do not execute if analysis has already run */ | |
244 | if (fFinishedLatch.getCount() == 0) { | |
245 | return; | |
246 | } | |
247 | ||
248 | /* Do not execute if analysis already running */ | |
249 | synchronized (syncObj) { | |
250 | if (fStarted) { | |
251 | return; | |
252 | } | |
253 | fStarted = true; | |
254 | } | |
255 | ||
256 | /* | |
257 | * Actual analysis will be run on a separate thread | |
258 | */ | |
259 | fJob = new Job(NLS.bind(Messages.TmfAbstractAnalysisModule_RunningAnalysis, getName())) { | |
260 | @Override | |
261 | protected IStatus run(final IProgressMonitor monitor) { | |
262 | try { | |
263 | monitor.beginTask("", IProgressMonitor.UNKNOWN); //$NON-NLS-1$ | |
264 | broadcast(new TmfStartAnalysisSignal(TmfAbstractAnalysisModule.this, TmfAbstractAnalysisModule.this)); | |
265 | fAnalysisCancelled = !executeAnalysis(monitor); | |
266 | } catch (TmfAnalysisException e) { | |
267 | Activator.logError("Error executing analysis with trace " + getTrace().getName(), e); //$NON-NLS-1$ | |
268 | } finally { | |
269 | synchronized (syncObj) { | |
270 | monitor.done(); | |
271 | setAnalysisCompleted(); | |
272 | } | |
273 | } | |
274 | if (!fAnalysisCancelled) { | |
275 | return Status.OK_STATUS; | |
276 | } | |
baa96b1d BH |
277 | // Reset analysis so that it can be executed again. |
278 | resetAnalysis(); | |
c068a752 GB |
279 | return Status.CANCEL_STATUS; |
280 | } | |
281 | ||
282 | @Override | |
283 | protected void canceling() { | |
284 | TmfAbstractAnalysisModule.this.canceling(); | |
285 | } | |
286 | ||
287 | }; | |
288 | fJob.schedule(); | |
289 | } | |
290 | ||
291 | @Override | |
292 | public IStatus schedule() { | |
293 | if (fTrace == null) { | |
294 | return new Status(IStatus.ERROR, Activator.PLUGIN_ID, String.format("No trace specified for analysis %s", getName())); //$NON-NLS-1$ | |
295 | } | |
296 | execute(); | |
297 | ||
298 | return Status.OK_STATUS; | |
299 | } | |
300 | ||
301 | @Override | |
d9810555 AM |
302 | public Iterable<IAnalysisOutput> getOutputs() { |
303 | return fOutputs; | |
c068a752 GB |
304 | } |
305 | ||
306 | @Override | |
307 | public void registerOutput(IAnalysisOutput output) { | |
308 | if (!fOutputs.contains(output)) { | |
309 | fOutputs.add(output); | |
310 | } | |
311 | } | |
312 | ||
313 | @Override | |
314 | public boolean waitForCompletion(IProgressMonitor monitor) { | |
315 | try { | |
316 | while (!fFinishedLatch.await(1, TimeUnit.MILLISECONDS)) { | |
317 | if (monitor.isCanceled()) { | |
318 | fAnalysisCancelled = true; | |
319 | return false; | |
320 | } | |
321 | } | |
322 | } catch (InterruptedException e) { | |
323 | Activator.logError("Error while waiting for module completion", e); //$NON-NLS-1$ | |
324 | } | |
325 | return !fAnalysisCancelled; | |
326 | } | |
327 | ||
328 | /** | |
329 | * Signal handler for trace closing | |
330 | * | |
331 | * @param signal | |
332 | * Trace closed signal | |
333 | */ | |
334 | @TmfSignalHandler | |
335 | public void traceClosed(TmfTraceClosedSignal signal) { | |
336 | /* Is the closing trace the one that was requested? */ | |
337 | if (signal.getTrace() == fTrace) { | |
338 | cancel(); | |
339 | fTrace = null; | |
340 | } | |
341 | } | |
342 | ||
343 | /** | |
344 | * Returns a full help text to display | |
345 | * | |
346 | * @return Full help text for the module | |
347 | */ | |
348 | protected String getFullHelpText() { | |
349 | return NLS.bind(Messages.TmfAbstractAnalysisModule_AnalysisModule, getName()); | |
350 | } | |
351 | ||
352 | /** | |
353 | * Gets a short help text, to display as header to other help text | |
354 | * | |
355 | * @param trace | |
356 | * The trace to show help for | |
357 | * | |
358 | * @return Short help text describing the module | |
359 | */ | |
360 | protected String getShortHelpText(ITmfTrace trace) { | |
361 | return NLS.bind(Messages.TmfAbstractAnalysisModule_AnalysisForTrace, getName(), trace.getName()); | |
362 | } | |
363 | ||
364 | /** | |
365 | * Gets the help text specific for a trace who does not have required | |
366 | * characteristics for module to execute | |
367 | * | |
368 | * @param trace | |
369 | * The trace to apply the analysis to | |
370 | * @return Help text | |
371 | */ | |
372 | protected String getTraceCannotExecuteHelpText(ITmfTrace trace) { | |
373 | return Messages.TmfAbstractAnalysisModule_AnalysisCannotExecute; | |
374 | } | |
375 | ||
376 | @Override | |
377 | public String getHelpText() { | |
378 | return getFullHelpText(); | |
379 | } | |
380 | ||
381 | @Override | |
382 | public String getHelpText(ITmfTrace trace) { | |
383 | if (trace == null) { | |
384 | return getHelpText(); | |
385 | } | |
386 | String text = getShortHelpText(trace); | |
387 | if (!canExecute(trace)) { | |
388 | text = text + getTraceCannotExecuteHelpText(trace); | |
389 | } | |
390 | return text; | |
391 | } | |
392 | ||
393 | } |