JAL-2446 added getFeatureGroups

[jalview.git] / src / jalview / datamodel / features / SequenceFeatures.java

diff --git a/src/jalview/datamodel/features/SequenceFeatures.java b/src/jalview/datamodel/features/SequenceFeatures.java
index 227d8fd..a61eff2 100644 (file)
--- a/src/jalview/datamodel/features/SequenceFeatures.java
+++ b/src/jalview/datamodel/features/SequenceFeatures.java
@@ -5,8 +5,10 @@ import jalview.datamodel.SequenceFeature;
 import java.util.ArrayList;
 import java.util.Collections;
 import java.util.HashMap;
+import java.util.HashSet;
 import java.util.List;
 import java.util.Map;
+import java.util.Set;
 
 /**
  * A class that stores sequence features in a way that supports efficient
@@ -207,4 +209,20 @@ public class SequenceFeatures
     }
     return false;
   }
+
+  /**
+   * Returns a set of the distinct feature groups present in the collection. The
+   * set may include null.
+   * 
+   * @return
+   */
+  public Set<String> getFeatureGroups()
+  {
+    Set<String> groups = new HashSet<String>();
+    for (FeatureStore featureSet : featureStore.values())
+    {
+      groups.addAll(featureSet.getFeatureGroups());
+    }
+    return groups;
+  }
 }