X-Git-Url: http://source.jalview.org/gitweb/?a=blobdiff_plain;f=src%2Fjalview%2Fio%2FBioJsHTMLOutput.java;h=86b4c2855283ca0a9292b9817fe807ba14c2f942;hb=5d59bb58f5bcecc6b240d125e13bbe1f1868f681;hp=6869292e97a02345e7d07b42a99ceb8f8d187200;hpb=9e5effcac65476958e619298a1a939254f77bc4b;p=jalview.git diff --git a/src/jalview/io/BioJsHTMLOutput.java b/src/jalview/io/BioJsHTMLOutput.java index 6869292..86b4c28 100644 --- a/src/jalview/io/BioJsHTMLOutput.java +++ b/src/jalview/io/BioJsHTMLOutput.java @@ -1,111 +1,129 @@ package jalview.io; -import static java.nio.file.Files.readAllBytes; -import static java.nio.file.Paths.get; -import jalview.datamodel.AlignmentI; -import jalview.datamodel.SequenceI; +import jalview.exceptions.NoFileSelectedException; import jalview.gui.AlignViewport; import jalview.gui.AlignmentPanel; import jalview.gui.FeatureRenderer; -import jalview.gui.SequenceRenderer; -import jalview.json.binding.v1.BioJsAlignmentPojo; -import jalview.json.binding.v1.BioJsSeqPojo; import jalview.util.MessageManager; +import java.io.BufferedReader; import java.io.IOException; +import java.io.InputStreamReader; import java.io.PrintWriter; - -import org.jsoup.Jsoup; -import org.jsoup.nodes.Document; -import org.jsoup.nodes.Element; - -import com.json.JSONException; -import com.json.JSONObject; +import java.net.URL; public class BioJsHTMLOutput { private AlignViewport av; - private FeatureRenderer fr; - public BioJsHTMLOutput(AlignmentPanel ap, SequenceRenderer sr, + + public BioJsHTMLOutput(AlignmentPanel ap, FeatureRenderer fr1) { - this.av = ap.av; - this.fr = new FeatureRenderer(ap); - fr.transferSettings(fr1); - exportAsBioJsHtml(); + if (ap != null) + { + + this.av = ap.av; + av.setFeatureRenderer(new FeatureRenderer(ap)); + } + exportJalviewAlignmentAsBioJsHtmlFile(); } - private void exportAsBioJsHtml() + private void exportJalviewAlignmentAsBioJsHtmlFile() { try { - JalviewFileChooser jvFileChooser = getJalviewFileChooserOption(); - int fileChooserOpt = jvFileChooser.showSaveDialog(null); - if (fileChooserOpt == JalviewFileChooser.APPROVE_OPTION) - { - jalview.bin.Cache.setProperty("LAST_DIRECTORY", jvFileChooser - .getSelectedFile().getParent()); - String selectedFile = jvFileChooser.getSelectedFile().getPath(); - String generartedBioJs = generateBioJsAlignmentData(av - .getAlignment()); - PrintWriter out = new java.io.PrintWriter(new java.io.FileWriter( - selectedFile)); - out.print(generartedBioJs); - out.close(); - jalview.util.BrowserLauncher.openURL("file:///" + selectedFile); - } - } catch (Exception ex) + String outputFile = getOutputFile(); + String jalviewAlignmentJson = JSONFile.getJSONData(av); + String bioJSTemplateString = getBioJsTemplateAsString(this); + String generatedBioJsWithJalviewAlignmentAsJson = bioJSTemplateString + .replaceAll( +"#sequenceData#", jalviewAlignmentJson) + .toString(); + + PrintWriter out = new java.io.PrintWriter(new java.io.FileWriter( + outputFile)); + out.print(generatedBioJsWithJalviewAlignmentAsJson); + out.flush(); + out.close(); + jalview.util.BrowserLauncher.openURL("file:///" + outputFile); + } catch (NoFileSelectedException ex) + { + // do noting if no file was selected + } catch (Exception e) { - ex.printStackTrace(); + e.printStackTrace(); } } - private JalviewFileChooser getJalviewFileChooserOption() + public String getOutputFile() throws NoFileSelectedException { - JalviewFileChooser chooser = new JalviewFileChooser( + String selectedFile = null; + JalviewFileChooser jvFileChooser = new JalviewFileChooser( jalview.bin.Cache.getProperty("LAST_DIRECTORY"), new String[] { "html" }, new String[] { "HTML files" }, "HTML files"); + jvFileChooser.setFileView(new JalviewFileView()); - chooser.setFileView(new JalviewFileView()); // TODO uncomment when supported by MassageManager - // chooser.setDialogTitle(MessageManager.getString("label.save_as_biojs_html")); - chooser.setDialogTitle("save as BioJs HTML"); - chooser.setToolTipText(MessageManager.getString("action.save")); - - return chooser; - } + jvFileChooser.setDialogTitle(MessageManager + .getString("label.save_as_biojs_html")); + jvFileChooser.setDialogTitle("save as BioJs HTML"); + jvFileChooser.setToolTipText(MessageManager.getString("action.save")); - private String generateBioJsAlignmentData(AlignmentI alignment) - throws IOException, JSONException - { - BioJsAlignmentPojo bjsAlignment = new BioJsAlignmentPojo(); - - int count = 0; - for (SequenceI seq : alignment.getSequences()) + int fileChooserOpt = jvFileChooser.showSaveDialog(null); + if (fileChooserOpt == JalviewFileChooser.APPROVE_OPTION) { - bjsAlignment.getSeqs().add( - new BioJsSeqPojo(String.valueOf(++count), seq.getName(), seq - .getSequenceAsString())); + jalview.bin.Cache.setProperty("LAST_DIRECTORY", jvFileChooser + .getSelectedFile().getParent()); + selectedFile = jvFileChooser.getSelectedFile().getPath(); } - - String seqs = new JSONObject(bjsAlignment).getString("seqs"); - String bioJSTemplate = new String( - readAllBytes(get("resources/templates/BioJSTemplate.txt"))); - - return bioJSTemplate.replaceAll("#sequenceData#", seqs); + else + { + throw new NoFileSelectedException("No file was selected."); + } + return selectedFile; } - public static void main(String[] args) throws IOException - { - Document doc = Jsoup.parse(new String( - readAllBytes(get("resources/templates/BioJSTemplate.txt")))); - Element content = doc.getElementById("seqData"); - System.out.println(content.val()); + public static String getBioJsTemplateAsString(Object currentObj) + throws IOException + { + InputStreamReader isReader = null; + BufferedReader buffReader = null; + StringBuilder sb = new StringBuilder(); + URL url = currentObj.getClass().getResource( + "/templates/BioJSTemplate.txt"); + if (url != null) + { + try + { + isReader = new InputStreamReader(url.openStream()); + buffReader = new BufferedReader(isReader); + String line; + String lineSeparator = System.getProperty("line.separator"); + while ((line = buffReader.readLine()) != null) + { + sb.append(line).append(lineSeparator); + } + + } catch (Exception ex) + { + ex.printStackTrace(); + } finally + { + if (isReader != null) + { + isReader.close(); + } + + if (buffReader != null) + { + buffReader.close(); + } + } + } + return sb.toString(); } - } -