1 /*******************************************************************************
2 * Jalview - A Sequence Alignment Editor and Viewer (Version 2.7)
3 * Copyright (C) 2011 J Procter, AM Waterhouse, J Engelhardt, LM Lui, G Barton, M Clamp, S Searle
5 * This file is part of Jalview.
7 * Jalview is free software: you can redistribute it and/or
8 * modify it under the terms of the GNU General Public License
9 * as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version.
11 * Jalview is distributed in the hope that it will be useful, but
12 * WITHOUT ANY WARRANTY; without even the implied warranty
13 * of MERCHANTABILITY or FITNESS FOR A PARTICULAR
14 * PURPOSE. See the GNU General Public License for more details.
16 * You should have received a copy of the GNU General Public License along with Jalview. If not, see <http://www.gnu.org/licenses/>.
17 *******************************************************************************/
21 package jalview.ext.jmol;
23 import jalview.api.FeatureRenderer;
24 import jalview.api.SequenceRenderer;
25 import jalview.datamodel.AlignmentI;
26 import jalview.datamodel.SequenceI;
27 import jalview.structure.StructureMapping;
28 import jalview.structure.StructureMappingcommandSet;
29 import jalview.structure.StructureSelectionManager;
30 import jalview.util.Comparison;
32 import java.awt.Color;
33 import java.util.ArrayList;
36 * Routines for generating Jmol commands for Jalview/Jmol binding another
42 public class JmolCommands
46 * Jmol utility which constructs the commands to colour chains by the given
49 * @returns Object[] { Object[] { <model being coloured>,
52 public static StructureMappingcommandSet[] getColourBySequenceCommand(
53 StructureSelectionManager ssm, String[] files,
54 SequenceI[][] sequence, SequenceRenderer sr, FeatureRenderer fr,
58 ArrayList<StructureMappingcommandSet> cset = new ArrayList<StructureMappingcommandSet>();
60 for (int pdbfnum = 0; pdbfnum < files.length; pdbfnum++)
62 StructureMapping[] mapping = ssm.getMapping(files[pdbfnum]);
63 StringBuffer command = new StringBuffer();
64 StructureMappingcommandSet smc;
65 ArrayList<String> str = new ArrayList<String>();
67 if (mapping == null || mapping.length < 1)
71 for (int s = 0; s < sequence[pdbfnum].length; s++)
73 for (int sp, m = 0; m < mapping.length; m++)
75 if (mapping[m].getSequence() == sequence[pdbfnum][s]
76 && (sp = alignment.findIndex(sequence[pdbfnum][s])) > -1)
78 SequenceI asp = alignment.getSequenceAt(sp);
79 for (int r = 0; r < asp.getLength(); r++)
81 // no mapping to gaps in sequence
82 if (jalview.util.Comparison.isGap(asp.getCharAt(r)))
86 int pos = mapping[m].getPDBResNum(asp.findPosition(r));
88 if (pos < 1 || pos == lastPos)
93 Color col = sr.getResidueBoxColour(sequence[pdbfnum][s], r);
96 col = fr.findFeatureColour(col, sequence[pdbfnum][s], r);
97 String newSelcom = (mapping[m].getChain() != " " ? ":"
98 + mapping[m].getChain() : "")
107 + col.getBlue() + "]";
108 if (command.length() > newSelcom.length()
109 && command.substring(
110 command.length() - newSelcom.length())
113 command = JmolCommands.condenseCommand(command, pos);
116 // TODO: deal with case when buffer is too large for Jmol to parse
117 // - execute command and flush
120 if (command.length() > 51200)
123 str.add(command.toString());
124 command.setLength(0);
126 command.append("select " + pos);
127 command.append(newSelcom);
135 str.add(command.toString());
136 command.setLength(0);
138 // Finally, add the command set ready to be returned.
139 cset.add(new StructureMappingcommandSet(JmolCommands.class,
140 files[pdbfnum], str.toArray(new String[str.size()])));
143 return cset.toArray(new StructureMappingcommandSet[cset.size()]);
146 public static StringBuffer condenseCommand(StringBuffer command, int pos)
149 // work back to last 'select'
150 int p = command.length(), q = p;
159 } while ((q = command.indexOf("select", p)) == -1 && p > 0);
161 StringBuffer sb = new StringBuffer(command.substring(0, q + 7));
163 command = command.delete(0, q + 7);
167 if (command.indexOf("-") > -1)
169 start = command.substring(0, command.indexOf("-"));
173 start = command.substring(0, command.indexOf(":"));
176 sb.append(start + "-" + pos + command.substring(command.indexOf(":")));