File: ReadQualityComparator.java

package info (click to toggle)
bbmap 39.20%2Bdfsg-3
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 26,024 kB
  • sloc: java: 312,743; sh: 18,099; python: 5,247; ansic: 2,074; perl: 96; makefile: 39; xml: 38
file content (55 lines) | stat: -rwxr-xr-x 1,478 bytes parent folder | download | duplicates (4)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
package sort;

import stream.Read;

/**
 * Sorts highest-quality reads first, and longest first on ties.
 * @author Brian Bushnell
 * @date Nov 9, 2016
 *
 */
public final class ReadQualityComparator extends ReadComparator {
	
	private ReadQualityComparator(){}
	
	@Override
	public int compare(Read a, Read b) {
		int x=compareInner(a, b);
		return ascending*x;
	}

	private static int compareInner(Read a, Read b) {
		if(a==b){return 0;}
		if(a==null){return 1;}
		if(b==null){return -1;}
		Read a2=a.mate, b2=b.mate;
		int alen=a.length()+a.mateLength();
		int blen=b.length()+b.mateLength();

		double aerrors=Read.expectedErrors(a.bases, a.quality, true, a.length());
		double berrors=Read.expectedErrors(b.bases, b.quality, true, b.length());
		if(a2!=null){aerrors+=Read.expectedErrors(a2.bases, a2.quality, true, a2.length());}
		if(b2!=null){berrors+=Read.expectedErrors(b2.bases, b2.quality, true, b2.length());}
		
		double arate=aerrors/alen;
		double brate=berrors/blen;
		
		if(arate!=brate){return arate>brate ? 1 : -1;}
		if(alen!=blen){return blen-alen;}
		int x=a.id.compareTo(b.id);
		if(x!=0){return x;}
		if(a.numericID>b.numericID){return 1;}
		if(b.numericID>a.numericID){return -1;}
		return 0;
	}
	
	public static final ReadQualityComparator comparator=new ReadQualityComparator();
	
	private int ascending=1;
	
	@Override
	public void setAscending(boolean asc){
		ascending=(asc ? 1 : -1);
	}
	
}