X-Git-Url: http://source.jalview.org/gitweb/?a=blobdiff_plain;ds=sidebyside;f=src%2Fjalview%2Fgui%2FJalview2XML.java;h=a90174121559ed7b2c5ddb0e557109b6703ec6cf;hb=e1124a62486322d2d28482529d2d6b850ce1719b;hp=008be21188ac3dbb809014943462fe666b309df9;hpb=c901bd1778574b5792a7ee168bfe3e1addd3e181;p=jalview.git

diff --git a/src/jalview/gui/Jalview2XML.java b/src/jalview/gui/Jalview2XML.java
index 008be21..a901741 100755
--- a/src/jalview/gui/Jalview2XML.java
+++ b/src/jalview/gui/Jalview2XML.java
@@ -1,466 +1,824 @@
+/*
+ * Jalview - A Sequence Alignment Editor and Viewer
+ * Copyright (C) 2005 AM Waterhouse, J Procter, G Barton, M Clamp, S Searle
+ *
+ * This program 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 2
+ * of the License, or (at your option) any later version.
+ *
+ * This program 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 this program; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA
+ */
 package jalview.gui;
 
+import jalview.io.*;
+
+import jalview.schemes.*;
+
 import java.io.*;
+
+import java.net.*;
+
 import java.util.*;
 import java.util.jar.*;
 
+import javax.swing.*;
+
 import jalview.binding.*;
-import jalview.schemes.*;
-import javax.swing.JInternalFrame;
 
 
+/**
+ * DOCUMENT ME!
+ *
+ * @author $author$
+ * @version $Revision$
+ */
 public class Jalview2XML
 {
+    // SAVES SEVERAL ALIGNEMENT WINDOWS TO SAME JARFILE
+    public static void SaveState(File statefile)
+    {
+        long creation = System.currentTimeMillis();
+        JInternalFrame[] frames = Desktop.desktop.getAllFrames();
 
-  public static void SaveState(File statefile)
-  {
-      long creation = System.currentTimeMillis();
-      JInternalFrame[] frames = Desktop.desktop.getAllFrames();
-      try{
-        JalviewState state = new JalviewState();
+        if (frames == null)
+        {
+            return;
+        }
 
-        Vector shortNames = new Vector();
-        for (int i = 0; i < frames.length; i++)
+        try
         {
-          if (frames[i] instanceof AlignFrame)
-          {
-            AlignFrame af = (AlignFrame) frames[i];
+            FileOutputStream fos = new FileOutputStream(statefile);
+            JarOutputStream jout = new JarOutputStream(fos);
 
-            String shortName = af.getTitle();
+            //NOTE UTF-8 MUST BE USED FOR WRITING UNICODE CHARS
+            ////////////////////////////////////////////////////
+            PrintWriter out = new PrintWriter(new OutputStreamWriter(jout,
+                        "UTF-8"));
 
-            shortName = shortName.replace('/', '_');
-            shortName = shortName.replace('\\', '_');
-            String jarFile = statefile.getParent()+File.separatorChar+shortName+".jar";
+            Vector shortNames = new Vector();
 
-            int count=1;
-            while( shortNames.contains(shortName) )
+            //REVERSE ORDER
+            for (int i = frames.length - 1; i > -1; i--)
             {
-              if(shortName.endsWith("_"+(count-1)))
-                shortName = shortName.substring(0, shortName.lastIndexOf("_"));
+                if (frames[i] instanceof AlignFrame)
+                {
+                    AlignFrame af = (AlignFrame) frames[i];
 
-              shortName = shortName.concat("_"+count);
-              count++;
+                    String shortName = af.getTitle();
+
+                    if (shortName.indexOf(File.separatorChar) > -1)
+                    {
+                        shortName = shortName.substring(shortName.lastIndexOf(
+                                    File.separatorChar) + 1);
+                    }
+
+                    int count = 1;
+
+                    while (shortNames.contains(shortName))
+                    {
+                        if (shortName.endsWith("_" + (count - 1)))
+                        {
+                            shortName = shortName.substring(0,
+                                    shortName.lastIndexOf("_"));
+                        }
+
+                        shortName = shortName.concat("_" + count);
+                        count++;
+                    }
+
+                    shortNames.addElement(shortName);
+
+                    if (!shortName.endsWith(".xml"))
+                    {
+                        shortName = shortName + ".xml";
+                    }
+
+                    SaveState(af, creation, shortName, jout, out);
+                }
             }
-            shortNames.addElement(shortName);
 
-            if(!shortName.endsWith(".xml"))
-              shortName = shortName+".xml";
+            out.close();
+            jout.close();
+        }
+        catch (Exception ex)
+        {
+            ex.printStackTrace();
+        }
+    }
+
+    // USE THIS METHOD TO SAVE A SINGLE ALIGNMENT WINDOW
+    public static void SaveAlignment(AlignFrame af, String jarFile,
+        String fileName)
+    {
+        try
+        {
+            FileOutputStream fos = new FileOutputStream(jarFile);
+            JarOutputStream jout = new JarOutputStream(fos);
+
+            //NOTE UTF-8 MUST BE USED FOR WRITING UNICODE CHARS
+            ////////////////////////////////////////////////////
+            PrintWriter out = new PrintWriter(new OutputStreamWriter(jout,
+                        "UTF-8"));
+
+            SaveState(af, System.currentTimeMillis(), fileName, jout, out);
+            out.close();
+            jout.close();
+        }
+        catch (Exception ex)
+        {
+          ex.printStackTrace();
+        }
+    }
+
+    /**
+     * DOCUMENT ME!
+     *
+     * @param af DOCUMENT ME!
+     * @param timeStamp DOCUMENT ME!
+     * @param fileName DOCUMENT ME!
+     * @param jout DOCUMENT ME!
+     * @param out DOCUMENT ME!
+     */
+    public static void SaveState(AlignFrame af, long timeStamp,
+        String fileName, JarOutputStream jout, PrintWriter out)
+    {
+        Vector seqids = new Vector();
+        Vector userColours = new Vector();
+
+        AlignViewport av = af.viewport;
+
+        JalviewModel object = new JalviewModel();
+        object.setVamsasModel(new VamsasModel());
+
+        object.setCreationDate(new java.util.Date(timeStamp));
+        object.setVersion(jalview.bin.Cache.getProperty("VERSION"));
+
+        jalview.datamodel.AlignmentI jal = af.viewport.alignment;
+
+        SequenceSet vamsasSet = new SequenceSet();
+        Sequence vamsasSeq;
+        JalviewModelSequence jms = new JalviewModelSequence();
+
+        vamsasSet.setGapChar(jal.getGapCharacter() + "");
+
+        JSeq jseq;
+
+        //SAVE SEQUENCES
+        int id = 0;
 
-            jarFile = statefile.getParent()+File.separatorChar+shortName+".jar";
+        for (int i = 0; i < jal.getHeight(); i++)
+        {
+            seqids.add(jal.getSequenceAt(i));
+
+            vamsasSeq = new Sequence();
+            vamsasSeq.setId(id + "");
+            vamsasSeq.setName(jal.getSequenceAt(i).getName());
+            vamsasSeq.setSequence(jal.getSequenceAt(i).getSequence());
+
+            jseq = new JSeq();
+            jseq.setStart(jal.getSequenceAt(i).getStart());
+            jseq.setEnd(jal.getSequenceAt(i).getEnd());
 
-            SaveState( af, creation, shortName, jarFile);
-            state.addFile(jarFile);
+            jseq.setColour(jal.getSequenceAt(i).getColor().getRGB());
+
+            jseq.setId(id);
+
+            jms.addJSeq(jseq);
+            vamsasSet.addSequence(vamsasSeq);
+            id++;
+        }
+
+        //SAVE TREES
+        ///////////////////////////////////
+        if (af.viewport.currentTree != null)
+        {
+          // FIND ANY ASSOCIATED TREES
+          // NOT IMPLEMENTED FOR HEADLESS STATE AT PRESENT
+          if (Desktop.desktop != null)
+          {
+            JInternalFrame[] frames = Desktop.desktop.getAllFrames();
+
+            for (int t = 0; t < frames.length; t++)
+            {
+              if (frames[t] instanceof TreePanel)
+              {
+                TreePanel tp = (TreePanel) frames[t];
+
+                if (tp.treeCanvas.av.alignment == jal)
+                {
+                  Tree tree = new Tree();
+                  tree.setTitle(tp.getTitle());
+                  tree.setCurrentTree( (af.viewport.currentTree == tp.getTree()));
+                  tree.setNewick(tp.getTree().toString());
+                  tree.setThreshold(tp.treeCanvas.threshold);
+
+                  tree.setFitToWindow(tp.fitToWindow.getState());
+                  tree.setFontName(tp.getTreeFont().getName());
+                  tree.setFontSize(tp.getTreeFont().getSize());
+                  tree.setFontStyle(tp.getTreeFont().getStyle());
+                  tree.setMarkUnlinked(tp.placeholdersMenu.getState());
+                  tree.setShowBootstrap(tp.bootstrapMenu.getState());
+                  tree.setShowDistances(tp.distanceMenu.getState());
+
+                  tree.setHeight(tp.getHeight());
+                  tree.setWidth(tp.getWidth());
+                  tree.setXpos(tp.getX());
+                  tree.setYpos(tp.getY());
+
+                  jms.addTree(tree);
+                }
+              }
+            }
           }
         }
 
-        PrintWriter out = new PrintWriter(new FileWriter(statefile));
-        state.marshal(out);
-        out.close();
-      }
-      catch(Exception ex){ex.printStackTrace();}
-  }
+        //SAVE ANNOTATIONS
+        if (jal.getAlignmentAnnotation() != null)
+        {
+            jalview.datamodel.AlignmentAnnotation[] aa = jal.getAlignmentAnnotation();
 
+            for (int i = 0; i < aa.length; i++)
+            {
+                if (aa[i].label.equals("Quality") ||
+                        aa[i].label.equals("Conservation") ||
+                        aa[i].label.equals("Consensus"))
+                {
+                    continue;
+                }
+
+                Annotation an = new Annotation();
+                an.setDescription(aa[i].description);
+                an.setGraph(aa[i].isGraph);
+                an.setLabel(aa[i].label);
+
+                AnnotationElement ae;
+
+                for (int a = 0; a < aa[i].annotations.length; a++)
+                {
+                    if ((aa[i] == null) || (aa[i].annotations[a] == null))
+                    {
+                        continue;
+                    }
+
+                    ae = new AnnotationElement();
+                    ae.setDescription(aa[i].annotations[a].description);
+                    ae.setDisplayCharacter(aa[i].annotations[a].displayCharacter);
+                    ae.setValue(aa[i].annotations[a].value);
+                    ae.setPosition(a);
+                    ae.setSecondaryStructure(aa[i].annotations[a].secondaryStructure +
+                        "");
+                    an.addAnnotationElement(ae);
+                }
+
+                vamsasSet.addAnnotation(an);
+            }
+        }
 
-  public static void SaveState( AlignFrame af, long timeStamp, String fileName, String jarFile)
-  {
+        //SAVE GROUPS
+        if (jal.getGroups() != null)
+        {
+            JGroup[] groups = new JGroup[jal.getGroups().size()];
 
-    Vector seqids = new Vector();
+            for (int i = 0; i < groups.length; i++)
+            {
+                groups[i] = new JGroup();
+
+                jalview.datamodel.SequenceGroup sg = (jalview.datamodel.SequenceGroup) jal.getGroups()
+                                                                                          .elementAt(i);
+                groups[i].setStart(sg.getStartRes());
+                groups[i].setEnd(sg.getEndRes());
+                groups[i].setName(sg.getName());
+                if(sg.cs!=null)
+                {
+                  if (sg.cs.conservationApplied())
+                  {
+                    groups[i].setConsThreshold(sg.cs.getConservationInc());
+
+                    if (sg.cs instanceof jalview.schemes.UserColourScheme)
+                    {
+                      groups[i].setColour(SetUserColourScheme(sg.cs,
+                          userColours,
+                          jms));
+                    }
+                    else
+                    {
+                      groups[i].setColour(ColourSchemeProperty.getColourName(sg.
+                          cs));
+                    }
+                  }
+                  else if (sg.cs instanceof jalview.schemes.UserColourScheme)
+                  {
+                    groups[i].setColour(SetUserColourScheme(sg.cs, userColours,
+                        jms));
+                  }
+                  else
+                  {
+                    groups[i].setColour(ColourSchemeProperty.getColourName(
+                        sg.cs));
+                  }
+
+                  groups[i].setPidThreshold(sg.cs.getThreshold());
+                }
+
+                groups[i].setOutlineColour(sg.getOutlineColour().getRGB());
+                groups[i].setDisplayBoxes(sg.getDisplayBoxes());
+                groups[i].setDisplayText(sg.getDisplayText());
+                groups[i].setColourText(sg.getColourText());
+
+                for (int s = 0; s < sg.getSize(); s++)
+                {
+                    jalview.datamodel.Sequence seq = (jalview.datamodel.Sequence) sg.getSequenceAt(s);
+                    int index = seqids.indexOf(seq);
+                    groups[i].addSeq(index);
+                }
+            }
 
-    AlignViewport av = af.viewport;
+            jms.setJGroup(groups);
+        }
 
-    JalviewModel object = new JalviewModel();
-    object.setVamsasModel( new VamsasModel() );
+        ///////////SAVE VIEWPORT
+        Viewport view = new Viewport();
+        view.setTitle(af.getTitle());
+        view.setXpos(af.getX());
+        view.setYpos(af.getY());
+        view.setWidth(af.getWidth());
+        view.setHeight(af.getHeight());
+        view.setStartRes(av.startRes);
+        view.setStartSeq(av.startSeq);
+
+        if (av.getGlobalColourScheme() instanceof jalview.schemes.UserColourScheme)
+        {
+            view.setBgColour(SetUserColourScheme(av.getGlobalColourScheme(),
+                    userColours, jms));
+        }
+        else
+        {
+            view.setBgColour(ColourSchemeProperty.getColourName(
+                    av.getGlobalColourScheme()));
+        }
 
-    object.setCreationDate( new java.util.Date(timeStamp) );
+        ColourSchemeI cs = av.getGlobalColourScheme();
 
-    jalview.datamodel.AlignmentI jal = af.viewport.alignment;
+        if(cs!=null)
+        {
+          if (cs.conservationApplied())
+          {
+            view.setConsThreshold(cs.getConservationInc());
+            if (cs instanceof jalview.schemes.UserColourScheme)
+              view.setBgColour(SetUserColourScheme(cs, userColours, jms));
+          }
 
-    SequenceSet vamsasSet = new SequenceSet();
-    Sequence vamsasSeq;
-    JalviewModelSequence jms = new JalviewModelSequence();
+          if (cs instanceof ResidueColourScheme)
+          {
+            view.setPidThreshold(cs.getThreshold());
+          }
+        }
 
-    vamsasSet.setGapChar(jal.getGapCharacter()+"");
+        view.setConservationSelected(av.getConservationSelected());
+        view.setPidSelected(av.getAbovePIDThreshold());
+        view.setFontName(av.font.getName());
+        view.setFontSize(av.font.getSize());
+        view.setFontStyle(av.font.getStyle());
+        view.setRenderGaps(av.renderGaps);
+        view.setShowAnnotation(av.getShowAnnotation());
+        view.setShowBoxes(av.getShowBoxes());
+        view.setShowColourText(av.getColourText());
+        view.setShowConservation(av.showConservation);
+        view.setShowFullId(av.getShowFullId());
+        view.setShowIdentity(av.showIdentity);
+        view.setShowQuality(av.showQuality);
+        view.setShowSequenceFeatures(av.showSequenceFeatures);
+        view.setShowText(av.getShowText());
+        view.setWrapAlignment(av.getWrapAlignment());
+
+        jms.addViewport(view);
+
+        object.setJalviewModelSequence(jms);
+        object.getVamsasModel().addSequenceSet(vamsasSet);
+
+        try
+        {
+            if (!fileName.endsWith(".xml"))
+            {
+                fileName = fileName + ".xml";
+            }
 
-    JSeq jseq;
+            JarEntry entry = new JarEntry(fileName);
+            jout.putNextEntry(entry);
 
+            object.marshal(out);
+        }
+        catch (Exception ex)
+        {
+            ex.printStackTrace();
+        }
+    }
 
-    //SAVE SEQUENCES
-    int id = 0;
-    for(int i=0; i<jal.getHeight(); i++)
+    static String SetUserColourScheme(jalview.schemes.ColourSchemeI cs,
+        Vector userColours, JalviewModelSequence jms)
     {
-      seqids.add(jal.getSequenceAt(i));
+        String id = null;
+        jalview.schemes.UserColourScheme ucs = (jalview.schemes.UserColourScheme) cs;
 
-      vamsasSeq = new Sequence( );
-      vamsasSeq.setId(id+"");
-      vamsasSeq.setName(jal.getSequenceAt(i).getName());
-      vamsasSeq.setSequence(jal.getSequenceAt(i).getSequence());
+        if (!userColours.contains(ucs))
+        {
+            userColours.add(ucs);
 
-      jseq = new JSeq();
-      jseq.setStart(jal.getSequenceAt(i).getStart());
-      jseq.setEnd(jal.getSequenceAt(i).getEnd());
+            java.awt.Color[] colours = ucs.getColours();
+            jalview.binding.UserColours uc = new jalview.binding.UserColours();
+            jalview.binding.UserColourScheme jbucs = new jalview.binding.UserColourScheme();
 
-      jseq.setColour(jal.getSequenceAt(i).getColor().getRGB());
+            for (int i = 0; i < colours.length; i++)
+            {
+                jalview.binding.Colour col = new jalview.binding.Colour();
+                col.setRGB(jalview.util.Format.getHexString(colours[i]));
+                jbucs.addColour(col);
+            }
 
-      jseq.setId( id );
+            id = "ucs" + userColours.indexOf(ucs);
+            uc.setId(id);
+            uc.setUserColourScheme(jbucs);
+            jms.addUserColours(uc);
+        }
 
-      jms.addJSeq(jseq);
-      vamsasSet.addSequence(vamsasSeq);
-      id++;
+        return id;
     }
 
-    //SAVE TREES
-    if(af.viewport.currentTree!=null)
+    static jalview.schemes.UserColourScheme GetUserColourScheme(
+        JalviewModelSequence jms, String id)
     {
-      object.getVamsasModel().addTree(af.viewport.currentTree.toString());
+        UserColours[] uc = jms.getUserColours();
+        UserColours colours = null;
+
+        for (int i = 0; i < uc.length; i++)
+        {
+            if (uc[i].getId().equals(id))
+            {
+                colours = uc[i];
+
+                break;
+            }
+        }
+
+        int csize = colours.getUserColourScheme().getColourCount();
+        java.awt.Color[] newColours = new java.awt.Color[csize];
+
+        for (int i = 0; i < csize; i++)
+        {
+            newColours[i] = new java.awt.Color(Integer.parseInt(
+                        colours.getUserColourScheme().getColour(i).getRGB(), 16));
+        }
+
+        return new jalview.schemes.UserColourScheme(newColours);
     }
 
-    //SAVE ANNOTATIONS
-    if(jal.getAlignmentAnnotation()!=null)
+    /**
+     * DOCUMENT ME!
+     *
+     * @param file DOCUMENT ME!
+     */
+    public static AlignFrame LoadJalviewAlign(String file)
     {
+        JalviewModel object = new JalviewModel();
+        jalview.gui.AlignFrame af = null;
 
-      jalview.datamodel.AlignmentAnnotation [] aa = jal.getAlignmentAnnotation();
-      for(int i=0; i<aa.length; i++)
-      {
-        if(aa[i].label.equals("Quality")
-        || aa[i].label.equals("Conservation")
-        || aa[i].label.equals("Consensus"))
-       {    continue;}
+        try
+        {
+            //UNMARSHALLER SEEMS TO CLOSE JARINPUTSTREAM, MOST ANNOYING
+            URL url = null;
 
-        Annotation an = new Annotation();
-        an.setDescription(aa[i].description);
-        an.setGraph(aa[i].isGraph);
-        an.setLabel(aa[i].label);
+            if (file.startsWith("http://"))
+            {
+                url = new URL(file);
+            }
 
-        AnnotationElement ae;
+            JarInputStream jin = null;
+            JarEntry jarentry = null;
+            int entryCount = 1;
 
-        for(int a=0; a<aa[i].annotations.length; a++)
+            do
+            {
+                if (url != null)
+                {
+                    jin = new JarInputStream(url.openStream());
+                }
+                else
+                {
+                    jin = new JarInputStream(new FileInputStream(file));
+                }
+
+                for (int i = 0; i < entryCount; i++)
+                {
+                    jarentry = jin.getNextJarEntry();
+                }
+
+                if (jarentry != null)
+                {
+                    InputStreamReader in = new InputStreamReader(jin, "UTF-8");
+                    object = (JalviewModel) object.unmarshal(in);
+                    af = LoadFromObject(object);
+                    entryCount++;
+                }
+            }
+            while (jarentry != null);
+        }
+        catch(java.net.UnknownHostException ex)
         {
-          if(aa[i]==null || aa[i].annotations[a]==null)
-            continue;
-
-          ae = new AnnotationElement();
-          ae.setDescription(aa[i].annotations[a].description);
-          ae.setDisplayCharacter(aa[i].annotations[a].displayCharacter);
-          ae.setValue(aa[i].annotations[a].value);
-          ae.setPosition(a);
-          ae.setSecondaryStructure(aa[i].annotations[a].secondaryStructure+"");
-          an.addAnnotationElement(ae);
+          ex.printStackTrace();
+          System.err.println("Couldn't locate Jalview XML file : " +
+              ex + "\n");
+           JOptionPane.showInternalMessageDialog(Desktop.desktop,
+                                    "Couldn't locate " + file,
+                                    "URL not found",
+                                    JOptionPane.WARNING_MESSAGE);
+
         }
+        catch (Exception ex)
+        {
+          ex.printStackTrace();
+            System.err.println("Exception whilst loading jalview XML file : " +
+                ex + "\n");
+             JOptionPane.showInternalMessageDialog(Desktop.desktop,
+                                      "Error loading  " + file,
+                                      "Error loading Jalview file",
+                                      JOptionPane.WARNING_MESSAGE);
 
-        vamsasSet.addAnnotation(an);
-      }
+        }
 
+        return af;
     }
 
-    //SAVE GROUPS
-    if(jal.getGroups()!=null)
+    static AlignFrame LoadFromObject(JalviewModel object)
     {
-      JGroup [] groups = new JGroup[jal.getGroups().size()];
-      for(int i=0; i<groups.length; i++)
-      {
-        groups[i] = new JGroup();
-        jalview.datamodel.SequenceGroup sg
-            = (jalview.datamodel.SequenceGroup)jal.getGroups().elementAt(i);
-        groups[i].setStart( sg.getStartRes() );
-        groups[i].setEnd( sg.getEndRes() );
-        groups[i].setName( sg.getName());
-
-        if(sg.cs instanceof ConservationColourScheme)
-          groups[i].setConsThreshold( ((ConservationColourScheme)sg.cs).inc);
-
-        if (sg.cs instanceof ResidueColourScheme)
-          groups[i].setPidThreshold( ( (ResidueColourScheme) sg.cs).getThreshold());
-       else if (sg.cs instanceof ScoreColourScheme)
-         groups[i].setPidThreshold( ( (ScoreColourScheme) sg.cs).getThreshold());
-
-
-        groups[i].setColour( ColourSchemeProperty.getColourName(sg.cs) );
-        groups[i].setOutlineColour(sg.getOutlineColour().getRGB());
-        groups[i].setDisplayBoxes(sg.getDisplayBoxes());
-        groups[i].setDisplayText(sg.getDisplayText());
-        groups[i].setColourText(sg.getColourText());
-
-        for(int s=0; s<sg.getSize(); s++)
+        Vector seqids = new Vector();
+        SequenceSet vamsasSet = object.getVamsasModel().getSequenceSet(0);
+        Sequence[] vamsasSeq = vamsasSet.getSequence();
+
+        JalviewModelSequence jms = object.getJalviewModelSequence();
+
+        //////////////////////////////////
+        //LOAD SEQUENCES
+        jalview.datamodel.Sequence[] jseqs = new jalview.datamodel.Sequence[vamsasSeq.length];
+        JSeq[] JSEQ = object.getJalviewModelSequence().getJSeq();
+
+        for (int i = 0; i < vamsasSeq.length; i++)
         {
-          jalview.datamodel.Sequence seq = (jalview.datamodel.Sequence)sg.getSequenceAt(s);
-          int index = seqids.indexOf(seq);
-          groups[i].addSeq(index);
+            jseqs[i] = new jalview.datamodel.Sequence(vamsasSeq[i].getName(),
+                    vamsasSeq[i].getSequence());
+
+            jseqs[i].setStart(JSEQ[i].getStart());
+            jseqs[i].setEnd(JSEQ[i].getEnd());
+            jseqs[i].setColor(new java.awt.Color(JSEQ[i].getColour()));
+            seqids.add(jseqs[i]);
         }
 
-      }
+        /////////////////////////////////
+        jalview.datamodel.Alignment al = new jalview.datamodel.Alignment(jseqs);
 
-      jms.setJGroup(groups);
+        /////////////////////////////////
+        //////////////////////////////////
+        //LOAD ANNOTATIONS
+        if (vamsasSet.getAnnotation() != null)
+        {
+            Annotation[] an = vamsasSet.getAnnotation();
 
-    }
+            for (int i = 0; i < an.length; i++)
+            {
+                AnnotationElement[] ae = an[i].getAnnotationElement();
+                jalview.datamodel.Annotation[] anot = new jalview.datamodel.Annotation[al.getWidth()];
+
+                for (int aa = 0; aa < ae.length; aa++)
+                {
+                    anot[ae[aa].getPosition()] = new jalview.datamodel.Annotation(ae[aa].getDisplayCharacter(),
+                            ae[aa].getDescription(),
+                            ae[aa].getSecondaryStructure().charAt(0),
+                            ae[aa].getValue());
+                }
+
+                jalview.datamodel.AlignmentAnnotation jaa = null;
+
+                if (an[i].getGraph())
+                {
+                    jaa = new jalview.datamodel.AlignmentAnnotation(an[i].getLabel(),
+                            an[i].getDescription(), anot, 0, 0, 1);
+                }
+                else
+                {
+                    jaa = new jalview.datamodel.AlignmentAnnotation(an[i].getLabel(),
+                            an[i].getDescription(), anot);
+                }
+
+                al.addAnnotation(jaa);
+            }
+        }
 
-    ///////////SAVE VIEWPORT
-    Viewport view = new Viewport();
-   view.setTitle(af.getTitle());
-   view.setXpos(af.getX());
-   view.setYpos(af.getY());
-   view.setWidth(af.getWidth());
-   view.setHeight(af.getHeight());
-   view.setStartRes(av.startRes);
-   view.setStartSeq(av.startSeq);
-   view.setBgColour( ColourSchemeProperty.getColourName
-                     ( av.getGlobalColourScheme()));
-
-   ColourSchemeI cs = av.getGlobalColourScheme();
-   if(cs instanceof ConservationColourScheme)
-     view.setConsThreshold( ((ConservationColourScheme)cs).inc);
-
-   if (cs instanceof ResidueColourScheme)
-     view.setPidThreshold( ( (ResidueColourScheme) cs).getThreshold());
-  else if (cs instanceof ScoreColourScheme)
-     view.setPidThreshold( ( (ScoreColourScheme) cs).getThreshold());
-
-
-   view.setConservationSelected(av.getConservationSelected());
-   view.setPidSelected(av.getAbovePIDThreshold());
-   view.setFontName(av.font.getFontName());
-   view.setFontSize(av.font.getSize());
-   view.setFontStyle(av.font.getStyle());
-   view.setRenderGaps(av.renderGaps);
-   view.setShowAnnotation(av.getShowAnnotation());
-   view.setShowBoxes(av.getShowBoxes());
-   view.setShowColourText(av.getColourText());
-   view.setShowConservation(av.showConservation);
-   view.setShowFullId(av.getShowFullId());
-   view.setShowIdentity(av.showIdentity);
-   view.setShowQuality(av.showQuality);
-   view.setShowSequenceFeatures(av.showSequenceFeatures);
-   view.setShowText(av.getShowText());
-   view.setWrapAlignment(av.getWrapAlignment());
-
-   jms.addViewport( view );
-
-    object.setJalviewModelSequence(jms);
-    object.getVamsasModel().addSequenceSet(vamsasSet);
-
-
-    try
-    {
-    // NAME OF FILE
-     FileOutputStream fos = new FileOutputStream(jarFile);
+        /////////////////////////////////
+        // LOAD VIEWPORT
+        Viewport[] views = jms.getViewport();
+        Viewport view = views[0]; // DEAL WITH MULTIPLE VIEWPORTS LATER
+
+        AlignFrame af = new AlignFrame(al);
 
-     //PUT IT IN THIS JARFILE
-     JarOutputStream jout = new JarOutputStream(fos);
+        //  af.changeColour() );
+        /////////////////////////
+        //LOAD GROUPS
+        if (jms.getJGroupCount() > 0)
+        {
+            JGroup[] groups = jms.getJGroup();
 
-     //NOTE UTF-8 MUST BE USED FOR WRITING UNICODE CHARS
-     ////////////////////////////////////////////////////
-     PrintWriter out = new PrintWriter(new OutputStreamWriter(jout, "UTF-8"));
+            for (int i = 0; i < groups.length; i++)
+            {
+                ColourSchemeI cs = null;
+
+                if (groups[i].getColour() != null)
+                {
+                    if (groups[i].getColour().startsWith("ucs"))
+                    {
+                        cs = GetUserColourScheme(jms, groups[i].getColour());
+                    }
+                    else
+                    {
+                        cs = ColourSchemeProperty.getColour(al,
+                                groups[i].getColour());
+                    }
+
+                    cs.setThreshold(groups[i].getPidThreshold(), true);
+                }
+
+                Vector seqs = new Vector();
+                int[] ids = groups[i].getSeq();
+
+                for (int s = 0; s < ids.length; s++)
+                {
+                    seqs.addElement((jalview.datamodel.SequenceI) seqids.elementAt(
+                            ids[s]));
+                }
+
+                jalview.datamodel.SequenceGroup sg = new jalview.datamodel.SequenceGroup(seqs,
+                        groups[i].getName(), cs, groups[i].getDisplayBoxes(),
+                        groups[i].getDisplayText(), groups[i].getColourText(),
+                        groups[i].getStart(), groups[i].getEnd());
+
+                sg.setOutlineColour(new java.awt.Color(
+                        groups[i].getOutlineColour()));
+
+                if (groups[i].getConsThreshold() != 0)
+                {
+                    jalview.analysis.Conservation c = new jalview.analysis.Conservation("All",
+                            ResidueProperties.propHash, 3, sg.sequences, 0,
+                            sg.getWidth() - 1);
+                    c.calculate();
+                    c.verdict(false, 25);
+                    sg.cs.setConservation(c);
+                }
+
+                al.addGroup(sg);
+            }
+        }
 
-     if(!fileName.endsWith(".xml"))
-              fileName = fileName+".xml";
 
-     JarEntry entry = new JarEntry(fileName);
-     jout.putNextEntry(entry);
+        af.setBounds(view.getXpos(), view.getYpos(), view.getWidth(),
+            view.getHeight());
+        af.viewport.setStartRes(view.getStartRes());
+        af.viewport.setStartSeq(view.getStartSeq());
+        af.viewport.showConservation = view.getShowConservation();
+        af.viewport.showQuality = view.getShowQuality();
+        af.viewport.showIdentity = view.getShowIdentity();
+        af.viewport.setAbovePIDThreshold(view.getPidSelected());
+        af.abovePIDThreshold.setSelected(view.getPidSelected());
+        af.viewport.setColourText(view.getShowColourText());
+        af.colourTextMenuItem.setSelected(view.getShowColourText());
+        af.viewport.setConservationSelected(view.getConservationSelected());
+        af.conservationMenuItem.setSelected(view.getConservationSelected());
 
-      object.marshal(out);
-      out.close();
-      fos.close();
-      jout.close();
-    }
-    catch (Exception ex)
-    {
-      ex.printStackTrace();
-    }
-  }
-
-  public static void LoadState(String file)
-  {
-    try{
-       JalviewState state = new JalviewState();
-       BufferedReader in = new BufferedReader(new FileReader(file));
-       state = (JalviewState)state.unmarshal(in);
-       Enumeration en = state.enumerateFile();
-       while( en.hasMoreElements() )
-       {
-         LoadJalviewAlign( en.nextElement().toString());
-       }
-    }
-    catch(Exception ex)
-    {   ex.printStackTrace(); return; }
-  }
-
-  public static void LoadJalviewAlign(String file)
-  {
-    JalviewModel object = new JalviewModel();
-    Vector seqids = new Vector();
-    try{
-      FileInputStream fis = new FileInputStream(file);
-      JarInputStream jin = new JarInputStream(fis);
-      jin.getNextJarEntry();
-      InputStreamReader in = new InputStreamReader(jin, "UTF-8");
-      object = (JalviewModel) object.unmarshal(in);
-    }
-    catch(Exception ex)
-    {   System.out.println(ex); return; }
-
-
-     SequenceSet vamsasSet = object.getVamsasModel().getSequenceSet(0);
-     Sequence [] vamsasSeq = vamsasSet.getSequence();
-
-     JalviewModelSequence jms = object.getJalviewModelSequence();
-     //////////////////////////////////
-     //LOAD SEQUENCES
-     jalview.datamodel.Sequence [] jseqs = new jalview.datamodel.Sequence[vamsasSeq.length];
-     JSeq [] JSEQ = object.getJalviewModelSequence().getJSeq();
-     for(int i=0; i<vamsasSeq.length; i++)
-     {
-       jseqs[i] = new jalview.datamodel.Sequence(vamsasSeq[i].getName(),
-                                                 vamsasSeq[i].getSequence());
-
-       jseqs[i].setStart( JSEQ[i].getStart());
-       jseqs[i].setEnd( JSEQ[i].getEnd());
-       jseqs[i].setColor( new java.awt.Color(JSEQ[i].getColour()) );
-       seqids.add(jseqs[i]);
-     }
-
-     /////////////////////////////////
-     jalview.datamodel.Alignment al = new jalview.datamodel.Alignment(jseqs);
-     /////////////////////////////////
-
-     //////////////////////////////////
-     //LOAD ANNOTATIONS
-     if(vamsasSet.getAnnotation()!=null)
-     {
-       Annotation[] an = vamsasSet.getAnnotation();
-       for (int i = 0; i < an.length; i++)
-       {
-         AnnotationElement[] ae = an[i].getAnnotationElement();
-         jalview.datamodel.Annotation anot[]
-             = new jalview.datamodel.Annotation[al.getWidth()];
-         for (int aa = 0; aa < ae.length; aa++)
-         {
-           anot[ae[aa].getPosition()]
-               = new jalview.datamodel.Annotation(
-                   ae[aa].getDisplayCharacter(),
-                   ae[aa].getDescription(),
-                   ae[aa].getSecondaryStructure().charAt(0),
-                   ae[aa].getValue()
-               );
-         }
-
-
-          jalview.datamodel.AlignmentAnnotation jaa = null;
-          if(an[i].getGraph())
-            jaa = new jalview.datamodel.AlignmentAnnotation(an[i].getLabel(),
-                                        an[i].getDescription(), anot,
-                                        0,0,1);
-          else
-            jaa = new jalview.datamodel.AlignmentAnnotation(an[i].getLabel(),
-                                        an[i].getDescription(), anot);
-          al.addAnnotation(jaa);
-       }
-     }
-
-      /////////////////////////////////
-      // LOAD VIEWPORT
-      Viewport[] views = jms.getViewport();
-      Viewport view = views[0]; // DEAL WITH MULTIPLE VIEWPORTS LATER
-
-      AlignFrame af = new AlignFrame(al);
-    //  af.changeColour() );
-
-    /////////////////////////
-    //LOAD GROUPS
-    if (jms.getJGroupCount()>0)
-    {
-      JGroup[] groups = jms.getJGroup();
-      for (int i = 0; i < groups.length; i++)
-      {
-       ColourSchemeI cs = ColourSchemeProperty.getColour(al, groups[i].getColour() );
-       if (cs instanceof ResidueColourScheme)
-         ( (ResidueColourScheme) cs).setThreshold(groups[i].getPidThreshold());
-       else if (cs instanceof ScoreColourScheme)
-      ( (ScoreColourScheme) cs).setThreshold(groups[i].getPidThreshold());
-
-        jalview.datamodel.SequenceGroup sg = new jalview.datamodel.SequenceGroup
-                (groups[i].getName(),
-                 cs,
-                 groups[i].getDisplayBoxes(),
-                 groups[i].getDisplayText(),
-                 groups[i].getColourText(),
-                 groups[i].getStart(),
-                 groups[i].getEnd()) ;
-        sg.setOutlineColour( new java.awt.Color(groups[i].getOutlineColour()));
-        int [] ids = groups[i].getSeq();
-        for(int s=0; s<ids.length; s++)
-            sg.addSequence( (jalview.datamodel.SequenceI)seqids.elementAt( ids[s] ));
-
-        if(groups[i].getConsThreshold()!=0)
+        af.viewport.setShowFullId(view.getShowFullId());
+        af.fullSeqId.setSelected(view.getShowFullId());
+
+        af.viewport.setFont(new java.awt.Font(view.getFontName(),
+                view.getFontStyle(), view.getFontSize()));
+        af.alignPanel.fontChanged();
+
+        af.viewport.setRenderGaps(view.getRenderGaps());
+        af.renderGapsMenuItem.setSelected(view.getRenderGaps());
+
+        af.viewport.setWrapAlignment(view.getWrapAlignment());
+        af.wrapMenuItem.setSelected(view.getWrapAlignment());
+
+        if (view.getWrapAlignment())
+        {
+            af.alignPanel.setWrapAlignment(view.getWrapAlignment());
+        }
+        else
         {
-          jalview.analysis.Conservation c = new jalview.analysis.Conservation("All",
-              ResidueProperties.propHash, 3,
-              sg.sequences, 0,
-              sg.getWidth() - 1);
-          c.calculate();
-          c.verdict(false, 25);
-          cs = new ConservationColourScheme(c, cs);
-          sg.cs = cs;
+            af.annotationPanelMenuItem.setState(view.getShowAnnotation());
+            af.viewport.setShowAnnotation(view.getShowAnnotation());
+            af.alignPanel.setAnnotationVisible(view.getShowAnnotation());
         }
 
+        af.viewport.setShowBoxes(view.getShowBoxes());
+        af.viewBoxesMenuItem.setSelected(view.getShowBoxes());
+        af.viewport.setShowText(view.getShowText());
+        af.viewTextMenuItem.setSelected(view.getShowText());
 
-        al.addGroup(sg);
-      }
-    }
+        ColourSchemeI cs = null;
 
+        if (view.getBgColour() != null)
+        {
+            if (view.getBgColour().startsWith("ucs"))
+            {
+                cs = GetUserColourScheme(jms, view.getBgColour());
+            }
+            else
+            {
+                cs = ColourSchemeProperty.getColour(al, view.getBgColour());
+            }
 
-    Desktop.addInternalFrame(af, view.getTitle(), AlignFrame.NEW_WINDOW_WIDTH, AlignFrame.NEW_WINDOW_HEIGHT);
-    af.setBounds(view.getXpos(), view.getYpos(), view.getWidth(), view.getHeight());
-    af.viewport.setStartRes( view.getStartRes() );
-    af.viewport.setStartSeq( view.getStartSeq() );
-    af.viewport.setAbovePIDThreshold( view.getPidSelected() );
-    af.viewport.setColourText( view.getShowColourText() );
-    af.viewport.setConservationSelected( view.getConservationSelected());
-    af.viewport.setShowFullId( view.getShowFullId());
-    af.viewport.setFont( new java.awt.Font( view.getFontName(),
-                                            view.getFontStyle(),
-                                            view.getFontSize()));
-    af.alignPanel.fontChanged();
-    af.viewport.setRenderGaps( view.getRenderGaps() );
+            if(cs!=null)
+            {
+              cs.setThreshold(view.getPidThreshold(), true);
+              cs.setConsensus(af.viewport.vconsensus);
+            }
+        }
 
-    af.viewport.setWrapAlignment( view.getWrapAlignment() );
-    if(view.getWrapAlignment())
-    {
-      af.alignPanel.setWrapAlignment(view.getWrapAlignment());
-    }
-    else
-    {
-      af.viewport.setShowAnnotation(view.getShowAnnotation());
-      af.alignPanel.setAnnotationVisible(view.getShowAnnotation());
-    }
+        af.setColourSelected(view.getBgColour());
+        af.viewport.setGlobalColourScheme(cs);
+        af.viewport.setColourAppliesToAllGroups(false);
+        af.changeColour(cs);
+        if (view.getConservationSelected() && cs!=null)
+        {
+          cs.setConservationInc(view.getConsThreshold());
+        }
+
+        af.viewport.setColourAppliesToAllGroups(true);
 
-    af.viewport.setShowBoxes( view.getShowBoxes() );
-    af.viewport.setShowText( view.getShowText() );
+        if (view.getShowSequenceFeatures())
+        {
+            af.viewport.showSequenceFeatures = true;
+            af.sequenceFeatures.setSelected(true);
+            new SequenceFeatureFetcher(al, af.alignPanel);
+            al.featuresAdded = true;
+        }
 
-    af.viewport.setGlobalColourScheme( ColourSchemeProperty.getColour(al, view.getBgColour()));
-    af.viewport.setColourAppliesToAllGroups(false);
-    af.changeColour( af.viewport.getGlobalColourScheme() );
-    af.viewport.setColourAppliesToAllGroups(true);
+        Desktop.addInternalFrame(af, view.getTitle(),
+             view.getWidth(), view.getHeight());
 
-    ColourSchemeI cs = af.viewport.getGlobalColourScheme();
-    if(cs instanceof ConservationColourScheme)
-      ( (ConservationColourScheme) cs).inc = view.getConsThreshold();
+        //LOAD TREES
+        ///////////////////////////////////////
+        if (jms.getTreeCount() > 0)
+        {
+            try
+            {
+                for (int t = 0; t < jms.getTreeCount(); t++)
+                {
 
-    if (cs instanceof ResidueColourScheme)
-      ( (ResidueColourScheme) cs).setThreshold(view.getPidThreshold());
-    else if (cs instanceof ScoreColourScheme)
-      ( (ScoreColourScheme) cs).setThreshold(view.getPidThreshold());
+                    Tree tree = jms.getTree(t);
 
-    //LOAD TREES - THIS RELEASE ONLY ALLOWS 1 TREE
-    if(object.getVamsasModel().getTreeCount()>0)
-    {
-      try{
-        af.ShowNewickTree(new jalview.io.NewickFile(
-            (String)object.getVamsasModel().getTree(0)), "Tree");
-      }catch(Exception ex){ex.printStackTrace();}
+                    TreePanel tp = af.ShowNewickTree(new jalview.io.NewickFile(
+                                tree.getNewick()), tree.getTitle(),
+                                tree.getWidth(), tree.getHeight(),
+                                tree.getXpos(), tree.getYpos());
 
-    }
+                    tp.fitToWindow.setState(tree.getFitToWindow());
+                    tp.fitToWindow_actionPerformed(null);
 
-  }
-}
+                    if(tree.getFontName()!=null)
+                      tp.setTreeFont(new java.awt.Font(tree.getFontName(),
+                                          tree.getFontStyle(),
+                                          tree.getFontSize()));
+                    else
+                      tp.setTreeFont(new java.awt.Font(view.getFontName(),
+                                            view.getFontStyle(),
+                                            tree.getFontSize()));
+
+
+                    tp.placeholdersMenu.setState(tree.getMarkUnlinked());
+                    tp.placeholdersMenu_actionPerformed(null);
+                    tp.bootstrapMenu.setState(tree.getShowBootstrap());
+                    tp.bootstrapMenu_actionPerformed(null);
+                    tp.distanceMenu.setState(tree.getShowDistances());
+                    tp.distanceMenu_actionPerformed(null);
+                    tp.treeCanvas.threshold = tree.getThreshold();
 
+                    if (tree.getCurrentTree())
+                      af.viewport.setCurrentTree(tp.getTree());
+                }
+
+            }
+            catch (Exception ex)
+            {
+                ex.printStackTrace();
+            }
 
+        }
 
+        return af;
+    }
+}