X-Git-Url: http://source.jalview.org/gitweb/?a=blobdiff_plain;f=src%2Fjalview%2Fext%2Fensembl%2FEnsemblFeatures.java;h=56498590928b507ac15f510d9b18fd68d875b4a6;hb=57738a1f3c19b1c3a00bd3ac5108f8cd0af32f99;hp=744191dfe112993c049363414ff5f77a457df56f;hpb=783fbd0b01eea0b485a8664221413ef5ade10e21;p=jalview.git

diff --git a/src/jalview/ext/ensembl/EnsemblFeatures.java b/src/jalview/ext/ensembl/EnsemblFeatures.java
index 744191d..5649859 100644
--- a/src/jalview/ext/ensembl/EnsemblFeatures.java
+++ b/src/jalview/ext/ensembl/EnsemblFeatures.java
@@ -27,6 +27,7 @@ import jalview.datamodel.SequenceFeature;
 import jalview.datamodel.SequenceI;
 import jalview.io.gff.SequenceOntologyI;
 import jalview.util.JSONUtils;
+import jalview.util.Platform;
 
 import java.io.BufferedReader;
 import java.io.IOException;
@@ -35,10 +36,8 @@ import java.net.URL;
 import java.util.ArrayList;
 import java.util.Iterator;
 import java.util.List;
+import java.util.Map;
 
-import org.json.simple.JSONArray;
-import org.json.simple.JSONObject;
-import org.json.simple.parser.JSONParser;
 import org.json.simple.parser.ParseException;
 
 /**
@@ -92,16 +91,14 @@ class EnsemblFeatures extends EnsemblRestClient
   public AlignmentI getSequenceRecords(String query) throws IOException
   {
     // TODO: use a vararg String... for getSequenceRecords instead?
+
     List<String> queries = new ArrayList<>();
     queries.add(query);
-    BufferedReader fp = getSequenceReader(queries);
-    if (fp == null)
-    {
+    SequenceI seq = parseFeaturesJson(queries);
+    if (seq == null)
       return null;
-    }
-
-    SequenceI seq = parseFeaturesJson(fp);
     return new Alignment(new SequenceI[] { seq });
+
   }
 
   /**
@@ -111,30 +108,33 @@ class EnsemblFeatures extends EnsemblRestClient
    * @param br
    * @return
    */
-  private SequenceI parseFeaturesJson(BufferedReader br)
+  @SuppressWarnings("unchecked")
+  private SequenceI parseFeaturesJson(List<String> queries)
   {
     SequenceI seq = new Sequence("Dummy", "");
-
-    JSONParser jp = new JSONParser();
     try
     {
-      JSONArray responses = (JSONArray) jp.parse(br);
-      Iterator rvals = responses.iterator();
+      Iterator<Object> rvals = (Iterator<Object>) getJSON(null, queries, -1,
+              MODE_ITERATOR, null);
+      if (rvals == null)
+      {
+        return null;
+      }
       while (rvals.hasNext())
       {
         try
         {
-          JSONObject obj = (JSONObject) rvals.next();
+          Map<String, Object> obj = (Map<String, Object>) rvals.next();
           String type = obj.get("feature_type").toString();
           int start = Integer.parseInt(obj.get("start").toString());
           int end = Integer.parseInt(obj.get("end").toString());
           String source = obj.get("source").toString();
           String strand = obj.get("strand").toString();
+          Object phase = obj.get("phase");
           String alleles = JSONUtils
-                  .arrayToList((JSONArray) obj.get("alleles"));
-          String clinSig = JSONUtils
-                  .arrayToList(
-                          (JSONArray) obj.get("clinical_significance"));
+                  .arrayToStringList((List<Object>) obj.get("alleles"));
+          String clinSig = JSONUtils.arrayToStringList(
+                  (List<Object>) obj.get("clinical_significance"));
 
           /*
            * convert 'variation' to 'sequence_variant', and 'cds' to 'CDS'
@@ -149,12 +149,16 @@ class EnsemblFeatures extends EnsemblRestClient
           {
             type = SequenceOntologyI.CDS;
           }
-          
+
           String desc = getFirstNotNull(obj, "alleles", "external_name",
                   JSON_ID);
           SequenceFeature sf = new SequenceFeature(type, desc, start, end,
                   source);
           sf.setStrand("1".equals(strand) ? "+" : "-");
+          if (phase != null)
+          {
+            sf.setPhase(phase.toString());
+          }
           setFeatureAttribute(sf, obj, "id");
           setFeatureAttribute(sf, obj, "Parent");
           setFeatureAttribute(sf, obj, "consequence_type");
@@ -162,6 +166,7 @@ class EnsemblFeatures extends EnsemblRestClient
           sf.setValue("clinical_significance", clinSig);
 
           seq.addSequenceFeature(sf);
+
         } catch (Throwable t)
         {
           // ignore - keep trying other features
@@ -169,6 +174,7 @@ class EnsemblFeatures extends EnsemblRestClient
       }
     } catch (ParseException | IOException e)
     {
+      e.printStackTrace();
       // ignore
     }
 
@@ -184,17 +190,16 @@ class EnsemblFeatures extends EnsemblRestClient
    * @param keys
    * @return
    */
-  protected String getFirstNotNull(JSONObject obj, String... keys)
+  @SuppressWarnings("unchecked")
+  protected String getFirstNotNull(Map<String, Object> obj, String... keys)
   {
-    String desc = null;
-
     for (String key : keys)
     {
       Object val = obj.get(key);
       if (val != null)
       {
-        String s = val instanceof JSONArray
-                ? JSONUtils.arrayToList((JSONArray) val)
+        String s = val instanceof List<?>
+                ? JSONUtils.arrayToStringList((List<Object>) val)
                 : val.toString();
         if (!s.isEmpty())
         {
@@ -202,7 +207,7 @@ class EnsemblFeatures extends EnsemblRestClient
         }
       }
     }
-    return desc;
+    return null;
   }
 
   /**
@@ -213,8 +218,8 @@ class EnsemblFeatures extends EnsemblRestClient
    * @param obj
    * @param key
    */
-  protected void setFeatureAttribute(SequenceFeature sf, JSONObject obj,
-          String key)
+  protected void setFeatureAttribute(SequenceFeature sf,
+          Map<String, Object> obj, String key)
   {
     Object object = obj.get(key);
     if (object != null)