Merge branch 'bug/JAL-2791exportFilteredFeature' into merge/JAL-2791

[jalview.git] / src / jalview / ext / ensembl / Species.java

diff --git a/src/jalview/ext/ensembl/Species.java b/src/jalview/ext/ensembl/Species.java
index 350d0d5..cc5465e 100644 (file)
--- a/src/jalview/ext/ensembl/Species.java
+++ b/src/jalview/ext/ensembl/Species.java
@@ -20,6 +20,9 @@
  */
 package jalview.ext.ensembl;
 
+import java.util.HashSet;
+import java.util.Set;
+
 /**
  * Selected species identifiers used by Ensembl
  * 
@@ -34,10 +37,22 @@ enum Species
    */
   human(true), mouse(true), s_cerevisiae(true), cow(false), pig(false),
   rat(true), celegans(true), sheep(false), horse(false), gorilla(false),
-  rabbit(false), gibbon(false), dog(false), orangutan(false),
-  xenopus(true), chimpanzee(false), cat(false), zebrafish(true), chicken(
-          true), dmelanogaster(true);
+  rabbit(false), gibbon(false), dog(false), orangutan(false), xenopus(true),
+  chimpanzee(false), cat(false), zebrafish(true), chicken(true),
+  dmelanogaster(true);
+
+  static Set<Species> modelOrganisms = new HashSet<>();
 
+  static
+  {
+    for (Species s : values())
+    {
+      if (s.isModelOrganism())
+      {
+        modelOrganisms.add(s);
+      }
+    }
+  }
   boolean modelOrganism;
 
   private Species(boolean model)
@@ -49,4 +64,9 @@ enum Species
   {
     return modelOrganism;
   }
+
+  public static Set<Species> getModelOrganisms()
+  {
+    return modelOrganisms;
+  }
 }