Now an InternalFrame with menus
[jalview.git] / src / jalview / gui / PCAPanel.java
index 474a158..71891aa 100755 (executable)
@@ -28,6 +28,12 @@ import java.awt.*;
 import java.awt.event.*;\r
 \r
 import java.util.*;\r
+import javax.swing.*;\r
+import java.io.FileOutputStream;\r
+import org.jibble.epsgraphics.EpsGraphics2D;\r
+import javax.imageio.ImageIO;\r
+import java.awt.image.BufferedImage;\r
+import java.awt.print.*;\r
 \r
 \r
 /**\r
@@ -50,38 +56,78 @@ public class PCAPanel extends GPCAPanel implements Runnable
      * @param av DOCUMENT ME!\r
      * @param s DOCUMENT ME!\r
      */\r
-    public PCAPanel(AlignViewport av, SequenceI[] s)\r
+    public PCAPanel(AlignViewport av)\r
     {\r
         this.av = av;\r
 \r
+        boolean sameLength = true;\r
+\r
         if ((av.getSelectionGroup() != null) &&\r
                 (av.getSelectionGroup().getSize() > 3))\r
         {\r
             seqs = new Sequence[av.getSelectionGroup().getSize()];\r
-\r
-            for (int i = 0; i < s.length; i++)\r
+            int length = av.getSelectionGroup().getSequenceAt(0).getLength();\r
+            for (int i = 0; i < av.getSelectionGroup().getSize(); i++)\r
             {\r
                 seqs[i] = av.getSelectionGroup().getSequenceAt(i);\r
+                if(seqs[i].getLength()!=length)\r
+                {\r
+                  sameLength = false;\r
+                  break;\r
+                }\r
             }\r
         }\r
-\r
-        if (s == null)\r
+        else\r
         {\r
             seqs = new Sequence[av.getAlignment().getHeight()];\r
+            int length = av.alignment.getSequenceAt(0).getLength();\r
 \r
             for (int i = 0; i < av.getAlignment().getHeight(); i++)\r
             {\r
                 seqs[i] = av.getAlignment().getSequenceAt(i);\r
+                if(seqs[i].getLength()!=length)\r
+               {\r
+                 sameLength = false;\r
+                 break;\r
+               }\r
+\r
             }\r
         }\r
 \r
+        if (!sameLength)\r
+        {\r
+          JOptionPane.showMessageDialog(Desktop.desktop,\r
+                                        "The sequences must be aligned before calculating PCA.\n" +\r
+                                        "Try using the Pad function in the edit menu,\n" +\r
+                                        "or one of the multiple sequence alignment web services.",\r
+                                        "Sequences not aligned",\r
+                                        JOptionPane.WARNING_MESSAGE);\r
+\r
+          return;\r
+        }\r
+\r
+\r
+        Desktop.addInternalFrame(this, "Principal component analysis",\r
+                               400, 400);\r
+\r
 \r
         rc = new RotatableCanvas(av);\r
-        add(rc, BorderLayout.CENTER);\r
+        this.getContentPane().add(rc, BorderLayout.CENTER);\r
         Thread worker = new Thread(this);\r
         worker.start();\r
     }\r
 \r
+    public void bgcolour_actionPerformed(ActionEvent e)\r
+    {\r
+      Color col = JColorChooser.showDialog(this, "Select Background Colour",\r
+                rc.bgColour);\r
+\r
+      if(col!=null)\r
+        rc.bgColour = col;\r
+      rc.repaint();\r
+    }\r
+\r
+\r
 \r
 \r
     /**\r
@@ -89,46 +135,60 @@ public class PCAPanel extends GPCAPanel implements Runnable
      */\r
     public void run()\r
     {\r
-                pca = new PCA(seqs);\r
-                pca.run();\r
+      try{\r
+        pca = new PCA(seqs);\r
+        pca.run();\r
 \r
-                // Now find the component coordinates\r
-                int ii = 0;\r
+        // Now find the component coordinates\r
+        int ii = 0;\r
 \r
-                while ((ii < seqs.length) && (seqs[ii] != null))\r
-                {\r
-                    ii++;\r
-                }\r
+        while ( (ii < seqs.length) && (seqs[ii] != null))\r
+        {\r
+          ii++;\r
+        }\r
 \r
-                double[][] comps = new double[ii][ii];\r
+        double[][] comps = new double[ii][ii];\r
 \r
-                for (int i = 0; i < ii; i++)\r
-                {\r
-                    if (pca.getEigenvalue(i) > 1e-4)\r
-                    {\r
-                        comps[i] = pca.component(i);\r
-                    }\r
-                }\r
+        for (int i = 0; i < ii; i++)\r
+        {\r
+          if (pca.getEigenvalue(i) > 1e-4)\r
+          {\r
+            comps[i] = pca.component(i);\r
+          }\r
+        }\r
 \r
-                //////////////////\r
-                xCombobox.setSelectedIndex(0);\r
-                yCombobox.setSelectedIndex(1);\r
-                zCombobox.setSelectedIndex(2);\r
+        //////////////////\r
+        xCombobox.setSelectedIndex(0);\r
+        yCombobox.setSelectedIndex(1);\r
+        zCombobox.setSelectedIndex(2);\r
 \r
-                top = pca.getM().rows - 1;\r
+        top = pca.getM().rows - 1;\r
 \r
-                Vector points = new Vector();\r
-                float[][] scores = pca.getComponents(top - 1, top - 2, top - 3, 100);\r
+        Vector points = new Vector();\r
+        float[][] scores = pca.getComponents(top - 1, top - 2, top - 3, 100);\r
 \r
-                for (int i = 0; i < pca.getM().rows; i++)\r
-                {\r
-                    SequencePoint sp = new SequencePoint(seqs[i], scores[i]);\r
-                    points.addElement(sp);\r
-                }\r
+        for (int i = 0; i < pca.getM().rows; i++)\r
+        {\r
+          SequencePoint sp = new SequencePoint(seqs[i], scores[i]);\r
+          points.addElement(sp);\r
+        }\r
+\r
+        rc.setPoints(points, pca.getM().rows);\r
+        rc.repaint();\r
+        seqs = null;\r
+      }\r
+     catch(OutOfMemoryError er)\r
+      { JOptionPane.showInternalMessageDialog(Desktop.desktop,\r
+                                              "Out of memory calculating PCA!!"\r
+                                              +\r
+                                              "\nSee help files for increasing Java Virtual Machine memory."\r
+                                              , "Out of memory",\r
+                                              JOptionPane.WARNING_MESSAGE);\r
+        System.out.println("PCAPanel: "+er);\r
+        System.gc();\r
+\r
+      }\r
 \r
-                rc.setPoints(points, pca.getM().rows);\r
-                rc.repaint();\r
-                seqs = null;\r
     }\r
 \r
     /**\r
@@ -187,4 +247,204 @@ public class PCAPanel extends GPCAPanel implements Runnable
     {\r
         doDimensionChange();\r
     }\r
-}\r
+\r
+\r
+    public void outputValues_actionPerformed(ActionEvent e)\r
+    {\r
+      CutAndPasteTransfer cap = new CutAndPasteTransfer();\r
+      Desktop.addInternalFrame(cap, "PCA details", 500,\r
+                500);\r
+\r
+      cap.setText(pca.getDetails());\r
+     }\r
+\r
+    public void showLabels_actionPerformed(ActionEvent e)\r
+    {\r
+      rc.showLabels(showLabels.getState());\r
+    }\r
+\r
+    public void print_actionPerformed(ActionEvent e)\r
+    {\r
+      PCAPrinter printer = new PCAPrinter();\r
+      printer.start();\r
+    }\r
+\r
+\r
+    class PCAPrinter extends Thread implements Printable\r
+    {\r
+      public void run()\r
+      {\r
+        PrinterJob printJob = PrinterJob.getPrinterJob();\r
+        PageFormat pf = printJob.pageDialog(printJob.defaultPage());\r
+\r
+        printJob.setPrintable(this, pf);\r
+\r
+        if (printJob.printDialog())\r
+        {\r
+          try\r
+          {\r
+            printJob.print();\r
+          }\r
+          catch (Exception PrintException)\r
+          {\r
+            PrintException.printStackTrace();\r
+          }\r
+        }\r
+      }\r
+\r
+      public int print(Graphics pg, PageFormat pf, int pi)\r
+          throws PrinterException\r
+      {\r
+        pg.translate( (int) pf.getImageableX(), (int) pf.getImageableY());\r
+\r
+        rc.drawBackground(pg, rc.bgColour);\r
+        rc.drawScene(pg);\r
+        if (rc.drawAxes == true)\r
+        {\r
+          rc.drawAxes(pg);\r
+        }\r
+\r
+        if (pi == 0)\r
+          return Printable.PAGE_EXISTS;\r
+        else\r
+          return Printable.NO_SUCH_PAGE;\r
+      }\r
+    }\r
+\r
+\r
+\r
+\r
+    /**\r
+     * DOCUMENT ME!\r
+     *\r
+     * @param e DOCUMENT ME!\r
+     */\r
+    public void eps_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
+      if (renderStyle.equalsIgnoreCase("text"))\r
+      {\r
+        accurateText = false;\r
+      }\r
+\r
+      int width = rc.getWidth();\r
+      int height = rc.getHeight();\r
+\r
+      try\r
+      {\r
+        jalview.io.JalviewFileChooser chooser = new jalview.io.JalviewFileChooser(\r
+            jalview.bin.Cache.getProperty(\r
+                "LAST_DIRECTORY"), new String[]\r
+            {"eps"},\r
+            new String[]\r
+            {"Encapsulated Postscript"},\r
+            "Encapsulated Postscript");\r
+        chooser.setFileView(new jalview.io.JalviewFileView());\r
+        chooser.setDialogTitle("Create EPS file from PCA");\r
+        chooser.setToolTipText("Save");\r
+\r
+        int value = chooser.showSaveDialog(this);\r
+\r
+        if (value != jalview.io.JalviewFileChooser.APPROVE_OPTION)\r
+        {\r
+          return;\r
+        }\r
+\r
+        jalview.bin.Cache.setProperty("LAST_DIRECTORY",\r
+                                      chooser.getSelectedFile().getParent());\r
+\r
+        FileOutputStream out = new FileOutputStream(chooser.getSelectedFile());\r
+        EpsGraphics2D pg = new EpsGraphics2D("PCA", out, 0, 0, width,\r
+                                             height);\r
+\r
+        pg.setAccurateTextMode(accurateText);\r
+\r
+        rc.drawBackground(pg, rc.bgColour);\r
+        rc.drawScene(pg);\r
+\r
+        if (rc.drawAxes == true)\r
+        {\r
+          rc.drawAxes(pg);\r
+        }\r
+\r
+\r
+        pg.flush();\r
+        pg.close();\r
+      }\r
+      catch (Exception ex)\r
+      {\r
+        ex.printStackTrace();\r
+      }\r
+    }\r
+\r
+    /**\r
+     * DOCUMENT ME!\r
+     *\r
+     * @param e DOCUMENT ME!\r
+     */\r
+    public void png_actionPerformed(ActionEvent e)\r
+    {\r
+      int width = rc.getWidth();\r
+      int height = rc.getHeight();\r
+\r
+      try\r
+      {\r
+        jalview.io.JalviewFileChooser chooser = new jalview.io.JalviewFileChooser(\r
+            jalview.bin.Cache.getProperty(\r
+                "LAST_DIRECTORY"), new String[]\r
+            {"png"},\r
+            new String[]\r
+            {"Portable network graphics"},\r
+            "Portable network graphics");\r
+\r
+        chooser.setFileView(new jalview.io.JalviewFileView());\r
+        chooser.setDialogTitle("Create PNG image from PCA");\r
+        chooser.setToolTipText("Save");\r
+\r
+        int value = chooser.showSaveDialog(this);\r
+\r
+        if (value != jalview.io.JalviewFileChooser.APPROVE_OPTION)\r
+        {\r
+          return;\r
+        }\r
+\r
+        jalview.bin.Cache.setProperty("LAST_DIRECTORY",\r
+                                      chooser.getSelectedFile().getParent());\r
+\r
+        FileOutputStream out = new FileOutputStream(chooser.getSelectedFile());\r
+\r
+        BufferedImage bi = new BufferedImage(width, height,\r
+                                             BufferedImage.TYPE_INT_RGB);\r
+        Graphics big = bi.getGraphics();\r
+\r
+        big.drawImage(rc.img,0,0,this);\r
+\r
+        ImageIO.write(bi, "png", out);\r
+        out.close();\r
+      }\r
+      catch (Exception ex)\r
+      {\r
+        ex.printStackTrace();\r
+      }\r
+    }\r
+\r
+  }\r