4 import jalview.bin.Cache;
5 import jalview.datamodel.AlignmentI;
6 import jalview.datamodel.HiddenMarkovModel;
7 import jalview.datamodel.Sequence;
8 import jalview.datamodel.SequenceGroup;
9 import jalview.datamodel.SequenceI;
10 import jalview.gui.AlignFrame;
11 import jalview.gui.AlignViewport;
12 import jalview.gui.JvOptionPane;
13 import jalview.gui.Preferences;
14 import jalview.io.DataSourceType;
15 import jalview.io.FileParse;
16 import jalview.io.HMMFile;
17 import jalview.util.MessageManager;
18 import jalview.ws.params.ArgumentI;
21 import java.io.FileNotFoundException;
22 import java.io.IOException;
23 import java.util.ArrayList;
24 import java.util.List;
26 import javax.swing.JOptionPane;
28 public class HMMBuildThread implements Runnable
30 HMMERCommands cmds = new HMMERCommands();
33 AlignViewport viewport;
37 List<ArgumentI> params;
40 boolean forGroup = false;
49 * This is used for validation purposes. Do not use!
53 public HMMBuildThread(AlignmentI alignment)
55 this.alignment = alignment;
59 public HMMBuildThread(AlignFrame af, List<ArgumentI> args)
62 viewport = af.getViewport();
67 * Builds a HMM from an alignment, then imports and adds it to the alignment.
72 barID = System.currentTimeMillis();
75 af.setProgressBar(MessageManager.getString("status.running_hmmbuild"),
78 cmds.HMMERFOLDER = Cache.getProperty(Preferences.HMMER_PATH);
80 List<SequenceGroup> groups = new ArrayList<>();
83 for (ArgumentI arg : params)
85 String name = arg.getName();
86 if (MessageManager.getString("label.hmmbuild_for").equals(name))
88 String value = arg.getValue();
89 if ("Alignment".equals(value))
91 alignment = viewport.getAlignment();
94 else if ("All groups and alignment".equals(value))
96 alignment = viewport.getAlignment();
97 groups.addAll(viewport.getAlignment().getGroups());
100 else if ("All groups".equals(value))
103 groups = viewport.getAlignment().getGroups();
106 else if ("Selected group".equals(value))
109 groups.add(viewport.getSelectionGroup());
113 else if (MessageManager.getString("label.use_reference")
116 if (!af.getViewport().hasReferenceAnnotation())
121 MessageManager.getString("status.running_hmmbuild"),
124 JvOptionPane.showInternalMessageDialog(af, MessageManager
125 .getString("warn.no_reference_annotation"));
133 alignment = viewport.getAlignment();
136 if (alignment != null)
142 if (alignment == null)
144 alignment = viewport.getAlignment();
147 if (groups != null && groups.size() > 0)
149 for (SequenceGroup grp : groups)
160 af.setProgressBar(MessageManager.getString("status.running_hmmbuild"),
165 private void runHMMBuild()
167 if (alignment == null && group == null)
169 JOptionPane.showMessageDialog(af,
170 MessageManager.getString("warn.no_sequence_data"));
174 hmmTemp = File.createTempFile("hmm", ".hmm");
175 hmmTemp.deleteOnExit();
176 stoTemp = File.createTempFile("output", ".sto");
177 stoTemp.deleteOnExit();
178 } catch (IOException e1)
180 e1.printStackTrace();
188 List<SequenceI> seqs = null;
191 seqs = group.getHMMConsensusSequences(true);
196 array = group.getSelectionAsNewSequences(alignment);
200 seqs = alignment.getHMMConsensusSequences(true);
201 if (!alignment.isAligned())
205 array = alignment.getSequencesArray();
208 cmds.setHmmSeqs(seqs);
210 if (array.length < 1)
214 JOptionPane.showMessageDialog(af,
215 MessageManager.getString("warn.no_sequence_data"));
219 SequenceI[] newArr = new SequenceI[array.length];
221 for (SequenceI seq : array)
223 newArr[index] = new Sequence(seq);
227 cmds.uniquifySequences(newArr);
228 cmds.exportData(newArr, stoTemp, null, null);
229 jalview.analysis.SeqsetUtils.deuniquify(cmds.hash, array);
231 } catch (FileNotFoundException e)
233 // TODO Auto-generated catch block
239 boolean ran = runCommand();
244 } catch (IOException | InterruptedException e)
246 // TODO Auto-generated catch block
253 } catch (IOException | InterruptedException e)
255 // TODO Auto-generated catch block
258 } catch (Exception e)
267 * Executes the hmmbuild command in the command line.
270 * @throws IOException
271 * @throws InterruptedException
273 private boolean runCommand() throws IOException, InterruptedException
275 File file = new File(cmds.HMMERFOLDER + "/hmmbuild");
276 if (!file.canExecute())
278 file = new File(cmds.HMMERFOLDER + "/hmmbuild.exe");
280 if (!file.canExecute())
284 JvOptionPane.showInternalMessageDialog(af,
285 MessageManager.getString("warn.hmmbuild_failed"));
291 String command = cmds.HMMERFOLDER + cmds.HMMBUILD + cmds.SPACE;
296 for (ArgumentI arg : params)
298 String argName = arg.getName();
302 name = arg.getValue();
305 case "Use Reference Annotation":
306 command += "--hand ";
316 if (name == null || "".equals(name))
320 name = group.getName();
326 name = af.getTitle();
328 if (name == null || "".equals(name))
335 command += "-n " + name.replace(' ', '_') + cmds.SPACE;
336 if (!alignment.isNucleotide())
338 command += cmds.FORCEAMINO; // TODO check for rna
342 command += cmds.FORCEDNA;
345 command += hmmTemp.getAbsolutePath()
346 + cmds.SPACE + stoTemp.getAbsolutePath() + cmds.SPACE;
347 return cmds.runCommand(command);
351 * Imports the .hmm file produced by hmmbuild.
353 * @throws IOException
354 * @throws InterruptedException
356 private void importData() throws IOException, InterruptedException
358 HMMFile file = new HMMFile(new FileParse(hmmTemp.getAbsolutePath(),
359 DataSourceType.FILE));
360 SequenceI[] seqs = file.getSeqsAsArray();
361 SequenceI seq = seqs[0];
362 seq.createDatasetSequence();
363 HiddenMarkovModel hmm = file.getHMM();
366 seq.insertCharAt(0, group.getStartRes(), '-');
367 seq.insertCharAt(group.getEndRes() + 1,
368 alignment.getWidth() - group.getEndRes(), '-');
369 seq.updateHMMMapping();
370 SequenceI topSeq = group.getSequencesInOrder(alignment)[0];
371 int topIndex = alignment.findIndex(topSeq);
372 alignment.insertSequenceAt(topIndex, seq);
373 group.setSeqrep(seq);
374 group.addSequence(seq, false);
378 alignment.insertSequenceAt(0, seq);
381 viewport.alignmentChanged(viewport.getAlignPanel());
382 viewport.getAlignPanel().adjustAnnotationHeight();
383 viewport.updateSequenceIdColours();
385 if (viewport.getAlignPanel().alignFrame.getSelectedHMM() == null)
387 viewport.getAlignPanel().alignFrame.setSelectedHMMSequence(seq);
394 * Runs hmmbuild, and waits for the results to be imported before continuing
396 public void hmmbuildWaitTillComplete()
398 Thread loader = new Thread(this);
401 while (loader.isAlive())
406 } catch (Exception ex)