JAL-1835 made BioJSON embedding to html exports configurable, and trapped import...

[jalview.git] / src / jalview / io / HtmlFile.java

diff --git a/src/jalview/io/HtmlFile.java b/src/jalview/io/HtmlFile.java
index 73aa5d2..4805b8e 100644 (file)
--- a/src/jalview/io/HtmlFile.java
+++ b/src/jalview/io/HtmlFile.java
@@ -21,17 +21,19 @@
 
 package jalview.io;
 
+import jalview.api.ComplexAlignFile;
+import jalview.api.FeaturesDisplayedI;
+import jalview.datamodel.ColumnSelection;
+import jalview.datamodel.SequenceI;
+import jalview.schemes.ColourSchemeI;
+
 import java.io.IOException;
+import java.io.StringReader;
 
 import org.jsoup.Jsoup;
 import org.jsoup.nodes.Document;
 import org.jsoup.nodes.Element;
 
-import jalview.api.ComplexAlignFile;
-import jalview.datamodel.ColumnSelection;
-import jalview.datamodel.SequenceI;
-import jalview.schemes.ColourSchemeI;
-
 public class HtmlFile extends AlignFile implements ComplexAlignFile
 {
   public static final String FILE_EXT = "html";
@@ -46,6 +48,8 @@ public class HtmlFile extends AlignFile implements ComplexAlignFile
 
   private SequenceI[] hiddenSequences;
 
+  private FeaturesDisplayedI displayedFeatures;
+
   public HtmlFile()
   {
     super();
@@ -64,6 +68,8 @@ public class HtmlFile extends AlignFile implements ComplexAlignFile
   @Override
   public void parse() throws IOException
   {
+    Element content = null;
+    Document doc = null;
     try
     {
       StringBuilder htmlData = new StringBuilder();
@@ -72,12 +78,23 @@ public class HtmlFile extends AlignFile implements ComplexAlignFile
       {
         htmlData.append(currentLine);
       }
+      doc = Jsoup.parse(htmlData.toString());
+    } catch (OutOfMemoryError oom)
+    {
+      errormessage = "Not enough memory to process HTML document";
+      throw new IOException(errormessage);
+    }
 
-      Document doc = Jsoup.parse(htmlData.toString());
-      Element content = doc.getElementById("seqData");
-      String alignmentJsonString = content.val();
-
-      JSONFile jsonFile = new JSONFile().parse(alignmentJsonString);
+    try
+    {
+      content = doc.getElementById("seqData");
+      if (content == null)
+      {
+        errormessage = "The html document is not embedded with BioJSON data";
+        throw new IOException(errormessage);
+      }
+      JSONFile jsonFile = new JSONFile().parse(new StringReader(content
+              .val()));
       this.seqs = jsonFile.getSeqs();
       this.seqGroups = jsonFile.getSeqGroups();
       this.annotations = jsonFile.getAnnotations();
@@ -85,9 +102,10 @@ public class HtmlFile extends AlignFile implements ComplexAlignFile
       this.colourScheme = jsonFile.getColourScheme();
       this.hiddenSequences = jsonFile.getHiddenSequences();
       this.columnSelection = jsonFile.getColumnSelection();
+      this.displayedFeatures = jsonFile.getDisplayedFeatures();
     } catch (Exception e)
     {
-      e.printStackTrace();
+      throw e;
     }
   }
 
@@ -139,4 +157,10 @@ public class HtmlFile extends AlignFile implements ComplexAlignFile
     this.hiddenSequences = hiddenSequences;
   }
 
+  @Override
+  public FeaturesDisplayedI getDisplayedFeatures()
+  {
+    return displayedFeatures;
+  }
+
 }