Sequence colour in viewport
[jalview.git] / src / jalview / gui / AlignViewport.java
index 743c8dd..b275611 100755 (executable)
@@ -51,9 +51,6 @@ public class AlignViewport
     boolean renderGaps = true;
     boolean showSequenceFeatures = false;
     boolean showAnnotation = true;
-    boolean showConservation = true;
-    boolean showQuality = true;
-    boolean showIdentity = true;
     boolean colourAppliesToAllGroups = true;
     ColourSchemeI globalColourScheme = null;
     boolean conservationColourSelected = false;
@@ -112,13 +109,9 @@ public class AlignViewport
 
     boolean gatherViewsHere = false;
 
+    Stack historyList = new Stack();
+    Stack redoList = new Stack();
 
-    public AlignViewport(AlignmentI al, boolean dataset)
-    {
-      isDataset = dataset;
-      setAlignment(al);
-      init();
-    }
     /**
      * Creates a new AlignViewport object.
      *
@@ -155,14 +148,10 @@ public class AlignViewport
 
       showJVSuffix = Cache.getDefault("SHOW_JVSUFFIX", true);
       showAnnotation = Cache.getDefault("SHOW_ANNOTATIONS", true);
-      showConservation = Cache.getDefault("SHOW_CONSERVATION", true);
-
-      showQuality = Cache.getDefault("SHOW_QUALITY", true);
-      showIdentity = Cache.getDefault("SHOW_IDENTITY", true);
 
       autoCalculateConsensus = Cache.getDefault("AUTO_CALC_CONSENSUS", true);
 
-      padGaps = Cache.getDefault("PAD_GAPS", false);
+      padGaps = Cache.getDefault("PAD_GAPS", true);
 
        String fontName = Cache.getDefault("FONT_NAME", "SansSerif");
        String fontStyle = Cache.getDefault("FONT_STYLE", Font.PLAIN + "") ;
@@ -181,7 +170,6 @@ public class AlignViewport
 
        setFont(new Font(fontName, style, Integer.parseInt(fontSize)));
 
-
        alignment.setGapCharacter( Cache.getDefault("GAP_SYMBOL", "-").charAt(0) );
 
 
@@ -199,12 +187,13 @@ public class AlignViewport
                 AlignmentAnnotation.BAR_GRAPH);
             conservation.hasText = true;
 
-            if (showConservation)
+
+            if (Cache.getDefault("SHOW_CONSERVATION", true))
             {
               alignment.addAnnotation(conservation);
             }
 
-            if (showQuality)
+            if (Cache.getDefault("SHOW_QUALITY", true))
             {
               quality = new AlignmentAnnotation("Quality",
                                                 "Alignment Quality based on Blosum62 scores",
@@ -212,6 +201,7 @@ public class AlignViewport
                                                 0f,
                                                 11f,
                                                 AlignmentAnnotation.BAR_GRAPH);
+              quality.hasText = true;
 
               alignment.addAnnotation(quality);
             }
@@ -222,11 +212,10 @@ public class AlignViewport
                                                AlignmentAnnotation.BAR_GRAPH);
           consensus.hasText = true;
 
-           if (showIdentity)
+           if (Cache.getDefault("SHOW_IDENTITY", true))
            {
              alignment.addAnnotation(consensus);
            }
-
         }
 
         if (jalview.bin.Cache.getProperty("DEFAULT_COLOUR") != null)
@@ -245,6 +234,8 @@ public class AlignViewport
                 globalColourScheme.setConsensus(hconsensus);
             }
         }
+
+        wrapAlignment = jalview.bin.Cache.getDefault("WRAP_ALIGNMENT", false);
     }
 
 
@@ -278,24 +269,43 @@ public class AlignViewport
       {
         try
         {
-          long t = System.currentTimeMillis();
+          updatingConservation = true;
+
+          while (UPDATING_CONSERVATION)
+          {
+            try
+            {
+              if (ap != null)
+              {
+                ap.repaint();
+              }
+              Thread.sleep(200);
+            }
+            catch (Exception ex)
+            {
+              ex.printStackTrace();
+            }
+          }
+
+          UPDATING_CONSERVATION = true;
+
+
+          int alWidth = alignment.getWidth();
+          if(alWidth<0)
+            return;
 
           Conservation cons = new jalview.analysis.Conservation("All",
               jalview.schemes.ResidueProperties.propHash, 3,
-              alignment.getSequences(), 0, alignment.getWidth() - 1);
+              alignment.getSequences(), 0, alWidth -1);
 
           cons.calculate();
           cons.verdict(false, ConsPercGaps);
 
-          if (showQuality)
+          if (quality!=null)
           {
             cons.findQuality();
           }
 
-          System.out.println("Conservation took " + (System.currentTimeMillis() - t) +
-                             "ms");
-
-          int alWidth = alignment.getWidth();
           String sequence = cons.getConsSequence().getSequence();
           float minR;
           float minG;
@@ -319,8 +329,9 @@ public class AlignViewport
 
           conservation.annotations = new Annotation[alWidth];
 
-          if (showQuality)
+          if (quality!=null)
           {
+            quality.graphMax = cons.qualityRange[1].floatValue();
             quality.annotations = new Annotation[alWidth];
             qmin = cons.qualityRange[0].floatValue();
             qmax = cons.qualityRange[1].floatValue();
@@ -349,7 +360,7 @@ public class AlignViewport
                                          minB + (maxB * vprop)));
 
             // Quality calc
-            if (showQuality)
+            if (quality!=null)
             {
               value = ( (Double) cons.quality.get(i)).floatValue();
               vprop = value - qmin;
@@ -361,16 +372,13 @@ public class AlignViewport
                   minB + (maxB * vprop)));
             }
           }
-
-          if (quality != null)
-          {
-            quality.graphMax = cons.qualityRange[1].floatValue();
-          }
         }
         catch (OutOfMemoryError error)
         {
           javax.swing.SwingUtilities.invokeLater(new Runnable()
           {
+
+
             public void run()
             {
               javax.swing.JOptionPane.showInternalMessageDialog(Desktop.desktop,
@@ -382,16 +390,22 @@ public class AlignViewport
             }
           });
 
+          conservation = null;
+          quality = null;
+
           System.out.println("Conservation calculation: " + error);
           System.gc();
 
         }
 
+        UPDATING_CONSERVATION = false;
+        updatingConservation = false;
+
         if(ap!=null)
         {
           ap.repaint();
         }
-        updatingConservation = false;
+
       }
     }
 
@@ -402,6 +416,10 @@ public class AlignViewport
 
     boolean consUpdateNeeded = false;
 
+    static boolean UPDATING_CONSENSUS = false;
+
+    static boolean UPDATING_CONSERVATION = false;
+
     boolean updatingConsensus = false;
 
     boolean updatingConservation = false;
@@ -409,42 +427,22 @@ public class AlignViewport
     /**
      * DOCUMENT ME!
      */
-    public void updateConservation(AlignmentPanel ap)
+    public void updateConservation(final AlignmentPanel ap)
     {
-      if (alignment.isNucleotide())
+      if (alignment.isNucleotide() || conservation==null)
         return;
 
-      updatingConservation = true;
-
-      if (conservationThread == null || !conservationThread.isAlive())
-      {
-        conservationThread = new ConservationThread(ap);
-        conservationThread.start();
-      }
-      else
-      {
-        consUpdateNeeded = true;
-        System.out.println("come back later");
-      }
+      conservationThread = new ConservationThread(ap);
+      conservationThread.start();
     }
 
     /**
      * DOCUMENT ME!
      */
-    public void updateConsensus(AlignmentPanel ap)
+    public void updateConsensus(final AlignmentPanel ap)
     {
-      updatingConsensus = true;
-
-      if (consensusThread == null || !consensusThread.isAlive())
-      {
-        consensusThread = new ConsensusThread(ap);
-        consensusThread.start();
-      }
-      else
-      {
-        consUpdateNeeded = true;
-        System.out.println("come back later");
-      }
+      consensusThread = new ConsensusThread(ap);
+      consensusThread.start();
     }
 
 
@@ -457,23 +455,43 @@ public class AlignViewport
       }
       public void run()
       {
+        updatingConsensus = true;
+        while (UPDATING_CONSENSUS)
+        {
+          try
+          {
+            if (ap != null)
+            {
+              ap.repaint();
+            }
+
+            Thread.sleep(200);
+          }
+          catch (Exception ex)
+          {
+            ex.printStackTrace();
+          }
+        }
+
+
+        UPDATING_CONSENSUS = true;
+
         try
         {
           int aWidth = alignment.getWidth();
+          if(aWidth<0)
+            return;
 
           consensus.annotations = null;
           consensus.annotations = new Annotation[aWidth];
 
-          long t = System.currentTimeMillis();
+
           hconsensus = new Hashtable[aWidth];
           AAFrequency.calculate(alignment.getSequencesArray(),
                                 0,
                                 alignment.getWidth(),
                                 hconsensus);
 
-          System.out.println("Consensus took " +
-                             (System.currentTimeMillis() - t) + "ms");
-
           for (int i = 0; i < aWidth; i++)
           {
             float value = 0;
@@ -504,6 +522,10 @@ public class AlignViewport
         }
         catch (OutOfMemoryError error)
         {
+          alignment.deleteAnnotation(consensus);
+
+          consensus = null;
+          hconsensus = null;
           javax.swing.SwingUtilities.invokeLater(new Runnable()
           {
             public void run()
@@ -520,14 +542,13 @@ public class AlignViewport
           System.out.println("Consensus calculation: " + error);
           System.gc();
         }
+        UPDATING_CONSENSUS = false;
+        updatingConsensus = false;
 
         if (ap != null)
         {
           ap.repaint();
         }
-
-
-        updatingConsensus = false;
       }
     }
     /**
@@ -548,6 +569,7 @@ public class AlignViewport
             seqs.append(consensus.annotations[i].displayCharacter);
         }
       }
+
       SequenceI sq = new Sequence("Consensus", seqs.toString());
       sq.setDescription("Percentage Identity Consensus "+((ignoreGapsInConsensusCalculation) ? " without gaps" : ""));
       return sq;
@@ -1503,4 +1525,79 @@ public class AlignViewport
       return sequenceSetID;
     }
 
+    public void alignmentChanged(AlignmentPanel ap)
+    {
+        if (padGaps)
+          alignment.padGaps();
+
+        if (hconsensus != null && autoCalculateConsensus)
+        {
+          updateConsensus(ap);
+          updateConservation(ap);
+        }
+
+        //Reset endRes of groups if beyond alignment width
+        int alWidth = alignment.getWidth();
+        Vector groups = alignment.getGroups();
+        if(groups!=null)
+        {
+          for(int i=0; i<groups.size(); i++)
+          {
+            SequenceGroup sg = (SequenceGroup)groups.elementAt(i);
+            if(sg.getEndRes()>alWidth)
+              sg.setEndRes(alWidth-1);
+          }
+        }
+
+        if(selectionGroup!=null && selectionGroup.getEndRes()>alWidth)
+          selectionGroup.setEndRes(alWidth-1);
+
+        resetAllColourSchemes();
+
+        alignment.adjustSequenceAnnotations();
+    }
+
+
+    void resetAllColourSchemes()
+    {
+      ColourSchemeI cs = globalColourScheme;
+      if(cs!=null)
+      {
+        if (cs instanceof ClustalxColourScheme)
+        {
+          ( (ClustalxColourScheme) cs).
+              resetClustalX(alignment.getSequences(),
+                            alignment.getWidth());
+        }
+
+        cs.setConsensus(hconsensus);
+        if (cs.conservationApplied())
+        {
+          Alignment al = (Alignment) alignment;
+          Conservation c = new Conservation("All",
+                                            ResidueProperties.propHash, 3,
+                                            al.getSequences(), 0,
+                                            al.getWidth() - 1);
+          c.calculate();
+          c.verdict(false, ConsPercGaps);
+
+          cs.setConservation(c);
+        }
+      }
+
+      int s, sSize = alignment.getGroups().size();
+      for(s=0; s<sSize; s++)
+      {
+        SequenceGroup sg = (SequenceGroup)alignment.getGroups().elementAt(s);
+        if(sg.cs!=null && sg.cs instanceof ClustalxColourScheme)
+        {
+          ((ClustalxColourScheme)sg.cs).resetClustalX(
+              sg.getSequences(true), sg.getWidth());
+        }
+        sg.recalcConservation();
+      }
+    }
+
+
+
 }