X-Git-Url: http://source.jalview.org/gitweb/?a=blobdiff_plain;f=src%2Fjalview%2Fdatamodel%2FAlignment.java;h=04977e852fcd85d83afaee169af0423b76b7468f;hb=797df64fa2a0a30773d0f48f5494d4155e5a8be3;hp=90a8b318e0fdd42a477c626f8d94c358ab671190;hpb=174230b4233d9ce80f94527768d2cd2f76da11ab;p=jalview.git diff --git a/src/jalview/datamodel/Alignment.java b/src/jalview/datamodel/Alignment.java index 90a8b31..04977e8 100755 --- a/src/jalview/datamodel/Alignment.java +++ b/src/jalview/datamodel/Alignment.java @@ -1,850 +1,1296 @@ /* - * Jalview - A Sequence Alignment Editor and Viewer - * Copyright (C) 2006 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 + * Jalview - A Sequence Alignment Editor and Viewer (Version 2.7) + * Copyright (C) 2011 J Procter, AM Waterhouse, J Engelhardt, LM Lui, G Barton, M Clamp, S Searle + * + * This file is part of Jalview. + * + * Jalview 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 3 of the License, or (at your option) any later version. + * + * Jalview 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 Jalview. If not, see . */ package jalview.datamodel; -import jalview.analysis.*; - -import jalview.util.*; - import java.util.*; -/** Data structure to hold and manipulate a multiple sequence alignment +import jalview.analysis.*; + +/** + * Data structure to hold and manipulate a multiple sequence alignment + */ +/** + * @author JimP + * */ public class Alignment implements AlignmentI { - protected Alignment dataset; - protected Vector sequences; - protected Vector groups = new Vector(); - protected char gapCharacter = '-'; - protected int type = NUCLEOTIDE; - public static final int PROTEIN = 0; - public static final int NUCLEOTIDE = 1; + protected Alignment dataset; - /** DOCUMENT ME!! */ - public AlignmentAnnotation[] annotations; + protected Vector sequences; - HiddenSequences hiddenSequences = new HiddenSequences(this); + protected Vector groups = new Vector(); - private void initAlignment(SequenceI[] seqs) { - int i=0; + protected char gapCharacter = '-'; - if( jalview.util.Comparison.isNucleotide(seqs)) - type = NUCLEOTIDE; - else - type = PROTEIN; + protected int type = NUCLEOTIDE; - sequences = new Vector(); + public static final int PROTEIN = 0; - for (i = 0; i < seqs.length; i++) - { - sequences.addElement(seqs[i]); - } + public static final int NUCLEOTIDE = 1; + + public boolean hasRNAStructure = false; + + /** DOCUMENT ME!! */ + public AlignmentAnnotation[] annotations; + + HiddenSequences hiddenSequences = new HiddenSequences(this); + + public Hashtable alignmentProperties; + private void initAlignment(SequenceI[] seqs) + { + int i = 0; + + if (jalview.util.Comparison.isNucleotide(seqs)) + { + type = NUCLEOTIDE; } - /** Make an alignment from an array of Sequences. - * - * @param sequences - */ - public Alignment(SequenceI[] seqs) + else { - initAlignment(seqs); + type = PROTEIN; } - /** - * Make a new alignment from an array of SeqCigars - * @param seqs SeqCigar[] - */ - public Alignment(SeqCigar[] alseqs) { - SequenceI[] seqs = SeqCigar.createAlignmentSequences(alseqs, gapCharacter, new ColumnSelection(), null); - initAlignment(seqs); + + sequences = new Vector(); + + for (i = 0; i < seqs.length; i++) + { + sequences.addElement(seqs[i]); } - /** - * Make a new alignment from an CigarArray - * JBPNote - can only do this when compactAlignment does not contain hidden regions. - * JBPNote - must also check that compactAlignment resolves to a set of SeqCigars - or construct them appropriately. - * @param compactAlignment CigarArray - */ - public static AlignmentI createAlignment(CigarArray compactAlignment) { - throw new Error("Alignment(CigarArray) not yet implemented"); - // this(compactAlignment.refCigars); + + } + + /** + * Make an alignment from an array of Sequences. + * + * @param sequences + */ + public Alignment(SequenceI[] seqs) + { + initAlignment(seqs); + } + + /** + * Make a new alignment from an array of SeqCigars + * + * @param seqs + * SeqCigar[] + */ + public Alignment(SeqCigar[] alseqs) + { + SequenceI[] seqs = SeqCigar.createAlignmentSequences(alseqs, + gapCharacter, new ColumnSelection(), null); + initAlignment(seqs); + } + + /** + * Make a new alignment from an CigarArray JBPNote - can only do this when + * compactAlignment does not contain hidden regions. JBPNote - must also check + * that compactAlignment resolves to a set of SeqCigars - or construct them + * appropriately. + * + * @param compactAlignment + * CigarArray + */ + public static AlignmentI createAlignment(CigarArray compactAlignment) + { + throw new Error("Alignment(CigarArray) not yet implemented"); + // this(compactAlignment.refCigars); + } + + /** + * DOCUMENT ME! + * + * @return DOCUMENT ME! + */ + public Vector getSequences() + { + return sequences; + } + + public SequenceI[] getSequencesArray() + { + if (sequences == null) + return null; + SequenceI[] reply = new SequenceI[sequences.size()]; + for (int i = 0; i < sequences.size(); i++) + { + reply[i] = (SequenceI) sequences.elementAt(i); } + return reply; + } - /** - * DOCUMENT ME! - * - * @return DOCUMENT ME! - */ - public Vector getSequences() + /** + * DOCUMENT ME! + * + * @param i + * DOCUMENT ME! + * + * @return DOCUMENT ME! + */ + public SequenceI getSequenceAt(int i) + { + if (i>-1 && i < sequences.size()) { - return sequences; + return (SequenceI) sequences.elementAt(i); } - public SequenceI [] getSequencesArray() + return null; + } + + /** + * Adds a sequence to the alignment. Recalculates maxLength and size. + * + * @param snew + */ + public void addSequence(SequenceI snew) + { + if (dataset != null) { - SequenceI [] reply = new SequenceI[sequences.size()]; - for(int i=0; i -1 && i < getHeight()) + { + sequences.removeElementAt(i); + hiddenSequences.adjustHeightSequenceDeleted(i); } + } - /** Adds a sequence to the alignment. Recalculates maxLength and size. - * - * @param snew - */ - public void addSequence(SequenceI snew) + /** */ + public SequenceGroup findGroup(SequenceI s) + { + for (int i = 0; i < this.groups.size(); i++) { - if(dataset!=null) + SequenceGroup sg = (SequenceGroup) groups.elementAt(i); + + if (sg.getSequences(null).contains(s)) { - if(snew.getDatasetSequence()!=null) - { - System.out.println(snew.getName()); - getDataset().addSequence(snew.getDatasetSequence()); - } - else - { - Sequence ds = new Sequence(snew.getName(), - AlignSeq.extractGaps("-. ", - snew.getSequence()), - snew.getStart(), - snew.getEnd()); - - snew.setDatasetSequence(ds); - getDataset().addSequence(ds); - } + return sg; } - - sequences.addElement(snew); } + return null; + } + + /** + * DOCUMENT ME! + * + * @param s + * DOCUMENT ME! + * + * @return DOCUMENT ME! + */ + public SequenceGroup[] findAllGroups(SequenceI s) + { + Vector temp = new Vector(); - /** Adds a sequence to the alignment. Recalculates maxLength and size. - * - * @param snew - */ - public void setSequenceAt(int i, SequenceI snew) + int gSize = groups.size(); + for (int i = 0; i < gSize; i++) { - SequenceI oldseq = getSequenceAt(i); - deleteSequence(oldseq); + SequenceGroup sg = (SequenceGroup) groups.elementAt(i); + if (sg == null || sg.getSequences(null) == null) + { + this.deleteGroup(sg); + gSize--; + continue; + } - sequences.setElementAt(snew, i); + if (sg.getSequences(null).contains(s)) + { + temp.addElement(sg); + } } - /** - * DOCUMENT ME! - * - * @return DOCUMENT ME! - */ - public Vector getGroups() + SequenceGroup[] ret = new SequenceGroup[temp.size()]; + + for (int i = 0; i < temp.size(); i++) { - return groups; - } - /** Takes out columns consisting entirely of gaps (-,.," ") - */ - public void removeGaps() { - removeGaps((ShiftList)null); + ret[i] = (SequenceGroup) temp.elementAt(i); } - /** - * remove gaps in alignment - recording any frame shifts in shiftrecord - * intended to be passed to ColumnSelection.compensateForEdits(shiftrecord) - * @param shiftrecord - */ - public void removeGaps(ShiftList shiftrecord) { - SequenceI[] seqs = getVisibleAndRepresentedSeqs(); - int j, jSize = seqs.length; - int width = 0; - for (int i = 0; i < jSize; i++) + return ret; + } + + /** */ + public void addGroup(SequenceGroup sg) + { + if (!groups.contains(sg)) + { + if (hiddenSequences.getSize() > 0) + { + int i, iSize = sg.getSize(); + for (i = 0; i < iSize; i++) { - if (seqs[i].getLength() > width) + if (!sequences.contains(sg.getSequenceAt(i))) { - width = seqs[i].getLength(); + sg.deleteSequence(sg.getSequenceAt(i), false); + iSize--; + i--; } } - int startCol = -1, endCol = -1; - boolean delete = true; - for (int i = 0; i < width; i++) + if (sg.getSize() < 1) { - delete = true; - - for (j = 0; j < jSize; j++) - { - if (seqs[j].getLength() > i) - { - if (!jalview.util.Comparison.isGap(seqs[j].getCharAt(i))) - { - if(delete) - endCol = i; - - delete = false; - break; - } - } - } - - if(delete && startCol==-1) - { - startCol = i; - } - - - if (!delete && startCol > -1) - { - deleteColumns(seqs, startCol, endCol); - if (shiftrecord!=null) { - shiftrecord.addShift(startCol, 1+endCol-startCol); - } - width -= (endCol - startCol); - i -= (endCol - startCol); - startCol = -1; - endCol = -1; - } + return; } + } - if (delete && startCol > -1) - { - deleteColumns(seqs, startCol, endCol); - if (shiftrecord!=null) { - shiftrecord.addShift(startCol, 1+endCol-startCol); - } - } + groups.addElement(sg); } + } - /** Removes a range of columns (start to end inclusive). - * - * @param seqs Sequences to remove columns from - * @param start Start column in the alignment - * @param end End column in the alignment - */ - public void deleteColumns(SequenceI [] seqs, int start, int end) + /** + * remove any annotation that references gp + * + * @param gp + * (if null, removes all group associated annotation) + */ + private void removeAnnotationForGroup(SequenceGroup gp) + { + if (annotations == null || annotations.length == 0) { - for(int i=0; iseqs[j].getLength()) - { - sequences.removeElement(seqs[j]); - j--; - jSize--; - } - else - { - seqs[j].setStart(newstart); - seqs[j].setSequence(seqs[j].getSequence().substring(i)); - } + todelete[p++] = annotations[i]; } - } - - /** - * DOCUMENT ME! - * - * @param i DOCUMENT ME! - */ - public void trimRight(int i) - { - SequenceI[] seqs = getVisibleAndRepresentedSeqs(); - int j, jSize = seqs.length; - for (j = 0; j < jSize; j++) + else { - int newend = seqs[j].findPosition(i); - - seqs[j].setEnd(newend); - if(seqs[j].getLength()>i) - seqs[j].setSequence(seqs[j].getSequence().substring(0, i + 1)); + tokeep[k++] = annotations[i]; } + } } - - /** - * DOCUMENT ME! - * - * @param s DOCUMENT ME! - */ - public void deleteSequence(SequenceI s) + else { - for (int i = 0; i < getHeight(); i++) + for (i = 0, p = 0, k = 0; i < annotations.length; i++) + { + if (annotations[i].groupRef == gp) { - if (getSequenceAt(i) == s) - { - deleteSequence(i); - } + todelete[p++] = annotations[i]; } + else + { + tokeep[k++] = annotations[i]; + } + } } - - /** - * DOCUMENT ME! - * - * @param i DOCUMENT ME! - */ - public void deleteSequence(int i) + if (p > 0) { - sequences.removeElementAt(i); + // clear out the group associated annotation. + for (i = 0; i < p; i++) + { + unhookAnnotation(todelete[i]); + todelete[i] = null; + } + t = new AlignmentAnnotation[k]; + for (i = 0; i < k; i++) + { + t[i] = tokeep[i]; + } + annotations = t; } + } - - /** */ - public SequenceGroup findGroup(SequenceI s) + public void deleteAllGroups() + { + if (annotations != null) { - for (int i = 0; i < this.groups.size(); i++) - { - SequenceGroup sg = (SequenceGroup) groups.elementAt(i); - - if (sg.getSequences(false).contains(s)) - { - return sg; - } - } - - return null; + removeAnnotationForGroup(null); } + groups.removeAllElements(); + } - /** - * DOCUMENT ME! - * - * @param s DOCUMENT ME! - * - * @return DOCUMENT ME! - */ - public SequenceGroup[] findAllGroups(SequenceI s) + /** */ + public void deleteGroup(SequenceGroup g) + { + if (groups.contains(g)) { - Vector temp = new Vector(); + removeAnnotationForGroup(g); + groups.removeElement(g); + } + } - int gSize = groups.size(); - for (int i = 0; i < gSize; i++) - { - SequenceGroup sg = (SequenceGroup) groups.elementAt(i); - if(sg==null || sg.getSequences(false)==null) - { - this.deleteGroup(sg); - gSize--; - continue; - } + /** */ + public SequenceI findName(String name) + { + return findName(name, false); + } - if (sg.getSequences(false).contains(s)) - { - temp.addElement(sg); - } - } + /* + * (non-Javadoc) + * + * @see jalview.datamodel.AlignmentI#findName(java.lang.String, boolean) + */ + public SequenceI findName(String token, boolean b) + { + return findName(null, token, b); + } - SequenceGroup[] ret = new SequenceGroup[temp.size()]; + /* + * (non-Javadoc) + * + * @see jalview.datamodel.AlignmentI#findName(SequenceI, java.lang.String, + * boolean) + */ + public SequenceI findName(SequenceI startAfter, String token, boolean b) + { - for (int i = 0; i < temp.size(); i++) + int i = 0; + SequenceI sq = null; + String sqname = null; + if (startAfter != null) + { + // try to find the sequence in the alignment + boolean matched = false; + while (i < sequences.size()) + { + if (getSequenceAt(i++) == startAfter) { - ret[i] = (SequenceGroup) temp.elementAt(i); + matched = true; + break; } + } + if (!matched) + { + i = 0; + } + } + while (i < sequences.size()) + { + sq = getSequenceAt(i); + sqname = sq.getName(); + if (sqname.equals(token) // exact match + || (b && // allow imperfect matches - case varies + (sqname.equalsIgnoreCase(token)))) + { + return getSequenceAt(i); + } - return ret; + i++; } + return null; + } + public SequenceI[] findSequenceMatch(String name) + { + Vector matches = new Vector(); + int i = 0; - /** */ - public void addGroup(SequenceGroup sg) + while (i < sequences.size()) { - if (!groups.contains(sg)) - { - groups.addElement(sg); - } + if (getSequenceAt(i).getName().equals(name)) + { + matches.addElement(getSequenceAt(i)); + } + i++; } - /** - * DOCUMENT ME! - */ - public void deleteAllGroups() + SequenceI[] result = new SequenceI[matches.size()]; + for (i = 0; i < result.length; i++) { - groups.removeAllElements(); + result[i] = (SequenceI) matches.elementAt(i); + } - int i = 0; + return result; - while (i < sequences.size()) - { - SequenceI s = getSequenceAt(i); - s.setColor(java.awt.Color.white); - i++; - } - } + } - /** */ - public void deleteGroup(SequenceGroup g) - { - if (groups.contains(g)) - { - groups.removeElement(g); - } - } + /* + * (non-Javadoc) + * + * @see jalview.datamodel.AlignmentI#findIndex(jalview.datamodel.SequenceI) + */ + public int findIndex(SequenceI s) + { + int i = 0; - /** */ - public SequenceI findName(String name) + while (i < sequences.size()) { - int i = 0; + if (s == getSequenceAt(i)) + { + return i; + } - while (i < sequences.size()) - { - if (getSequenceAt(i).getName().equals(name)) - { - return getSequenceAt(i); - } + i++; + } - i++; - } + return -1; + } - return null; - } + /* + * (non-Javadoc) + * + * @see + * jalview.datamodel.AlignmentI#findIndex(jalview.datamodel.SearchResults) + */ + public int findIndex(SearchResults results) + { + int i = 0; - public SequenceI [] findSequenceMatch(String name) + while (i < sequences.size()) { - Vector matches = new Vector(); - int i = 0; - - while (i < sequences.size()) + if (results.involvesSequence(getSequenceAt(i))) { - if (getSequenceAt(i).getName().equals(name)) - { - matches.addElement(getSequenceAt(i)); - } - i++; + return i; } + i++; + } + return -1; + } - SequenceI [] result = new SequenceI[matches.size()]; - for(i=0; i maxLength) + { + maxLength = getSequenceAt(i).getLength(); + } } + return maxLength; + } - /** */ - public int findIndex(SequenceI s) - { - int i = 0; + /** + * DOCUMENT ME! + * + * @param gc + * DOCUMENT ME! + */ + public void setGapCharacter(char gc) + { + gapCharacter = gc; - while (i < sequences.size()) - { - if (s == getSequenceAt(i)) - { - return i; - } + for (int i = 0; i < sequences.size(); i++) + { + Sequence seq = (Sequence) sequences.elementAt(i); + seq.setSequence(seq.getSequenceAsString().replace('.', gc) + .replace('-', gc).replace(' ', gc)); + } + } - i++; - } + /** + * DOCUMENT ME! + * + * @return DOCUMENT ME! + */ + public char getGapCharacter() + { + return gapCharacter; + } - return -1; - } + /* + * (non-Javadoc) + * + * @see jalview.datamodel.AlignmentI#isAligned() + */ + public boolean isAligned() + { + return isAligned(false); + } - /** - * DOCUMENT ME! - * - * @return DOCUMENT ME! - */ - public int getHeight() + /* + * (non-Javadoc) + * + * @see jalview.datamodel.AlignmentI#isAligned(boolean) + */ + public boolean isAligned(boolean includeHidden) + { + int width = getWidth(); + if (hiddenSequences == null || hiddenSequences.getSize() == 0) { - return sequences.size(); + includeHidden = true; // no hidden sequences to check against. } - - /** - * DOCUMENT ME! - * - * @return DOCUMENT ME! - */ - public int getWidth() + for (int i = 0; i < sequences.size(); i++) { - int maxLength = -1; - - for (int i = 0; i < sequences.size(); i++) + if (includeHidden || !hiddenSequences.isHidden(getSequenceAt(i))) + { + if (getSequenceAt(i).getLength() != width) { - if (getSequenceAt(i).getLength() > maxLength) - { - maxLength = getSequenceAt(i).getLength(); - } + return false; } + } + } + + return true; + } + + /* + * (non-Javadoc) + * + * @seejalview.datamodel.AlignmentI#deleteAnnotation(jalview.datamodel. + * AlignmentAnnotation) + */ + public boolean deleteAnnotation(AlignmentAnnotation aa) + { + return deleteAnnotation(aa, true); + } + + public boolean deleteAnnotation(AlignmentAnnotation aa, boolean unhook) + { + int aSize = 1; - return maxLength; + if (annotations != null) + { + aSize = annotations.length; } - /** - * DOCUMENT ME! - * - * @return DOCUMENT ME! - */ - public int getMaxIdLength() + if (aSize < 1) { - int max = 0; - int i = 0; + return false; + } - while (i < sequences.size()) - { - SequenceI seq = getSequenceAt(i); - String tmp = seq.getName() + "/" + seq.getStart() + "-" + - seq.getEnd(); + AlignmentAnnotation[] temp = new AlignmentAnnotation[aSize - 1]; - if (tmp.length() > max) - { - max = tmp.length(); - } + boolean swap = false; + int tIndex = 0; - i++; - } + for (int i = 0; i < aSize; i++) + { + if (annotations[i] == aa) + { + swap = true; + continue; + } + if (tIndex < temp.length) + temp[tIndex++] = annotations[i]; + } - return max; + if (swap) + { + annotations = temp; + if (unhook) { + unhookAnnotation(aa); + } } + return swap; + } - /** - * DOCUMENT ME! - * - * @param gc DOCUMENT ME! - */ - public void setGapCharacter(char gc) + /** + * remove any object references associated with this annotation + * + * @param aa + */ + private void unhookAnnotation(AlignmentAnnotation aa) + { + if (aa.sequenceRef != null) { - gapCharacter = gc; + aa.sequenceRef.removeAlignmentAnnotation(aa); + } + if (aa.groupRef != null) + { + // probably need to do more here in the future (post 2.5.0) + aa.groupRef = null; + } + } - for (int i = 0; i < sequences.size(); i++) + /* + * (non-Javadoc) + * + * @seejalview.datamodel.AlignmentI#addAnnotation(jalview.datamodel. + * AlignmentAnnotation) + */ + public void addAnnotation(AlignmentAnnotation aa) + { + addAnnotation(aa, -1); + } + + /* + * (non-Javadoc) + * + * @seejalview.datamodel.AlignmentI#addAnnotation(jalview.datamodel. + * AlignmentAnnotation, int) + */ + public void addAnnotation(AlignmentAnnotation aa, int pos) + { + if(aa.getRNAStruc()!= null){ + hasRNAStructure=true; + } + + int aSize = 1; + if (annotations != null) + { + aSize = annotations.length + 1; + } + + AlignmentAnnotation[] temp = new AlignmentAnnotation[aSize]; + int i = 0; + if (pos == -1 || pos >= aSize) + { + temp[aSize - 1] = aa; + } + else + { + temp[pos] = aa; + } + if (aSize > 1) + { + int p = 0; + for (i = 0; i < (aSize - 1); i++, p++) + { + if (p == pos) + { + p++; + } + if (p < temp.length) { - Sequence seq = (Sequence) sequences.elementAt(i); - seq.setSequence( seq.getSequence().replace('.', gc) ); - seq.setSequence( seq.getSequence().replace('-', gc) ); - seq.setSequence( seq.getSequence().replace(' ', gc) ); + temp[p] = annotations[i]; } + } } - /** - * DOCUMENT ME! - * - * @return DOCUMENT ME! - */ - public char getGapCharacter() + annotations = temp; + } + + public void setAnnotationIndex(AlignmentAnnotation aa, int index) + { + if (aa == null || annotations == null || annotations.length - 1 < index) { - return gapCharacter; + return; } - /** - * DOCUMENT ME! - * - * @return DOCUMENT ME! - */ - public Vector getAAFrequency() + int aSize = annotations.length; + AlignmentAnnotation[] temp = new AlignmentAnnotation[aSize]; + + temp[index] = aa; + + for (int i = 0; i < aSize; i++) { - return AAFrequency.calculate(sequences, 0, getWidth()); + if (i == index) + { + continue; + } + + if (i < index) + { + temp[i] = annotations[i]; + } + else + { + temp[i] = annotations[i - 1]; + } } - /** - * DOCUMENT ME! - * - * @return DOCUMENT ME! - */ - public boolean isAligned() - { - int width = getWidth(); + annotations = temp; + } - for (int i = 0; i < sequences.size(); i++) - { - if (getSequenceAt(i).getLength() != width) - { - return false; - } - } + /** + * DOCUMENT ME! + * + * @return DOCUMENT ME! + */ + public AlignmentAnnotation[] getAlignmentAnnotation() + { + return annotations; + } - return true; + public void setNucleotide(boolean b) + { + if (b) + { + type = NUCLEOTIDE; } + else + { + type = PROTEIN; + } + } - /** - * DOCUMENT ME! - * - * @param aa DOCUMENT ME! - */ - public void deleteAnnotation(AlignmentAnnotation aa) + public boolean isNucleotide() + { + if (type == NUCLEOTIDE) { - int aSize = 1; + return true; + } + else + { + return false; + } + } + + public boolean hasRNAStructure(){ + //TODO can it happen that structure is removed from alignment? + return hasRNAStructure; + } - if (annotations != null) + public void setDataset(Alignment data) + { + if (dataset == null && data == null) + { + // Create a new dataset for this alignment. + // Can only be done once, if dataset is not null + // This will not be performed + SequenceI[] seqs = new SequenceI[getHeight()]; + SequenceI currentSeq; + for (int i = 0; i < getHeight(); i++) + { + currentSeq = getSequenceAt(i); + if (currentSeq.getDatasetSequence() != null) + { + seqs[i] = (Sequence) currentSeq.getDatasetSequence(); + } + else { - aSize = annotations.length; + seqs[i] = currentSeq.createDatasetSequence(); } + } - AlignmentAnnotation[] temp = new AlignmentAnnotation[aSize - 1]; + dataset = new Alignment(seqs); + } + else if (dataset == null && data != null) + { + dataset = data; + } + dataset.addAlignmentRef(); + } - int tIndex = 0; + /** + * reference count for number of alignments referencing this one. + */ + int alignmentRefs = 0; - for (int i = 0; i < aSize; i++) - { - if (annotations[i] == aa) - { - continue; - } + /** + * increase reference count to this alignment. + */ + private void addAlignmentRef() + { + alignmentRefs++; + } - temp[tIndex] = annotations[i]; - tIndex++; - } + public Alignment getDataset() + { + return dataset; + } - annotations = temp; - } + public boolean padGaps() + { + boolean modified = false; + // Remove excess gaps from the end of alignment + int maxLength = -1; - public void adjustSequenceAnnotations() + SequenceI current; + for (int i = 0; i < sequences.size(); i++) { - if(annotations!=null) + current = getSequenceAt(i); + for (int j = current.getLength(); j > maxLength; j--) { - for (int a = 0; a < annotations.length; a++) + if (j > maxLength + && !jalview.util.Comparison.isGap(current.getCharAt(j))) { - if (annotations[a].sequenceRef != null) - { - annotations[a].adjustForAlignment(); - } + maxLength = j; + break; } } } - /** - * DOCUMENT ME! - * - * @param aa DOCUMENT ME! - */ - public void addAnnotation(AlignmentAnnotation aa) - { - int aSize = 1; - if (annotations != null) - { - aSize = annotations.length + 1; - } + maxLength++; - AlignmentAnnotation[] temp = new AlignmentAnnotation[aSize]; + int cLength; + for (int i = 0; i < sequences.size(); i++) + { + current = getSequenceAt(i); + cLength = current.getLength(); - temp[aSize-1] = aa; + if (cLength < maxLength) + { + current.insertCharAt(cLength, maxLength - cLength, gapCharacter); + modified = true; + } + else if (current.getLength() > maxLength) + { + current.deleteChars(maxLength, current.getLength()); + } + } + return modified; + } - int i = 0; + /** + * Justify the sequences to the left or right by deleting and inserting gaps + * before the initial residue or after the terminal residue + * + * @param right + * true if alignment padded to right, false to justify to left + * @return true if alignment was changed + */ + public boolean justify(boolean right) + { + boolean modified = false; - if (aSize > 1) + // Remove excess gaps from the end of alignment + int maxLength = -1; + int ends[] = new int[sequences.size() * 2]; + SequenceI current; + for (int i = 0; i < sequences.size(); i++) + { + current = getSequenceAt(i); + // This should really be a sequence method + ends[i * 2] = current.findIndex(current.getStart()); + ends[i * 2 + 1] = current.findIndex(current.getStart() + + current.getLength()); + boolean hitres = false; + for (int j = 0, rs = 0, ssiz = current.getLength(); j < ssiz; j++) + { + if (!jalview.util.Comparison.isGap(current.getCharAt(j))) { - for (i = 0; i < (aSize-1); i++) + if (!hitres) + { + ends[i * 2] = j; + hitres = true; + } + else + { + ends[i * 2 + 1] = j; + if (j - ends[i * 2] > maxLength) { - temp[i] = annotations[i]; + maxLength = j - ends[i * 2]; } + } } - - annotations = temp; + } } - public void setAnnotationIndex(AlignmentAnnotation aa, int index) + maxLength++; + // now edit the flanking gaps to justify to either left or right + int cLength, extent, diff; + for (int i = 0; i < sequences.size(); i++) { - if(aa==null || annotations==null || annotations.length-1 ends[i * 2 + 1]) + { + current.deleteChars(ends[i * 2 + 1] + 1, extent); + modified = true; + } + if (ends[i * 2] > diff) + { + current.deleteChars(0, ends[i * 2] - diff); + modified = true; + } else - temp[i] = annotations[i-1]; + { + if (ends[i * 2] < diff) + { + current.insertCharAt(0, diff - ends[i * 2], gapCharacter); + modified = true; + } + } + } + else + { + // left justify + if (ends[i * 2] > 0) + { + current.deleteChars(0, ends[i * 2]); + modified = true; + ends[i * 2 + 1] -= ends[i * 2]; + extent -= ends[i * 2]; + } + if (extent > maxLength) + { + current.deleteChars(maxLength + 1, extent); + modified = true; + } + else + { + if (extent < maxLength) + { + current.insertCharAt(extent, maxLength - extent, gapCharacter); + modified = true; + } + } } - - annotations = temp; } + return modified; + } + + public HiddenSequences getHiddenSequences() + { + return hiddenSequences; + } - /** - * DOCUMENT ME! - * - * @return DOCUMENT ME! - */ - public AlignmentAnnotation[] getAlignmentAnnotation() + public CigarArray getCompactAlignment() + { + SeqCigar alseqs[] = new SeqCigar[sequences.size()]; + for (int i = 0; i < sequences.size(); i++) { - return annotations; + alseqs[i] = new SeqCigar((SequenceI) sequences.elementAt(i)); } + CigarArray cal = new CigarArray(alseqs); + cal.addOperation(CigarArray.M, getWidth()); + return cal; + } + + public void setProperty(Object key, Object value) + { + if (alignmentProperties == null) + alignmentProperties = new Hashtable(); + + alignmentProperties.put(key, value); + } + + public Object getProperty(Object key) + { + if (alignmentProperties != null) + return alignmentProperties.get(key); + else + return null; + } + + public Hashtable getProperties() + { + return alignmentProperties; + } - public void setNucleotide(boolean b) + AlignedCodonFrame[] codonFrameList = null; + + /* + * (non-Javadoc) + * + * @see + * jalview.datamodel.AlignmentI#addCodonFrame(jalview.datamodel.AlignedCodonFrame + * ) + */ + public void addCodonFrame(AlignedCodonFrame codons) + { + if (codons == null) + return; + if (codonFrameList == null) { - if(b) - type = NUCLEOTIDE; - else - type = PROTEIN; + codonFrameList = new AlignedCodonFrame[] + { codons }; + return; } + AlignedCodonFrame[] t = new AlignedCodonFrame[codonFrameList.length + 1]; + System.arraycopy(codonFrameList, 0, t, 0, codonFrameList.length); + t[codonFrameList.length] = codons; + codonFrameList = t; + } + + /* + * (non-Javadoc) + * + * @see jalview.datamodel.AlignmentI#getCodonFrame(int) + */ + public AlignedCodonFrame getCodonFrame(int index) + { + return codonFrameList[index]; + } - public boolean isNucleotide() + /* + * (non-Javadoc) + * + * @see + * jalview.datamodel.AlignmentI#getCodonFrame(jalview.datamodel.SequenceI) + */ + public AlignedCodonFrame[] getCodonFrame(SequenceI seq) + { + if (seq == null || codonFrameList == null) + return null; + Vector cframes = new Vector(); + for (int f = 0; f < codonFrameList.length; f++) { - if(type==NUCLEOTIDE) - return true; - else - return false; + if (codonFrameList[f].involvesSequence(seq)) + cframes.addElement(codonFrameList[f]); } + if (cframes.size() == 0) + return null; + AlignedCodonFrame[] cfr = new AlignedCodonFrame[cframes.size()]; + cframes.copyInto(cfr); + return cfr; + } - public void setDataset(Alignment data) + /* + * (non-Javadoc) + * + * @see jalview.datamodel.AlignmentI#getCodonFrames() + */ + public AlignedCodonFrame[] getCodonFrames() + { + return codonFrameList; + } + + /* + * (non-Javadoc) + * + * @seejalview.datamodel.AlignmentI#removeCodonFrame(jalview.datamodel. + * AlignedCodonFrame) + */ + public boolean removeCodonFrame(AlignedCodonFrame codons) + { + if (codons == null || codonFrameList == null) + return false; + boolean removed = false; + int i = 0, iSize = codonFrameList.length; + while (i < iSize) { - if(dataset==null && data==null) + if (codonFrameList[i] == codons) { - // Create a new dataset for this alignment. - // Can only be done once, if dataset is not null - // This will not be performed - Sequence[] seqs = new Sequence[getHeight()]; - for (int i = 0; i < getHeight(); i++) + removed = true; + if (i + 1 < iSize) { - if(getSequenceAt(i).getDatasetSequence()!=null) - { - seqs[i] = (Sequence)getSequenceAt(i).getDatasetSequence(); - } - else - { - seqs[i] = new Sequence(getSequenceAt(i).getName(), - AlignSeq.extractGaps( - jalview.util.Comparison.GapChars, - getSequenceAt(i).getSequence() - ), - getSequenceAt(i).getStart(), - getSequenceAt(i).getEnd()); - seqs[i].sequenceFeatures = getSequenceAt(i).getSequenceFeatures(); - getSequenceAt(i).setSequenceFeatures(null); - getSequenceAt(i).setDatasetSequence(seqs[i]); - } + System.arraycopy(codonFrameList, i + 1, codonFrameList, i, iSize + - i - 1); } - - dataset = new Alignment(seqs); + iSize--; } - else if(dataset==null && data!=null) + else { - dataset = data; + i++; } } + return removed; + } - public Alignment getDataset() + public void append(AlignmentI toappend) + { + // TODO test this method for a future 2.5 release + // currently tested for use in jalview.gui.SequenceFetcher + boolean samegap = toappend.getGapCharacter() == getGapCharacter(); + char oldc = toappend.getGapCharacter(); + boolean hashidden = toappend.getHiddenSequences() != null + && toappend.getHiddenSequences().hiddenSequences != null; + // get all sequences including any hidden ones + Vector sqs = (hashidden) ? toappend.getHiddenSequences() + .getFullAlignment().getSequences() : toappend.getSequences(); + if (sqs != null) { - return dataset; - } - - public boolean padGaps() { - boolean modified=false; - - //Remove excess gaps from the end of alignment - int maxLength = -1; - - SequenceI current; - for (int i = 0; i < sequences.size(); i++) + Enumeration sq = sqs.elements(); + while (sq.hasMoreElements()) { - current = getSequenceAt(i); - for (int j = current.getLength(); j > maxLength; j--) + SequenceI addedsq = (SequenceI) sq.nextElement(); + if (!samegap) { - if (j > maxLength && !jalview.util.Comparison.isGap( - current.getCharAt(j))) + char[] oldseq = addedsq.getSequence(); + for (int c = 0; c < oldseq.length; c++) { - maxLength = j; - break; + if (oldseq[c] == oldc) + { + oldseq[c] = gapCharacter; + } } } + addSequence(addedsq); } - - maxLength++; - - for (int i = 0; i < sequences.size(); - i++) - { - current = getSequenceAt(i); - - if (current.getLength() < maxLength) - { - current.insertCharAt(maxLength - 1, gapCharacter); - modified=true; - } - else if(current.getLength() > maxLength) - { - current.deleteChars(maxLength, current.getLength()); - } - } - return modified; } - - public HiddenSequences getHiddenSequences() + AlignmentAnnotation[] alan = toappend.getAlignmentAnnotation(); + for (int a = 0; alan != null && a < alan.length; a++) { - return hiddenSequences; + addAnnotation(alan[a]); } - SequenceI [] getVisibleAndRepresentedSeqs() + AlignedCodonFrame[] acod = toappend.getCodonFrames(); + for (int a = 0; acod != null && a < acod.length; a++) { - if(hiddenSequences==null || hiddenSequences.getSize()<1) - return getSequencesArray(); - - Vector seqs = new Vector(); - SequenceI seq; - SequenceGroup hidden; - for (int i = 0; i < sequences.size(); i++) + this.addCodonFrame(acod[a]); + } + Vector sg = toappend.getGroups(); + if (sg != null) + { + Enumeration el = sg.elements(); + while (el.hasMoreElements()) + { + addGroup((SequenceGroup) el.nextElement()); + } + } + if (toappend.getHiddenSequences() != null) + { + HiddenSequences hs = toappend.getHiddenSequences(); + if (hiddenSequences == null) + { + hiddenSequences = new HiddenSequences(this); + } + if (hs.hiddenSequences != null) { - seq = (SequenceI) sequences.elementAt(i); - seqs.addElement(seq); - hidden = seq.getHiddenSequences(); - if(hidden!=null) + for (int s = 0; s < hs.hiddenSequences.length; s++) { - for(int j=0; j