added colSel as a parameter formatAdaptor print function to allow editing of annotati...
[jalview.git] / src / jalview / io / FormatAdapter.java
index 0c8554a..24c9109 100755 (executable)
-/*\r
-* Jalview - A Sequence Alignment Editor and Viewer\r
-* Copyright (C) 2005 AM Waterhouse, J Procter, G Barton, M Clamp, S Searle\r
-*\r
-* This program is free software; you can redistribute it and/or\r
-* modify it under the terms of the GNU General Public License\r
-* as published by the Free Software Foundation; either version 2\r
-* of the License, or (at your option) any later version.\r
-*\r
-* This program is distributed in the hope that it will be useful,\r
-* but WITHOUT ANY WARRANTY; without even the implied warranty of\r
-* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the\r
-* GNU General Public License for more details.\r
-*\r
-* You should have received a copy of the GNU General Public License\r
-* along with this program; if not, write to the Free Software\r
-* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA\r
-*/\r
-package jalview.io;\r
-\r
-import jalview.datamodel.*;\r
-\r
-import java.util.Vector;\r
-\r
-\r
-/**\r
- * DOCUMENT ME!\r
- *\r
- * @author $author$\r
- * @version $Revision$\r
- */\r
-public class FormatAdapter extends AppletFormatAdapter\r
-{\r
-    /**\r
-     * DOCUMENT ME!\r
-     *\r
-     * @param format DOCUMENT ME!\r
-     * @param seqs DOCUMENT ME!\r
-     *\r
-     * @return DOCUMENT ME!\r
-     */\r
-    public String formatSequences(String format, Vector seqs)\r
-    {\r
-        SequenceI[] s = new SequenceI[seqs.size()];\r
-\r
-        for (int i = 0; i < seqs.size(); i++)\r
-            s[i] = (SequenceI) seqs.elementAt(i);\r
-\r
-        try\r
-        {\r
-            AlignFile afile = null;\r
-\r
-            if (format.equalsIgnoreCase("FASTA"))\r
-            {\r
-                afile = new FastaFile();\r
-                afile.addJVSuffix(\r
-                    jalview.bin.Cache.getDefault("FASTA_JVSUFFIX", true));\r
-            }\r
-            else if (format.equalsIgnoreCase("MSF"))\r
-            {\r
-              afile = new MSFfile();\r
-              afile.addJVSuffix(\r
-                  jalview.bin.Cache.getDefault("MSF_JVSUFFIX", true));\r
-            }\r
-            else if (format.equalsIgnoreCase("PileUp"))\r
-            {\r
-                afile = new PileUpfile();\r
-                afile.addJVSuffix(\r
-                    jalview.bin.Cache.getDefault("PILEUP_JVSUFFIX", true));\r
-            }\r
-            else if (format.equalsIgnoreCase("CLUSTAL"))\r
-            {\r
-                afile = new ClustalFile();\r
-                afile.addJVSuffix(\r
-                    jalview.bin.Cache.getDefault("CLUSTAL_JVSUFFIX", true));\r
-            }\r
-            else if (format.equalsIgnoreCase("BLC"))\r
-            {\r
-                afile = new BLCFile();\r
-                afile.addJVSuffix(\r
-                    jalview.bin.Cache.getDefault("BLC_JVSUFFIX", true));\r
-            }\r
-            else if (format.equalsIgnoreCase("PIR"))\r
-            {\r
-                afile = new PIRFile();\r
-                afile.addJVSuffix(\r
-                    jalview.bin.Cache.getDefault("PIR_JVSUFFIX", true));\r
-            }\r
-            else if (format.equalsIgnoreCase("PFAM"))\r
-            {\r
-                afile = new PfamFile();\r
-                afile.addJVSuffix(\r
-                    jalview.bin.Cache.getDefault("PFAM_JVSUFFIX", true));\r
-            }\r
-\r
-            afile.setSeqs(s);\r
-\r
-            return afile.print();\r
-        }\r
-        catch (Exception e)\r
-        {\r
-            System.err.println("Failed to write alignment as a '" + format +\r
-                "' file\n");\r
-            e.printStackTrace();\r
-        }\r
-\r
-        return null;\r
-    }\r
-}\r
+/*
+ * Jalview - A Sequence Alignment Editor and Viewer
+ * Copyright (C) 2007 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.io;
+
+import jalview.datamodel.*;
+
+/**
+ * DOCUMENT ME!
+ *
+ * @author $author$
+ * @version $Revision$
+ */
+public class FormatAdapter
+    extends AppletFormatAdapter
+{
+  
+  public String formatSequences(String format,
+                                SequenceI[] seqs,
+                                String[] omitHiddenColumns)
+  {
+
+    return formatSequences(format, replaceStrings(seqs, omitHiddenColumns));
+  }
+
+  /**
+   * create sequences with each seuqence string replaced with the one given in omitHiddenCOlumns
+   * @param seqs
+   * @param omitHiddenColumns
+   * @return new sequences
+   */
+  public SequenceI[] replaceStrings(SequenceI[] seqs, String[] omitHiddenColumns)
+  {
+    if (omitHiddenColumns != null)
+    {
+      SequenceI[] tmp = new SequenceI[seqs.length];
+      for (int i = 0; i < seqs.length; i++)
+      {
+        tmp[i] = new Sequence(
+            seqs[i].getName(), omitHiddenColumns[i],
+            seqs[i].getStart(), seqs[i].getEnd());
+        tmp[i].setDescription(seqs[i].getDescription());
+      }
+      seqs = tmp;
+    }
+    return seqs;
+  }
+
+  /**
+   * Format a vector of sequences as a flat alignment file.
+   *
+   * @param format Format string as givien in the AppletFormatAdaptor list (exact match to name of class implementing file io for that format)
+   * @param seqs vector of sequences to write
+   *
+   * @return String containing sequences in desired format
+   */
+  public String formatSequences(String format,
+                                SequenceI[] seqs)
+  {
+
+    try
+    {
+      AlignFile afile = null;
+
+      if (format.equalsIgnoreCase("FASTA"))
+      {
+        afile = new FastaFile();
+        afile.addJVSuffix(
+            jalview.bin.Cache.getDefault("FASTA_JVSUFFIX", true));
+      }
+      else if (format.equalsIgnoreCase("MSF"))
+      {
+        afile = new MSFfile();
+        afile.addJVSuffix(
+            jalview.bin.Cache.getDefault("MSF_JVSUFFIX", true));
+      }
+      else if (format.equalsIgnoreCase("PileUp"))
+      {
+        afile = new PileUpfile();
+        afile.addJVSuffix(
+            jalview.bin.Cache.getDefault("PILEUP_JVSUFFIX", true));
+      }
+      else if (format.equalsIgnoreCase("CLUSTAL"))
+      {
+        afile = new ClustalFile();
+        afile.addJVSuffix(
+            jalview.bin.Cache.getDefault("CLUSTAL_JVSUFFIX", true));
+      }
+      else if (format.equalsIgnoreCase("BLC"))
+      {
+        afile = new BLCFile();
+        afile.addJVSuffix(
+            jalview.bin.Cache.getDefault("BLC_JVSUFFIX", true));
+      }
+      else if (format.equalsIgnoreCase("PIR"))
+      {
+        afile = new PIRFile();
+        afile.addJVSuffix(
+            jalview.bin.Cache.getDefault("PIR_JVSUFFIX", true));
+      }
+      else if (format.equalsIgnoreCase("PFAM"))
+      {
+        afile = new PfamFile();
+        afile.addJVSuffix(
+            jalview.bin.Cache.getDefault("PFAM_JVSUFFIX", true));
+      }
+
+      afile.setSeqs(seqs);
+
+      return afile.print();
+    }
+    catch (Exception e)
+    {
+      System.err.println("Failed to write alignment as a '" + format +
+                         "' file\n");
+      e.printStackTrace();
+    }
+
+    return null;
+  }
+  public boolean getCacheSuffixDefault(String format)
+  {
+    if (isValidFormat(format))
+      return jalview.bin.Cache.getDefault(format.toUpperCase()+"_JVSUFFIX", true);
+    return false;
+  }
+  public String formatSequences(String format, AlignmentI alignment, String[] omitHidden, ColumnSelection colSel)
+  {
+    return formatSequences(format, alignment, omitHidden, getCacheSuffixDefault(format), colSel);
+  }
+    /**
+   * hack function to replace seuqences with visible sequence strings before generating a
+   * string of the alignment in the given format.
+   * @param format
+     * @param alignment
+     * @param omitHidden sequence strings to write out in order of sequences in alignment
+     * @param colSel defines hidden columns that are edited out of annotation 
+   * @return string representation of the alignment formatted as format
+   */
+  public String formatSequences(String format, AlignmentI alignment, String[] omitHidden, boolean suffix, ColumnSelection colSel)
+  {
+    AlignFile afile = null;
+    if (omitHidden!=null)
+    {
+      // 
+      Alignment alv = new Alignment(replaceStrings(alignment.getSequencesArray(), omitHidden));
+      AlignmentAnnotation[] ala = alignment.getAlignmentAnnotation();
+      for (int i=0; i<ala.length; i++)
+      {
+        AlignmentAnnotation na = new AlignmentAnnotation(ala[i]);
+        colSel.makeVisibleAnnotation(na);
+        alv.addAnnotation(na);
+      }
+      return this.formatSequences(format, alv, suffix);
+    }
+    return this.formatSequences(format, alignment, suffix);
+  }
+}