JAL-629 Tidy up tests and replaced methods before merge to develop

[jalview.git] / src / jalview / datamodel / SeqDistanceContactMatrix.java

diff --git a/src/jalview/datamodel/SeqDistanceContactMatrix.java b/src/jalview/datamodel/SeqDistanceContactMatrix.java
index 4b23f04..e37a5b2 100644 (file)
--- a/src/jalview/datamodel/SeqDistanceContactMatrix.java
+++ b/src/jalview/datamodel/SeqDistanceContactMatrix.java
@@ -1,5 +1,10 @@
 package jalview.datamodel;
 
+import java.awt.Color;
+import java.util.BitSet;
+import java.util.HashMap;
+import java.util.List;
+
 /**
  * Dummy contact matrix based on sequence distance
  * 
@@ -8,6 +13,8 @@ package jalview.datamodel;
  */
 public class SeqDistanceContactMatrix implements ContactMatrixI
 {
+  private static final String SEQUENCE_DISTANCE = "SEQUENCE_DISTANCE";
+
   private int width = 0;
 
   public SeqDistanceContactMatrix(int width)
@@ -88,4 +95,69 @@ public class SeqDistanceContactMatrix implements ContactMatrixI
     return null;
   }
 
+  @Override
+  public String getAnnotDescr()
+  {
+    return "Sequence distance matrix";
+  }
+
+  @Override
+  public String getAnnotLabel()
+  {
+    return "Sequence Distance";
+  }
+
+  @Override
+  public String getType()
+  {
+    return SEQUENCE_DISTANCE;
+  }
+
+  @Override
+  public int getWidth()
+  {
+    return width;
+  }
+
+  @Override
+  public int getHeight()
+  {
+    return width;
+  }
+  private List<BitSet> groups=null;
+  @Override
+  public void updateGroups(List<BitSet> colGroups)
+  {
+    groups = colGroups;
+  }
+  @Override
+  public boolean hasGroups()
+  {
+    return groups!=null;
+  }
+  @Override
+  public List<BitSet> getGroups()
+  {
+    return groups;
+  }  
+
+  HashMap<BitSet,Color> colorMap = new HashMap<>();
+  @Override 
+  public Color getColourForGroup(BitSet bs)
+  {
+    if (bs==null) {
+      return Color.white;
+    }
+    Color groupCol=colorMap.get(bs);
+    if (groupCol==null)
+    {
+      return Color.white;
+    }
+    return groupCol;
+  }
+  @Override 
+  public void setColorForGroup(BitSet bs,Color color)
+  {
+    colorMap.put(bs,color);
+  }
 }