X-Git-Url: http://source.jalview.org/gitweb/?a=blobdiff_plain;f=src%2Fjalview%2Fanalysis%2FGrouping.java;h=165a3df6c8b19f505f1b137fe0a3f5dabf77ec00;hb=db93a1adcbe0a4eaaf06e0a70ade0d6c5c1961c3;hp=e4c9cea07c0843b0add0edcd62e20c94e72de747;hpb=47168f025aefdaa044802bd5f8f510ffe43a4808;p=jalview.git

diff --git a/src/jalview/analysis/Grouping.java b/src/jalview/analysis/Grouping.java
index e4c9cea..165a3df 100644
--- a/src/jalview/analysis/Grouping.java
+++ b/src/jalview/analysis/Grouping.java
@@ -1,6 +1,6 @@
 /*
- * Jalview - A Sequence Alignment Editor and Viewer (Version 2.8.2)
- * Copyright (C) 2014 The Jalview Authors
+ * Jalview - A Sequence Alignment Editor and Viewer (Version 2.9.0b2)
+ * Copyright (C) 2015 The Jalview Authors
  * 
  * This file is part of Jalview.
  * 
@@ -20,11 +20,14 @@
  */
 package jalview.analysis;
 
+import jalview.datamodel.ColumnSelection;
 import jalview.datamodel.SequenceGroup;
 import jalview.datamodel.SequenceI;
 
-import java.util.Hashtable;
+import java.util.ArrayList;
+import java.util.HashMap;
 import java.util.List;
+import java.util.Map;
 import java.util.Vector;
 
 /**
@@ -50,9 +53,9 @@ public class Grouping
           String[] selectedChars, List<SequenceGroup> list)
   {
     // TODO: determine how to get/recover input data for group generation
-    Hashtable gps = new Hashtable();
+    Map<String, List<SequenceI>> gps = new HashMap<String, List<SequenceI>>();
     int width = 0, i;
-    Hashtable pgroup = new Hashtable();
+    Map<String, SequenceGroup> pgroup = new HashMap<String, SequenceGroup>();
     if (list != null)
     {
       for (SequenceGroup sg : list)
@@ -66,33 +69,117 @@ public class Grouping
     for (i = 0; i < sequences.length; i++)
     {
       String schar = selectedChars[i];
-      SequenceGroup pgp = (SequenceGroup) pgroup
-              .get(((Object) sequences[i]).toString());
+      SequenceGroup pgp = pgroup.get(((Object) sequences[i]).toString());
       if (pgp != null)
       {
         schar = pgp.getName() + ":" + schar;
       }
-      Vector svec = (Vector) gps.get(schar);
+      List<SequenceI> svec = gps.get(schar);
       if (svec == null)
       {
-        svec = new Vector();
+        svec = new ArrayList<SequenceI>();
         gps.put(schar, svec);
       }
       if (width < sequences[i].getLength())
       {
         width = sequences[i].getLength();
       }
-      svec.addElement(sequences[i]);
+      svec.add(sequences[i]);
     }
     // make some groups
-    java.util.Enumeration sge = gps.keys();
     SequenceGroup[] groups = new SequenceGroup[gps.size()];
     i = 0;
-    while (sge.hasMoreElements())
+    for (String key : gps.keySet())
     {
-      String key = (String) sge.nextElement();
-      SequenceGroup group = new SequenceGroup((Vector) gps.get(key),
-              "Subseq: " + key, null, true, true, false, 0, width - 1);
+      SequenceGroup group = new SequenceGroup(gps.get(key), "Subseq: "
+              + key, null, true, true, false, 0, width - 1);
+
+      groups[i++] = group;
+    }
+    gps.clear();
+    pgroup.clear();
+    return groups;
+  }
+
+  /**
+   * Divide the given sequences based on the equivalence of characters at
+   * selected columns If exgroups is provided, existing groups will be
+   * subdivided.
+   * 
+   * @param sequences
+   * @param columnSelection
+   * @param list
+   * @return
+   */
+  public static SequenceGroup[] makeGroupsFromCols(SequenceI[] sequences,
+          ColumnSelection cs, List<SequenceGroup> list)
+  {
+    // TODO: determine how to get/recover input data for group generation
+    Map<String, List<SequenceI>> gps = new HashMap<String, List<SequenceI>>();
+    Map<String, SequenceGroup> pgroup = new HashMap<String, SequenceGroup>();
+    if (list != null)
+    {
+      for (SequenceGroup sg : list)
+      {
+        for (SequenceI sq : sg.getSequences(null))
+        {
+          pgroup.put(sq.toString(), sg);
+        }
+      }
+    }
+
+    /*
+     * get selected columns (in the order they were selected);
+     * note this could include right-to-left ranges
+     */
+    int[] spos = new int[cs.getSelected().size()];
+    int width = -1;
+    int i = 0;
+    for (Integer pos : cs.getSelected())
+    {
+      spos[i++] = pos.intValue();
+    }
+
+    for (i = 0; i < sequences.length; i++)
+    {
+      int slen = sequences[i].getLength();
+      if (width < slen)
+      {
+        width = slen;
+      }
+
+      SequenceGroup pgp = pgroup.get(((Object) sequences[i]).toString());
+      StringBuilder schar = new StringBuilder();
+      if (pgp != null)
+      {
+        schar.append(pgp.getName() + ":");
+      }
+      for (int p : spos)
+      {
+        if (p >= slen)
+        {
+          schar.append("~");
+        }
+        else
+        {
+          schar.append(sequences[i].getCharAt(p));
+        }
+      }
+      List<SequenceI> svec = gps.get(schar.toString());
+      if (svec == null)
+      {
+        svec = new ArrayList<SequenceI>();
+        gps.put(schar.toString(), svec);
+      }
+      svec.add(sequences[i]);
+    }
+    // make some groups
+    SequenceGroup[] groups = new SequenceGroup[gps.size()];
+    i = 0;
+    for (String key : gps.keySet())
+    {
+      SequenceGroup group = new SequenceGroup(gps.get(key), "Subseq: "
+              + key, null, true, true, false, 0, width - 1);
 
       groups[i++] = group;
     }