JSON refactoring
[jalview.git] / src / jalview / io / JSONFile.java
index 7071a9d..3f1b1a7 100644 (file)
@@ -53,6 +53,7 @@ import jalview.schemes.JalviewColourScheme;
 import jalview.schemes.ResidueColourScheme;
 import jalview.util.ColorUtils;
 import jalview.util.Format;
+import jalview.util.JSONUtils;
 import jalview.viewmodel.seqfeatures.FeaturesDisplayed;
 
 import java.awt.Color;
@@ -62,12 +63,9 @@ import java.util.ArrayList;
 import java.util.Hashtable;
 import java.util.Iterator;
 import java.util.List;
+import java.util.Map;
 import java.util.Vector;
 
-import org.json.simple.JSONArray;
-import org.json.simple.JSONObject;
-import org.json.simple.parser.JSONParser;
-
 public class JSONFile extends AlignFile implements ComplexAlignFile
 {
   private static String version = new BuildDetails().getVersion();
@@ -220,8 +218,7 @@ public class JSONFile extends AlignFile implements ComplexAlignFile
           jsonAlignmentPojo.getSeqGroups().add(seqGrpPojo);
         }
       }
-      org.json.JSONObject generatedJSon = new org.json.JSONObject(
-              jsonAlignmentPojo);
+      org.json.JSONObject generatedJSon = new org.json.JSONObject(jsonAlignmentPojo);
       jsonOutput = generatedJSon.toString();
       return jsonOutput.replaceAll("xstart", "xStart").replaceAll("xend",
               "xEnd");
@@ -398,17 +395,15 @@ public class JSONFile extends AlignFile implements ComplexAlignFile
   {
     try
     {
-      JSONParser jsonParser = new JSONParser();
-      JSONObject alignmentJsonObj = (JSONObject) jsonParser
-              .parse(jsonAlignmentString);
-      JSONArray seqJsonArray = (JSONArray) alignmentJsonObj.get("seqs");
-      JSONArray alAnnotJsonArray = (JSONArray) alignmentJsonObj
+      Map<String, Object> alignmentJsonObj = (Map<String, Object>) JSONUtils.parse(jsonAlignmentString);
+      List<Object> seqJsonArray = (List<Object>) alignmentJsonObj.get("seqs");
+      List<Object> alAnnotJsonArray = (List<Object>) alignmentJsonObj
               .get("alignAnnotation");
-      JSONArray jsonSeqArray = (JSONArray) alignmentJsonObj
+      List<Object> jsonSeqArray = (List<Object>) alignmentJsonObj
               .get("seqFeatures");
-      JSONArray seqGrpJsonArray = (JSONArray) alignmentJsonObj
+      List<Object> seqGrpJsonArray = (List<Object>) alignmentJsonObj
               .get("seqGroups");
-      JSONObject jvSettingsJsonObj = (JSONObject) alignmentJsonObj
+      Map<String, Object> jvSettingsJsonObj = (Map<String, Object>) alignmentJsonObj
               .get("appSettings");
 
       if (jvSettingsJsonObj != null)
@@ -424,10 +419,9 @@ public class JSONFile extends AlignFile implements ComplexAlignFile
 
       hiddenSequences = new ArrayList<>();
       seqMap = new Hashtable<>();
-      for (Iterator<JSONObject> sequenceIter = seqJsonArray
-              .iterator(); sequenceIter.hasNext();)
+      for (Iterator<Object> sequenceIter = seqJsonArray.iterator(); sequenceIter.hasNext();)
       {
-        JSONObject sequence = sequenceIter.next();
+        Map<String, Object> sequence = (Map<String, Object>) sequenceIter.next();
         String sequcenceString = sequence.get("seq").toString();
         String sequenceName = sequence.get("name").toString();
         String seqUniqueId = sequence.get("id").toString();
@@ -445,10 +439,9 @@ public class JSONFile extends AlignFile implements ComplexAlignFile
 
       parseFeatures(jsonSeqArray);
 
-      for (Iterator<JSONObject> seqGrpIter = seqGrpJsonArray
-              .iterator(); seqGrpIter.hasNext();)
+      for (Iterator<Object> seqGrpIter = seqGrpJsonArray.iterator(); seqGrpIter.hasNext();)
       {
-        JSONObject seqGrpObj = seqGrpIter.next();
+        Map<String, Object> seqGrpObj = (Map<String, Object>)seqGrpIter.next();
         String grpName = seqGrpObj.get("groupName").toString();
         String colourScheme = seqGrpObj.get("colourScheme").toString();
         String description = (seqGrpObj.get("description") == null) ? null
@@ -464,16 +457,15 @@ public class JSONFile extends AlignFile implements ComplexAlignFile
         int startRes = Integer
                 .valueOf(seqGrpObj.get("startRes").toString());
         int endRes = Integer.valueOf(seqGrpObj.get("endRes").toString());
-        JSONArray sequenceRefs = (JSONArray) seqGrpObj.get("sequenceRefs");
+        List<Object> sequenceRefs = (List<Object>) seqGrpObj.get("sequenceRefs");
 
         ArrayList<SequenceI> grpSeqs = new ArrayList<>();
         if (sequenceRefs.size() > 0)
         {
-          Iterator<String> seqHashIter = sequenceRefs.iterator();
+          Iterator<Object> seqHashIter = sequenceRefs.iterator();
           while (seqHashIter.hasNext())
           {
-            String seqHash = seqHashIter.next();
-            Sequence sequence = seqMap.get(seqHash);
+            Sequence sequence = seqMap.get(seqHashIter.next());
             if (sequence != null)
             {
               grpSeqs.add(sequence);
@@ -490,17 +482,15 @@ public class JSONFile extends AlignFile implements ComplexAlignFile
 
       }
 
-      for (Iterator<JSONObject> alAnnotIter = alAnnotJsonArray
-              .iterator(); alAnnotIter.hasNext();)
+      for (Iterator<Object> alAnnotIter = alAnnotJsonArray.iterator(); alAnnotIter.hasNext();)
       {
-        JSONObject alAnnot = alAnnotIter.next();
-        JSONArray annotJsonArray = (JSONArray) alAnnot.get("annotations");
+        Map<String, Object> alAnnot = (Map<String, Object>) alAnnotIter.next();
+        List<Object> annotJsonArray = (List<Object>) alAnnot.get("annotations");
         Annotation[] annotations = new Annotation[annotJsonArray.size()];
         int count = 0;
-        for (Iterator<JSONObject> annotIter = annotJsonArray
-                .iterator(); annotIter.hasNext();)
+        for (Iterator<Object> annotIter = annotJsonArray.iterator(); annotIter.hasNext();)
         {
-          JSONObject annot = annotIter.next();
+          Map<String, Object> annot = (Map<String, Object>) annotIter.next();
           if (annot == null)
           {
             annotations[count] = null;
@@ -536,7 +526,7 @@ public class JSONFile extends AlignFile implements ComplexAlignFile
         alignAnnot.graph = (alAnnot.get("graphType") == null) ? 0
                 : Integer.valueOf(alAnnot.get("graphType").toString());
 
-        JSONObject diplaySettings = (JSONObject) alAnnot
+        Map<String, Object> diplaySettings = (Map<String, Object>) alAnnot
                 .get("annotationSettings");
         if (diplaySettings != null)
         {
@@ -604,7 +594,7 @@ public class JSONFile extends AlignFile implements ComplexAlignFile
     return this;
   }
 
-  public void parseHiddenSeqRefsAsList(JSONObject jvSettingsJson)
+  public void parseHiddenSeqRefsAsList(Map<String, Object> jvSettingsJson)
   {
     hiddenSeqRefs = new ArrayList<>();
     String hiddenSeqs = (String) jvSettingsJson.get("hiddenSeqs");
@@ -618,7 +608,7 @@ public class JSONFile extends AlignFile implements ComplexAlignFile
     }
   }
 
-  public void parseHiddenCols(JSONObject jvSettingsJson)
+  public void parseHiddenCols(Map<String, Object> jvSettingsJson)
   {
     String hiddenCols = (String) jvSettingsJson.get("hiddenCols");
     if (hiddenCols != null && !hiddenCols.isEmpty())
@@ -635,15 +625,15 @@ public class JSONFile extends AlignFile implements ComplexAlignFile
   }
 
   @SuppressWarnings("unchecked")
-  private void parseFeatures(JSONArray jsonSeqFeatures)
+  private void parseFeatures(List<Object> jsonSeqFeatures)
   {
     if (jsonSeqFeatures != null)
     {
       displayedFeatures = new FeaturesDisplayed();
-      for (Iterator<JSONObject> seqFeatureItr = jsonSeqFeatures
+      for (Iterator<Object> seqFeatureItr = jsonSeqFeatures
               .iterator(); seqFeatureItr.hasNext();)
       {
-        JSONObject jsonFeature = seqFeatureItr.next();
+        Map<String, Object> jsonFeature = (Map<String, Object>) seqFeatureItr.next();
         Long begin = (Long) jsonFeature.get("xStart");
         Long end = (Long) jsonFeature.get("xEnd");
         String type = (String) jsonFeature.get("type");
@@ -665,14 +655,13 @@ public class JSONFile extends AlignFile implements ComplexAlignFile
         SequenceFeature sequenceFeature = new SequenceFeature(type,
                 description, featureBegin, featureEnd, score, featureGrp);
 
-        JSONArray linksJsonArray = (JSONArray) jsonFeature.get("links");
+        List<Object> linksJsonArray = (List<Object>) jsonFeature.get("links");
         if (linksJsonArray != null && linksJsonArray.size() > 0)
         {
-          Iterator<String> linkList = linksJsonArray.iterator();
+          Iterator<Object> linkList = linksJsonArray.iterator();
           while (linkList.hasNext())
           {
-            String link = linkList.next();
-            sequenceFeature.addLink(link);
+            sequenceFeature.addLink((String) linkList.next());
           }
         }