X-Git-Url: http://source.jalview.org/gitweb/?a=blobdiff_plain;f=src%2Fjalview%2Fschemes%2FRNAHelicesColourChooser.java;h=cef7eb6627ae4f573492672d1bbd5e6861981c9d;hb=6fb502e4a8749d1d98f5442c3b8156ab47996b77;hp=e575b91dc058277b63fce6e632ade28bfad0ab1a;hpb=44f6e9fd1e6df7c78950ef2c778e86add4ec498e;p=jalview.git

diff --git a/src/jalview/schemes/RNAHelicesColourChooser.java b/src/jalview/schemes/RNAHelicesColourChooser.java
index e575b91..cef7eb6 100644
--- a/src/jalview/schemes/RNAHelicesColourChooser.java
+++ b/src/jalview/schemes/RNAHelicesColourChooser.java
@@ -1,13 +1,13 @@
 /*
- * Jalview - A Sequence Alignment Editor and Viewer (Version 2.7)
- * Copyright (C) 2011 J Procter, AM Waterhouse, J Engelhardt, LM Lui, G Barton, M Clamp, S Searle
+ * Jalview - A Sequence Alignment Editor and Viewer (Version 2.8)
+ * Copyright (C) 2012 J Procter, AM Waterhouse, LM Lui, J Engelhardt, G Barton, M Clamp, S Searle
  * 
  * 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 
@@ -15,11 +15,13 @@
  * 
  * You should have received a copy of the GNU General Public License along with Jalview.  If not, see <http://www.gnu.org/licenses/>.
  */
-package jalview.gui;
+package jalview.schemes;
 
 import java.util.*;
 import java.awt.event.*;
 
+import jalview.api.AlignViewportI;
+import jalview.api.AlignmentViewPanel;
 import jalview.datamodel.*;
 import jalview.schemes.*;
 
@@ -33,9 +35,9 @@ import jalview.schemes.*;
 public class RNAHelicesColourChooser
 {
 
-  AlignViewport av;
+  AlignViewportI av;
 
-  AlignmentPanel ap;
+  AlignmentViewPanel ap;
 
   ColourSchemeI oldcs;
 
@@ -45,17 +47,15 @@ public class RNAHelicesColourChooser
 
   boolean adjusting = false;
 
-  public RNAHelicesColourChooser(AlignViewport av, final AlignmentPanel ap)
+  public RNAHelicesColourChooser(AlignViewportI av,
+          final AlignmentViewPanel ap)
   {
     oldcs = av.getGlobalColourScheme();
     if (av.getAlignment().getGroups() != null)
     {
       oldgroupColours = new Hashtable();
-      Vector allGroups = ap.av.getAlignment().getGroups();
-      SequenceGroup sg;
-      for (int g = 0; g < allGroups.size(); g++)
+      for (SequenceGroup sg : ap.getAlignment().getGroups())
       {
-        sg = (SequenceGroup) allGroups.get(g);
         if (sg.cs != null)
         {
           oldgroupColours.put(sg, sg.cs);
@@ -97,8 +97,24 @@ public class RNAHelicesColourChooser
       return;
     }
 
-    currentAnnotation = av.getAlignment().getAlignmentAnnotation()[0];// annotations.getSelectedIndex()];
-
+    // This loop will find the first rna structure annotation by which to colour
+    //  the sequences.
+    AlignmentAnnotation[] annotations = av.getAlignment().getAlignmentAnnotation();
+    for (int i = 0; i < annotations.length; i++) {
+    	
+    	// is this a sensible way of determining type of annotation?
+    	if (annotations[i].getRNAStruc() != null) { 
+    		currentAnnotation = annotations[i];
+    		break;
+    	}
+    }
+    if (currentAnnotation == null)   
+    {
+    	System.err.println("Jalview is about to try and colour by RNAHelices even"
+    			+ " though there are no RNA secondary structure annotations present!");
+    	currentAnnotation = av.getAlignment().getAlignmentAnnotation()[0];// annotations.getSelectedIndex()];
+    }
+    
     RNAHelicesColour rhc = null;
 
     rhc = new RNAHelicesColour(currentAnnotation);
@@ -107,12 +123,8 @@ public class RNAHelicesColourChooser
 
     if (av.getAlignment().getGroups() != null)
     {
-      Vector allGroups = ap.av.getAlignment().getGroups();
-      SequenceGroup sg;
-      for (int g = 0; g < allGroups.size(); g++)
+      for (SequenceGroup sg : ap.getAlignment().getGroups())
       {
-        sg = (SequenceGroup) allGroups.get(g);
-
         if (sg.cs == null)
         {
           continue;
@@ -131,11 +143,8 @@ public class RNAHelicesColourChooser
     av.setGlobalColourScheme(oldcs);
     if (av.getAlignment().getGroups() != null)
     {
-      Vector allGroups = ap.av.getAlignment().getGroups();
-      SequenceGroup sg;
-      for (int g = 0; g < allGroups.size(); g++)
+      for (SequenceGroup sg : ap.getAlignment().getGroups())
       {
-        sg = (SequenceGroup) allGroups.get(g);
         sg.cs = (ColourSchemeI) oldgroupColours.get(sg);
       }
     }