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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
|
package structures;
import shared.KillSwitch;
import shared.Parse;
import shared.Tools;
import stream.Read;
public class Quantizer {
public static boolean parse(String arg, String a, String b){
if(a.equals("quantize")){
if(b!=null && b.equalsIgnoreCase("sticky")){
STICKY=true;
return true;
}
}else if(a.equals("quantizesticky")){
STICKY=Parse.parseBoolean(b);
return true;
}
if(b==null || b.length()<1 || Character.isLetter(b.charAt(0))){
return Parse.parseBoolean(b);
}
return setArray(b);
}
private static boolean setArray(String s){
final byte[] array;
if(s.charAt(0)=='/'){
int quant=Integer.parseInt(s.substring(1));
assert(quant>0 && quant<128);
if(quant==1){return false;}
ByteBuilder bb=new ByteBuilder();
for(int i=0, max=Read.MAX_CALLED_QUALITY(); i<=max; i+=quant){
bb.append((byte)i);
}
array=bb.toBytes();
}else{
array=Parse.parseByteArray(s, ",");
}
setArray(array);
return true;
}
private static void setArray(byte[] a){
quantizeArray=a;
qualityRemapArray=makeQualityRemapArray(quantizeArray);
}
public static void quantize(Read r1, Read r2){
quantize(r1.quality);
if(r2!=null){quantize(r2.quality);}
}
public static void quantize(byte[] quals){
if(quals==null){return;}
byte prev=0;
for(int i=0; i<quals.length; i++){
final byte qOld=quals[i];
byte q=qualityRemapArray[qOld];
if(STICKY && q!=prev && prev>0 && q>0 && Tools.absdif(qOld, prev)<=Tools.absdif(qOld, q)){q=prev;}
quals[i]=q;
prev=q;
}
}
private static final byte[] makeQualityRemapArray(byte[] quantizeArray) {
byte[] array=KillSwitch.allocByte1D(128);
for(int i=0; i<array.length; i++){
byte q=0;
for(byte x : quantizeArray){
if((i>0 && q==0 && x>0) || Tools.absdif(x, i)<=Tools.absdif(q, i)){q=x;}
}
array[i]=q;
}
return array;
}
// private static byte[] quantizeArray={0, 8, 13, 22, 27, 32, 37}; //Old
private static byte[] quantizeArray={0, 14, 21, 27, 32, 36};
private static byte[] qualityRemapArray=makeQualityRemapArray(quantizeArray);
private static boolean STICKY=true;
}
|
