X-Git-Url: http://source.jalview.org/gitweb/?a=blobdiff_plain;ds=sidebyside;f=src%2Fjalview%2Fdatamodel%2FSequenceGroup.java;h=f640ebbb66a19906392b18c37217c3038dd4f595;hb=25ad831b7890848eff8f1a1676932774efef2183;hp=6f6850bc1ecf657d461642b49159e4788ec02178;hpb=ba0bffb2a7cda833ff1dd2691f756294402b48cd;p=jalview.git

diff --git a/src/jalview/datamodel/SequenceGroup.java b/src/jalview/datamodel/SequenceGroup.java
index 6f6850b..f640ebb 100755
--- a/src/jalview/datamodel/SequenceGroup.java
+++ b/src/jalview/datamodel/SequenceGroup.java
@@ -1,6 +1,6 @@
 /*
  * Jalview - A Sequence Alignment Editor and Viewer
- * Copyright (C) 2005 AM Waterhouse, J Procter, G Barton, M Clamp, S Searle
+ * Copyright (C) 2006 AM Waterhouse, J Procter, G Barton, M Clamp, S Searle
  *
  * This program is free software; you can redistribute it and/or
  * modify it under the terms of the GNU General Public License
@@ -20,325 +20,597 @@ package jalview.datamodel;
 
 import jalview.analysis.*;
 
-import jalview.datamodel.*;
-
 import jalview.schemes.*;
 
 import java.awt.*;
 
-import java.util.Vector;
+import java.util.*;
+
 
+/**
+ * DOCUMENT ME!
+ *
+ * @author $author$
+ * @version $Revision$
+ */
 public class SequenceGroup
 {
-  String groupName;
-  Conservation conserve;
-  Vector aaFrequency;
-  boolean displayBoxes;
-  boolean displayText;
-  boolean colourText;
-  public Vector sequences = new Vector();
-  int width = -1;
-  public ColourSchemeI cs;
-  int startRes = 0;
-  int endRes = 0;
-  Color outlineColour = Color.black;
-
-  public SequenceGroup()
-  {
-    groupName = "Group";
-    this.displayBoxes = true;
-    this.displayText = true;
-    this.colourText = false;
-    cs = null;
-  }
-
-  public SequenceGroup(Vector sequences, String groupName, ColourSchemeI scheme,
-                       boolean displayBoxes, boolean displayText,
-                       boolean colourText,
-                       int start, int end)
-  {
-    this.sequences = sequences;
-    this.groupName = groupName;
-    this.displayBoxes = displayBoxes;
-    this.displayText = displayText;
-    this.colourText = colourText;
-    this.cs = scheme;
-    startRes = start;
-    endRes = end;
-    recalcConservation();
-  }
-
-  public SequenceGroup(String groupName, ColourSchemeI scheme,
-                       boolean displayBoxes, boolean displayText,
-                       boolean colourText,
-                       int start, int end)
-  {
-    this.groupName = groupName;
-    this.displayBoxes = displayBoxes;
-    this.displayText = displayText;
-    this.colourText = colourText;
-    this.cs = scheme;
-    startRes = start;
-    endRes = end;
-  }
-
-  public boolean adjustForRemoveLeft(int col)
-  {
-    // return value is true if the group still exists
-    if (startRes >= col)
-    {
-      startRes = startRes - col;
-    }
-
-    if (endRes >= col)
-    {
-      endRes = endRes - col;
-
-      if (startRes > endRes)
-      {
-        startRes = 0;
-      }
-    }
-    else
+    String groupName;
+    Conservation conserve;
+    Vector aaFrequency;
+    boolean displayBoxes;
+    boolean displayText;
+    boolean colourText;
+    private Vector sequences = new Vector();
+    int width = -1;
+
+    /** DOCUMENT ME!! */
+    public ColourSchemeI cs;
+    int startRes = 0;
+    int endRes = 0;
+    Color outlineColour = Color.black;
+    public int thresholdTextColour = 0;
+    public Color textColour = Color.black;
+    public Color textColour2 = Color.white;
+
+
+    /**
+     * Creates a new SequenceGroup object.
+     */
+    public SequenceGroup()
     {
-      // must delete this group!!
-      return false;
+        groupName = "Group";
+        this.displayBoxes = true;
+        this.displayText = true;
+        this.colourText = false;
+        cs = null;
     }
 
-    return true;
-  }
-
-  public boolean adjustForRemoveRight(int col)
-  {
-    if (startRes > col)
+    /**
+     * Creates a new SequenceGroup object.
+     *
+     * @param sequences DOCUMENT ME!
+     * @param groupName DOCUMENT ME!
+     * @param scheme DOCUMENT ME!
+     * @param displayBoxes DOCUMENT ME!
+     * @param displayText DOCUMENT ME!
+     * @param colourText DOCUMENT ME!
+     * @param start DOCUMENT ME!
+     * @param end DOCUMENT ME!
+     */
+    public SequenceGroup(Vector sequences, String groupName,
+        ColourSchemeI scheme, boolean displayBoxes, boolean displayText,
+        boolean colourText, int start, int end)
     {
-      // delete this group
-      return false;
+        this.sequences = sequences;
+        this.groupName = groupName;
+        this.displayBoxes = displayBoxes;
+        this.displayText = displayText;
+        this.colourText = colourText;
+        this.cs = scheme;
+        startRes = start;
+        endRes = end;
+        recalcConservation();
     }
 
-    if (endRes >= col)
+    /**
+     * Creates a new SequenceGroup object.
+     *
+     * @param groupName DOCUMENT ME!
+     * @param scheme DOCUMENT ME!
+     * @param displayBoxes DOCUMENT ME!
+     * @param displayText DOCUMENT ME!
+     * @param colourText DOCUMENT ME!
+     * @param start DOCUMENT ME!
+     * @param end DOCUMENT ME!
+     */
+    public SequenceGroup(String groupName, ColourSchemeI scheme,
+        boolean displayBoxes, boolean displayText, boolean colourText,
+        int start, int end)
     {
-      endRes = col;
+        this.groupName = groupName;
+        this.displayBoxes = displayBoxes;
+        this.displayText = displayText;
+        this.colourText = colourText;
+        this.cs = scheme;
+        startRes = start;
+        endRes = end;
     }
 
-    return true;
-  }
+    public SequenceI [] getSelectionAsNewSequences(AlignmentI align)
+    {
+      int iSize = sequences.size();
+      SequenceI [] seqs = new SequenceI[iSize];
+      SequenceI [] inorder = getSequencesInOrder(align);
 
-  public String getName()
-  {
-    return groupName;
-  }
 
-  public void setName(String name)
-  {
-    groupName = name;
-  }
+    for (int i = 0; i < iSize; i++)
+    {
+      SequenceI seq = inorder[i];
 
-  public Conservation getConservation()
-  {
-    return conserve;
-  }
+      seqs[i] = new Sequence(seq.getName(),
+                             seq.getSequence(startRes, endRes + 1),
+                             seq.findPosition(startRes),
+                             findEndRes(seq));
 
-  public void setConservation(Conservation c)
-  {
-    conserve = c;
-  }
+      seqs[i].setDescription(seq.getDescription());
+      seqs[i].setDBRef(seq.getDBRef());
+      seqs[i].setSequenceFeatures(seq.getSequenceFeatures());
+      if (seq.getDatasetSequence() != null)
+        seqs[i].setDatasetSequence(seq.getDatasetSequence());
 
-  public void addSequence(SequenceI s, boolean recalc)
-  {
-    if (!sequences.contains(s))
-      sequences.addElement(s);
+      if(seq.getAnnotation()!=null)
+      {
+        for(int a=0; a<seq.getAnnotation().length; a++)
+          seqs[i].addAlignmentAnnotation(seq.getAnnotation()[a]);
+      }
+    }
 
-    if(recalc)
-      recalcConservation();
-  }
+    return seqs;
 
+    }
 
-  public void recalcConservation()
-  {
-    if (cs != null)
+    /**
+     * If sequence ends in gaps, the end residue can
+     * be correctly calculated here
+     * @param seq SequenceI
+     * @return int
+     */
+    public int findEndRes(SequenceI seq)
     {
-      cs.setConsensus(AAFrequency.calculate(sequences, 0, getWidth()));
+      int eres = 0;
+      char ch;
+
+      for (int j = 0; j < endRes + 1 && j < seq.getLength(); j++)
+      {
+        ch = seq.getCharAt(j);
+        if (!jalview.util.Comparison.isGap( (ch)))
+        {
+          eres++;
+        }
+      }
+
+      if (eres > 0)
+      {
+        eres += seq.getStart() - 1;
+      }
+
+      return eres;
     }
 
-    if (cs instanceof ClustalxColourScheme)
+    public Vector getSequences(boolean includeHidden)
     {
-      ( (ClustalxColourScheme) cs).resetClustalX(sequences,getWidth());
+      if(!includeHidden)
+        return sequences;
+      else
+      {
+        Vector allSequences = new Vector();
+        SequenceI seq;
+        for (int i = 0; i < sequences.size(); i++)
+        {
+          seq = (SequenceI) sequences.elementAt(i);
+          allSequences.addElement(seq);
+          if (seq.getHiddenSequences() != null)
+          {
+            for (int h = 0; h < seq.getHiddenSequences().getSize(false); h++)
+            {
+              allSequences.addElement(
+                  seq.getHiddenSequences().getSequenceAt(h)
+                  );
+            }
+          }
+        }
+
+        return allSequences;
+      }
     }
 
+    public SequenceI[] getSequencesAsArray(boolean includeHidden)
+    {
+      Vector tmp = getSequences(includeHidden);
+      if(tmp==null)
+        return null;
+      SequenceI [] result = new SequenceI[tmp.size()];
+      for(int i=0; i<result.length; i++)
+        result[i] = (SequenceI)tmp.elementAt(i);
+
+      return result;
+    }
 
-    if ( cs  instanceof ConservationColourScheme)
+    /**
+     * DOCUMENT ME!
+     *
+     * @param col DOCUMENT ME!
+     *
+     * @return DOCUMENT ME!
+     */
+    public boolean adjustForRemoveLeft(int col)
     {
-      Conservation c = new Conservation(groupName,
-                                        ResidueProperties.propHash, 3, sequences,
-                                        0, getWidth());
-      c.calculate();
-      c.verdict(false, 25);
+        // return value is true if the group still exists
+        if (startRes >= col)
+        {
+            startRes = startRes - col;
+        }
 
-      ConservationColourScheme ccs = (ConservationColourScheme) cs;
-      ccs.conserve = c;
-      if (ccs.cs instanceof ClustalxColourScheme)
-      {
-        ( (ClustalxColourScheme) ccs.cs).resetClustalX(sequences, getWidth());
-      }
+        if (endRes >= col)
+        {
+            endRes = endRes - col;
 
+            if (startRes > endRes)
+            {
+                startRes = 0;
+            }
+        }
+        else
+        {
+            // must delete this group!!
+            return false;
+        }
+
+        return true;
     }
-  }
 
-  public void addOrRemove(SequenceI s, boolean recalc)
-  {
-    if (sequences.contains(s))
+    /**
+     * DOCUMENT ME!
+     *
+     * @param col DOCUMENT ME!
+     *
+     * @return DOCUMENT ME!
+     */
+    public boolean adjustForRemoveRight(int col)
     {
-      deleteSequence(s, recalc);
+        if (startRes > col)
+        {
+            // delete this group
+            return false;
+        }
+
+        if (endRes >= col)
+        {
+            endRes = col;
+        }
+
+        return true;
     }
-    else
+
+    /**
+     * DOCUMENT ME!
+     *
+     * @return DOCUMENT ME!
+     */
+    public String getName()
     {
-      addSequence(s, recalc);
+        return groupName;
     }
-  }
 
-  public void deleteSequence(SequenceI s, boolean recalc)
-  {
-    sequences.removeElement(s);
-    if(recalc)
-      recalcConservation();
-  }
+    /**
+     * DOCUMENT ME!
+     *
+     * @param name DOCUMENT ME!
+     */
+    public void setName(String name)
+    {
+        groupName = name;
+    }
 
-  public int getStartRes()
-  {
-    return startRes;
-  }
+    /**
+     * DOCUMENT ME!
+     *
+     * @return DOCUMENT ME!
+     */
+    public Conservation getConservation()
+    {
+        return conserve;
+    }
 
-  public int getEndRes()
-  {
-    return endRes;
-  }
+    /**
+     * DOCUMENT ME!
+     *
+     * @param c DOCUMENT ME!
+     */
+    public void setConservation(Conservation c)
+    {
+        conserve = c;
+    }
 
-  public void setStartRes(int i)
-  {
-    startRes = i;
-  }
+    /**
+     * DOCUMENT ME!
+     *
+     * @param s DOCUMENT ME!
+     * @param recalc DOCUMENT ME!
+     */
+    public void addSequence(SequenceI s, boolean recalc)
+    {
+        if (s!=null && !sequences.contains(s))
+        {
+            sequences.addElement(s);
+        }
 
-  public void setEndRes(int i)
-  {
-    endRes = i;
-  }
+        if (recalc)
+        {
+            recalcConservation();
+        }
+    }
 
-  public int getSize()
-  {
-    return sequences.size();
-  }
+    /**
+     * DOCUMENT ME!
+     */
+    public void recalcConservation()
+    {
+        if(cs == null)
+          return;
 
-  public SequenceI getSequenceAt(int i)
-  {
-    return (SequenceI) sequences.elementAt(i);
-  }
+        try
+        {
+          cs.setConsensus(AAFrequency.calculate(sequences, startRes, endRes+1));
+
+          if (cs instanceof ClustalxColourScheme)
+          {
+            ( (ClustalxColourScheme) cs).resetClustalX(sequences, getWidth());
+          }
+
+          if (cs.conservationApplied())
+          {
+            Conservation c = new Conservation(groupName,
+                                              ResidueProperties.propHash, 3, sequences,
+                                              startRes, endRes+1);
+            c.calculate();
+            c.verdict(false, 25);
+
+            cs.setConservation(c);
+
+            if (cs instanceof ClustalxColourScheme)
+            {
+              ( (ClustalxColourScheme) cs).resetClustalX(sequences,
+                                                         getWidth());
+            }
+          }
+        }
+        catch (java.lang.OutOfMemoryError err)
+        {
+          System.out.println("Out of memory loading groups: " + err);
+        }
 
-  public void setColourText(boolean state)
-  {
-    colourText = state;
-  }
+    }
 
-  public boolean getColourText()
-  {
-    return colourText;
-  }
+    /**
+     * DOCUMENT ME!
+     *
+     * @param s DOCUMENT ME!
+     * @param recalc DOCUMENT ME!
+     */
+    public void addOrRemove(SequenceI s, boolean recalc)
+    {
+        if (sequences.contains(s))
+        {
+            deleteSequence(s, recalc);
+        }
+        else
+        {
+            addSequence(s, recalc);
+        }
+    }
 
-  public void setDisplayText(boolean state)
-  {
-    displayText = state;
-  }
+    /**
+     * DOCUMENT ME!
+     *
+     * @param s DOCUMENT ME!
+     * @param recalc DOCUMENT ME!
+     */
+    public void deleteSequence(SequenceI s, boolean recalc)
+    {
+        sequences.removeElement(s);
 
-  public boolean getDisplayText()
-  {
-    return displayText;
-  }
+        if (recalc)
+        {
+            recalcConservation();
+        }
+    }
 
-  public void setDisplayBoxes(boolean state)
-  {
-    displayBoxes = state;
-  }
+    /**
+     * DOCUMENT ME!
+     *
+     * @return DOCUMENT ME!
+     */
+    public int getStartRes()
+    {
+        return startRes;
+    }
 
-  public boolean getDisplayBoxes()
-  {
-    return displayBoxes;
-  }
+    /**
+     * DOCUMENT ME!
+     *
+     * @return DOCUMENT ME!
+     */
+    public int getEndRes()
+    {
+        return endRes;
+    }
 
-  public int getWidth()
-  {
-    // MC This needs to get reset when characters are inserted and deleted
-    if (sequences.size() > 0)
+    /**
+     * DOCUMENT ME!
+     *
+     * @param i DOCUMENT ME!
+     */
+    public void setStartRes(int i)
     {
-      width = ( (SequenceI) sequences.elementAt(0)).getLength();
+        startRes = i;
     }
 
-    for (int i = 1; i < sequences.size(); i++)
+    /**
+     * DOCUMENT ME!
+     *
+     * @param i DOCUMENT ME!
+     */
+    public void setEndRes(int i)
     {
-      SequenceI seq = (SequenceI) sequences.elementAt(i);
+        endRes = i;
+    }
 
-      if (seq.getLength() > width)
+    /**
+     * DOCUMENT ME!
+     *
+     * @return DOCUMENT ME!
+     */
+    public int getSize(boolean includeHidden)
+    {
+      if(!includeHidden)
+        return sequences.size();
+      else
       {
-        width = seq.getLength();
+        int total = sequences.size();
+        SequenceI seq;
+        for (int i = 0; i < sequences.size(); i++)
+        {
+          seq = (SequenceI) sequences.elementAt(i);
+          if (seq.getHiddenSequences() != null)
+          {
+            total += seq.getHiddenSequences().getSize(false);
+          }
+        }
+        return total;
       }
     }
 
-    return width;
-  }
+    /**
+     * DOCUMENT ME!
+     *
+     * @param i DOCUMENT ME!
+     *
+     * @return DOCUMENT ME!
+     */
+    public SequenceI getSequenceAt(int i)
+    {
+        return (SequenceI) sequences.elementAt(i);
+    }
+
+    /**
+     * DOCUMENT ME!
+     *
+     * @param state DOCUMENT ME!
+     */
+    public void setColourText(boolean state)
+    {
+        colourText = state;
+    }
 
-  public void setOutlineColour(Color c)
-  {
-    outlineColour = c;
-  }
+    /**
+     * DOCUMENT ME!
+     *
+     * @return DOCUMENT ME!
+     */
+    public boolean getColourText()
+    {
+        return colourText;
+    }
 
-  public Color getOutlineColour()
-  {
-    return outlineColour;
-  }
+    /**
+     * DOCUMENT ME!
+     *
+     * @param state DOCUMENT ME!
+     */
+    public void setDisplayText(boolean state)
+    {
+        displayText = state;
+    }
 
-  /**
-   *
-   * returns the sequences in the group ordered by the ordering given by al
-   *
-   * @param al Alignment
-   * @return SequenceI[]
-   */
-  public SequenceI[] getSequencesInOrder(Alignment al)
-  {
-    int sz;
-    java.util.Hashtable orderedSeqs = new java.util.Hashtable();
-    SequenceI[] seqs = new SequenceI[sz = sequences.size()];
+    /**
+     * DOCUMENT ME!
+     *
+     * @return DOCUMENT ME!
+     */
+    public boolean getDisplayText()
+    {
+        return displayText;
+    }
 
-    for (int i = 0; i < sz; i++)
+    /**
+     * DOCUMENT ME!
+     *
+     * @param state DOCUMENT ME!
+     */
+    public void setDisplayBoxes(boolean state)
     {
-      SequenceI seq = (SequenceI) sequences.elementAt(i);
-      int index = al.findIndex(seq);
-      orderedSeqs.put(index + "", seq);
+        displayBoxes = state;
     }
 
-    int index = 0;
+    /**
+     * DOCUMENT ME!
+     *
+     * @return DOCUMENT ME!
+     */
+    public boolean getDisplayBoxes()
+    {
+        return displayBoxes;
+    }
 
-    for (int i = 0; i < sz; i++)
+    /**
+     * DOCUMENT ME!
+     *
+     * @return DOCUMENT ME!
+     */
+    public int getWidth()
     {
-      SequenceI seq = null;
+        // MC This needs to get reset when characters are inserted and deleted
+        if (sequences.size() > 0)
+        {
+            width = ((SequenceI) sequences.elementAt(0)).getLength();
+        }
 
-      while (seq == null)
-      {
-        if (orderedSeqs.containsKey(index + ""))
+        for (int i = 1; i < sequences.size(); i++)
         {
-          seq = (SequenceI) orderedSeqs.get(index + "");
-          index++;
+            SequenceI seq = (SequenceI) sequences.elementAt(i);
 
-          break;
+            if (seq.getLength() > width)
+            {
+                width = seq.getLength();
+            }
         }
-        else
+
+        return width;
+    }
+
+    /**
+     * DOCUMENT ME!
+     *
+     * @param c DOCUMENT ME!
+     */
+    public void setOutlineColour(Color c)
+    {
+        outlineColour = c;
+    }
+
+    /**
+     * DOCUMENT ME!
+     *
+     * @return DOCUMENT ME!
+     */
+    public Color getOutlineColour()
+    {
+        return outlineColour;
+    }
+
+    /**
+     *
+     * returns the sequences in the group ordered by the ordering given by al
+     *
+     * @param al Alignment
+     * @return SequenceI[]
+     */
+    public SequenceI[] getSequencesInOrder(AlignmentI al)
+    {
+        int sSize = sequences.size();
+        int alHeight = al.getHeight();
+
+        SequenceI[] seqs = new SequenceI[sSize];
+
+        int index = 0;
+        for (int i = 0; i < alHeight && index<sSize; i++)
         {
-          index++;
+          if(sequences.contains( al.getSequenceAt(i) ) )
+            seqs[index++] = al.getSequenceAt(i);
         }
-      }
 
-      seqs[index] = seq;
+        return seqs;
     }
-
-    return seqs;
-  }
 }