saves and loads annotations with seqrefs
[jalview.git] / src / jalview / gui / Jalview2XML.java
index 7e2d1dd..6384125 100755 (executable)
  */\r
 package jalview.gui;\r
 \r
-import jalview.io.*;\r
 \r
 import jalview.schemes.*;\r
 \r
+import jalview.gui.*;\r
+\r
 import java.io.*;\r
 \r
 import java.net.*;\r
@@ -32,7 +33,11 @@ import java.util.jar.*;
 \r
 import javax.swing.*;\r
 \r
-import jalview.binding.*;\r
+import org.exolab.castor.xml.*;\r
+\r
+import jalview.schemabinding.version2.*;\r
+\r
+\r
 \r
 \r
 /**\r
@@ -44,7 +49,7 @@ import jalview.binding.*;
 public class Jalview2XML\r
 {\r
     // SAVES SEVERAL ALIGNEMENT WINDOWS TO SAME JARFILE\r
-    public static void SaveState(File statefile)\r
+    public void SaveState(File statefile)\r
     {\r
         long creation = System.currentTimeMillis();\r
         JInternalFrame[] frames = Desktop.desktop.getAllFrames();\r
@@ -116,7 +121,7 @@ public class Jalview2XML
     }\r
 \r
     // USE THIS METHOD TO SAVE A SINGLE ALIGNMENT WINDOW\r
-    public static void SaveAlignment(AlignFrame af, String jarFile,\r
+    public void SaveAlignment(AlignFrame af, String jarFile,\r
         String fileName)\r
     {\r
         try\r
@@ -148,7 +153,7 @@ public class Jalview2XML
      * @param jout DOCUMENT ME!\r
      * @param out DOCUMENT ME!\r
      */\r
-    public static void SaveState(AlignFrame af, long timeStamp,\r
+    public void SaveState(AlignFrame af, long timeStamp,\r
         String fileName, JarOutputStream jout, PrintWriter out)\r
     {\r
         Vector seqids = new Vector();\r
@@ -157,7 +162,7 @@ public class Jalview2XML
         AlignViewport av = af.viewport;\r
 \r
         JalviewModel object = new JalviewModel();\r
-        object.setVamsasModel(new jalview.binding.VamsasModel());\r
+        object.setVamsasModel(new jalview.schemabinding.version2.VamsasModel());\r
 \r
         object.setCreationDate(new java.util.Date(timeStamp));\r
         object.setVersion(jalview.bin.Cache.getProperty("VERSION"));\r
@@ -311,11 +316,28 @@ public class Jalview2XML
 \r
                 Annotation an = new Annotation();\r
                 an.setDescription(aa[i].description);\r
+\r
+                if(aa[i].sequenceRef!=null)\r
+                 {\r
+                   an.setSequenceRef(aa[i].sequenceRef.getName());\r
+                 }\r
+\r
                 if(aa[i].graph>0)\r
                 {\r
                   an.setGraph(true);\r
                   an.setGraphType(aa[i].graph);\r
+                  if(aa[i].getThreshold()!=null)\r
+                  {\r
+                    ThresholdLine line = new ThresholdLine();\r
+                    line.setLabel(aa[i].getThreshold().label);\r
+                    line.setValue(aa[i].getThreshold().value);\r
+                    line.setColour(aa[i].getThreshold().colour.getRGB());\r
+                    an.setThresholdLine(line);\r
+                  }\r
                 }\r
+                else\r
+                  an.setGraph(false);\r
+\r
                 an.setLabel(aa[i].label);\r
 \r
                 AnnotationElement ae;\r
@@ -373,6 +395,12 @@ public class Jalview2XML
                           cs));\r
                     }\r
                   }\r
+                  else if(sg.cs instanceof jalview.schemes.AnnotationColourGradient)\r
+                  {\r
+                    groups[i].setColour(\r
+                        ColourSchemeProperty.getColourName(\r
+                      ( (jalview.schemes.AnnotationColourGradient) sg.cs).getBaseColour()));\r
+                  }\r
                   else if (sg.cs instanceof jalview.schemes.UserColourScheme)\r
                   {\r
                     groups[i].setColour(SetUserColourScheme(sg.cs, userColours,\r
@@ -419,6 +447,26 @@ public class Jalview2XML
             view.setBgColour(SetUserColourScheme(av.getGlobalColourScheme(),\r
                     userColours, jms));\r
         }\r
+        else if(av.getGlobalColourScheme() instanceof jalview.schemes.AnnotationColourGradient)\r
+        {\r
+           jalview.schemes.AnnotationColourGradient acg\r
+              = (jalview.schemes.AnnotationColourGradient)av.getGlobalColourScheme();\r
+\r
+            AnnotationColours ac = new AnnotationColours();\r
+            ac.setAboveThreshold(acg.getAboveThreshold());\r
+            ac.setThreshold(acg.getAnnotationThreshold());\r
+            ac.setAnnotation(acg.getAnnotation());\r
+            if(acg.getBaseColour() instanceof jalview.schemes.UserColourScheme)\r
+              ac.setColourScheme(SetUserColourScheme(acg.getBaseColour(),\r
+                    userColours, jms));\r
+            else\r
+              ac.setColourScheme(ColourSchemeProperty.getColourName(acg.getBaseColour()));\r
+\r
+            ac.setMaxColour(acg.getMaxColour().getRGB());\r
+            ac.setMinColour(acg.getMinColour().getRGB());\r
+            view.setAnnotationColours(ac);\r
+            view.setBgColour("AnnotationColourGradient");\r
+        }\r
         else\r
         {\r
             view.setBgColour(ColourSchemeProperty.getColourName(\r
@@ -461,19 +509,21 @@ public class Jalview2XML
 \r
         if(av.featuresDisplayed!=null)\r
         {\r
-          jalview.binding.FeatureSettings fs = new jalview.binding.FeatureSettings();\r
+          jalview.schemabinding.version2.FeatureSettings fs\r
+              = new jalview.schemabinding.version2.FeatureSettings();\r
 \r
-          Enumeration e = af.alignPanel.seqPanel.seqCanvas.getFeatureRenderer().featureColours.keys();\r
-          while (e.hasMoreElements())\r
+          String [] renderOrder =\r
+              af.alignPanel.seqPanel.seqCanvas.getFeatureRenderer().renderOrder;\r
+          for(int ro=0; ro<renderOrder.length; ro++)\r
           {\r
-            String type = e.nextElement().toString();\r
             Setting setting = new Setting();\r
-            setting.setType(type);\r
+            setting.setType(renderOrder[ro]);\r
             setting.setColour(\r
-                af.alignPanel.seqPanel.seqCanvas.getFeatureRenderer().getColour(type).getRGB()\r
+                af.alignPanel.seqPanel.seqCanvas.getFeatureRenderer().getColour(renderOrder[ro]).getRGB()\r
                 );\r
+\r
             setting.setDisplay(\r
-                av.featuresDisplayed.contains(type)\r
+                av.featuresDisplayed.containsKey(renderOrder[ro])\r
                 );\r
 \r
             fs.addSetting(setting);\r
@@ -507,7 +557,7 @@ public class Jalview2XML
         }\r
     }\r
 \r
-    static String SetUserColourScheme(jalview.schemes.ColourSchemeI cs,\r
+    String SetUserColourScheme(jalview.schemes.ColourSchemeI cs,\r
         Vector userColours, JalviewModelSequence jms)\r
     {\r
         String id = null;\r
@@ -518,12 +568,12 @@ public class Jalview2XML
             userColours.add(ucs);\r
 \r
             java.awt.Color[] colours = ucs.getColours();\r
-            jalview.binding.UserColours uc = new jalview.binding.UserColours();\r
-            jalview.binding.UserColourScheme jbucs = new jalview.binding.UserColourScheme();\r
+            jalview.schemabinding.version2.UserColours uc = new jalview.schemabinding.version2.UserColours();\r
+            jalview.schemabinding.version2.UserColourScheme jbucs = new jalview.schemabinding.version2.UserColourScheme();\r
 \r
             for (int i = 0; i < colours.length; i++)\r
             {\r
-                jalview.binding.Colour col = new jalview.binding.Colour();\r
+                jalview.schemabinding.version2.Colour col = new jalview.schemabinding.version2.Colour();\r
                 col.setRGB(jalview.util.Format.getHexString(colours[i]));\r
                 jbucs.addColour(col);\r
             }\r
@@ -537,7 +587,7 @@ public class Jalview2XML
         return id;\r
     }\r
 \r
-    static jalview.schemes.UserColourScheme GetUserColourScheme(\r
+   jalview.schemes.UserColourScheme GetUserColourScheme(\r
         JalviewModelSequence jms, String id)\r
     {\r
         UserColours[] uc = jms.getUserColours();\r
@@ -570,9 +620,9 @@ public class Jalview2XML
      *\r
      * @param file DOCUMENT ME!\r
      */\r
-    public static AlignFrame LoadJalviewAlign(String file)\r
+    public AlignFrame LoadJalviewAlign(String file)\r
     {\r
-        JalviewModel object = new JalviewModel();\r
+\r
         jalview.gui.AlignFrame af = null;\r
 \r
         try\r
@@ -603,12 +653,29 @@ public class Jalview2XML
                 for (int i = 0; i < entryCount; i++)\r
                 {\r
                     jarentry = jin.getNextJarEntry();\r
-                }\r
+\r
+                  }\r
+\r
+\r
+                  class NoDescIDResolver implements IDResolver\r
+                  {\r
+                    public Object resolve(String idref)\r
+                    {\r
+                      System.out.println(idref+" used");\r
+                      return null;\r
+                    }\r
+                  }\r
+\r
 \r
                 if (jarentry != null)\r
                 {\r
                     InputStreamReader in = new InputStreamReader(jin, "UTF-8");\r
-                    object = (JalviewModel) object.unmarshal(in);\r
+                    JalviewModel object = new JalviewModel();\r
+\r
+                    Unmarshaller unmar = new Unmarshaller(object);\r
+                    unmar.setValidation(false);\r
+                    object = (JalviewModel) unmar.unmarshal( in );\r
+\r
                     af = LoadFromObject(object);\r
                     entryCount++;\r
                 }\r
@@ -628,6 +695,14 @@ public class Jalview2XML
         }\r
         catch (Exception ex)\r
         {\r
+          //Is Version 1 Jar file?\r
+          af =  new Jalview2XML_V1().LoadJalviewAlign(file);\r
+\r
+          if(af!=null)\r
+          {\r
+            System.out.println("Successfuly loaded archive file");\r
+            return af;\r
+          }\r
           ex.printStackTrace();\r
             System.err.println("Exception whilst loading jalview XML file : " +\r
                 ex + "\n");\r
@@ -641,7 +716,7 @@ public class Jalview2XML
         return af;\r
     }\r
 \r
-    static AlignFrame LoadFromObject(JalviewModel object)\r
+    AlignFrame LoadFromObject(JalviewModel object)\r
     {\r
         Vector seqids = new Vector();\r
         SequenceSet vamsasSet = object.getVamsasModel().getSequenceSet(0);\r
@@ -679,12 +754,12 @@ public class Jalview2XML
             Features[] features = JSEQ[i].getFeatures();\r
             for (int f = 0; f < features.length; f++)\r
             {\r
-              //features[f].getBegin()\r
               jalview.datamodel.SequenceFeature sf\r
                   = new jalview.datamodel.SequenceFeature(features[f].getType(),\r
                   features[f].getDescription(), features[f].getStatus(),\r
-                  features[f].getBegin(), features[f].getEnd());\r
-              sf.setFeatureGroup(features[f].getFeatureGroup());\r
+                  features[f].getBegin(), features[f].getEnd(),\r
+                  features[f].getFeatureGroup());\r
+\r
               al.getSequenceAt(i).getDatasetSequence().addSequenceFeature(sf);\r
             }\r
           }\r
@@ -701,7 +776,8 @@ public class Jalview2XML
 \r
           }\r
         }\r
-        /////////////////////////////////\r
+\r
+         /////////////////////////////////\r
         //////////////////////////////////\r
         //LOAD ANNOTATIONS\r
         if (vamsasSet.getAnnotation() != null)\r
@@ -717,7 +793,7 @@ public class Jalview2XML
                 {\r
                     anot[ae[aa].getPosition()] = new jalview.datamodel.Annotation(ae[aa].getDisplayCharacter(),\r
                             ae[aa].getDescription(),\r
-                            ae[aa].getSecondaryStructure().charAt(0),\r
+                            ae[aa].getSecondaryStructure().length()==0?' ':ae[aa].getSecondaryStructure().charAt(0),\r
                             ae[aa].getValue());\r
                 }\r
 \r
@@ -725,9 +801,20 @@ public class Jalview2XML
 \r
                 if (an[i].getGraph())\r
                 {\r
-                    jaa = new jalview.datamodel.AlignmentAnnotation(an[i].getLabel(),\r
-                            an[i].getDescription(), anot, 0, 0,\r
-                            an[i].getGraphType());\r
+                  jaa = new jalview.datamodel.AlignmentAnnotation(an[i].getLabel(),\r
+                      an[i].getDescription(), anot, 0, 0,\r
+                      an[i].getGraphType());\r
+\r
+                  if (an[i].getThresholdLine() != null)\r
+                  {\r
+                    jaa.setThreshold(new jalview.datamodel.GraphLine(\r
+                                  an[i].getThresholdLine().getValue(),\r
+                                  an[i].getThresholdLine().getLabel(),\r
+                                new java.awt.Color(an[i].getThresholdLine().getColour()))\r
+                             );\r
+\r
+                  }\r
+\r
                 }\r
                 else\r
                 {\r
@@ -735,10 +822,19 @@ public class Jalview2XML
                             an[i].getDescription(), anot);\r
                 }\r
 \r
+                if(an[i].getSequenceRef()!=null)\r
+                {\r
+                  jaa.createSequenceMapping(\r
+                      al.findName(an[i].getSequenceRef()), 1\r
+                      );\r
+                  al.findName(an[i].getSequenceRef()).addAlignmentAnnotation(jaa);\r
+                }\r
+\r
                 al.addAnnotation(jaa);\r
             }\r
         }\r
 \r
+\r
         /////////////////////////////////\r
         // LOAD VIEWPORT\r
         Viewport[] views = jms.getViewport();\r
@@ -770,7 +866,8 @@ public class Jalview2XML
                                 groups[i].getColour());\r
                     }\r
 \r
-                    cs.setThreshold(groups[i].getPidThreshold(), true);\r
+                    if(cs!=null)\r
+                      cs.setThreshold(groups[i].getPidThreshold(), true);\r
                 }\r
 \r
                 Vector seqs = new Vector();\r
@@ -805,6 +902,7 @@ public class Jalview2XML
         }\r
 \r
 \r
+\r
         af.setBounds(view.getXpos(), view.getYpos(), view.getWidth(),\r
             view.getHeight());\r
         af.viewport.setStartRes(view.getStartRes());\r
@@ -857,6 +955,96 @@ public class Jalview2XML
             {\r
                 cs = GetUserColourScheme(jms, view.getBgColour());\r
             }\r
+            else if(view.getBgColour().startsWith("Annotation"))\r
+            {\r
+              //int find annotation\r
+              for (int i = 0; i < af.viewport.alignment.getAlignmentAnnotation().length; i++)\r
+              {\r
+                if (af.viewport.alignment.getAlignmentAnnotation()[i].label.\r
+                    equals(view.getAnnotationColours().getAnnotation()))\r
+                {\r
+                  if (af.viewport.alignment.getAlignmentAnnotation()[i].\r
+                      getThreshold() == null)\r
+                  {\r
+                    af.viewport.alignment.getAlignmentAnnotation()[i].\r
+                        setThreshold(\r
+                            new jalview.datamodel.GraphLine(\r
+                                view.getAnnotationColours().getThreshold(),\r
+                                "Threshold", java.awt.Color.black)\r
+\r
+                        );\r
+                  }\r
+\r
+\r
+                  if (view.getAnnotationColours().getColourScheme().equals(\r
+                      "None"))\r
+                  {\r
+                    cs = new AnnotationColourGradient(\r
+                        af.viewport.alignment.getAlignmentAnnotation()[i],\r
+                        new java.awt.Color(view.getAnnotationColours().\r
+                                           getMinColour()),\r
+                        new java.awt.Color(view.getAnnotationColours().\r
+                                           getMaxColour()),\r
+                        view.getAnnotationColours().getAboveThreshold());\r
+                  }\r
+                  else if (view.getAnnotationColours().getColourScheme().\r
+                           startsWith("ucs"))\r
+                  {\r
+                     cs = new AnnotationColourGradient(\r
+                         af.viewport.alignment.getAlignmentAnnotation()[i],\r
+                         GetUserColourScheme(jms, view.getAnnotationColours().\r
+                                                        getColourScheme()),\r
+                         view.getAnnotationColours().getAboveThreshold()\r
+                         );\r
+                   }\r
+                   else\r
+                   {\r
+                     cs = new AnnotationColourGradient(\r
+                         af.viewport.alignment.getAlignmentAnnotation()[i],\r
+                         ColourSchemeProperty.getColour(al,\r
+                         view.getAnnotationColours().getColourScheme()),\r
+                         view.getAnnotationColours().getAboveThreshold()\r
+                         );\r
+                   }\r
+\r
+                  // Also use these settings for all the groups\r
+                  if (al.getGroups() != null)\r
+                    for (int g = 0; g < al.getGroups().size(); g++)\r
+                    {\r
+                      jalview.datamodel.SequenceGroup sg\r
+                          = (jalview.datamodel.SequenceGroup)al.getGroups().elementAt(g);\r
+\r
+                      if(sg.cs == null)\r
+                        continue;\r
+\r
+\r
+                  /*    if (view.getAnnotationColours().getColourScheme().equals("None"))\r
+                      {\r
+                        sg.cs = new AnnotationColourGradient(\r
+                            af.viewport.alignment.getAlignmentAnnotation()[i],\r
+                            new java.awt.Color(view.getAnnotationColours().\r
+                                               getMinColour()),\r
+                            new java.awt.Color(view.getAnnotationColours().\r
+                                               getMaxColour()),\r
+                            view.getAnnotationColours().getAboveThreshold());\r
+                      }\r
+                      else*/\r
+                      {\r
+                        sg.cs = new AnnotationColourGradient(\r
+                            af.viewport.alignment.getAlignmentAnnotation()[i],\r
+                            sg.cs,\r
+                            view.getAnnotationColours().getAboveThreshold()\r
+                            );\r
+                      }\r
+\r
+                    }\r
+\r
+\r
+                  break;\r
+                }\r
+\r
+              }\r
+            }\r
             else\r
             {\r
                 cs = ColourSchemeProperty.getColour(al, view.getBgColour());\r
@@ -889,20 +1077,24 @@ public class Jalview2XML
         if(jms.getFeatureSettings()!=null)\r
         {\r
           af.viewport.featuresDisplayed = new Hashtable();\r
-         // af.alignPanel.seqPanel.seqCanvas.getFeatureRenderer().featuresDisplayed!=null)\r
-         // jalview.binding.FeatureSettings fs = new jalview.binding.FeatureSettings();\r
+          String [] renderOrder = new String[jms.getFeatureSettings().getSettingCount()];\r
          for(int fs=0; fs<jms.getFeatureSettings().getSettingCount(); fs++)\r
          {\r
            Setting setting = jms.getFeatureSettings().getSetting(fs);\r
 \r
            af.alignPanel.seqPanel.seqCanvas.getFeatureRenderer().setColour(setting.getType(),\r
                new java.awt.Color(setting.getColour()));\r
+\r
+           renderOrder[fs] = setting.getType();\r
+\r
            if(setting.getDisplay())\r
              af.viewport.featuresDisplayed.put(\r
                  setting.getType(), new Integer(setting.getColour()));\r
          }\r
+         af.alignPanel.seqPanel.seqCanvas.getFeatureRenderer().renderOrder = renderOrder;\r
         }\r
 \r
+\r
         Desktop.addInternalFrame(af, view.getTitle(),\r
              view.getWidth(), view.getHeight());\r
 \r
@@ -955,3 +1147,4 @@ public class Jalview2XML
         return af;\r
     }\r
 }\r
+\r