X-Git-Url: http://source.jalview.org/gitweb/?a=blobdiff_plain;f=src%2Fjalview%2Fext%2Fensembl%2FEnsemblLookup.java;h=f6b3a47b6c07e73a55d474db4a9bb22b6b85d1e7;hb=321caefc5a40cd735c93e0bfa450e0e04abc485d;hp=0d1b5540c9728a4fbb47f71ff5aa3c5f2307344f;hpb=14b1b2a878190d5fffda298c4b9a901c72c74ad3;p=jalview.git

diff --git a/src/jalview/ext/ensembl/EnsemblLookup.java b/src/jalview/ext/ensembl/EnsemblLookup.java
index 0d1b554..f6b3a47 100644
--- a/src/jalview/ext/ensembl/EnsemblLookup.java
+++ b/src/jalview/ext/ensembl/EnsemblLookup.java
@@ -32,14 +32,15 @@ import java.net.URL;
 import java.util.Arrays;
 import java.util.Collections;
 import java.util.List;
-import java.util.function.Function;
 
 import org.json.simple.JSONObject;
 import org.json.simple.parser.JSONParser;
 import org.json.simple.parser.ParseException;
 
 /**
- * A client for the Ensembl lookup REST endpoint
+ * A client for the Ensembl /lookup REST endpoint, used to find the gene
+ * identifier given a gene, transcript or protein identifier, or to extract the
+ * species or chromosomal coordinates from the same service response
  * 
  * @author gmcarstairs
  */
@@ -47,14 +48,6 @@ public class EnsemblLookup extends EnsemblRestClient
 {
   private static final String SPECIES = "species";
 
-  private static final String PARENT = "Parent";
-
-  private static final String OBJECT_TYPE_TRANSLATION = "Translation";
-  private static final String OBJECT_TYPE_TRANSCRIPT = "Transcript";
-  private static final String ID = "id";
-  private static final String OBJECT_TYPE_GENE = "Gene";
-  private static final String OBJECT_TYPE = "object_type";
-
   /**
    * Default constructor (to use rest.ensembl.org)
    */
@@ -89,17 +82,26 @@ public class EnsemblLookup extends EnsemblRestClient
   protected URL getUrl(List<String> ids) throws MalformedURLException
   {
     String identifier = ids.get(0);
-    return getUrl(identifier);
+    return getUrl(identifier, null);
   }
 
   /**
+   * Gets the url for lookup of the given identifier, optionally with objectType
+   * also specified in the request
+   * 
    * @param identifier
+   * @param objectType
    * @return
    */
-  protected URL getUrl(String identifier)
+  protected URL getUrl(String identifier, String objectType)
   {
     String url = getDomain() + "/lookup/id/" + identifier
             + CONTENT_TYPE_JSON;
+    if (objectType != null)
+    {
+      url += "&" + OBJECT_TYPE + "=" + objectType;
+    }
+
     try
     {
       return new URL(url);
@@ -128,20 +130,15 @@ public class EnsemblLookup extends EnsemblRestClient
   }
 
   /**
-   * Calls the Ensembl lookup REST endpoint and returns
-   * <ul>
-   * <li>the 'id' for the identifier if its type is "Gene"</li>
-   * <li>the 'Parent' if its type is 'Transcript'</li>
-   * <ul>
-   * If the type is 'Translation', does a recursive call to this method, passing
-   * in the 'Parent' (transcript id).
+   * Returns the gene id related to the given identifier (which may be for a
+   * gene, transcript or protein)
    * 
    * @param identifier
    * @return
    */
   public String getGeneId(String identifier)
   {
-    return (String) getResult(identifier, br -> parseGeneId(br));
+    return parseGeneId(getResult(identifier, null));
   }
 
   /**
@@ -153,34 +150,45 @@ public class EnsemblLookup extends EnsemblRestClient
    */
   public String getSpecies(String identifier)
   {
-    return (String) getResult(identifier, br -> getAttribute(br, SPECIES));
+    String species = null;
+    JSONObject json = getResult(identifier, null);
+    if (json != null)
+    {
+      Object o = json.get(SPECIES);
+      if (o != null)
+      {
+        species = o.toString();
+      }
+    }
+    return species;
   }
 
   /**
-   * Calls the /lookup/id rest service and delegates parsing of the JSON
-   * response to the supplied parser
+   * Calls the /lookup/id rest service and returns the response as a JSONObject,
+   * or null if any error
    * 
    * @param identifier
-   * @param parser
+   * @param objectType
+   *          (optional)
    * @return
    */
-  protected Object getResult(String identifier,
-          Function<BufferedReader, Object> parser)
+  protected JSONObject getResult(String identifier, String objectType)
   {
     List<String> ids = Arrays.asList(new String[] { identifier });
 
     BufferedReader br = null;
     try
     {
-      URL url = getUrl(identifier);
+      URL url = getUrl(identifier, objectType);
       if (url != null)
       {
         br = getHttpResponse(url, ids);
       }
-      return br == null ? null : parser.apply(br);
-    } catch (IOException e)
+      return br == null ? null : (JSONObject) (new JSONParser().parse(br));
+    } catch (IOException | ParseException e)
     {
-      // ignore
+      System.err.println("Error parsing " + identifier + " lookup response "
+              + e.getMessage());
       return null;
     } finally
     {
@@ -198,74 +206,42 @@ public class EnsemblLookup extends EnsemblRestClient
   }
 
   /**
-   * Answers the value of 'attribute' from the JSON response, or null if not
-   * found
-   * 
-   * @param br
-   * @param attribute
-   * @return
-   */
-  protected String getAttribute(BufferedReader br, String attribute)
-  {
-    String value = null;
-    JSONParser jp = new JSONParser();
-    try
-    {
-      JSONObject val = (JSONObject) jp.parse(br);
-      value = val.get(attribute).toString();
-    } catch (ParseException | NullPointerException | IOException e)
-    {
-      // ignore
-    }
-    return value;
-  }
-
-  /**
    * Parses the JSON response and returns the gene identifier, or null if not
    * found. If the returned object_type is Gene, returns the id, if Transcript
    * returns the Parent. If it is Translation (peptide identifier), then the
-   * Parent is the transcript identifier, so we redo the search with this value.
+   * Parent is the transcript identifier, so we redo the search with this value,
+   * specifying that object_type should be Transcript.
    * 
-   * @param br
+   * @param jsonObject
    * @return
    */
-  protected String parseGeneId(BufferedReader br)
+  protected String parseGeneId(JSONObject json)
   {
+    if (json == null)
+    {
+      // e.g. lookup failed with 404 not found
+      return null;
+    }
+
     String geneId = null;
-    JSONParser jp = new JSONParser();
-    try
+    String type = json.get(OBJECT_TYPE).toString();
+    if (OBJECT_TYPE_GENE.equalsIgnoreCase(type))
     {
-      JSONObject val = (JSONObject) jp.parse(br);
-      String type = val.get(OBJECT_TYPE).toString();
-      if (OBJECT_TYPE_GENE.equalsIgnoreCase(type))
-      {
-        geneId = val.get(ID).toString();
-      }
-      else if (OBJECT_TYPE_TRANSCRIPT.equalsIgnoreCase(type))
-      {
-        geneId = val.get(PARENT).toString();
-      }
-      else if (OBJECT_TYPE_TRANSLATION.equalsIgnoreCase(type))
-      {
-        String transcriptId = val.get(PARENT).toString();
-        try
-        {
-          geneId = getGeneId(transcriptId);
-        } catch (StackOverflowError e)
-        {
-          /*
-           * unlikely data condition error!
-           */
-          System.err
-                  .println("** Ensembl lookup "
-                          + getUrl(transcriptId).toString()
-                          + " looping on Parent!");
-        }
-      }
-    } catch (ParseException | IOException e)
+      // got the gene - just returns its id
+      geneId = json.get(JSON_ID).toString();
+    }
+    else if (OBJECT_TYPE_TRANSCRIPT.equalsIgnoreCase(type))
+    {
+      // got the transcript - return its (Gene) Parent
+      geneId = json.get(PARENT).toString();
+    }
+    else if (OBJECT_TYPE_TRANSLATION.equalsIgnoreCase(type))
     {
-      // ignore
+      // got the protein - look up its Parent, restricted to type Transcript
+      String transcriptId = json.get(PARENT).toString();
+      geneId = parseGeneId(getResult(transcriptId, OBJECT_TYPE_TRANSCRIPT));
     }
+
     return geneId;
   }
 
@@ -278,7 +254,7 @@ public class EnsemblLookup extends EnsemblRestClient
    */
   public GeneLociI getGeneLoci(String geneId)
   {
-    return (GeneLociI) getResult(geneId, br -> parseGeneLoci(br));
+    return parseGeneLoci(getResult(geneId, OBJECT_TYPE_GENE));
   }
 
   /**
@@ -286,21 +262,24 @@ public class EnsemblLookup extends EnsemblRestClient
    * seq_region_name, start, end and returns an object that wraps them, or null
    * if unsuccessful
    * 
-   * @param br
+   * @param json
    * @return
    */
-  GeneLociI parseGeneLoci(BufferedReader br)
+  GeneLociI parseGeneLoci(JSONObject json)
   {
-    JSONParser jp = new JSONParser();
+    if (json == null)
+    {
+      return null;
+    }
+
     try
     {
-      JSONObject val = (JSONObject) jp.parse(br);
-      final String species = val.get("species").toString();
-      final String assembly = val.get("assembly_name").toString();
-      final String chromosome = val.get("seq_region_name").toString();
-      String strand = val.get("strand").toString();
-      int start = Integer.parseInt(val.get("start").toString());
-      int end = Integer.parseInt(val.get("end").toString());
+      final String species = json.get("species").toString();
+      final String assembly = json.get("assembly_name").toString();
+      final String chromosome = json.get("seq_region_name").toString();
+      String strand = json.get("strand").toString();
+      int start = Integer.parseInt(json.get("start").toString());
+      int end = Integer.parseInt(json.get("end").toString());
       int fromEnd = end - start + 1;
       boolean reverseStrand = "-1".equals(strand);
       int toStart = reverseStrand ? end : start;
@@ -337,10 +316,10 @@ public class EnsemblLookup extends EnsemblRestClient
           return map;
         }
       };
-    } catch (ParseException | NullPointerException | IOException
-            | NumberFormatException | ClassCastException e)
+    } catch (NullPointerException | NumberFormatException e)
     {
       Cache.log.error("Error looking up gene loci: " + e.getMessage());
+      e.printStackTrace();
     }
     return null;
   }