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"));
132 if (alignment != null)
138 if (alignment == null)
140 alignment = viewport.getAlignment();
143 if (groups != null && groups.size() > 0)
145 for (SequenceGroup grp : groups)
156 af.setProgressBar(MessageManager.getString("status.running_hmmbuild"),
161 private void runHMMBuild()
163 if (alignment == null && group == null)
165 JOptionPane.showMessageDialog(af,
166 MessageManager.getString("warn.no_sequence_data"));
170 hmmTemp = File.createTempFile("hmm", ".hmm");
171 hmmTemp.deleteOnExit();
172 stoTemp = File.createTempFile("output", ".sto");
173 stoTemp.deleteOnExit();
174 } catch (IOException e1)
176 e1.printStackTrace();
184 List<SequenceI> seqs = null;
187 seqs = group.getHMMConsensusSequences(true);
192 array = group.getSelectionAsNewSequences(alignment);
196 seqs = alignment.getHMMConsensusSequences(true);
197 if (!alignment.isAligned())
201 array = alignment.getSequencesArray();
204 cmds.setHmmSeqs(seqs);
206 if (array.length < 1)
210 JOptionPane.showMessageDialog(af,
211 MessageManager.getString("warn.no_sequence_data"));
215 SequenceI[] newArr = new SequenceI[array.length];
217 for (SequenceI seq : array)
219 newArr[index] = new Sequence(seq);
223 cmds.uniquifySequences(newArr);
224 cmds.exportData(newArr, stoTemp, null, null);
225 jalview.analysis.SeqsetUtils.deuniquify(cmds.hash, array);
227 } catch (FileNotFoundException e)
229 // TODO Auto-generated catch block
235 boolean ran = runCommand();
240 } catch (IOException | InterruptedException e)
242 // TODO Auto-generated catch block
249 } catch (IOException | InterruptedException e)
251 // TODO Auto-generated catch block
254 } catch (Exception e)
263 * Executes the hmmbuild command in the command line.
266 * @throws IOException
267 * @throws InterruptedException
269 private boolean runCommand() throws IOException, InterruptedException
271 File file = new File(cmds.HMMERFOLDER + "/hmmbuild");
272 if (!file.canExecute())
274 file = new File(cmds.HMMERFOLDER + "/hmmbuild.exe");
276 if (!file.canExecute())
280 JvOptionPane.showInternalMessageDialog(af,
281 MessageManager.getString("warn.hmmbuild_failed"));
287 String command = cmds.HMMERFOLDER + cmds.HMMBUILD + cmds.SPACE;
292 for (ArgumentI arg : params)
294 String argName = arg.getName();
298 name = arg.getValue();
301 case "Use Reference Annotation":
302 command += "--hand ";
312 if (name == null || "".equals(name))
316 name = group.getName();
322 name = af.getTitle();
324 if (name == null || "".equals(name))
331 command += "-n " + name.replace(' ', '_') + cmds.SPACE;
332 if (!alignment.isNucleotide())
334 command += cmds.FORCEAMINO; // TODO check for rna
338 command += cmds.FORCEDNA;
341 command += hmmTemp.getAbsolutePath()
342 + cmds.SPACE + stoTemp.getAbsolutePath() + cmds.SPACE;
343 return cmds.runCommand(command);
347 * Imports the .hmm file produced by hmmbuild.
349 * @throws IOException
350 * @throws InterruptedException
352 private void importData() throws IOException, InterruptedException
354 HMMFile file = new HMMFile(new FileParse(hmmTemp.getAbsolutePath(),
355 DataSourceType.FILE));
356 SequenceI[] seqs = file.getSeqsAsArray();
357 SequenceI seq = seqs[0];
358 seq.createDatasetSequence();
359 HiddenMarkovModel hmm = file.getHMM();
362 seq.insertCharAt(0, group.getStartRes(), '-');
363 seq.insertCharAt(group.getEndRes() + 1,
364 alignment.getWidth() - group.getEndRes(), '-');
365 seq.updateHMMMapping();
366 SequenceI topSeq = group.getSequencesInOrder(alignment)[0];
367 int topIndex = alignment.findIndex(topSeq);
368 alignment.insertSequenceAt(topIndex, seq);
369 group.setSeqrep(seq);
370 group.addSequence(seq, false);
374 alignment.insertSequenceAt(0, seq);
377 viewport.alignmentChanged(viewport.getAlignPanel());
378 viewport.getAlignPanel().adjustAnnotationHeight();
379 viewport.updateSequenceIdColours();
381 if (viewport.getAlignPanel().alignFrame.getSelectedHMM() == null)
383 viewport.getAlignPanel().alignFrame.setSelectedHMMSequence(seq);
390 * Runs hmmbuild, and waits for the results to be imported before continuing
392 public void hmmbuildWaitTillComplete()
394 Thread loader = new Thread(this);
397 while (loader.isAlive())
402 } catch (Exception ex)