2 * Jalview - A Sequence Alignment Editor and Viewer ($$Version-Rel$$)
3 * Copyright (C) $$Year-Rel$$ The Jalview Authors
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
10 * of the License, or (at your option) any later version.
12 * Jalview is distributed in the hope that it will be useful, but
13 * WITHOUT ANY WARRANTY; without even the implied warranty
14 * of MERCHANTABILITY or FITNESS FOR A PARTICULAR
15 * PURPOSE. See the GNU General Public License for more details.
17 * You should have received a copy of the GNU General Public License
18 * along with Jalview. If not, see <http://www.gnu.org/licenses/>.
19 * The Jalview Authors are detailed in the 'AUTHORS' file.
21 package jalview.datamodel;
23 import jalview.analysis.AAFrequency;
24 import jalview.analysis.Conservation;
25 import jalview.renderer.ResidueShader;
26 import jalview.renderer.ResidueShaderI;
27 import jalview.schemes.ColourSchemeI;
29 import java.awt.Color;
30 import java.util.ArrayList;
31 import java.util.List;
35 * Collects a set contiguous ranges on a set of sequences
40 public class SequenceGroup implements AnnotatedCollectionI
46 Conservation conserve;
48 boolean displayBoxes = true;
50 boolean displayText = true;
52 boolean colourText = false;
55 * after Olivier's non-conserved only character display
57 boolean showNonconserved = false;
62 private List<SequenceI> sequences = new ArrayList<SequenceI>();
65 * representative sequence for this group (if any)
67 private SequenceI seqrep = null;
72 * Colourscheme applied to group if any
74 public ResidueShaderI cs;
76 // start column (base 0)
79 // end column (base 0)
82 public Color outlineColour = Color.black;
84 public Color idColour = null;
86 public int thresholdTextColour = 0;
88 public Color textColour = Color.black;
90 public Color textColour2 = Color.white;
93 * consensus calculation property
95 private boolean ignoreGapsInConsensus = true;
98 * consensus calculation property
100 private boolean showSequenceLogo = false;
103 * flag indicating if logo should be rendered normalised
105 private boolean normaliseSequenceLogo;
108 * @return the includeAllConsSymbols
110 public boolean isShowSequenceLogo()
112 return showSequenceLogo;
116 * Creates a new SequenceGroup object.
118 public SequenceGroup()
120 groupName = "JGroup:" + this.hashCode();
121 cs = new ResidueShader();
125 * Creates a new SequenceGroup object.
130 * @param displayBoxes
134 * first column of group
136 * last column of group
138 public SequenceGroup(List<SequenceI> sequences, String groupName,
139 ColourSchemeI scheme, boolean displayBoxes, boolean displayText,
140 boolean colourText, int start, int end)
143 this.sequences = sequences;
144 this.groupName = groupName;
145 this.displayBoxes = displayBoxes;
146 this.displayText = displayText;
147 this.colourText = colourText;
148 this.cs = new ResidueShader(scheme);
151 recalcConservation();
159 public SequenceGroup(SequenceGroup seqsel)
164 sequences = new ArrayList<SequenceI>();
165 sequences.addAll(seqsel.sequences);
166 if (seqsel.groupName != null)
168 groupName = new String(seqsel.groupName);
170 displayBoxes = seqsel.displayBoxes;
171 displayText = seqsel.displayText;
172 colourText = seqsel.colourText;
173 startRes = seqsel.startRes;
174 endRes = seqsel.endRes;
176 if (seqsel.description != null)
178 description = new String(seqsel.description);
180 hidecols = seqsel.hidecols;
181 hidereps = seqsel.hidereps;
182 idColour = seqsel.idColour;
183 outlineColour = seqsel.outlineColour;
184 seqrep = seqsel.seqrep;
185 textColour = seqsel.textColour;
186 textColour2 = seqsel.textColour2;
187 thresholdTextColour = seqsel.thresholdTextColour;
188 width = seqsel.width;
189 ignoreGapsInConsensus = seqsel.ignoreGapsInConsensus;
190 if (seqsel.conserve != null)
192 recalcConservation(); // safer than
193 // aaFrequency = (Vector) seqsel.aaFrequency.clone(); // ??
198 public SequenceI[] getSelectionAsNewSequences(AlignmentI align)
200 int iSize = sequences.size();
201 SequenceI[] seqs = new SequenceI[iSize];
202 SequenceI[] inorder = getSequencesInOrder(align);
204 for (int i = 0, ipos = 0; i < inorder.length; i++)
206 SequenceI seq = inorder[i];
208 seqs[ipos] = seq.getSubSequence(startRes, endRes + 1);
209 if (seqs[ipos] != null)
211 seqs[ipos].setDescription(seq.getDescription());
212 seqs[ipos].setDBRefs(seq.getDBRefs());
213 seqs[ipos].setSequenceFeatures(seq.getSequenceFeatures());
214 if (seq.getDatasetSequence() != null)
216 seqs[ipos].setDatasetSequence(seq.getDatasetSequence());
219 if (seq.getAnnotation() != null)
221 AlignmentAnnotation[] alann = align.getAlignmentAnnotation();
222 // Only copy annotation that is either a score or referenced by the
223 // alignment's annotation vector
224 for (int a = 0; a < seq.getAnnotation().length; a++)
226 AlignmentAnnotation tocopy = seq.getAnnotation()[a];
229 boolean found = false;
230 for (int pos = 0; pos < alann.length; pos++)
232 if (alann[pos] == tocopy)
243 AlignmentAnnotation newannot = new AlignmentAnnotation(
244 seq.getAnnotation()[a]);
245 newannot.restrict(startRes, endRes);
246 newannot.setSequenceRef(seqs[ipos]);
247 newannot.adjustForAlignment();
248 seqs[ipos].addAlignmentAnnotation(newannot);
258 if (iSize != inorder.length)
260 SequenceI[] nseqs = new SequenceI[iSize];
261 System.arraycopy(seqs, 0, nseqs, 0, iSize);
269 * If sequence ends in gaps, the end residue can be correctly calculated here
275 public int findEndRes(SequenceI seq)
280 for (int j = 0; j < endRes + 1 && j < seq.getLength(); j++)
282 ch = seq.getCharAt(j);
283 if (!jalview.util.Comparison.isGap((ch)))
291 eres += seq.getStart() - 1;
298 public List<SequenceI> getSequences()
304 public List<SequenceI> getSequences(
305 Map<SequenceI, SequenceCollectionI> hiddenReps)
307 if (hiddenReps == null)
309 // TODO: need a synchronizedCollection here ?
314 List<SequenceI> allSequences = new ArrayList<SequenceI>();
315 for (SequenceI seq : sequences)
317 allSequences.add(seq);
318 if (hiddenReps.containsKey(seq))
320 SequenceCollectionI hsg = hiddenReps.get(seq);
321 for (SequenceI seq2 : hsg.getSequences())
323 if (seq2 != seq && !allSequences.contains(seq2))
325 allSequences.add(seq2);
335 public SequenceI[] getSequencesAsArray(
336 Map<SequenceI, SequenceCollectionI> map)
338 List<SequenceI> tmp = getSequences(map);
343 return tmp.toArray(new SequenceI[tmp.size()]);
352 * @return DOCUMENT ME!
354 public boolean adjustForRemoveLeft(int col)
356 // return value is true if the group still exists
359 startRes = startRes - col;
364 endRes = endRes - col;
366 if (startRes > endRes)
373 // must delete this group!!
386 * @return DOCUMENT ME!
388 public boolean adjustForRemoveRight(int col)
407 * @return DOCUMENT ME!
409 public String getName()
414 public String getDescription()
425 public void setName(String name)
428 // TODO: URGENT: update dependent objects (annotation row)
431 public void setDescription(String desc)
439 * @return DOCUMENT ME!
441 public Conservation getConservation()
452 public void setConservation(Conservation c)
458 * Add s to this sequence group. If aligment sequence is already contained in
459 * group, it will not be added again, but recalculation may happen if the flag
463 * alignment sequence to be added
465 * true means Group's conservation should be recalculated
467 public void addSequence(SequenceI s, boolean recalc)
469 synchronized (sequences)
471 if (s != null && !sequences.contains(s))
478 recalcConservation();
484 * Max Gaps Threshold (percent) for performing a conservation calculation
486 private int consPercGaps = 25;
489 * @return Max Gaps Threshold for performing a conservation calculation
491 public int getConsPercGaps()
497 * set Max Gaps Threshold (percent) for performing a conservation calculation
499 * @param consPercGaps
501 public void setConsPercGaps(int consPercGaps)
503 this.consPercGaps = consPercGaps;
507 * calculate residue conservation and colourschemes for group - but only if
508 * necessary. returns true if the calculation resulted in a visible change to
511 public boolean recalcConservation()
513 return recalcConservation(false);
517 * calculate residue conservation for group - but only if necessary. returns
518 * true if the calculation resulted in a visible change to group
521 * when set, colourschemes for this group are not refreshed after
524 public boolean recalcConservation(boolean defer)
526 if (cs == null && consensus == null && conservation == null)
530 // TODO: try harder to detect changes in state in order to minimise
531 // recalculation effort
535 ProfilesI cnsns = AAFrequency.calculate(sequences, startRes,
536 endRes + 1, showSequenceLogo);
537 if (consensus != null)
539 _updateConsensusRow(cnsns, sequences.size());
544 cs.setConsensus(cnsns);
548 if ((conservation != null)
549 || (cs != null && cs.conservationApplied()))
551 Conservation c = new Conservation(groupName, sequences, startRes,
554 c.verdict(false, consPercGaps);
555 if (conservation != null)
557 _updateConservationRow(c);
561 if (cs.conservationApplied())
563 cs.setConservation(c);
566 // eager update - will cause a refresh of overview regardless
569 if (cs != null && !defer)
571 // TODO: JAL-2034 should cs.alignmentChanged modify return state
572 cs.alignmentChanged(context != null ? context : this, null);
579 } catch (java.lang.OutOfMemoryError err)
582 System.out.println("Out of memory loading groups: " + err);
587 private void _updateConservationRow(Conservation c)
589 if (conservation == null)
594 conservation.label = "Conservation for " + getName();
595 conservation.description = "Conservation for group " + getName()
596 + " less than " + consPercGaps + "% gaps";
597 // preserve width if already set
598 int aWidth = (conservation.annotations != null) ? (endRes < conservation.annotations.length ? conservation.annotations.length
601 conservation.annotations = null;
602 conservation.annotations = new Annotation[aWidth]; // should be alignment
604 c.completeAnnotations(conservation, null, startRes, endRes + 1);
607 public ProfilesI consensusData = null;
609 private void _updateConsensusRow(ProfilesI cnsns, long nseq)
611 if (consensus == null)
615 consensus.label = "Consensus for " + getName();
616 consensus.description = "Percent Identity";
617 consensusData = cnsns;
618 // preserve width if already set
619 int aWidth = (consensus.annotations != null) ? (endRes < consensus.annotations.length ? consensus.annotations.length
622 consensus.annotations = null;
623 consensus.annotations = new Annotation[aWidth]; // should be alignment width
625 AAFrequency.completeConsensus(consensus, cnsns, startRes, endRes + 1,
626 ignoreGapsInConsensus, showSequenceLogo, nseq); // TODO: setting
629 // ignoreGapsInConsensusCalculation);
634 * sequence to either add or remove from group
636 * flag passed to delete/addSequence to indicate if group properties
637 * should be recalculated
639 public void addOrRemove(SequenceI s, boolean recalc)
641 synchronized (sequences)
643 if (sequences.contains(s))
645 deleteSequence(s, recalc);
649 addSequence(s, recalc);
660 * true means recalculate conservation
662 public void deleteSequence(SequenceI s, boolean recalc)
664 synchronized (sequences)
670 recalcConservation();
678 * @return the first column selected by this group. Runs from 0<=i<N_cols
681 public int getStartRes()
688 * @return the groups last selected column. Runs from 0<=i<N_cols
691 public int getEndRes()
697 * Set the first column selected by this group. Runs from 0<=i<N_cols
701 public void setStartRes(int i)
707 * Set the groups last selected column. Runs from 0<=i<N_cols
711 public void setEndRes(int i)
717 * @return number of sequences in group
721 return sequences.size();
726 * @return the ith sequence
728 public SequenceI getSequenceAt(int i)
730 return sequences.get(i);
737 public void setColourText(boolean state)
745 * @return DOCUMENT ME!
747 public boolean getColourText()
758 public void setDisplayText(boolean state)
766 * @return DOCUMENT ME!
768 public boolean getDisplayText()
779 public void setDisplayBoxes(boolean state)
781 displayBoxes = state;
787 * @return DOCUMENT ME!
789 public boolean getDisplayBoxes()
795 * computes the width of current set of sequences and returns it
797 * @return DOCUMENT ME!
800 public int getWidth()
802 synchronized (sequences)
804 // MC This needs to get reset when characters are inserted and deleted
805 boolean first = true;
806 for (SequenceI seq : sequences)
808 if (first || seq.getLength() > width)
810 width = seq.getLength();
824 public void setOutlineColour(Color c)
832 * @return DOCUMENT ME!
834 public Color getOutlineColour()
836 return outlineColour;
841 * returns the sequences in the group ordered by the ordering given by al.
842 * this used to return an array with null entries regardless, new behaviour is
843 * below. TODO: verify that this does not affect use in applet or application
847 * @return SequenceI[] intersection of sequences in group with al, ordered by
848 * al, or null if group does not intersect with al
850 public SequenceI[] getSequencesInOrder(AlignmentI al)
852 return getSequencesInOrder(al, true);
856 * return an array representing the intersection of the group with al,
857 * optionally returning an array the size of al.getHeight() where nulls mark
858 * the non-intersected sequences
862 * @return null or array
864 public SequenceI[] getSequencesInOrder(AlignmentI al, boolean trim)
866 synchronized (sequences)
868 int sSize = sequences.size();
869 int alHeight = al.getHeight();
871 SequenceI[] seqs = new SequenceI[(trim) ? sSize : alHeight];
874 for (int i = 0; i < alHeight && index < sSize; i++)
876 if (sequences.contains(al.getSequenceAt(i)))
878 seqs[(trim) ? index : i] = al.getSequenceAt(i);
890 if (index < seqs.length)
892 SequenceI[] dummy = seqs;
893 seqs = new SequenceI[index];
896 seqs[index] = dummy[index];
905 * @return the idColour
907 public Color getIdColour()
914 * the idColour to set
916 public void setIdColour(Color idColour)
918 this.idColour = idColour;
922 * @return the representative sequence for this group
925 public SequenceI getSeqrep()
931 * set the representative sequence for this group. Note - this affects the
932 * interpretation of the Hidereps attribute.
935 * the seqrep to set (null means no sequence representative)
938 public void setSeqrep(SequenceI seqrep)
940 this.seqrep = seqrep;
945 * @return true if group has a sequence representative
948 public boolean hasSeqrep()
950 return seqrep != null;
954 * visibility of rows or represented rows covered by group
956 private boolean hidereps = false;
959 * set visibility of sequences covered by (if no sequence representative is
960 * defined) or represented by this group.
964 public void setHidereps(boolean visibility)
966 hidereps = visibility;
971 * @return true if sequences represented (or covered) by this group should be
974 public boolean isHidereps()
980 * visibility of columns intersecting this group
982 private boolean hidecols = false;
985 * set intended visibility of columns covered by this group
989 public void setHideCols(boolean visibility)
991 hidecols = visibility;
996 * @return true if columns covered by group should be hidden
998 public boolean isHideCols()
1004 * create a new sequence group from the intersection of this group with an
1005 * alignment Hashtable of hidden representatives
1011 * @return new group containing sequences common to this group and alignment
1013 public SequenceGroup intersect(AlignmentI alignment,
1014 Map<SequenceI, SequenceCollectionI> map)
1016 SequenceGroup sgroup = new SequenceGroup(this);
1017 SequenceI[] insect = getSequencesInOrder(alignment);
1018 sgroup.sequences = new ArrayList<SequenceI>();
1019 for (int s = 0; insect != null && s < insect.length; s++)
1021 if (map == null || map.containsKey(insect[s]))
1023 sgroup.sequences.add(insect[s]);
1030 * @return the showUnconserved
1032 public boolean getShowNonconserved()
1034 return showNonconserved;
1038 * @param showNonconserved
1039 * the showUnconserved to set
1041 public void setShowNonconserved(boolean displayNonconserved)
1043 this.showNonconserved = displayNonconserved;
1046 AlignmentAnnotation consensus = null, conservation = null;
1049 * flag indicating if consensus histogram should be rendered
1051 private boolean showConsensusHistogram;
1054 * set this alignmentAnnotation object as the one used to render consensus
1059 public void setConsensus(AlignmentAnnotation aan)
1061 if (consensus == null)
1069 * @return automatically calculated consensus row note: the row is a stub if a
1070 * consensus calculation has not yet been performed on the group
1072 public AlignmentAnnotation getConsensus()
1074 // TODO get or calculate and get consensus annotation row for this group
1075 int aWidth = this.getWidth();
1083 if (consensus == null)
1085 consensus = new AlignmentAnnotation("", "", new Annotation[1], 0f,
1086 100f, AlignmentAnnotation.BAR_GRAPH);
1087 consensus.hasText = true;
1088 consensus.autoCalculated = true;
1089 consensus.groupRef = this;
1090 consensus.label = "Consensus for " + getName();
1091 consensus.description = "Percent Identity";
1097 * set this alignmentAnnotation object as the one used to render consensus
1102 public void setConservationRow(AlignmentAnnotation aan)
1104 if (conservation == null)
1111 * get the conservation annotation row for this group
1113 * @return autoCalculated annotation row
1115 public AlignmentAnnotation getConservationRow()
1117 if (conservation == null)
1119 conservation = new AlignmentAnnotation("", "", new Annotation[1], 0f,
1120 11f, AlignmentAnnotation.BAR_GRAPH);
1123 conservation.hasText = true;
1124 conservation.autoCalculated = true;
1125 conservation.groupRef = this;
1126 conservation.label = "Conservation for " + getName();
1127 conservation.description = "Conservation for group " + getName()
1128 + " less than " + consPercGaps + "% gaps";
1129 return conservation;
1134 * @return true if annotation rows have been instantiated for this group
1136 public boolean hasAnnotationRows()
1138 return consensus != null || conservation != null;
1141 public SequenceI getConsensusSeq()
1144 StringBuffer seqs = new StringBuffer();
1145 for (int i = 0; i < consensus.annotations.length; i++)
1147 if (consensus.annotations[i] != null)
1149 if (consensus.annotations[i].description.charAt(0) == '[')
1151 seqs.append(consensus.annotations[i].description.charAt(1));
1155 seqs.append(consensus.annotations[i].displayCharacter);
1160 SequenceI sq = new Sequence("Group" + getName() + " Consensus",
1162 sq.setDescription("Percentage Identity Consensus "
1163 + ((ignoreGapsInConsensus) ? " without gaps" : ""));
1167 public void setIgnoreGapsConsensus(boolean state)
1169 if (this.ignoreGapsInConsensus != state && consensus != null)
1171 ignoreGapsInConsensus = state;
1172 recalcConservation();
1174 ignoreGapsInConsensus = state;
1177 public boolean getIgnoreGapsConsensus()
1179 return ignoreGapsInConsensus;
1183 * @param showSequenceLogo
1184 * indicates if a sequence logo is shown for consensus annotation
1186 public void setshowSequenceLogo(boolean showSequenceLogo)
1188 // TODO: decouple calculation from settings update
1189 if (this.showSequenceLogo != showSequenceLogo && consensus != null)
1191 this.showSequenceLogo = showSequenceLogo;
1192 recalcConservation();
1194 this.showSequenceLogo = showSequenceLogo;
1199 * @param showConsHist
1200 * flag indicating if the consensus histogram for this group should
1203 public void setShowConsensusHistogram(boolean showConsHist)
1206 if (showConsensusHistogram != showConsHist && consensus != null)
1208 this.showConsensusHistogram = showConsHist;
1209 recalcConservation();
1211 this.showConsensusHistogram = showConsHist;
1215 * @return the showConsensusHistogram
1217 public boolean isShowConsensusHistogram()
1219 return showConsensusHistogram;
1223 * set flag indicating if logo should be normalised when rendered
1227 public void setNormaliseSequenceLogo(boolean norm)
1229 normaliseSequenceLogo = norm;
1232 public boolean isNormaliseSequenceLogo()
1234 return normaliseSequenceLogo;
1239 * returns a new array with all annotation involving this group
1241 public AlignmentAnnotation[] getAlignmentAnnotation()
1243 // TODO add in other methods like 'getAlignmentAnnotation(String label),
1245 ArrayList<AlignmentAnnotation> annot = new ArrayList<AlignmentAnnotation>();
1246 synchronized (sequences)
1248 for (SequenceI seq : sequences)
1250 AlignmentAnnotation[] aa = seq.getAnnotation();
1253 for (AlignmentAnnotation al : aa)
1255 if (al.groupRef == this)
1262 if (consensus != null)
1264 annot.add(consensus);
1266 if (conservation != null)
1268 annot.add(conservation);
1271 return annot.toArray(new AlignmentAnnotation[0]);
1275 public Iterable<AlignmentAnnotation> findAnnotation(String calcId)
1277 List<AlignmentAnnotation> aa = new ArrayList<AlignmentAnnotation>();
1282 for (AlignmentAnnotation a : getAlignmentAnnotation())
1284 if (calcId.equals(a.getCalcId()))
1293 * Returns a list of annotations that match the specified sequenceRef, calcId
1294 * and label, ignoring null values.
1296 * @return list of AlignmentAnnotation objects
1299 public Iterable<AlignmentAnnotation> findAnnotations(SequenceI seq,
1300 String calcId, String label)
1302 ArrayList<AlignmentAnnotation> aa = new ArrayList<AlignmentAnnotation>();
1303 for (AlignmentAnnotation ann : getAlignmentAnnotation())
1305 if (ann.getCalcId() != null && ann.getCalcId().equals(calcId)
1306 && ann.sequenceRef != null && ann.sequenceRef == seq
1307 && ann.label != null && ann.label.equals(label))
1316 * Answer true if any annotation matches the calcId passed in (if not null).
1321 public boolean hasAnnotation(String calcId)
1323 if (calcId != null && !"".equals(calcId))
1325 for (AlignmentAnnotation a : getAlignmentAnnotation())
1327 if (a.getCalcId() == calcId)
1337 * Remove all sequences from the group (leaving other properties unchanged).
1341 synchronized (sequences)
1347 private AnnotatedCollectionI context;
1350 * Sets the alignment or group context for this group
1353 * @throws IllegalArgumentException
1354 * if setting the context would result in a circular reference chain
1356 public void setContext(AnnotatedCollectionI ctx)
1358 AnnotatedCollectionI ref = ctx;
1361 if (ref == this || ref.getContext() == ctx)
1363 throw new IllegalArgumentException(
1364 "Circular reference in SequenceGroup.context");
1366 ref = ref.getContext();
1374 * @see jalview.datamodel.AnnotatedCollectionI#getContext()
1377 public AnnotatedCollectionI getContext()
1382 public void setColourScheme(ColourSchemeI scheme)
1386 cs = new ResidueShader();
1388 cs.setColourScheme(scheme);
1391 public void setGroupColourScheme(ResidueShaderI scheme)
1396 public ColourSchemeI getColourScheme()
1398 return cs == null ? null : cs.getColourScheme();
1401 public ResidueShaderI getGroupColourScheme()
1407 public boolean isNucleotide()
1409 if (context != null) {
1410 return context.isNucleotide();
1417 * @return true if seq is a member of the group
1420 public boolean contains(SequenceI seq1)
1422 return sequences.contains(seq1);
1428 * @return true if startRes<=apos and endRes>=apos and seq is in the group
1430 public boolean contains(SequenceI seq, int apos)
1432 return (startRes <= apos && endRes >= apos) && sequences.contains(seq);