Highlight hidden columns after revealing
[jalview.git] / src / jalview / gui / TreePanel.java
index 8d13e93..190f15c 100755 (executable)
@@ -38,7 +38,7 @@ import java.util.*;
 \r
 import javax.imageio.*;\r
 \r
-import javax.swing.*;\r
+import java.beans.PropertyChangeEvent;\r
 \r
 \r
 /**\r
@@ -49,13 +49,11 @@ import javax.swing.*;
  */\r
 public class TreePanel extends GTreePanel\r
 {\r
-    SequenceI[] seq;\r
     String type;\r
     String pwtype;\r
-    int start;\r
-    int end;\r
     TreeCanvas treeCanvas;\r
     NJTree tree;\r
+    AlignViewport av;\r
 \r
     /**\r
      * Creates a new TreePanel object.\r
@@ -67,14 +65,13 @@ public class TreePanel extends GTreePanel
      * @param s DOCUMENT ME!\r
      * @param e DOCUMENT ME!\r
      */\r
-    public TreePanel(AlignViewport av, Vector seqVector, String type,\r
-                     String pwtype, int s, int e)\r
+    public TreePanel(AlignViewport av, String type, String pwtype)\r
     {\r
       super();\r
-      initTreePanel(av, seqVector, type, pwtype, s, e, null);\r
+      initTreePanel(av, type, pwtype, null);\r
 \r
       // We know this tree has distances. JBPNote TODO: prolly should add this as a userdefined default\r
-      treeCanvas.setShowDistances(true);\r
+      showDistances(true);\r
     }\r
 \r
     /**\r
@@ -86,30 +83,51 @@ public class TreePanel extends GTreePanel
      * @param type DOCUMENT ME!\r
      * @param pwtype DOCUMENT ME!\r
      */\r
-    public TreePanel(AlignViewport av, Vector seqVector, NewickFile newtree,\r
-                     String type, String pwtype)\r
+    public TreePanel(AlignViewport av,\r
+                     String type,\r
+                     String pwtype,\r
+                     NewickFile newtree)\r
     {\r
       super();\r
-      initTreePanel(av, seqVector, type, pwtype, 0, seqVector.size(), newtree);\r
+      initTreePanel(av, type, pwtype, newtree);\r
+    }\r
+\r
+    public AlignmentI getAlignment()\r
+    {\r
+      return treeCanvas.av.getAlignment();\r
     }\r
 \r
 \r
-    void initTreePanel(AlignViewport av, Vector seqVector, String type,\r
-                       String pwtype, int s, int e, NewickFile newTree)\r
+    void initTreePanel(AlignViewport av, String type,  String pwtype,\r
+                       NewickFile newTree)\r
     {\r
 \r
       this.type = type;\r
       this.pwtype = pwtype;\r
 \r
-      start = s;\r
-      end = e;\r
+      treeCanvas = new TreeCanvas(av, scrollPane);\r
+      scrollPane.setViewportView(treeCanvas);\r
 \r
-      seq = new Sequence[seqVector.size()];\r
-      seqVector.toArray(seq);\r
+      av.addPropertyChangeListener(new java.beans.PropertyChangeListener()\r
+      {\r
+        public void propertyChange(PropertyChangeEvent evt)\r
+        {\r
+          if (evt.getPropertyName().equals("alignment"))\r
+          {\r
+            if(tree==null)\r
+              System.out.println("tree is null");\r
+            if(evt.getNewValue()==null)\r
+              System.out.println("new value is null");\r
 \r
+            tree.UpdatePlaceHolders( (Vector) evt.getNewValue());\r
+\r
+            repaint();\r
+          }\r
+        }\r
+      });\r
+\r
+      this.av = av;\r
 \r
-      treeCanvas = new TreeCanvas(av, scrollPane);\r
-      scrollPane.setViewportView(treeCanvas);\r
 \r
       TreeLoader tl = new TreeLoader(newTree);\r
       tl.start();\r
@@ -123,69 +141,91 @@ public class TreePanel extends GTreePanel
       public TreeLoader(NewickFile newtree)\r
       {\r
         this.newtree = newtree;\r
+        if (newtree != null)\r
+        {\r
+          // Must be outside run(), as Jalview2XML tries to\r
+          // update distance/bootstrap visibility at the same time\r
+          showBootstrap(newtree.HasBootstrap());\r
+          showDistances(newtree.HasDistances());\r
+        }\r
       }\r
 \r
       public void run()\r
       {\r
-        NJTree tree;\r
+\r
         if(newtree!=null)\r
-          tree = new NJTree(seq, newtree);\r
+          tree = new NJTree(av.alignment.getSequencesArray(),\r
+                            newtree);\r
         else\r
-          tree = new NJTree(seq, type, pwtype, start, end);\r
+        {\r
+          int start, end;\r
+          SequenceI [] seqs;\r
+          String [] seqStrings = null;\r
+          if (av.hasHiddenColumns)\r
+          {\r
+            seqStrings = av.getSelectionAsString();\r
+          }\r
+\r
+          if(av.getSelectionGroup()==null)\r
+          {\r
+            start = 0;\r
+            end = av.alignment.getWidth();\r
+            seqs = av.alignment.getSequencesArray();\r
+          }\r
+          else\r
+          {\r
+            start = av.getSelectionGroup().getStartRes();\r
+            end = av.getSelectionGroup().getEndRes()+1;\r
+            seqs = av.getSelectionGroup().getSequencesInOrder(av.alignment);\r
+          }\r
+\r
+          tree = new NJTree(seqs, seqStrings, type, pwtype, start, end);\r
+        }\r
+\r
 \r
         tree.reCount(tree.getTopNode());\r
         tree.findHeight(tree.getTopNode());\r
         treeCanvas.setTree(tree);\r
-        if(newtree!=null)\r
-        {\r
-          treeCanvas.setShowBootstrap(newtree.HasBootstrap());\r
-          treeCanvas.setShowDistances(newtree.HasDistances());\r
-        }\r
-\r
-        setViewStateFromTreeCanvas(treeCanvas);\r
 \r
         treeCanvas.repaint();\r
 \r
+        av.setCurrentTree(tree);\r
+\r
       }\r
     }\r
 \r
+    public void showDistances(boolean b)\r
+    {\r
+      treeCanvas.setShowDistances(b);\r
+      distanceMenu.setSelected(b);\r
+    }\r
 \r
-    /**\r
-     * DOCUMENT ME!\r
-     *\r
-     * @return DOCUMENT ME!\r
-     */\r
-    public NJTree getTree()\r
+    public void showBootstrap(boolean b)\r
     {\r
-        return tree;\r
+      treeCanvas.setShowBootstrap(b);\r
+      bootstrapMenu.setSelected(b);\r
     }\r
 \r
-    /**\r
-     * DOCUMENT ME!\r
-     *\r
-     * @param tree DOCUMENT ME!\r
-     */\r
-    private void setViewStateFromTreeCanvas(TreeCanvas tree)\r
+    public void showPlaceholders(boolean b)\r
     {\r
-        // update view menu state from treeCanvas render state\r
-        distanceMenu.setSelected(tree.showDistances);\r
-        bootstrapMenu.setSelected(tree.showBootstrap);\r
-        placeholdersMenu.setSelected(tree.markPlaceholders);\r
+      placeholdersMenu.setState(b);\r
+      treeCanvas.setMarkPlaceholders(b);\r
     }\r
 \r
+\r
+\r
+\r
     /**\r
      * DOCUMENT ME!\r
      *\r
-     * @param tree DOCUMENT ME!\r
+     * @return DOCUMENT ME!\r
      */\r
-    private void setTreeCanvasFromViewState(TreeCanvas tree)\r
+    public NJTree getTree()\r
     {\r
-        // update view menu state from treeCanvas render state\r
-        tree.showDistances = distanceMenu.isSelected();\r
-        tree.showBootstrap = bootstrapMenu.isSelected();\r
-        tree.markPlaceholders = placeholdersMenu.isSelected();\r
+        return tree;\r
     }\r
 \r
+\r
     /**\r
      * DOCUMENT ME!\r
      *\r
@@ -218,7 +258,7 @@ public class TreePanel extends GTreePanel
         Desktop.addInternalFrame(cap, buffer.toString(), 500, 100);\r
 \r
         jalview.io.NewickFile fout = new jalview.io.NewickFile(tree.getTopNode());\r
-        cap.setText(fout.print(false, true));\r
+        cap.setText(fout.print(tree.isHasBootstrap(), tree.isHasDistances(), tree.isHasRootDistance()));\r
     }\r
 \r
     /**\r
@@ -245,7 +285,7 @@ public class TreePanel extends GTreePanel
             try\r
             {\r
                 jalview.io.NewickFile fout = new jalview.io.NewickFile(tree.getTopNode());\r
-                String output = fout.print(false, true); // distances only\r
+                String output = fout.print(tree.isHasBootstrap(), tree.isHasDistances(), tree.isHasRootDistance());\r
                 java.io.PrintWriter out = new java.io.PrintWriter(new java.io.FileWriter(\r
                             choice));\r
                 out.println(output);\r
@@ -269,6 +309,16 @@ public class TreePanel extends GTreePanel
         treeCanvas.startPrinting();\r
     }\r
 \r
+\r
+    public void originalSeqData_actionPerformed(ActionEvent e)\r
+    {\r
+      CutAndPasteTransfer cap = new CutAndPasteTransfer();\r
+      cap.setText(tree.printOriginalSequenceData());\r
+      Desktop.addInternalFrame(cap, "Original Data",\r
+          400, 400);\r
+    }\r
+\r
+\r
     /**\r
      * DOCUMENT ME!\r
      *\r
@@ -285,33 +335,25 @@ public class TreePanel extends GTreePanel
      *\r
      * @param e DOCUMENT ME!\r
      */\r
-    public void fontSize_actionPerformed(ActionEvent e)\r
+    public void font_actionPerformed(ActionEvent e)\r
     {\r
         if (treeCanvas == null)\r
         {\r
             return;\r
         }\r
 \r
-        String size = fontSize.getText().substring(fontSize.getText().indexOf("-") +\r
-                1);\r
-\r
-        Object selection = JOptionPane.showInternalInputDialog(Desktop.desktop,\r
-                "Select font size", "Font size", JOptionPane.QUESTION_MESSAGE,\r
-                null,\r
-                new String[]\r
-                {\r
-                    "1", "2", "4", "6", "8", "10", "12", "14", "16", "18", "20"\r
-                }, "Font Size - " + size);\r
-\r
-        if (selection != null)\r
-        {\r
-            fontSize.setText("Font Size - " + selection);\r
+        new FontChooser(this);\r
+    }\r
 \r
-            int i = Integer.parseInt(selection.toString());\r
-            treeCanvas.setFontSize(i);\r
-        }\r
+    public Font getTreeFont()\r
+    {\r
+        return treeCanvas.font;\r
+    }\r
 \r
-        scrollPane.setViewportView(treeCanvas);\r
+    public void setTreeFont(Font font)\r
+    {\r
+      if(treeCanvas!=null)\r
+      treeCanvas.setFont(font);\r
     }\r
 \r
     /**\r
@@ -351,6 +393,32 @@ public class TreePanel extends GTreePanel
      */\r
     public void epsTree_actionPerformed(ActionEvent e)\r
     {\r
+      boolean accurateText = true;\r
+\r
+      String renderStyle = jalview.bin.Cache.getDefault("EPS_RENDERING",\r
+          "Prompt each time");\r
+\r
+    // If we need to prompt, and if the GUI is visible then\r
+    // Prompt for EPS rendering style\r
+      if (renderStyle.equalsIgnoreCase("Prompt each time")\r
+          && !\r
+          (System.getProperty("java.awt.headless") != null\r
+           && System.getProperty("java.awt.headless").equals("true")))\r
+      {\r
+        EPSOptions eps = new EPSOptions();\r
+        renderStyle = eps.getValue();\r
+\r
+        if (renderStyle==null || eps.cancelled)\r
+          return;\r
+\r
+\r
+      }\r
+\r
+      if (renderStyle.equalsIgnoreCase("text"))\r
+      {\r
+        accurateText = false;\r
+      }\r
+\r
         int width = treeCanvas.getWidth();\r
         int height = treeCanvas.getHeight();\r
 \r
@@ -372,11 +440,13 @@ public class TreePanel extends GTreePanel
             }\r
 \r
             jalview.bin.Cache.setProperty("LAST_DIRECTORY",\r
-                chooser.getSelectedFile().getParent());\r
+                                          chooser.getSelectedFile().getParent());\r
 \r
             FileOutputStream out = new FileOutputStream(chooser.getSelectedFile());\r
             EpsGraphics2D pg = new EpsGraphics2D("Tree", out, 0, 0, width,\r
-                    height);\r
+                                                 height);\r
+\r
+            pg.setAccurateTextMode(accurateText);\r
 \r
             treeCanvas.draw(pg, width, height);\r
 \r