JAL-1432 updated copyright notices
[jalview.git] / src / jalview / appletgui / PCAPanel.java
index 5d32373..46343ce 100755 (executable)
@@ -1,44 +1,39 @@
 /*
- * Jalview - A Sequence Alignment Editor and Viewer (Version 2.6)
- * Copyright (C) 2010 J Procter, AM Waterhouse, G Barton, M Clamp, S Searle
+ * Jalview - A Sequence Alignment Editor and Viewer (Version 2.8.0b1)
+ * Copyright (C) 2014 The Jalview Authors
  * 
  * This file is part of Jalview.
  * 
  * Jalview is free software: you can redistribute it and/or
  * modify it under the terms of the GNU General Public License 
  * as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version.
- * 
+ *  
  * Jalview is distributed in the hope that it will be useful, but 
  * WITHOUT ANY WARRANTY; without even the implied warranty 
  * of MERCHANTABILITY or FITNESS FOR A PARTICULAR 
  * PURPOSE.  See the GNU General Public License for more details.
  * 
  * You should have received a copy of the GNU General Public License along with Jalview.  If not, see <http://www.gnu.org/licenses/>.
+ * The Jalview Authors are detailed in the 'AUTHORS' file.
  */
 package jalview.appletgui;
 
-import java.util.*;
-
 import java.awt.*;
 import java.awt.event.*;
 
-import jalview.analysis.*;
 import jalview.datamodel.*;
+import jalview.viewmodel.PCAModel;
 
 public class PCAPanel extends EmbmenuFrame implements Runnable,
         ActionListener, ItemListener
 {
-  PCA pca;
-
-  int top;
-
   RotatableCanvas rc;
 
   AlignViewport av;
 
-  SequenceI[] seqs;
+  PCAModel pcaModel;
 
-  AlignmentView seqstrings;
+  int top = 0;
 
   public PCAPanel(AlignViewport av)
   {
@@ -58,14 +53,18 @@ public class PCAPanel extends EmbmenuFrame implements Runnable,
     }
 
     this.av = av;
-    seqstrings = av.getAlignmentView(av.getSelectionGroup() != null);
-    if (av.getSelectionGroup() == null)
+    boolean selected = av.getSelectionGroup() != null
+            && av.getSelectionGroup().getSize() > 0;
+    AlignmentView seqstrings = av.getAlignmentView(selected);
+    boolean nucleotide = av.getAlignment().isNucleotide();
+    SequenceI[] seqs;
+    if (!selected)
     {
-      seqs = av.alignment.getSequencesArray();
+      seqs = av.getAlignment().getSequencesArray();
     }
     else
     {
-      seqs = av.getSelectionGroup().getSequencesInOrder(av.alignment);
+      seqs = av.getSelectionGroup().getSequencesInOrder(av.getAlignment());
     }
     SeqCigar sq[] = seqstrings.getSequences();
     int length = sq[0].getWidth();
@@ -79,13 +78,14 @@ public class PCAPanel extends EmbmenuFrame implements Runnable,
         return;
       }
     }
+    pcaModel = new PCAModel(seqstrings, seqs, nucleotide);
 
     rc = new RotatableCanvas(av);
     embedMenuIfNeeded(rc);
     add(rc, BorderLayout.CENTER);
 
     jalview.bin.JalviewLite.addFrame(this, "Principal component analysis",
-            400, 400);
+            475, 400);
 
     Thread worker = new Thread(this);
     worker.start();
@@ -96,46 +96,32 @@ public class PCAPanel extends EmbmenuFrame implements Runnable,
    */
   public void run()
   {
-    pca = new PCA(seqstrings.getSequenceStrings(' '));
-    pca.run();
-
-    // Now find the component coordinates
-    int ii = 0;
-
-    while ((ii < seqs.length) && (seqs[ii] != null))
-    {
-      ii++;
-    }
-
-    double[][] comps = new double[ii][ii];
-
-    for (int i = 0; i < ii; i++)
+    // TODO progress indicator
+    calcSettings.setEnabled(false);
+    rc.setEnabled(false);
+    try
     {
-      if (pca.getEigenvalue(i) > 1e-4)
-      {
-        comps[i] = pca.component(i);
-      }
-    }
-
-    // ////////////////
-    xCombobox.select(0);
-    yCombobox.select(1);
-    zCombobox.select(2);
-
-    top = pca.getM().rows - 1;
-
-    Vector points = new Vector();
-    float[][] scores = pca.getComponents(top - 1, top - 2, top - 3, 100);
-
-    for (int i = 0; i < pca.getM().rows; i++)
+      nuclSetting.setState(pcaModel.isNucleotide());
+      protSetting.setState(!pcaModel.isNucleotide());
+      pcaModel.run();
+      // ////////////////
+      xCombobox.select(0);
+      yCombobox.select(1);
+      zCombobox.select(2);
+
+      pcaModel.updateRc(rc);
+      // rc.invalidate();
+      top = pcaModel.getTop();
+    } catch (OutOfMemoryError x)
     {
-      SequencePoint sp = new SequencePoint(seqs[i], scores[i]);
-      points.addElement(sp);
+      System.err.println("Out of memory when calculating PCA.");
+      return;
     }
+    calcSettings.setEnabled(true);
 
-    rc.setPoints(points, pca.getM().rows);
+    // TODO revert progress indicator
+    rc.setEnabled(true);
     rc.repaint();
-    seqs = null;
     this.repaint();
   }
 
@@ -149,13 +135,7 @@ public class PCAPanel extends EmbmenuFrame implements Runnable,
     int dim1 = top - xCombobox.getSelectedIndex();
     int dim2 = top - yCombobox.getSelectedIndex();
     int dim3 = top - zCombobox.getSelectedIndex();
-
-    float[][] scores = pca.getComponents(dim1, dim2, dim3, 100);
-    for (int i = 0; i < pca.getM().rows; i++)
-    {
-      ((SequencePoint) rc.points.elementAt(i)).coord = scores[i];
-    }
-
+    pcaModel.updateRcView(dim1, dim2, dim3);
     rc.img = null;
     rc.rotmat.setIdentity();
     rc.initAxes();
@@ -168,7 +148,14 @@ public class PCAPanel extends EmbmenuFrame implements Runnable,
     {
       showOriginalData();
     }
-    else
+    if (evt.getSource() == resetButton)
+    {
+      xCombobox.select(0);
+      yCombobox.select(1);
+      zCombobox.select(2);
+      doDimensionChange();
+    }
+    if (evt.getSource() == values)
     {
       values_actionPerformed();
     }
@@ -188,6 +175,26 @@ public class PCAPanel extends EmbmenuFrame implements Runnable,
     {
       zCombobox_actionPerformed();
     }
+    else if (evt.getSource() == labels)
+    {
+      labels_itemStateChanged(evt);
+    }
+    else if (evt.getSource() == nuclSetting)
+    {
+      if (!pcaModel.isNucleotide())
+      {
+        pcaModel.setNucleotide(true);
+        new Thread(this).start();
+      }
+    }
+    else if (evt.getSource() == protSetting)
+    {
+      if (pcaModel.isNucleotide())
+      {
+        pcaModel.setNucleotide(false);
+        new Thread(this).start();
+      }
+    }
   }
 
   protected void xCombobox_actionPerformed()
@@ -213,7 +220,7 @@ public class PCAPanel extends EmbmenuFrame implements Runnable,
     frame.add(cap);
     jalview.bin.JalviewLite.addFrame(frame, "PCA details", 500, 500);
 
-    cap.setText(pca.getDetails());
+    cap.setText(pcaModel.getDetails());
   }
 
   void showOriginalData()
@@ -234,7 +241,8 @@ public class PCAPanel extends EmbmenuFrame implements Runnable,
     {
     }
     ;
-    Object[] alAndColsel = seqstrings.getAlignmentAndColumnSelection(gc);
+    Object[] alAndColsel = pcaModel.getSeqtrings()
+            .getAlignmentAndColumnSelection(gc);
 
     if (alAndColsel != null && alAndColsel[0] != null)
     {
@@ -265,6 +273,8 @@ public class PCAPanel extends EmbmenuFrame implements Runnable,
 
   protected Choice zCombobox = new Choice();
 
+  protected Button resetButton = new Button();
+
   FlowLayout flowLayout1 = new FlowLayout();
 
   BorderLayout borderLayout1 = new BorderLayout();
@@ -275,8 +285,14 @@ public class PCAPanel extends EmbmenuFrame implements Runnable,
 
   Menu menu2 = new Menu();
 
+  Menu calcSettings = new Menu();
+
   protected CheckboxMenuItem labels = new CheckboxMenuItem();
 
+  protected CheckboxMenuItem protSetting = new CheckboxMenuItem();
+
+  protected CheckboxMenuItem nuclSetting = new CheckboxMenuItem();
+
   MenuItem values = new MenuItem();
 
   MenuItem inputData = new MenuItem();
@@ -298,14 +314,22 @@ public class PCAPanel extends EmbmenuFrame implements Runnable,
     yCombobox.addItemListener(this);
     xCombobox.setFont(new java.awt.Font("Verdana", 0, 12));
     xCombobox.addItemListener(this);
+    resetButton.setFont(new java.awt.Font("Verdana", 0, 12));
+    resetButton.setLabel("Reset");
+    resetButton.addActionListener(this);
     this.setMenuBar(menuBar1);
     menu1.setLabel("File");
     menu2.setLabel("View");
+    calcSettings.setLabel("Change Parameters");
     labels.setLabel("Labels");
     labels.addItemListener(this);
     values.setLabel("Output Values...");
     values.addActionListener(this);
     inputData.setLabel("Input Data...");
+    nuclSetting.setLabel("Nucleotide matrix");
+    nuclSetting.addItemListener(this);
+    protSetting.setLabel("Protein matrix");
+    protSetting.addItemListener(this);
     this.add(jPanel2, BorderLayout.SOUTH);
     jPanel2.add(jLabel1, null);
     jPanel2.add(xCombobox, null);
@@ -313,11 +337,15 @@ public class PCAPanel extends EmbmenuFrame implements Runnable,
     jPanel2.add(yCombobox, null);
     jPanel2.add(jLabel3, null);
     jPanel2.add(zCombobox, null);
+    jPanel2.add(resetButton, null);
     menuBar1.add(menu1);
     menuBar1.add(menu2);
+    menuBar1.add(calcSettings);
     menu2.add(labels);
     menu1.add(values);
     menu1.add(inputData);
+    calcSettings.add(nuclSetting);
+    calcSettings.add(protSetting);
     inputData.addActionListener(this);
   }