JAL-1556 initial implementation - no tests as yet but ‘works’

[jalview.git] / src / jalview / gui / AnnotationLabels.java

diff --git a/src/jalview/gui/AnnotationLabels.java b/src/jalview/gui/AnnotationLabels.java
index 7462bc8..8930629 100755 (executable)
--- a/src/jalview/gui/AnnotationLabels.java
+++ b/src/jalview/gui/AnnotationLabels.java
@@ -41,6 +41,7 @@ import java.util.Arrays;
 import java.util.Collections;
 import java.util.Iterator;
 import java.util.Locale;
+import java.util.concurrent.Callable;
 
 import javax.swing.JCheckBoxMenuItem;
 import javax.swing.JMenuItem;
@@ -56,6 +57,7 @@ import jalview.datamodel.Alignment;
 import jalview.datamodel.AlignmentAnnotation;
 import jalview.datamodel.Annotation;
 import jalview.datamodel.ContactMatrixI;
+import jalview.datamodel.GroupSet;
 import jalview.datamodel.HiddenColumns;
 import jalview.datamodel.Sequence;
 import jalview.datamodel.SequenceGroup;
@@ -64,6 +66,7 @@ import jalview.io.DataSourceType;
 import jalview.io.FileFormat;
 import jalview.io.FormatAdapter;
 import jalview.io.NewickFile;
+import jalview.schemes.AnnotationColourGradient;
 import jalview.util.Comparison;
 import jalview.util.MessageManager;
 import jalview.util.Platform;
@@ -422,34 +425,131 @@ public class AnnotationLabels extends JPanel
         consclipbrd.addActionListener(this);
         pop.add(consclipbrd);
       }
-      if (aa[selectedRow].graph == AlignmentAnnotation.CONTACT_MAP
-              && PAEContactMatrix.PAEMATRIX
-                      .equals(aa[selectedRow].getCalcId()))
+
+      addColourOrFilterByOptions(ap,aa[selectedRow],pop);
+      
+      if (aa[selectedRow].graph == AlignmentAnnotation.CONTACT_MAP)
       {
-        final PAEContactMatrix cm = (PAEContactMatrix) av
-                .getContactMatrix(aa[selectedRow]);
-        if (cm != null && cm.getNewick() != null
-                && cm.getNewick().length() > 0)
+        addContactMatrixOptions(ap,aa[selectedRow],pop);
+          // Set/adjust threshold for grouping ?
+          // colour alignment by this [type]
+          // select/hide columns by this row
+          
+        }
+      }
+    
+    pop.show(this, evt.getX(), evt.getY());
+  }
+
+  static void addColourOrFilterByOptions(final AlignmentPanel ap,
+          final AlignmentAnnotation alignmentAnnotation, final JPopupMenu pop)
+  {
+    JMenuItem item;
+    item = new JMenuItem(MessageManager.getString("label.colour_by_annotation"));
+    item.addActionListener(new ActionListener()
+    {
+      
+      @Override
+      public void actionPerformed(ActionEvent e)
+      {
+        AnnotationColourChooser.displayFor(ap.av, ap,alignmentAnnotation,false);
+      };
+    });
+    pop.add(item);
+    if (alignmentAnnotation.sequenceRef!=null)
+    {
+      item = new JMenuItem(MessageManager.getString("label.colour_by_annotation")+" ("+MessageManager.getString("label.per_seq")+")");
+      item.addActionListener(new ActionListener()
+      {
+        @Override
+        public void actionPerformed(ActionEvent e)
         {
-          item = new JMenuItem("Show Tree for Matrix");
-          item.addActionListener(new ActionListener()
-          {
+          AnnotationColourChooser.displayFor(ap.av, ap,alignmentAnnotation,true);
+        };
+      });
+      pop.add(item);
+    }
+    item = new JMenuItem(MessageManager.getString("action.select_by_annotation"));
+    item.addActionListener(new ActionListener()
+    {
+      
+      @Override
+      public void actionPerformed(ActionEvent e)
+      {
+        AnnotationColumnChooser.displayFor(ap.av,ap,alignmentAnnotation);
+      };
+    });
+    pop.add(item);
+  }
+  static void addContactMatrixOptions(final AlignmentPanel ap,
+          final AlignmentAnnotation alignmentAnnotation, final JPopupMenu pop)
+  {
+    
+    final ContactMatrixI cm = ap.av.getContactMatrix(alignmentAnnotation);
+    JMenuItem item;
+    if (cm != null)
+    {
+      pop.addSeparator();
 
-            @Override
-            public void actionPerformed(ActionEvent e)
-            {
+      if (cm.hasGroups())
+      {
+        JCheckBoxMenuItem chitem = new JCheckBoxMenuItem("Show Groups on Matrix");
+        boolean showGroups = alignmentAnnotation.isShowGroupsForContactMatrix();
+        final AlignmentAnnotation sel_row=alignmentAnnotation;
+        chitem.setState(showGroups);
+        chitem.addActionListener(new ActionListener()
+        {
 
-              ap.alignFrame.showContactMapTree(aa[selectedRow], cm);
+          @Override
+          public void actionPerformed(ActionEvent e)
+          {
+            sel_row.setShowGroupsForContactMatrix(chitem.getState());
+            ap.getAnnotationPanel()
+            .paint(ap.getAnnotationPanel().getGraphics());
+          }
+        });
+        pop.add(chitem);
+      }
+      if (cm.hasTree())
+      {
+        item = new JMenuItem("Show Tree for Matrix");
+        item.addActionListener(new ActionListener()
+        {
 
-            }
-          });
-          pop.addSeparator();
-          pop.add(item);
-        }
+          @Override
+          public void actionPerformed(ActionEvent e)
+          {
 
+            ap.alignFrame.showContactMapTree(alignmentAnnotation, cm);
+
+          }
+        });
+        pop.add(item);
+      }
+      else
+      {
+        item = new JMenuItem("Calculate Tree for Matrix");
+        item.addActionListener(new ActionListener()
+        {
+          // TODO - refactor to analysis background thread
+          @Override
+          public void actionPerformed(ActionEvent e)
+          {
+            new Thread(new Runnable()
+            {
+              @Override
+              public void run()
+              {
+                AlignmentAnnotation alan = alignmentAnnotation;
+                cm.setGroupSet(GroupSet.makeGroups(cm, 5f, true));
+                ap.alignFrame.showContactMapTree(alan, cm);
+              }
+            }).start();
+          }
+        });
+        pop.add(item);
       }
     }
-    pop.show(this, evt.getX(), evt.getY());
   }
 
   /**