X-Git-Url: http://source.jalview.org/gitweb/?a=blobdiff_plain;f=src%2Fjalview%2Futil%2FMapList.java;h=0d71bb49b360993afd3ddc21a0ab58791c693ee3;hb=d89bd711a41eb3c1cd66d01c0067e1ddb5261ef9;hp=4a59f88aa50066da55870f06ec3651a68fb968db;hpb=539bd323123b239afdac5ef7b7706d6ccfc6a41d;p=jalview.git diff --git a/src/jalview/util/MapList.java b/src/jalview/util/MapList.java index 4a59f88..0d71bb4 100644 --- a/src/jalview/util/MapList.java +++ b/src/jalview/util/MapList.java @@ -1,222 +1,496 @@ /* - * 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 + * Jalview - A Sequence Alignment Editor and Viewer ($$Version-Rel$$) + * Copyright (C) $$Year-Rel$$ The Jalview Authors + * + * 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. - * - * 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. - * + * + * 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 this program; if not, write to the Free Software - * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA + * along with Jalview. If not, see . + * The Jalview Authors are detailed in the 'AUTHORS' file. */ package jalview.util; -import java.util.*; +import java.util.ArrayList; +import java.util.Arrays; +import java.util.List; /** - * MapList - * Simple way of bijectively mapping a non-contiguous linear range to another non-contiguous linear range + * A simple way of bijectively mapping a non-contiguous linear range to another + * non-contiguous linear range. + * * Use at your own risk! + * * TODO: efficient implementation of private posMap method - * TODO: test/ensure that sense of from and to ratio start position is conserved (codon start position recovery) - * TODO: optimize to use int[][] arrays rather than vectors. + * + * TODO: test/ensure that sense of from and to ratio start position is conserved + * (codon start position recovery) */ public class MapList { - /* (non-Javadoc) - * @see java.lang.Object#clone() + + /* + * Subregions (base 1) described as { [start1, end1], [start2, end2], ...} + */ + private List fromShifts; + + /* + * Same format as fromShifts, for the 'mapped to' sequence + */ + private List toShifts; + + /* + * number of steps in fromShifts to one toRatio unit + */ + private int fromRatio; + + /* + * number of steps in toShifts to one fromRatio + */ + private int toRatio; + + /* + * lowest and highest value in the from Map + */ + private int fromLowest; + + private int fromHighest; + + /* + * lowest and highest value in the to Map */ - protected Object clone() throws CloneNotSupportedException { - // TODO Auto-generated method stub - return super.clone(); + private int toLowest; + + private int toHighest; + + /** + * Constructor + */ + public MapList() + { + fromShifts = new ArrayList<>(); + toShifts = new ArrayList<>(); } - /* (non-Javadoc) - * @see java.lang.Object#equals(java.lang.Object) + + /** + * Two MapList objects are equal if they are the same object, or they both + * have populated shift ranges and all values are the same. */ - public boolean equals(MapList obj) { - if (obj==this) - return true; - if (obj!=null && obj.fromRatio==fromRatio && obj.toRatio==toRatio - && obj.fromShifts!=null && obj.toShifts!=null) { - int i,iSize=fromShifts.size(),j,jSize=obj.fromShifts.size(); - if (iSize!=jSize) - return false; - for (i=0,iSize=fromShifts.size(),j=0, jSize=obj.fromShifts.size(); i getFromRanges() + { + return fromShifts; + } + + /** + * Returns the 'to' ranges as {[start1, end1], [start2, end2], ...} + * + * @return + */ + public List getToRanges() + { + return toShifts; + } + + /** + * Flattens a list of [start, end] into a single [start1, end1, start2, + * end2,...] array. + * + * @param shifts + * @return + */ + protected static int[] getRanges(List shifts) + { + int[] rnges = new int[2 * shifts.size()]; + int i = 0; + for (int[] r : shifts) + { + rnges[i++] = r[0]; + rnges[i++] = r[1]; + } + return rnges; + } + + /** + * + * @return length of mapped phrase in from */ - int[] toRange=null; public int getFromRatio() { return fromRatio; } + + /** + * + * @return length of mapped phrase in to + */ public int getToRatio() { return toRatio; } - public int getFromLowest() { - return fromRange[0]; - } - public int getFromHighest() { - return fromRange[1]; + + public int getFromLowest() + { + return fromLowest; } - public int getToLowest() { - return toRange[0]; + + public int getFromHighest() + { + return fromHighest; } - public int getToHighest() { - return toRange[1]; + + public int getToLowest() + { + return toLowest; } - private void ensureRange(int[] limits, int pos) { - if (limits[0]>pos) - limits[0]=pos; - if (limits[1] fromRange, List toRange, int fromRatio, + int toRatio) + { + this(); + fromRange = coalesceRanges(fromRange); + toRange = coalesceRanges(toRange); + this.fromShifts = fromRange; + this.toShifts = toRange; + this.fromRatio = fromRatio; + this.toRatio = toRatio; + + fromLowest = Integer.MAX_VALUE; + fromHighest = Integer.MIN_VALUE; + for (int[] range : fromRange) + { + if (range.length != 2) + { + // throw new IllegalArgumentException(range); + System.err.println("Invalid format for fromRange " + + Arrays.toString(range) + " may cause errors"); + } + fromLowest = Math.min(fromLowest, Math.min(range[0], range[1])); + fromHighest = Math.max(fromHighest, Math.max(range[0], range[1])); + } - fromShifts = new Vector(); - for (int i=0;i + * The main use case for this method is when mapping cDNA sequence to its + * protein product, based on CDS feature ranges which derive from spliced + * exons, but are contiguous on the cDNA sequence. For example + * + *
+   *   CDS 1-20  // from exon1
+   *   CDS 21-35 // from exon2
+   *   CDS 36-71 // from exon3
+   * 'coalesce' to range 1-71
+   * 
+ * + * @param ranges + * @return the same list (if unchanged), else a new merged list, leaving the + * input list unchanged + */ + public static List coalesceRanges(final List ranges) + { + if (ranges == null || ranges.size() < 2) + { + return ranges; } - toShifts = new Vector(); - for (int i=0;i merged = new ArrayList<>(); + int[] lastRange = ranges.get(0); + int lastDirection = lastRange[1] >= lastRange[0] ? 1 : -1; + lastRange = new int[] { lastRange[0], lastRange[1] }; + merged.add(lastRange); + boolean first = true; + + for (final int[] range : ranges) { - ensureRange(toRange, to[i]); - ensureRange(toRange, to[i+1]); - toShifts.addElement(new int[] - {to[i], to[i + 1]}); + if (first) + { + first = false; + continue; + } + + int direction = range[1] >= range[0] ? 1 : -1; + + /* + * if next range is in the same direction as last and contiguous, + * just update the end position of the last range + */ + boolean sameDirection = range[1] == range[0] + || direction == lastDirection; + boolean extending = range[0] == lastRange[1] + lastDirection; + if (sameDirection && extending) + { + lastRange[1] = range[1]; + changed = true; + } + else + { + lastRange = new int[] { range[0], range[1] }; + merged.add(lastRange); + // careful: merging [5, 5] after [7, 6] should keep negative direction + lastDirection = (range[1] == range[0]) ? lastDirection : direction; + } } - this.fromRatio=fromRatio; - this.toRatio=toRatio; + + return changed ? merged : ranges; } + /** * get all mapped positions from 'from' to 'to' - * @return int[][] { int[] { fromStart, fromFinish, toStart, toFinish }, int [fromFinish-fromStart+2] { toStart..toFinish mappings}} + * + * @return int[][] { int[] { fromStart, fromFinish, toStart, toFinish }, int + * [fromFinish-fromStart+2] { toStart..toFinish mappings}} */ - public int[][] makeFromMap() + protected int[][] makeFromMap() { + // TODO not used - remove?? return posMap(fromShifts, fromRatio, toShifts, toRatio); } + /** * get all mapped positions from 'to' to 'from' + * * @return int[to position]=position mapped in from */ - public int[][] makeToMap() + protected int[][] makeToMap() { - return posMap(toShifts,toRatio, fromShifts, fromRatio); + // TODO not used - remove?? + return posMap(toShifts, toRatio, fromShifts, fromRatio); } + /** * construct an int map for intervals in intVals - * @param intVals - * @return int[] { from, to pos in range }, int[range.to-range.from+1] returning mapped position + * + * @param shiftTo + * @return int[] { from, to pos in range }, int[range.to-range.from+1] + * returning mapped position */ - private int[][] posMap(Vector intVals, int ratio, Vector toIntVals, - int toRatio) + private int[][] posMap(List shiftTo, int ratio, + List shiftFrom, int toRatio) { - int iv=0,ivSize = intVals.size(); - if (iv>=ivSize) + // TODO not used - remove?? + int iv = 0, ivSize = shiftTo.size(); + if (iv >= ivSize) { return null; } - int[] intv=(int[]) intVals.elementAt(iv++); - int from=intv[0],to=intv[1]; + int[] intv = shiftTo.get(iv++); + int from = intv[0], to = intv[1]; if (from > to) { from = intv[1]; - to=intv[0]; + to = intv[0]; } - while (ivto) + if (intv[0] > to) { - to=intv[0]; + to = intv[0]; } - if (intv[1]>to) + if (intv[1] > to) { - to=intv[1]; + to = intv[1]; } } - int tF=0,tT=0; - int mp[][] = new int[to-from+2][]; + int tF = 0, tT = 0; + int mp[][] = new int[to - from + 2][]; for (int i = 0; i < mp.length; i++) { - int[] m = shift(i+from,intVals,ratio,toIntVals, toRatio); + int[] m = shift(i + from, shiftTo, ratio, shiftFrom, toRatio); if (m != null) { if (i == 0) { - tF=tT=m[0]; + tF = tT = m[0]; } else { if (m[0] < tF) { - tF=m[0]; + tF = m[0]; } if (m[0] > tT) { - tT=m[0]; + tT = m[0]; } } } mp[i] = m; } - int[][] map = new int[][] - { - new int[] - { - from, to, tF, tT}, new int[to - from + 2]}; + int[][] map = new int[][] { new int[] { from, to, tF, tT }, + new int[to - from + 2] }; map[0][2] = tF; map[0][3] = tT; @@ -225,7 +499,7 @@ public class MapList { if (mp[i] != null) { - map[1][i] = mp[i][0]-tF; + map[1][i] = mp[i][0] - tF; } else { @@ -234,28 +508,29 @@ public class MapList } return map; } + /** * addShift - * @param pos start position for shift (in original reference frame) - * @param shift length of shift - * - public void addShift(int pos, int shift) - { - int sidx = 0; - int[] rshift=null; - while (sidx shiftTo, int fromRatio, + List shiftFrom, int toRatio) { - int[] fromCount = countPos(fromShifts, pos); - if (fromCount==null) + // TODO: javadoc; tests + int[] fromCount = countPositions(shiftTo, pos); + if (fromCount == null) { return null; } - int fromRemainder=(fromCount[0]-1) % fromRatio; - int toCount = 1+(((fromCount[0]-1) / fromRatio) * toRatio); - int[] toPos = countToPos(toShifts, toCount); - if (toPos==null) + int fromRemainder = (fromCount[0] - 1) % fromRatio; + int toCount = 1 + (((fromCount[0] - 1) / fromRatio) * toRatio); + int[] toPos = traverseToPosition(shiftFrom, toCount); + if (toPos == null) { - return null; // throw new Error("Bad Mapping!"); + return null; } - //System.out.println(fromCount[0]+" "+fromCount[1]+" "+toCount); - return new int[] - { - toPos[0], fromRemainder, toPos[1]}; + return new int[] { toPos[0], fromRemainder, toPos[1] }; } + /** - * count how many positions pos is along the series of intervals. - * @param intVals + * Counts how many positions pos is along the series of intervals. Returns an + * array of two values: + *
    + *
  • the number of positions traversed (inclusive) to reach {@code pos}
  • + *
  • +1 if the last interval traversed is forward, -1 if in a negative + * direction
  • + *
+ * Returns null if {@code pos} does not lie in any of the given intervals. + * + * @param intervals + * a list of start-end intervals * @param pos - * @return number of positions or null if pos is not within intervals + * a position that may lie in one (or more) of the intervals + * @return */ - private int[] countPos(Vector intVals, int pos) + protected static int[] countPositions(List intervals, int pos) { - int count=0,intv[],iv=0,ivSize=intVals.size(); - while (iv= intv[0] && pos <= intv[1]) { - return new int[] - { - count + pos - intv[0] + 1, +1}; + return new int[] { count + pos - intv[0] + 1, +1 }; } else { - count+=intv[1]-intv[0]+1; + count += intv[1] - intv[0] + 1; } } else { + /* + * reverse interval + */ if (pos >= intv[1] && pos <= intv[0]) { - return new int[] - { - count + intv[0] - pos + 1, -1}; + return new int[] { count + intv[0] - pos + 1, -1 }; } else { - count+=intv[0]-intv[1]+1; + count += intv[0] - intv[1] + 1; } } } return null; } + /** - * count out pos positions into a series of intervals and return the position - * @param intVals - * @param pos - * @return position pos in interval set + * Reads through the given intervals until {@code count} positions have been + * traversed, and returns an array consisting of two values: + *
    + *
  • the value at the {@code count'th} position
  • + *
  • +1 if the last interval read is forwards, -1 if reverse direction
  • + *
+ * Returns null if the ranges include less than {@code count} positions, or if + * {@code count < 1}. + * + * @param intervals + * a list of [start, end] ranges + * @param count + * the number of positions to traverse + * @return */ - private int[] countToPos(Vector intVals, int pos) + protected static int[] traverseToPosition(List intervals, + final int count) { - int count = 0, diff = 0, iv=0,ivSize=intVals.size(), intv[] = + int traversed = 0; + int ivSize = intervals.size(); + int iv = 0; + + if (count < 1) { - 0, 0}; - while (iv= 0) { - if (pos <= count + 1 + diff) + if (count <= traversed + 1 + diff) { - return new int[] - { - pos - count - 1 + intv[0], +1}; + return new int[] { intv[0] + (count - traversed - 1), +1 }; } else { - count+=1+diff; + traversed += 1 + diff; } } else { - if (pos <= count + 1 - diff) + if (count <= traversed + 1 - diff) { - return new int[] - { - intv[0] - (pos - count - 1), -1}; + return new int[] { intv[0] - (count - traversed - 1), -1 }; } else { - count+=1-diff; + traversed += 1 - diff; } } } - return null;//(diff<0) ? (intv[1]-1) : (intv[0]+1); + return null; } + /** * find series of intervals mapping from start-end in the From map. - * @param start position in to map - * @param end position in to map - * @return series of ranges in from map + * + * @param start + * position mapped 'to' + * @param end + * position mapped 'to' + * @return series of [start, end] ranges in sequence mapped 'from' */ - public int[] locateInFrom(int start, int end) { + public int[] locateInFrom(int start, int end) + { + return locateInFrom2(start, end); + // inefficient implementation - int fromStart[] = shiftTo(start); - int fromEnd[] = shiftTo(end); - if (fromStart==null || fromEnd==null) - return null; - int iv[] = getIntervals(fromShifts, fromStart, fromEnd,fromRatio); - return iv; + // int fromStart[] = shiftTo(start); + // needs to be inclusive of end of symbol position + // int fromEnd[] = shiftTo(end); + // return getIntervals(fromShifts, fromStart, fromEnd, fromRatio); } /** * find series of intervals mapping from start-end in the to map. - * @param start position in from map - * @param end position in from map - * @return series of ranges in to map + * + * @param start + * position mapped 'from' + * @param end + * position mapped 'from' + * @return series of [start, end] ranges in sequence mapped 'to' */ - public int[] locateInTo(int start, int end) { - // inefficient implementation - int toStart[] = shiftFrom(start); - int toEnd[] = shiftFrom(end); - if (toStart==null || toEnd==null) - return null; - int iv[] = getIntervals(toShifts, toStart, toEnd, toRatio); - return iv; + public int[] locateInTo(int start, int end) + { + return locateInTo2(start, end); + + // int toStart[] = shiftFrom(start); + // int toEnd[] = shiftFrom(end); + // return getIntervals(toShifts, toStart, toEnd, toRatio); } + /** - * like shift - except returns the intervals in the given vector of shifts which were spanned - * in traversing fromStart to fromEnd - * @param fromShifts2 + * like shift - except returns the intervals in the given vector of shifts + * which were spanned in traversing fromStart to fromEnd + * + * @param shiftFrom * @param fromStart * @param fromEnd * @param fromRatio2 - * @return + * @return series of from,to intervals from from first position of starting + * region to final position of ending region inclusive */ - private int[] getIntervals(Vector fromShifts2, int[] fromStart, int[] fromEnd, int fromRatio2) - { - // correct for word direction for start and end - int startpos = fromStart[0]+fromStart[2]*(fromRatio2-1); - int endpos = fromEnd[0]+fromEnd[2]*(fromRatio2-1); - int intv=0,intvSize= fromShifts2.size(); - int iv[],i=0,fs=-1,fe=-1; // containing intervals - while (intv=iv[0] && startpos<=iv[1]) { + protected static int[] getIntervals(List shiftFrom, + int[] fromStart, int[] fromEnd, int fromRatio2) + { + if (fromStart == null || fromEnd == null) + { + return null; + } + int startpos, endpos; + startpos = fromStart[0]; // first position in fromStart + endpos = fromEnd[0]; // last position in fromEnd + int endindx = (fromRatio2 - 1); // additional positions to get to last + // position from endpos + int intv = 0, intvSize = shiftFrom.size(); + int iv[], i = 0, fs = -1, fe_s = -1, fe = -1; // containing intervals + // search intervals to locate ones containing startpos and count endindx + // positions on from endpos + while (intv < intvSize && (fs == -1 || fe == -1)) + { + iv = shiftFrom.get(intv++); + if (fe_s > -1) + { + endpos = iv[0]; // start counting from beginning of interval + endindx--; // inclusive of endpos + } + if (iv[0] <= iv[1]) + { + if (fs == -1 && startpos >= iv[0] && startpos <= iv[1]) + { fs = i; } - if (fe==-1 && endpos>=iv[0] && endpos<=iv[1]) { - fe = i; + if (endpos >= iv[0] && endpos <= iv[1]) + { + if (fe_s == -1) + { + fe_s = i; + } + if (fe_s != -1) + { + if (endpos + endindx <= iv[1]) + { + fe = i; + endpos = endpos + endindx; // end of end token is within this + // interval + } + else + { + endindx -= iv[1] - endpos; // skip all this interval too + } + } } - } else { - if (fs==-1 && startpos<=iv[0] && startpos>=iv[1]) { + } + else + { + if (fs == -1 && startpos <= iv[0] && startpos >= iv[1]) + { fs = i; } - if (fe==-1 && endpos<=iv[0] && endpos>=iv[1]) { - fe = i; + if (endpos <= iv[0] && endpos >= iv[1]) + { + if (fe_s == -1) + { + fe_s = i; + } + if (fe_s != -1) + { + if (endpos - endindx >= iv[1]) + { + fe = i; + endpos = endpos - endindx; // end of end token is within this + // interval + } + else + { + endindx -= endpos - iv[1]; // skip all this interval too + } + } } } i++; } - if (fs==fe && fe==-1) + if (fs == fe && fe == -1) + { return null; - Vector ranges=new Vector(); - if (fs<=fe) { + } + List ranges = new ArrayList<>(); + if (fs <= fe) + { intv = fs; - i=fs; + i = fs; // truncate initial interval - iv = (int[]) fromShifts2.elementAt(intv++); - iv = new int[] { iv[0], iv[1]};// clone - if (i==fs) + iv = shiftFrom.get(intv++); + iv = new int[] { iv[0], iv[1] };// clone + if (i == fs) + { iv[0] = startpos; - while (i!=fe) { - ranges.addElement(iv); // add initial range - iv = (int[]) fromShifts2.elementAt(intv++); // get next interval - iv = new int[] { iv[0], iv[1]};// clone + } + while (i != fe) + { + ranges.add(iv); // add initial range + iv = shiftFrom.get(intv++); // get next interval + iv = new int[] { iv[0], iv[1] };// clone i++; } - if (i==fe) + if (i == fe) + { iv[1] = endpos; - ranges.addElement(iv); // add only - or final range - } else { + } + ranges.add(iv); // add only - or final range + } + else + { // walk from end of interval. - i=fromShifts2.size()-1; - while (i>fs) { + i = shiftFrom.size() - 1; + while (i > fs) + { i--; } - iv = (int[]) fromShifts2.elementAt(i); - iv = new int[] { iv[1], iv[0]};// reverse and clone + iv = shiftFrom.get(i); + iv = new int[] { iv[1], iv[0] };// reverse and clone // truncate initial interval - if (i==fs) + if (i == fs) { iv[0] = startpos; } - while (i!=fe) { - ranges.addElement(iv); // add (truncated) reversed interval - iv = (int[]) fromShifts2.elementAt(--i); + while (--i != fe) + { // fix apparent logic bug when fe==-1 + ranges.add(iv); // add (truncated) reversed interval + iv = shiftFrom.get(i); iv = new int[] { iv[1], iv[0] }; // reverse and clone } - if (i==fe) { + if (i == fe) + { // interval is already reversed iv[1] = endpos; } - ranges.addElement(iv); // add only - or final range + ranges.add(iv); // add only - or final range } // create array of start end intervals. int[] range = null; - if (ranges!=null && ranges.size()>0) + if (ranges != null && ranges.size() > 0) { - range = new int[ranges.size()*2]; + range = new int[ranges.size() * 2]; intv = 0; - intvSize=ranges.size(); - i=0; - while (intv= map.getFromLowest() + && getFromHighest() <= map.getFromHighest()) + || (getFromLowest() <= map.getFromLowest() + && getFromHighest() >= map.getFromHighest())); + } + else + { + return ((getToLowest() >= map.getToLowest() + && getToHighest() <= map.getToHighest()) + || (getToLowest() <= map.getToLowest() + && getToHighest() >= map.getToHighest())); + } + } + + /** + * String representation - for debugging, not guaranteed not to change + */ + @Override + public String toString() + { + StringBuilder sb = new StringBuilder(64); + sb.append("["); + for (int[] shift : fromShifts) + { + sb.append(" ").append(Arrays.toString(shift)); + } + sb.append(" ] "); + sb.append(fromRatio).append(":").append(toRatio); + sb.append(" to ["); + for (int[] shift : toShifts) + { + sb.append(" ").append(Arrays.toString(shift)); + } + sb.append(" ]"); + return sb.toString(); + } + + /** + * Extend this map list by adding the given map's ranges. There is no + * validation check that the ranges do not overlap existing ranges (or each + * other), but contiguous ranges are merged. + * + * @param map + */ + public void addMapList(MapList map) + { + if (this.equals(map)) + { + return; + } + this.fromLowest = Math.min(fromLowest, map.fromLowest); + this.toLowest = Math.min(toLowest, map.toLowest); + this.fromHighest = Math.max(fromHighest, map.fromHighest); + this.toHighest = Math.max(toHighest, map.toHighest); + + for (int[] range : map.getFromRanges()) + { + addRange(range, fromShifts); + } + for (int[] range : map.getToRanges()) + { + addRange(range, toShifts); + } + } + + /** + * Adds the given range to a list of ranges. If the new range just extends + * existing ranges, the current endpoint is updated instead. + * + * @param range + * @param addTo + */ + static void addRange(int[] range, List addTo) + { + /* + * list is empty - add to it! + */ + if (addTo.size() == 0) + { + addTo.add(range); + return; + } + + int[] last = addTo.get(addTo.size() - 1); + boolean lastForward = last[1] >= last[0]; + boolean newForward = range[1] >= range[0]; + + /* + * contiguous range in the same direction - just update endpoint + */ + if (lastForward == newForward && last[1] == range[0]) + { + last[1] = range[1]; + return; + } + + /* + * next range starts at +1 in forward sense - update endpoint + */ + if (lastForward && newForward && range[0] == last[1] + 1) + { + last[1] = range[1]; + return; + } + + /* + * next range starts at -1 in reverse sense - update endpoint + */ + if (!lastForward && !newForward && range[0] == last[1] - 1) + { + last[1] = range[1]; + return; + } + + /* + * just add the new range + */ + addTo.add(range); + } + + /** + * Returns true if mapping is from forward strand, false if from reverse + * strand. Result is just based on the first 'from' range that is not a single + * position. Default is true unless proven to be false. Behaviour is not well + * defined if the mapping has a mixture of forward and reverse ranges. + * + * @return + */ + public boolean isFromForwardStrand() + { + return isForwardStrand(getFromRanges()); + } + + /** + * Returns true if mapping is to forward strand, false if to reverse strand. + * Result is just based on the first 'to' range that is not a single position. + * Default is true unless proven to be false. Behaviour is not well defined if + * the mapping has a mixture of forward and reverse ranges. + * + * @return + */ + public boolean isToForwardStrand() + { + return isForwardStrand(getToRanges()); + } + /** - * test routine. not incremental. - * @param ml - * @param fromS - * @param fromE + * A helper method that returns true unless at least one range has start > + * end. Behaviour is undefined for a mixture of forward and reverse ranges. + * + * @param ranges + * @return */ - public static void testMap(MapList ml, int fromS, int fromE) + private boolean isForwardStrand(List ranges) { - for (int from = 1; from <= 25; from++) + boolean forwardStrand = true; + for (int[] range : ranges) { - int[] too=ml.shiftFrom(from); - System.out.print("ShiftFrom("+from+")=="); - if (too==null) + if (range[1] > range[0]) { - System.out.print("NaN\n"); + break; // forward strand confirmed } - else + else if (range[1] < range[0]) { - System.out.print(too[0]+" % "+too[1]+" ("+too[2]+")"); - System.out.print("\t+--+\t"); - int[] toofrom=ml.shiftTo(too[0]); - if (toofrom != null) - { - if (toofrom[0]!=from) - { - System.err.println("Mapping not reflexive:" + from + " " + too[0] + - "->" + toofrom[0]); - } - System.out.println("ShiftTo(" + too[0] + ")==" + toofrom[0] + " % " + - toofrom[1]+" ("+toofrom[2]+")"); - } - else - { - System.out.println("ShiftTo(" + too[0] + ")==" + - "NaN! - not Bijective Mapping!"); - } + forwardStrand = false; + break; // reverse strand confirmed } } - int mmap[][] = ml.makeFromMap(); - System.out.println("FromMap : (" + mmap[0][0] + " " + mmap[0][1] + " " + - mmap[0][2] + " " + mmap[0][3] + " "); - for (int i = 1; i <= mmap[1].length; i++) + return forwardStrand; + } + + /** + * + * @return true if from, or to is a three to 1 mapping + */ + public boolean isTripletMap() + { + return (toRatio == 3 && fromRatio == 1) + || (fromRatio == 3 && toRatio == 1); + } + + /** + * Returns a map which is the composite of this one and the input map. That + * is, the output map has the fromRanges of this map, and its toRanges are the + * toRanges of this map as transformed by the input map. + *

+ * Returns null if the mappings cannot be traversed (not all toRanges of this + * map correspond to fromRanges of the input), or if this.toRatio does not + * match map.fromRatio. + * + *

+   * Example 1:
+   *    this:   from [1-100] to [501-600]
+   *    input:  from [10-40] to [60-90]
+   *    output: from [10-40] to [560-590]
+   * Example 2 ('reverse strand exons'):
+   *    this:   from [1-100] to [2000-1951], [1000-951] // transcript to loci
+   *    input:  from [1-50]  to [41-90] // CDS to transcript
+   *    output: from [10-40] to [1960-1951], [1000-971] // CDS to gene loci
+   * 
+ * + * @param map + * @return + */ + public MapList traverse(MapList map) + { + if (map == null) { - if (mmap[1][i - 1] == -1) - { - System.out.print(i+"=XXX"); + return null; + } - } - else + /* + * compound the ratios by this rule: + * A:B with M:N gives A*M:B*N + * reduced by greatest common divisor + * so 1:3 with 3:3 is 3:9 or 1:3 + * 1:3 with 3:1 is 3:3 or 1:1 + * 1:3 with 1:3 is 1:9 + * 2:5 with 3:7 is 6:35 + */ + int outFromRatio = getFromRatio() * map.getFromRatio(); + int outToRatio = getToRatio() * map.getToRatio(); + int gcd = MathUtils.gcd(outFromRatio, outToRatio); + outFromRatio /= gcd; + outToRatio /= gcd; + + List toRanges = new ArrayList<>(); + for (int[] range : getToRanges()) + { + int fromLength = Math.abs(range[1] - range[0]) + 1; + int[] transferred = map.locateInTo(range[0], range[1]); + if (transferred == null || transferred.length % 2 != 0) { - System.out.print(i+"="+(mmap[0][2]+mmap[1][i-1])); + return null; } - if (i % 20==0) + + /* + * convert [start1, end1, start2, end2, ...] + * to [[start1, end1], [start2, end2], ...] + */ + int toLength = 0; + for (int i = 0; i < transferred.length;) { - System.out.print("\n"); + toRanges.add(new int[] { transferred[i], transferred[i + 1] }); + toLength += Math.abs(transferred[i + 1] - transferred[i]) + 1; + i += 2; } - else + + /* + * check we mapped the full range - if not, abort + */ + if (fromLength * map.getToRatio() != toLength * map.getFromRatio()) { - System.out.print(","); + return null; } } - //test range function - System.out.print("\nTest locateInFrom\n"); + + return new MapList(getFromRanges(), toRanges, outFromRatio, outToRatio); + } + + /** + * Answers true if the mapping is from one contiguous range to another, else + * false + * + * @return + */ + public boolean isContiguous() + { + return fromShifts.size() == 1 && toShifts.size() == 1; + } + + /** + * Returns the [start, end, start, end, ...] ranges in the 'from' range that + * map to positions between {@code start} and {@code end} in the 'to' range. + * Returns null if no mapped positions are found in start-end. + * + * @param start + * @param end + * @return + */ + public int[] locateInFrom2(int start, int end) + { + List ranges = mapBetween(start, end, toShifts, fromShifts, + toRatio, fromRatio); + + // TODO: or just return the List and adjust calling code to match + return ranges.isEmpty() ? null : MappingUtils.rangeListToArray(ranges); + } + + /** + * Returns the [start, end, start, end, ...] ranges in the 'to' range that map + * to the given start-end in the 'from' range. Returns null if either + * {@code start} or {@code end} is not a mapped 'from' range position. + * + * @param start + * @param end + * @return + */ + public int[] locateInTo2(int start, int end) + { + List ranges = mapBetween(start, end, fromShifts, toShifts, + fromRatio, toRatio); + + return ranges.isEmpty() ? null : MappingUtils.rangeListToArray(ranges); + } + + /** + * A helper method for navigating the mapping. Returns a (possibly empty) list + * of [start-end] positions in {@code ranges2} that map to positions in + * {@code ranges1} between {@code start} and {@code end}. + * + * @param start + * @param end + * @param ranges1 + * @param ranges2 + * @param wordLength1 + * @param wordLength2 + * @return + */ + final static List mapBetween(int start, int end, + List ranges1, List ranges2, int wordLength1, + int wordLength2) + { + if (end < start) + { + int tmp = end; + end = start; + start = tmp; + } + + /* + * first traverse ranges1 and record count of mapped positions + * to any that overlap start-end + */ + List overlaps = findOverlapPositions(ranges1, start, end); + if (overlaps.isEmpty()) { - int f=mmap[0][2],t=mmap[0][3]; - while (f mapped = new ArrayList<>(); + final int s1 = overlaps.size(); + final int s2 = ranges2.size(); + int ranges2Index = 0; + + /* + * count of mapped positions preceding ranges2[ranges2Index] + */ + int traversed = 0; + + /* + * for each [from-to] range in overlaps: + * - walk (what remains of) ranges2 + * - record the values at offsets [from-to] + * - stop when past 'to' positions (or at end of ranges2) + */ + for (int i = 0; i < s1; i++) + { + int[] overlap = overlaps.get(i); + final int toAdd = overlap[1] - overlap[0] + 1; + int added = 0; // how much of overlap has been 'found' + for (; added < toAdd && ranges2Index < s2; ranges2Index++) + { + int[] range2 = ranges2.get(ranges2Index); + int rangeStart = range2[0]; + int rangeEnd = range2[1]; + boolean reverseStrand = range2[1] < range2[0]; + int rangeLength = Math.abs(rangeEnd - rangeStart) + 1; + if (traversed + rangeLength <= overlap[0]) { - System.out.println("No range!"); + /* + * precedes overlap - keep looking + */ + traversed += rangeLength; + continue; } - System.out.print("\n"); - f++;t--; + int overlapStart = overlap[0] - traversed; + int overlapEnd = Math.min(overlapStart + toAdd - added - 1, + rangeLength - 1); + int mappedFrom = range2[0] + (reverseStrand ? - overlapStart : overlapStart); + int mappedTo = range2[0] + (reverseStrand ? - overlapEnd : overlapEnd); + mapped.add(new int[] { mappedFrom, mappedTo }); + int found = overlapEnd - overlapStart + 1; + added += found; + overlap[0] += found; + traversed += rangeLength; } } - System.out.print("\n"); - mmap = ml.makeToMap(); - System.out.println("ToMap : (" + mmap[0][0] + " " + mmap[0][1] + " " + - mmap[0][2] + " " + mmap[0][3] + " "); - for (int i = 1; i <= mmap[1].length; i++) + + return mapped; + } + + /** + * Converts the start-end positions (counted from zero) in the {@code ranges} + * list from one word length to another. Start-end positions are expanded if + * necessary to cover a whole word of length {@code wordLength1}. Positions + * are then divided by {@code wordLength1} and multiplied by + * {@code wordLength2} to give equivalent mapped words. + *

+ * Put simply, this converts peptide residue positions to the corresponding + * codon ranges, and codons - including partial codons - to the corresponding + * peptide positions; for example + * + *

+   * [1, 10] with word lengths 3:1 converts (as if bases [0-11]) to [1, 4]
+   * 
+ * + * @param ranges + * @param wordLength1 + * @param wordLength2 + * @return + */ + final static void mapWords(List ranges, int wordLength1, + int wordLength2) + { + if (wordLength1 == 1 && wordLength2 == 1) + { + return; // nothing to do here + } + int s = ranges.size(); + for (int i = 0; i < s; i++) { - if (mmap[1][i - 1] == -1) - { - System.out.print(i+"=XXX"); + int[] range = ranges.get(i); - } - else - { - System.out.print(i+"="+(mmap[0][2]+mmap[1][i-1])); - } - if (i % 20==0) - { - System.out.print("\n"); - } - else - { - System.out.print(","); - } + /* + * expand range start to the start of a word, + * and convert to wordLength2 + */ + range[0] -= range[0] % wordLength1; + range[0] = range[0] / wordLength1 * wordLength2; + + /* + * similar calculation for range end, adding + * (wordLength2 - 1) for end of mapped word + */ + range[1] -= range[1] % wordLength1; + range[1] = range[1] / wordLength1 * wordLength2; + range[1] += wordLength2 - 1; + } + } + + /** + * Helper method that returns a (possibly empty) list of offsets in + * {@code ranges} to subranges that overlap {@code start-end} (where start <= + * end}. The list returned holds counts of the number of positions traversed + * (exclusive) to reach the overlapping positions, not the overlapping values. + * Returns null if there are no overlaps. + * + * @param ranges + * @param start + * @param end + * @return + */ + final static List findOverlapPositions(List ranges, + int start, int end) + { + List positions = new ArrayList<>(); + int pos = 0; + int s = ranges.size(); + for (int i = 0; i < s; i++) + { + int[] range = ranges.get(i); + addOverlap(positions, pos, range, start, end); + pos += 1 + Math.abs(range[1] - range[0]); } - System.out.print("\n"); - //test range function - System.out.print("\nTest locateInTo\n"); + return positions; + } + + /** + * A helper method that checks whether {@code range} overlaps + * {@code start-end}, and if so adds the offset of the overlap in + * {@code range}, plus {@code pos}, to {@code positions}. + * + * @param positions + * a list of map offsets to add to + * @param pos + * the number of mapped positions already visited + * @param range + * a from-to range (may be forward or reverse) + * @param start + * position to test for overlap in range + * @param end + * position to test for overlap in range + * @return + */ + final static void addOverlap(List positions, int pos, int[] range, + int start, int end) + { + if (range[1] >= range[0]) { - int f=mmap[0][2],t=mmap[0][3]; - while (f= range[0]) + { + /* + * overlap + */ + int overlapStart = Math.max(start, range[0]); + int overlapStartOffset = pos + overlapStart - range[0]; + int overlapEnd = Math.min(end, range[1]); + int overlapEndOffset = pos + overlapEnd - range[0]; + int[] lastOverlap = positions.isEmpty() ? null + : positions.get(positions.size() - 1); + if (lastOverlap != null && overlapStartOffset == lastOverlap[1] + 1) { - for (int i=0; i= range[1]) + { + /* + * overlap + */ + int overlapStart = Math.max(start, range[1]); + int overlapEnd = Math.min(end, range[0]); + positions + .add(new int[] + { pos + range[0] - overlapEnd, + pos + range[0] - overlapStart }); + } + } } }