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
|
package com.wcohen.ss;
import com.wcohen.ss.api.*;
/**
* Smith-Waterman string distance, following Durban et al.
* Sec 2.3.
*/
public class SmithWaterman extends AbstractStringDistance
{
private CharMatchScore charMatchScore;
private double gapCost;
public SmithWaterman() {
this(CharMatchScore.DIST_21, 1.0 );
}
public SmithWaterman(CharMatchScore charMatchScore,double gapCost) {
this.charMatchScore = charMatchScore;
this.gapCost = gapCost;
}
public double score(StringWrapper s,StringWrapper t) {
MyMatrix mat = new MyMatrix( s, t );
return score(s,t,mat);
}
private double score(StringWrapper s,StringWrapper t,MyMatrix mat) {
double best = -Double.MAX_VALUE;
for (int i=0; i<=s.length(); i++) {
for (int j=0; j<=t.length(); j++) {
best = Math.max( best, mat.get(i,j) );
}
}
return best;
}
public String explainScore(StringWrapper s,StringWrapper t) {
MyMatrix mat = new MyMatrix( s, t );
double d = score(s,t,mat);
return mat.toString() + "\nScore = "+d;
}
private class MyMatrix extends MemoMatrix {
public MyMatrix(StringWrapper s,StringWrapper t) {
super(s,t);
}
public double compute(int i,int j) {
if (i==0) return 0;
if (j==0) return 0;
return max4( 0,
get(i-1,j-1) + charMatchScore.matchScore( sAt(i), tAt(j) ),
get(i-1, j) - gapCost,
get(i, j-1) - gapCost);
}
}
public String toString() { return "[SmithWaterman]"; }
static public void main(String[] argv) {
doMain(new SmithWaterman(), argv);
}
}
|