src/jalview/io/ClustalFile.java

   1 /*\r
   2  * Jalview - A Sequence Alignment Editor and Viewer\r
   3  * Copyright (C) 2005 AM Waterhouse, J Procter, G Barton, M Clamp, S Searle\r
   4  *\r
   5  * This program is free software; you can redistribute it and/or\r
   6  * modify it under the terms of the GNU General Public License\r
   7  * as published by the Free Software Foundation; either version 2\r
   8  * of the License, or (at your option) any later version.\r
   9  *\r
  10  * This program is distributed in the hope that it will be useful,\r
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of\r
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the\r
  13  * GNU General Public License for more details.\r
  14  *\r
  15  * You should have received a copy of the GNU General Public License\r
  16  * along with this program; if not, write to the Free Software\r
  17  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA\r
  18  */\r
  19 package jalview.io;\r
  20 \r
  21 import java.io.*;\r
  22 import java.util.*;\r
  23 \r
  24 import jalview.datamodel.*;\r
  25 import jalview.util.*;\r
  26 \r
  27 public class ClustalFile\r
  28     extends AlignFile\r
  29 {\r
  30 \r
  31   public ClustalFile()\r
  32   {\r
  33   }\r
  34 \r
  35   public ClustalFile(String inStr)\r
  36   {\r
  37     super(inStr);\r
  38   }\r
  39 \r
  40   public ClustalFile(String inFile, String type)\r
  41       throws IOException\r
  42   {\r
  43     super(inFile, type);\r
  44   }\r
  45 \r
  46   public void initData()\r
  47   {\r
  48     super.initData();\r
  49   }\r
  50 \r
  51   public void parse()\r
  52   {\r
  53     int i = 0;\r
  54     boolean flag = false;\r
  55 \r
  56     Vector headers = new Vector();\r
  57     Hashtable seqhash = new Hashtable();\r
  58     StringBuffer tempseq;\r
  59     String line, id;\r
  60     StringTokenizer str;\r
  61 \r
  62     try\r
  63     {\r
  64       while ( (line = nextLine()) != null)\r
  65       {\r
  66         if (line.indexOf(" ") != 0)\r
  67         {\r
  68           str = new StringTokenizer(line, " ");\r
  69 \r
  70           if (str.hasMoreTokens())\r
  71           {\r
  72             id = str.nextToken();\r
  73 \r
  74             if (id.equalsIgnoreCase("CLUSTAL"))\r
  75             {\r
  76               flag = true;\r
  77             }\r
  78             else\r
  79             {\r
  80               if (flag)\r
  81               {\r
  82                 if (seqhash.containsKey(id))\r
  83                 {\r
  84                   tempseq = (StringBuffer) seqhash.get(id);\r
  85                 }\r
  86                 else\r
  87                 {\r
  88                   tempseq = new StringBuffer();\r
  89                   seqhash.put(id, tempseq);\r
  90                 }\r
  91 \r
  92                 if (! (headers.contains(id)))\r
  93                 {\r
  94                   headers.addElement(id);\r
  95                 }\r
  96 \r
  97                 if (str.hasMoreTokens())\r
  98                 {\r
  99                   tempseq.append(str.nextToken());\r
 100                 }\r
 101               }\r
 102             }\r
 103           }\r
 104           else\r
 105             flag = true;\r
 106         }\r
 107       }\r
 108     }\r
 109     catch (IOException e)\r
 110     {\r
 111       System.err.println("Exception parsing clustal file " + e);\r
 112       e.printStackTrace();\r
 113     }\r
 114 \r
 115     if (flag)\r
 116     {\r
 117       this.noSeqs = headers.size();\r
 118 \r
 119       //Add sequences to the hash\r
 120       for (i = 0; i < headers.size(); i++)\r
 121       {\r
 122         if (seqhash.get(headers.elementAt(i)) != null)\r
 123         {\r
 124           if (maxLength < seqhash.get(headers.elementAt(i)).toString()\r
 125               .length())\r
 126           {\r
 127             maxLength = seqhash.get(headers.elementAt(i)).toString()\r
 128                 .length();\r
 129           }\r
 130 \r
 131           Sequence newSeq = parseId(headers.elementAt(i).toString());\r
 132           newSeq.setSequence( seqhash.get(headers.elementAt(i).toString()).toString() );\r
 133 \r
 134           seqs.addElement(newSeq);\r
 135         }\r
 136         else\r
 137         {\r
 138           System.err.println(\r
 139               "Clustal File Reader: Can't find sequence for " +\r
 140               headers.elementAt(i));\r
 141         }\r
 142       }\r
 143     }\r
 144   }\r
 145 \r
 146   public String print()\r
 147   {\r
 148     return print(getSeqsAsArray());\r
 149   }\r
 150 \r
 151   public String print(SequenceI[] s)\r
 152   {\r
 153     StringBuffer out = new StringBuffer("CLUSTAL\n\n");\r
 154 \r
 155     int max = 0;\r
 156     int maxid = 0;\r
 157 \r
 158     int i = 0;\r
 159 \r
 160     while ( (i < s.length) && (s[i] != null))\r
 161     {\r
 162       String tmp = printId(s[i]);\r
 163 \r
 164       if (s[i].getSequence().length() > max)\r
 165       {\r
 166         max = s[i].getSequence().length();\r
 167       }\r
 168 \r
 169       if (tmp.length() > maxid)\r
 170       {\r
 171         maxid = tmp.length();\r
 172       }\r
 173 \r
 174       i++;\r
 175     }\r
 176 \r
 177     if (maxid < 15)\r
 178     {\r
 179       maxid = 15;\r
 180     }\r
 181 \r
 182     maxid++;\r
 183 \r
 184     int len = 60;\r
 185     int nochunks = (max / len) + 1;\r
 186 \r
 187     for (i = 0; i < nochunks; i++)\r
 188     {\r
 189       int j = 0;\r
 190 \r
 191       while ( (j < s.length) && (s[j] != null))\r
 192       {\r
 193         out.append(new Format("%-" + maxid + "s").form( printId(s[j]) + " "));\r
 194 \r
 195         int start = i * len;\r
 196         int end = start + len;\r
 197 \r
 198         if ( (end < s[j].getSequence().length()) &&\r
 199             (start < s[j].getSequence().length()))\r
 200         {\r
 201           out.append(s[j].getSequence().substring(start, end));\r
 202         }\r
 203         else\r
 204         {\r
 205           if (start < s[j].getSequence().length())\r
 206           {\r
 207             out.append(s[j].getSequence().substring(start));\r
 208           }\r
 209         }\r
 210 \r
 211         out.append("\n");\r
 212         j++;\r
 213       }\r
 214 \r
 215       out.append("\n");\r
 216     }\r
 217 \r
 218     return out.toString();\r
 219   }\r
 220 }\r