JAL-1492 JAL-1397 make AnnotatedCollectionI context aware so groups can be coloured...

[jalview.git] / src / jalview / schemes / UserColourScheme.java

diff --git a/src/jalview/schemes/UserColourScheme.java b/src/jalview/schemes/UserColourScheme.java
index 82a5871..61aed43 100755 (executable)
--- a/src/jalview/schemes/UserColourScheme.java
+++ b/src/jalview/schemes/UserColourScheme.java
@@ -19,7 +19,11 @@
 package jalview.schemes;
 
 import java.awt.Color;
+import java.util.Map;
 import java.util.StringTokenizer;
+
+import jalview.datamodel.AnnotatedCollectionI;
+import jalview.datamodel.SequenceCollectionI;
 import jalview.datamodel.SequenceI;
 
 public class UserColourScheme extends ResidueColourScheme
@@ -38,7 +42,18 @@ public class UserColourScheme extends ResidueColourScheme
     super(ResidueProperties.aaIndex);
     colors = newColors;
   }
-
+  @Override
+  public ColourSchemeI applyTo(AnnotatedCollectionI sg,
+          Map<SequenceI, SequenceCollectionI> hiddenRepSequences)
+  {
+    UserColourScheme usc = new UserColourScheme(colors);
+    if (lowerCaseColours!=null) {
+      usc.schemeName = new String(schemeName);
+      usc.lowerCaseColours = new Color[lowerCaseColours.length];
+      System.arraycopy(lowerCaseColours, 0, usc.lowerCaseColours, 0, lowerCaseColours.length);
+    }
+    return usc;
+  }
   public UserColourScheme(String colour)
   {
     super(ResidueProperties.aaIndex);