X-Git-Url: http://source.jalview.org/gitweb/?a=blobdiff_plain;f=src%2Fjalview%2Fgui%2FAnnotationColumnChooser.java;h=87e2d7869558147a7b91ba6389bb2a658b33cf94;hb=f73031db2e38bfe6e3f1cd42f91b0f9b0130071a;hp=4abf49d147d4ce55c38b39d988c9852014ae7216;hpb=ead72caff9aef9b7103dbb15907aeae777200832;p=jalview.git

diff --git a/src/jalview/gui/AnnotationColumnChooser.java b/src/jalview/gui/AnnotationColumnChooser.java
index 4abf49d..87e2d78 100644
--- a/src/jalview/gui/AnnotationColumnChooser.java
+++ b/src/jalview/gui/AnnotationColumnChooser.java
@@ -1,5 +1,6 @@
 package jalview.gui;
 
+import jalview.datamodel.ColumnSelection;
 import jalview.schemes.AnnotationColourGradient;
 import jalview.util.MessageManager;
 
@@ -21,6 +22,8 @@ import net.miginfocom.swing.MigLayout;
 public class AnnotationColumnChooser extends AnnotationRowFilter
 {
 
+  private ColumnSelection oldColumnSelection;
+
   private JComboBox<String> annotations;
 
   JButton ok = new JButton();
@@ -51,7 +54,7 @@ public class AnnotationColumnChooser extends AnnotationRowFilter
     {
       return;
     }
-
+    setOldColumnSelection(av.getColumnSelection());
     adjusting = true;
 
     setAnnotations(new JComboBox<String>(
@@ -196,6 +199,7 @@ public class AnnotationColumnChooser extends AnnotationRowFilter
   public void reset()
   {
     av.getColumnSelection().clear();
+    av.setColumnSelection(this.getOldColumnSelection());
   }
 
   public void valueChanged(boolean updateAllAnnotation)
@@ -288,4 +292,18 @@ public class AnnotationColumnChooser extends AnnotationRowFilter
     ap.alignmentChanged();
     ap.paintAlignment(true);
   }
+
+  public ColumnSelection getOldColumnSelection()
+  {
+    return oldColumnSelection;
+  }
+
+  public void setOldColumnSelection(ColumnSelection currentColumnSelection)
+  {
+    if (currentColumnSelection != null)
+    {
+      this.oldColumnSelection = new ColumnSelection();
+      this.oldColumnSelection.setElementsFrom(currentColumnSelection);
+    }
+  }
 }