diff options
author | chris dyer <redpony@umd.edu> | 2010-12-28 20:56:42 -0500 |
---|---|---|
committer | chris dyer <redpony@umd.edu> | 2010-12-28 20:56:42 -0500 |
commit | 907fd790fe644836477443981d4d92a23385bdda (patch) | |
tree | 16cd6a075ce6d638a47c51507f06b07713aee456 /decoder | |
parent | 0f32c8d6fd42a0fff61a438dff41ddbb4a195b7e (diff) |
incorporate dwarf features
Diffstat (limited to 'decoder')
-rw-r--r-- | decoder/Makefile.am | 2 | ||||
-rw-r--r-- | decoder/cdec_ff.cc | 2 | ||||
-rw-r--r-- | decoder/dwarf.cc | 3209 | ||||
-rw-r--r-- | decoder/dwarf.h | 286 | ||||
-rw-r--r-- | decoder/ff_dwarf.cc | 893 | ||||
-rw-r--r-- | decoder/ff_dwarf.h | 100 | ||||
-rw-r--r-- | decoder/grammar.cc | 1 | ||||
-rw-r--r-- | decoder/rule_lexer.l | 13 | ||||
-rw-r--r-- | decoder/trule.h | 26 |
9 files changed, 4520 insertions, 12 deletions
diff --git a/decoder/Makefile.am b/decoder/Makefile.am index be04fb31..f0c5f73e 100644 --- a/decoder/Makefile.am +++ b/decoder/Makefile.am @@ -38,6 +38,8 @@ libcdec_a_SOURCES = \ maxtrans_blunsom.cc \ cdec_ff.cc \ cfg.cc \ + dwarf.cc \ + ff_dwarf.cc \ apply_fsa_models.cc \ rule_lexer.cc \ fst_translator.cc \ diff --git a/decoder/cdec_ff.cc b/decoder/cdec_ff.cc index a12b532f..c87396a6 100644 --- a/decoder/cdec_ff.cc +++ b/decoder/cdec_ff.cc @@ -14,6 +14,7 @@ #include "ff_register.h" #include "ff_charset.h" #include "ff_wordset.h" +#include "ff_dwarf.h" #ifdef HAVE_GLC #include <cdec/ff_glc.h> @@ -70,6 +71,7 @@ void register_feature_functions() { ff_registry.Register("LexicalTranslationTrigger", new FFFactory<LexicalTranslationTrigger>); ff_registry.Register("WordPairFeatures", new FFFactory<WordPairFeatures>); ff_registry.Register("WordSet", new FFFactory<WordSet>); + ff_registry.Register("Dwarf", new FFFactory<Dwarf>); #ifdef HAVE_GLC ff_registry.Register("ContextCRF", new FFFactory<Model1Features>); #endif diff --git a/decoder/dwarf.cc b/decoder/dwarf.cc new file mode 100644 index 00000000..7968fee2 --- /dev/null +++ b/decoder/dwarf.cc @@ -0,0 +1,3209 @@ +#include "dwarf.h" +#include "tdict.h" +#include "wordid.h" +#include "lattice.h" +#include "ff_dwarf.h" +#include <assert.h> +#include <algorithm> +#include <ostream> +#include <sstream> +#include <iostream> +#include <fstream> +#include <vector> +#include <map> +#include <set> +#include <boost/functional/hash.hpp> +#include <tr1/unordered_map> +#include <boost/tuple/tuple.hpp> + +using namespace std; +using namespace std::tr1; +using namespace boost::tuples; +using namespace boost; + +Alignment::Alignment() { + //unordered_map<std::vector<WordID>,int> XX; + _I=0; + _J=0; + kSOS = TD::Convert("<s>"); + kEOS = TD::Convert("</s>"); + kUNK = TD::Convert("**UNKNOWN**"); + SourceFWAntsIdxs = new int*[MAX_ARITY]; + SourceFWAntsAbsIdxs = new int*[MAX_ARITY]; + TargetFWAntsIdxs = new int*[MAX_ARITY]; + SourceAntsIdxs = new int*[MAX_ARITY]; + TargetAntsIdxs = new int*[MAX_ARITY]; + AntsAl = new int*[MAX_ARITY]; + for (int idx=0; idx<MAX_ARITY; idx++) { + SourceAntsIdxs[idx] = new int[40]; + SourceFWAntsIdxs[idx] = new int[40]; + SourceFWAntsAbsIdxs[idx] = new int[40]; + TargetAntsIdxs[idx] = new int[40]; + TargetFWAntsIdxs[idx] = new int[40]; + AntsAl[idx] = new int[40]; + } + for (int j=0; j<MAX_WORDS; j++) + for (int i=0; i<MAX_WORDS; i++) _matrix[j][i]=false; + for (int j=0; j<MAX_WORDS; j++) { + _tSpan[j][0]=MINIMUM_INIT; + _sSpan[j][1]=MAXIMUM_INIT; + } + for (int i=0; i<MAX_WORDS; i++) { + _sSpan[i][0]=MINIMUM_INIT; + _sSpan[i][1]=MAXIMUM_INIT; + } + alpha_oris=0.1; + alpha_orit=0.1; + alpha_doms=0.1; + alpha_domt=0.1; + beta_oris=0.1; + beta_orit=0.1; + beta_doms=0.1; + beta_domt=0.1; +} + +void Alignment::set(int j,int i) { +// create a link between j and i, update their corresponding span accordingly + if (DEBUG) cerr << "set(" << j << "," << i << ")" << endl; + assert(0<=j && j<MAX_WORDS); + assert(0<=i && i<MAX_WORDS); + if (0<=j && j<MAX_WORDS && 0<=i && i<MAX_WORDS) { + _matrix[j][i] = true; + _tSpan[j][0]=least(i,_tSpan[j][0]); + _tSpan[j][1]=most(i,_tSpan[j][1]); + _sSpan[i][0]=least(j,_sSpan[i][0]); + _sSpan[i][1]=most(j,_sSpan[i][1]); + } + _J=most(j+1,_J); + _I=most(i+1,_I); +} + +void Alignment::reset(int j,int i) { //probably won't be used, since the alignment is not dynamic +// remove the link between j and i, update their corresponding span accordingly + if (DEBUG) cerr << "reset(" << j << "," << i << ")" << endl; + assert(0<=j && j<MAX_WORDS); + assert(0<=i && i<MAX_WORDS); + _matrix[j][i] = false; + if (j==_sSpan[i][0] || j==_sSpan[i][1]) { + int min=MINIMUM_INIT; + int max=MAXIMUM_INIT; + for (int idx=_sSpan[i][0]; idx<=_sSpan[i][1]; idx++) { + if (_matrix[idx][i]) { + min=least(min,idx); + max=most(max,idx); + } + } + _sSpan[i][0]=min; + _sSpan[i][1]=max; + } + if (i==_tSpan[j][0] || i==_tSpan[j][1]) { + int min=MINIMUM_INIT; + int max=MAXIMUM_INIT; + for (int idx=_tSpan[j][0]; idx<=_tSpan[j][1]; idx++) { + if (_matrix[j][idx]) { + min=least(min,idx); + max=most(max,idx); + } + } + _tSpan[j][0]=min; + _tSpan[j][1]=max; + } +} + +int Alignment::targetOf(int j, int start) { + assert(j>=0); + if (start==-1) start = _tSpan[j][0]; + if (_tSpan[j][0]==MINIMUM_INIT) return -1; + for (int idx=start; idx<=_tSpan[j][1]; idx++) { + if (_matrix[j][idx]) return idx; + } + return -1; +} + +int Alignment::sourceOf(int i, int start) { + assert(i>=0); + if (start==-1) start = _sSpan[i][0]; + if (_sSpan[i][0]==MINIMUM_INIT) return -1; + for (int idx=start; idx<=_sSpan[i][1]; idx++) { + if (_matrix[idx][i]) return idx; + } + return -1; +} + +void Alignment::clearAls(int prevJ, int prevI) { + for (int j=0; j<=prevJ; j++) { + for (int i=0; i<prevI; i++) { + _matrix[j][i]=false; + } + } + for (int j=0; j<=prevJ; j++) { + _tSpan[j][0] = MINIMUM_INIT; + _tSpan[j][1] = MAXIMUM_INIT; + } + for (int i=0; i<=prevI; i++) { + _sSpan[i][0] = MINIMUM_INIT; + _sSpan[i][1] = MAXIMUM_INIT; + } + _J=0; + _I=0; +} + +int Alignment::DominanceSource(int fw1, int fw2) { + // Dominance of fw1 and fw2 + // 0 -> neither, 1 -> leftFirst, 2 -> rightFirst, 3 -> dontCare + if (DEBUG) cerr << "DominanceSource(" << fw1 << "," << fw2 << ")" << endl; + //cerr << TD::Convert(_f[fw1]) << "," << TD::Convert(_f[fw2]) << endl; + //cerr << AsString() << endl; + int dom = 0; + curr_al.push_back(fw1); curr_al.push_back(fw2); + if (doms_hash.find(curr_al)==doms_hash.end()) { + int* block = blockSource(fw1,fw2); + //cerr << "block = " << block[0] << "," << block[1] << "," << block[2] << "," << block[3] << endl; + if (block[0]==fw1) { + int tfw10 = _tSpan[fw1][0]; + int tfw11 = _tSpan[fw1][1]; + //cerr << "tfw = " << tfw10 << "," << tfw11 << endl; + if (tfw11<0) { + dom+=1; + } else { + if ((block[2]==tfw10 || block[3]==tfw11)) dom+=1; + } + } + if (block[1]==fw2) { + int tfw20 = _tSpan[fw2][0]; + int tfw21 = _tSpan[fw2][1]; + //cerr << "tfw = " << tfw20 << "," << tfw21 << endl; + if (tfw21<0) { + dom+=2; + } else { + if ((block[2]==tfw20 || block[3]==tfw21)) dom+=2; + } + } + delete block; + doms_hash.insert(pair<vector<int>,int>(curr_al,dom)); + } else { + dom = doms_hash[curr_al]; + } + if (DEBUG) cerr << " dom = " << dom << endl; + curr_al.pop_back(); curr_al.pop_back(); + return dom; +} + +vector<int> Alignment::DominanceSource4Sampler(int fw1, int fw2) { + if (DEBUG) cerr << "DominanceSource4Sampler(" << fw1 << "," << fw2 << ")" << endl; + int dom = 0; + int* block = blockSource(fw1,fw2); + //cerr << "block = " << block[0] << "," << block[1] << "," << block[2] << "," << block[3] << endl; + if (block[0]==fw1) { + int tfw10 = _tSpan[fw1][0]; + int tfw11 = _tSpan[fw1][1]; + //cerr << "tfw = " << tfw10 << "," << tfw11 << endl; + if (tfw11<0) { + dom+=1; + } else { + if ((block[2]==tfw10 || block[3]==tfw11)) dom+=1; + } + } + if (block[1]==fw2) { + int tfw20 = _tSpan[fw2][0]; + int tfw21 = _tSpan[fw2][1]; + //cerr << "tfw = " << tfw20 << "," << tfw21 << endl; + if (tfw21<0) { + dom+=2; + } else { + if ((block[2]==tfw20 || block[3]==tfw21)) dom+=2; + } + } + if (DEBUG) cerr << "doms = " << dom << endl; + vector<int> ret; + ret.push_back(dom); ret.push_back(block[0]); ret.push_back(block[1]); + ret.push_back(block[2]); ret.push_back(block[3]); + delete block; + return ret; +} + +int Alignment::DominanceTarget(int fw1, int fw2) { + int dom = 0; + curr_al.push_back(fw1); curr_al.push_back(fw2); + if (domt_hash.find(curr_al)==domt_hash.end()) { + int* block = blockTarget(fw1,fw2); + if (block[2]==fw1) { + int sfw10 = _sSpan[fw1][0]; + int sfw11 = _sSpan[fw1][1]; + if (sfw11<0) { + dom+=1; + } else { + if (block[0]==sfw10 || block[1]==sfw11) dom+=1; + } + } + if (block[3]==fw2) { + int sfw20 = _sSpan[fw2][0]; + int sfw21 = _sSpan[fw2][0]; + if (sfw21<0) { + dom+=2; + } else { + if (block[0]==sfw20 || block[1]==sfw21) dom+=2; + } + } + delete block; + domt_hash.insert(pair<vector<int>,int>(curr_al,dom)); + } else { + dom = domt_hash[curr_al]; + } + curr_al.pop_back(); curr_al.pop_back(); + return dom; +} + +vector<int> Alignment::DominanceTarget4Sampler(int fw1, int fw2) { + int dom = 0; + int* block = blockTarget(fw1,fw2); + if (block[2]==fw1) { + int sfw10 = _sSpan[fw1][0]; + int sfw11 = _sSpan[fw1][1]; + if (sfw11<0) { + dom+=1; + } else { + if (block[0]==sfw10 || block[1]==sfw11) dom+=1; + } + } + if (block[3]==fw2) { + int sfw20 = _sSpan[fw2][0]; + int sfw21 = _sSpan[fw2][0]; + if (sfw21<0) { + dom+=2; + } else { + if (block[0]==sfw20 || block[1]==sfw21) dom+=2; + } + } + vector<int> ret; + ret.push_back(dom); ret.push_back(block[0]); ret.push_back(block[1]); + ret.push_back(block[2]); ret.push_back(block[3]); + delete block; + return ret; +} + +void Alignment::OrientationSource(int fw, int* oril, int* orir, bool Lcompute, bool Rcompute) { + OrientationSource(fw,fw,oril,orir,Lcompute,Rcompute); +} + +vector<int> Alignment::OrientationSourceLeft4Sampler(int fw) { + return OrientationSourceLeft4Sampler(fw,fw); +} + +vector<int> Alignment::OrientationSourceLeft4Sampler(int fw0, int fw1) { + if (DEBUG) cerr << "OrientationSourceLeft4Sampler(" << fw0 << "," << fw1 << ")" << endl; + int oril = 0; + int N0=fw0-1; + while (N0>=0) { + if (minTSpan(N0)!=MINIMUM_INIT) break; + N0--; + } + int N1=fw1+1; + while (N1<_J) { + if (minTSpan(N1)!=MINIMUM_INIT) break; + N1++; + } + if (minTSpan(fw0)==MINIMUM_INIT && minTSpan(fw1)==MINIMUM_INIT) { + fw0 = N1; fw1 = N0; + } + if (DEBUG) cerr << "fw0=" << fw0 << ", fw1=" << fw1 << ", N0=" << N0 << ", N1=" << N1 << endl; + if (maxTSpan(N0)<minTSpan(fw0) || maxTSpan(fw0)<minTSpan(N0)) { + if (DEBUG) cerr << "N0=" << minTSpan(N0) << "-" << maxTSpan(N0); + if (DEBUG) cerr << "fw=" << minTSpan(fw0) << "-" << maxTSpan(fw0) << endl; + int *block = blockTarget(minTSpan(N0),maxTSpan(N0)); + if (block[0]<=fw0 && fw0<=block[1]) oril=5; + delete block; + if (oril==0) { + block = blockTarget(minTSpan(fw0),maxTSpan(fw0)); + if (block[0]<=N0 && N0<=block[1]) oril=5; + delete block; + } + if (oril==0) { + if (maxTSpan(N0)<minTSpan(fw0)) {// if N0 is monotone + oril=1; + block = blockTarget(maxTSpan(N0),minTSpan(fw0)-1); + if (block[0] <= fw0 && fw0 <= block[1]) oril+=2; + delete block; + } else { //if (maxTSpan(fw0)<minTSpan(N0)) { // if NO is non-monotone + oril=2; + block = blockTarget(maxTSpan(fw0)+1,minTSpan(N0)); + if (block[0] <= fw0 && fw0 <= block[1]) oril+=2; + delete block; + } + } + } else { + oril=5; + } + if (DEBUG) cerr << "oril = " << oril << endl; + int* block = blockSource(N0,fw0); + if (DEBUG) { + for (int i=0; i<4; i++) cerr << "block[" << i << "]=" << block[i] << endl; + } + vector<int> ret; + ret.push_back(oril); ret.push_back(block[0]); ret.push_back(block[1]); + ret.push_back(block[2]); ret.push_back(block[3]); + delete block; + return ret; +} + +vector<int> Alignment::OrientationSourceRight4Sampler(int fw) { + return OrientationSourceRight4Sampler(fw,fw); +} + +vector<int> Alignment::OrientationSourceRight4Sampler(int fw0, int fw1) { + if (DEBUG) cerr << "OrientationSourceLeft4Sampler(" << fw0 << "," << fw1 << ")" << endl; + int orir = 0; + int N0=fw0-1; + while (N0>=0) { + if (minTSpan(N0)!=MINIMUM_INIT) break; + N0--; + } + int N1=fw1+1; + while (N1<_J) { + if (minTSpan(N1)!=MINIMUM_INIT) break; + N1++; + } + if (minTSpan(fw0)==MINIMUM_INIT && minTSpan(fw1)==MINIMUM_INIT) { + fw0 = N1; fw1 = N0; + } + if (DEBUG) cerr << "fw0=" << fw0 << ", fw1=" << fw1 << ", N0=" << N0 << ", N1=" << N1 << endl; + if (maxTSpan(N1)<minTSpan(fw1) || maxTSpan(fw1)<minTSpan(N1)) { + int* block = blockTarget(minTSpan(N1),maxTSpan(N1)); + if (block[0]<=fw1 && fw1<=block[2]) orir=5; + delete block; + if (orir==0) { + block = blockTarget(minTSpan(fw1),maxTSpan(fw1)); + if (block[0]<=N1 && N1 <=block[1]) orir=5; + delete block; + } + if (DEBUG) cerr << "N1=" << minTSpan(N1) << "-" << maxTSpan(N1); + if (DEBUG) cerr << "fw1=" << minTSpan(fw1) << "-" << maxTSpan(fw1) << endl; + if (orir==0) { + if (maxTSpan(fw1)<minTSpan(N1)) { // if N1 is monotone + orir = 1; + block = blockTarget(maxTSpan(fw1)+1,minTSpan(N1)); + if (block[0] <= fw1 && fw1 <= block[1]) orir+=2; + delete block; + } else {// if (maxTSpan(N1)<minTSpan(fw1)) { // if N1 is non-monotone + orir = 2; + block = blockTarget(maxTSpan(N1),minTSpan(fw1)-1); + if (block[0] <= fw1 && fw1 <= block[1]) orir+=2; + delete block; + } + } + } else { + orir = 5; + } + if (DEBUG) cerr << "orir = " << orir << endl; + int* block = blockSource(fw1,N1); + vector<int> ret; + ret.push_back(orir); ret.push_back(block[0]); ret.push_back(block[1]); + ret.push_back(block[2]); ret.push_back(block[3]); + delete block; + return ret; +} + +void Alignment::OrientationSource(int fw0, int fw1, int* oril, int* orir, bool Lcompute, bool Rcompute) { + // Orientation + // A bit tricky since fw can be 1) unaligned 2) aligned to many + // 1 -> MA, 2 -> RA, 3 -> MG, 4 -> RG, 5 -> Other + if (DEBUG) cerr << "OrientationSource(" << fw0 << "," << fw1 << ")" << endl; + if (!Lcompute && !Rcompute) return; + curr_al.push_back(fw0); + curr_al.push_back(fw1); + *oril=0; + *orir=0; + int lr=0; + if (oris_hash.find(curr_al)==oris_hash.end()) { + // Find first aligned word N0 to the left of fw + int N0=fw0-1; + while (N0>=0) { + if (minTSpan(N0)!=MINIMUM_INIT) break; + N0--; + } + int N1=fw1+1; + while (N1<_J) { + if (minTSpan(N1)!=MINIMUM_INIT) break; + N1++; + } + if (minTSpan(fw0)==MINIMUM_INIT && minTSpan(fw1)==MINIMUM_INIT) { + fw0 = N1; fw1 = N0; + //cerr << "minTSpan(fw)==MINIMUM_INIT, thus fw0=" << fw0 << ", fw1=" << fw1 << endl; + } + if (DEBUG) cerr << "fw0=" << fw0 << ", fw1=" << fw1 << ", N0=" << N0 << ", N1=" << N1 << endl; + if (maxTSpan(N0)<minTSpan(fw0) || maxTSpan(fw0)<minTSpan(N0)) { + if (DEBUG) cerr << "N0=" << minTSpan(N0) << "-" << maxTSpan(N0); + if (DEBUG) cerr << "fw=" << minTSpan(fw0) << "-" << maxTSpan(fw0) << endl; + int *block = blockTarget(minTSpan(N0),maxTSpan(N0)); + if (block[0]<=fw0 && fw0<=block[1]) *oril=5; + delete block; + if (*oril==0) { + block = blockTarget(minTSpan(fw0),maxTSpan(fw0)); + if (block[0]<=N0 && N0<=block[1]) *oril=5; + delete block; + } + if (*oril==0) { + if (maxTSpan(N0)<minTSpan(fw0)) {// if N0 is monotone + *oril=1; + block = blockTarget(maxTSpan(N0),minTSpan(fw0)-1); + if (block[0] <= fw0 && fw0 <= block[1]) *oril+=2; + delete block; + } else { //if (maxTSpan(fw0)<minTSpan(N0)) { // if NO is non-monotone + *oril=2; + block = blockTarget(maxTSpan(fw0)+1,minTSpan(N0)); + if (block[0] <= fw0 && fw0 <= block[1]) *oril+=2; + delete block; + } + } + } else { + *oril=5; + } + if (DEBUG) cerr << "oril =" << *oril << endl; + // Right neighbor + if (maxTSpan(N1)<minTSpan(fw1) || maxTSpan(fw1)<minTSpan(N1)) { + int* block = blockTarget(minTSpan(N1),maxTSpan(N1)); + if (block[0]<=fw1 && fw1<=block[2]) *orir=5; + delete block; + if (*orir==0) { + block = blockTarget(minTSpan(fw1),maxTSpan(fw1)); + if (block[0]<=N1 && N1 <=block[1]) *orir=5; + delete block; + } + if (DEBUG) cerr << "N1=" << minTSpan(N1) << "-" << maxTSpan(N1); + if (DEBUG) cerr << "fw1=" << minTSpan(fw1) << "-" << maxTSpan(fw1) << endl; + if (*orir==0) { + if (maxTSpan(fw1)<minTSpan(N1)) { // if N1 is monotone + *orir = 1; + block = blockTarget(maxTSpan(fw1)+1,minTSpan(N1)); + if (block[0] <= fw1 && fw1 <= block[1]) *orir+=2; + delete block; + } else {// if (maxTSpan(N1)<minTSpan(fw1)) { // if N1 is non-monotone + *orir = 2; + block = blockTarget(maxTSpan(N1),minTSpan(fw1)-1); + if (block[0] <= fw1 && fw1 <= block[1]) *orir+=2; + delete block; + } + } + } else { + *orir = 5; + } + if (DEBUG) cerr << "orir =" << *orir << endl; + lr = link(*oril,*orir); + oris_hash.insert(pair<vector<int>,int>(curr_al,lr)); + } else { + lr = oris_hash[curr_al]; + } + if (DEBUG) cerr << "Lcompute=" << Lcompute << ", Rcompute=" << Rcompute << endl; + if (Lcompute) *oril = source(lr); + if (Rcompute) *orir = target(lr); + curr_al.pop_back(); + curr_al.pop_back(); +} + +int Alignment::OrientationSource(int* left, int* right) { + if (DEBUG) { + cerr << " OrientationSource("; + cerr << "left="<<left[0]<<","<<left[1]<<","<<left[2]<<","<<left[3]; + cerr << " right="<<right[0]<<","<<right[1]<<","<<right[2]<<","<<right[3]; + cerr << ")" << endl; + } + //if ((right[1]<=left[0]) return 5; + if (!(left[1]<right[0])) return 5; + int ori = 1; + if (right[3]<left[2]) ori=2; + int gapstart = left[3]+1; int gapend = right[2]-1; + if (ori==2) { gapstart = right[3]+1; gapend = left[2]-1; } + for (int j=gapstart; j<=gapend; j++) { + if (sourceOf(j)!=-1) { + ori+=2; break; + } + } + return ori; +} + +void Alignment::OrientationTarget(int fw, int *oril, int *orir, bool Lcompute, bool Rcompute) { + OrientationTarget(fw,fw,oril,orir,Lcompute,Rcompute); +} + +vector<int> Alignment::OrientationTargetLeft4Sampler(int fw) { + return OrientationTargetLeft4Sampler(fw,fw); +} + +vector<int> Alignment::OrientationTargetLeft4Sampler(int fw0, int fw1) { + if (DEBUG) cerr << "OrientationTargetLeft4Sampler " << fw0 << "," << fw1 << endl; + int oril=0; + int N0=fw0-1; + while (N0>=0) { + if (minSSpan(N0)!=MINIMUM_INIT) break; + N0--; + } + int N1=fw1+1; + while (N1<_I) { + if (minSSpan(N1)!=MINIMUM_INIT) break; + N1++; + } + if (minSSpan(fw0)==MINIMUM_INIT && minSSpan(fw1)==MINIMUM_INIT) { + fw0=N1; fw1=N0; + } + if (maxSSpan(N0)<minSSpan(fw0) || maxSSpan(fw0)<minSSpan(N0)) { + int *block = blockSource(minSSpan(N0),maxSSpan(N0)); + if (DEBUG) cerr << "block1[" << block[0] << "," << block[1] << "," << block[2] << "," << block[3] << endl; + if (block[2]<=fw0 && fw0<=block[3]) //source span of fw0 subsumes NO's or the other way around + oril=5; + delete block; + if (oril==0) { + block = blockSource(minSSpan(fw0), maxSSpan(fw0)); + if (DEBUG) cerr << "block2[" << block[0] << "," << block[1] << "," << block[2] << "," << block[3] << endl; + if (block[2] <= N0 && N0 <= block[3]) oril=5; + delete block; + } + if (oril==0) { + if (maxSSpan(N0)<minSSpan(fw0)) {// if N0 is monotone + oril=1; + block = blockSource(maxSSpan(N0),minSSpan(fw0)-1); + if (DEBUG) cerr << "block3[" << block[0] << "," << block[1] << "," << block[2] << "," << block[3] << endl; + if (block[2] <= fw0 && fw0 <= block[3]) oril+=2; + delete block; + } else { // (maxSSpan(fw0)<minSSpan(N0)) // if NO is non-monotone + oril=2; + block = blockSource(maxSSpan(fw0)+1,minSSpan(N0)); + if (DEBUG) cerr << "block4[" << block[0] << "," << block[1] << "," << block[2] << "," << block[3] << endl; + if (block[2] <= fw0 && fw0 <= block[3]) oril+=2; + delete block; + } + } + } else { //source span of fw0 subsumes NO's or the other way around + oril=5; + } + if (DEBUG) cerr << "oril = " << oril << endl; + int* block = blockSource(N0,fw0); + vector<int> ret; + ret.push_back(oril); ret.push_back(block[0]); ret.push_back(block[1]); + ret.push_back(block[2]); ret.push_back(block[3]); + delete block; + return ret; +} + +vector<int> Alignment::OrientationTargetRight4Sampler(int fw) { + return OrientationTargetRight4Sampler(fw,fw); +} + +vector<int> Alignment::OrientationTargetRight4Sampler(int fw0, int fw1) { + if (DEBUG) cerr << "OrientationTargetRight4Sampler " << fw0 << "," << fw1 << endl; + int orir=0; + int N0=fw0-1; + while (N0>=0) { + if (minSSpan(N0)!=MINIMUM_INIT) break; + N0--; + } + int N1=fw1+1; + while (N1<_I) { + if (minSSpan(N1)!=MINIMUM_INIT) break; + N1++; + } + if (minSSpan(fw0)==MINIMUM_INIT && minSSpan(fw1)==MINIMUM_INIT) { + fw0=N1; fw1=N0; + } + if (maxSSpan(N1)<minSSpan(fw1) || maxSSpan(fw1)<minSSpan(N1)) { + int *block = blockSource(minSSpan(N1),maxSSpan(N1)); + if (block[2]<=fw1 && fw1<=block[3]) orir=5; + delete block; + if (orir==0) { + block = blockSource(minSSpan(fw1),maxSSpan(fw1)); + if (block[2] <= N1 && N1 <= block[3]) orir=5; + delete block; + } + if (orir==0) { + if (maxSSpan(fw1)<minSSpan(N1)) { // if N1 is monotone + orir=1; + block = blockSource(maxSSpan(fw1)+1,minSSpan(N1)); + if (block[2] <= fw1 && fw1 <= block[3]) orir+=2; + delete block; + } else { //if (maxSSpan(N1)<minSSpan(fw1)) { // if N1 is non-monotone + orir=2; + block = blockSource(maxSSpan(N1),minSSpan(fw1)-1); + if (block[2] <= fw1 && fw1 <= block[3]) orir+=2; + delete block; + } + } + } else { + orir=5; + } + if (DEBUG) cerr << "orir = " << orir << endl; + int* block = blockSource(fw1,N1); + vector<int> ret; + ret.push_back(orir); ret.push_back(block[0]); ret.push_back(block[1]); + ret.push_back(block[2]); ret.push_back(block[3]); + delete block; + return ret; + +} + +void Alignment::OrientationTarget(int fw0, int fw1, int*oril, int*orir, bool Lcompute, bool Rcompute) { + if (DEBUG) cerr << "OrientationTarget " << fw0 << "," << fw1 << endl; + // Left Neighbor + if (!Lcompute && !Rcompute) return; + *oril=0; + *orir=0; + curr_al.push_back(fw0); + curr_al.push_back(fw1); + int lr = 0; + if (orit_hash.find(curr_al)==orit_hash.end()) { + // Find first aligned word N0 to the left of fw + //int fw0 = fw; int fw1 = fw; + int N0=fw0-1; + while (N0>=0) { + if (minSSpan(N0)!=MINIMUM_INIT) break; + N0--; + } + int N1=fw1+1; + while (N1<_I) { + if (minSSpan(N1)!=MINIMUM_INIT) break; + N1++; + } + if (minSSpan(fw0)==MINIMUM_INIT && minSSpan(fw1)==MINIMUM_INIT) { + fw0=N1; fw1=N0; + } + if (DEBUG) { + cerr << "fw0:" << fw0 << ", fw1:" << fw1 << ", N0:" << N0 << ", N1:" << N1 << endl ; + cerr << "minSSpan(N0)=" << minSSpan(N0) << " maxSSpan(N0)=" << maxSSpan(N0); + cerr << " minSSpan(fw0)="<< minSSpan(fw0) << " maxSSpan(fw0)=" << maxSSpan(fw0) << endl; + cerr << "minSSpan(fw1)=" << minSSpan(fw1) << " maxSSpan(fw1)=" << maxSSpan(fw1); + cerr << " minSSpan(N1)="<< minSSpan(N1) << " maxSSpan(N1)=" << maxSSpan(N1) << endl; + } + if (maxSSpan(N0)<minSSpan(fw0) || maxSSpan(fw0)<minSSpan(N0)) { + int *block = blockSource(minSSpan(N0),maxSSpan(N0)); + if (DEBUG) cerr << "block1[" << block[0] << "," << block[1] << "," << block[2] << "," << block[3] << endl; + if (block[2]<=fw0 && fw0<=block[3]) //source span of fw0 subsumes NO's or the other way around + *oril=5; + delete block; + if (*oril==0) { + block = blockSource(minSSpan(fw0), maxSSpan(fw0)); + if (DEBUG) cerr << "block2[" << block[0] << "," << block[1] << "," << block[2] << "," << block[3] << endl; + if (block[2] <= N0 && N0 <= block[3]) *oril=5; + delete block; + } + if (*oril==0) { + if (maxSSpan(N0)<minSSpan(fw0)) {// if N0 is monotone + *oril=1; + block = blockSource(maxSSpan(N0),minSSpan(fw0)-1); + if (DEBUG) cerr << "block3[" << block[0] << "," << block[1] << "," << block[2] << "," << block[3] << endl; + if (block[2] <= fw0 && fw0 <= block[3]) *oril+=2; + delete block; + } else { // (maxSSpan(fw0)<minSSpan(N0)) // if NO is non-monotone + *oril=2; + block = blockSource(maxSSpan(fw0)+1,minSSpan(N0)); + if (DEBUG) cerr << "block4[" << block[0] << "," << block[1] << "," << block[2] << "," << block[3] << endl; + if (block[2] <= fw0 && fw0 <= block[3]) *oril+=2; + delete block; + } + } + } else { //source span of fw0 subsumes NO's or the other way around + *oril=5; + } + if (DEBUG) cerr << "oril = " << *oril << endl; + // Right Neighbor + if (maxSSpan(N1)<minSSpan(fw1) || maxSSpan(fw1)<minSSpan(N1)) { + int *block = blockSource(minSSpan(N1),maxSSpan(N1)); + if (block[2]<=fw1 && fw1<=block[3]) *orir=5; + delete block; + if (*orir==0) { + block = blockSource(minSSpan(fw1),maxSSpan(fw1)); + if (block[2] <= N1 && N1 <= block[3]) *orir=5; + delete block; + } + if (*orir==0) { + if (maxSSpan(fw1)<minSSpan(N1)) { // if N1 is monotone + *orir=1; + block = blockSource(maxSSpan(fw1)+1,minSSpan(N1)); + if (block[2] <= fw1 && fw1 <= block[3]) *orir+=2; + delete block; + } else { //if (maxSSpan(N1)<minSSpan(fw1)) { // if N1 is non-monotone + *orir=2; + block = blockSource(maxSSpan(N1),minSSpan(fw1)-1); + if (block[2] <= fw1 && fw1 <= block[3]) *orir+=2; + delete block; + } + } + } else { + *orir=5; + } + if (DEBUG) cerr << "orir = " << *orir << endl; + lr = link(*oril,*orir); + orit_hash.insert(pair<vector<int>,int>(curr_al,lr)); + } else { + lr = orit_hash[curr_al]; + } + if (DEBUG) cerr << "Lcompute=" << Lcompute << ", Rcompute=" << Rcompute << endl; + if (DEBUG) cerr << "lr=" << lr << ", l=" << source(lr) << ", r=" << target(lr) << endl; + if (Lcompute>0) *oril=source(lr); + if (Rcompute>0) *orir=target(lr); + curr_al.pop_back(); + curr_al.pop_back(); +} + +int* Alignment::blockSource(int idx1, int idx2) { +// outputs a minimal block [s1,s2,t1,t2] that contains idx1 and idx2, where idx1 <= idx2 + if (DEBUG) cerr << "blockSource[" << idx1 << "," << idx2 << "]" << endl; + int *curr = new int[4]; + curr[0]=idx1; curr[1]=idx2; curr[2]=MINIMUM_INIT; curr[3]=MAXIMUM_INIT; + for (int j=curr[0]; j<=curr[1]; j++) { + curr[2] = least(curr[2],_tSpan[j][0]); + curr[3] = most(curr[3],_tSpan[j][1]); + } + int next[4]; + next[0]=curr[0]; next[1]=curr[1]; + for (int i=curr[2]; i<=curr[3]; i++) { + next[0] = least(next[0],_sSpan[i][0]); + next[1] = most(next[1],_sSpan[i][1]); + } + next[2] = curr[2]; next[3]= curr[3]; + int idx=1; + do { + // update the current + for (int j=next[0]; j<curr[0]; j++) { + curr[2] = least(curr[2],_tSpan[j][0]); + curr[3] = most(curr[3],_tSpan[j][1]); + } + for (int j=curr[1]+1; j<=next[1]; j++) { + curr[2] = least(curr[2],_tSpan[j][0]); + curr[3] = most(curr[3],_tSpan[j][1]); + } + curr[0] = next[0]; curr[1] = next[1]; + if (curr[2]==next[2] && curr[3]==next[3]) break; + // prepare for the next + for (int i=curr[2]; i<next[2]; i++) { + next[0]= least(next[0],_sSpan[i][0]); + next[1]= most(next[1],_sSpan[i][1]); + } + for (int i=next[3]+1; i<=curr[3]; i++) { + next[0] = least(next[0],_sSpan[i][0]); + next[1] = most(next[1],_sSpan[i][1]); + } + next[2] = curr[2]; next[3]= curr[3]; + idx++; + } while(1); + return curr; +} + +int* Alignment::blockTarget(int idx1, int idx2) { +// outputs a minimal [s1,s2,t1,t2] that contains idx1 and idx2, where idx1<=idx2 + int *curr = new int[4]; + curr[0]=MINIMUM_INIT; curr[1]=MAXIMUM_INIT; curr[2]=idx1; curr[3]=idx2; + for (int i=curr[2]; i<=curr[3]; i++) { + curr[0] = least(curr[0],_sSpan[i][0]); + curr[1] = most(curr[1],_sSpan[i][1]); + } + int next[4]; + next[2]=curr[2]; next[3]=curr[3]; + for (int j=curr[0]; j<=curr[1]; j++) { + next[2] = least(next[2],_tSpan[j][0]); + next[3] = most(next[3],_tSpan[j][1]); + } + next[0] = curr[0]; next[1]= curr[1]; + int idx=1; + do { + // update the current + for (int i=next[2]; i<curr[2]; i++) { + curr[0] = least(curr[0],_sSpan[i][0]); + curr[1] = most(curr[1],_sSpan[i][1]); + } + for (int i=curr[3]+1; i<=next[3]; i++) { + curr[0] = least(curr[0],_sSpan[i][0]); + curr[1] = most(curr[1],_sSpan[i][1]); + } + curr[2] = next[2]; curr[3] = next[3]; + if (curr[0]==next[0] && curr[1]==next[1]) break; + // prepare for the next + for (int j=curr[0]; j<next[0]; j++) { + next[2]= least(next[2],_tSpan[j][0]); + next[3]= most(next[3],_tSpan[j][1]); + } + for (int j=next[1]+1; j<=curr[1]; j++) { + next[2] = least(next[2],_tSpan[j][0]); + next[3] = most(next[3],_tSpan[j][1]); + } + next[0] = curr[0]; next[1]= curr[1]; + idx++; + } while(1); + return curr; +} + +int Alignment::firstSourceAligned(int start) { + for (int j=start; j<_J; j++) + if (_tSpan[j][0]!=MINIMUM_INIT) return j; + return -1; +} + +int Alignment::lastSourceAligned(int end) { + for (int j=end; j>=0; j--) + if (_tSpan[j][0]!=MINIMUM_INIT) return j; + return -1; +} + +int Alignment::firstTargetAligned(int start) { + for (int i=start; i<_I; i++) + if (_sSpan[i][0]!=MINIMUM_INIT) return i; + return -1; +} + +int Alignment::lastTargetAligned(int end) { + for (int i=end; i>=0; i--) + if (_sSpan[i][0]!=MINIMUM_INIT) return i; + return -1; +} + +void Alignment::BorderingSFWsOnly() { +// removes the record of all function word alignments, except those at the borders +// the number of alignments kept may be more than two +// i.e. where the leftmost / the rightmost alignments are unaligned. +// In such cases, this function continues keeping function word alignments until the +// first (or last) alignment words. + if (SourceFWIdxs[0]>2) { + int firstCut = 1; + for (int j=2; j<=SourceFWIdxs[0]; j++) { + if (SourceFWIdxs[3*j-2]>fas) break; + firstCut=j; + } + int lastCut = SourceFWIdxs[0]; + for (int j=SourceFWIdxs[0]-1; j>=0; j--) { + if (SourceFWIdxs[3*j-2]<las) break; + lastCut=j; + } + if (firstCut>=lastCut) return; + int delta = 0; + for (int j=lastCut; j<=SourceFWIdxs[0]; j++) { + delta++; + SourceFWIdxs[3*(firstCut+delta)-2]=SourceFWIdxs[3*j-2]; + SourceFWIdxs[3*(firstCut+delta)-1]=SourceFWIdxs[3*j-1]; + SourceFWIdxs[3*(firstCut+delta)] =SourceFWIdxs[3*j]; + } + SourceFWIdxs[0]=firstCut+delta; + } +} + +void Alignment::BorderingTFWsOnly() { +// similar to BorderingSFWsOnly() except this looks at the source side. + if (TargetFWIdxs[0]>2) { + int firstCut = 1; + for (int j=2; j<=TargetFWIdxs[0]; j++) { + if (TargetFWIdxs[3*j-2]>fat) break; + firstCut=j; + } + int lastCut = TargetFWIdxs[0]; + for (int j=TargetFWIdxs[0]-1; j>=0; j--) { + if (TargetFWIdxs[3*j-2]<lat) break; + lastCut=j; + } + if (firstCut>=lastCut) return; + int delta = 0; + for (int j=lastCut; j<=TargetFWIdxs[0]; j++) { + delta++; + TargetFWIdxs[3*(firstCut+delta)-2]=TargetFWIdxs[3*j-2]; + TargetFWIdxs[3*(firstCut+delta)-1]=TargetFWIdxs[3*j-1]; + TargetFWIdxs[3*(firstCut+delta)] =TargetFWIdxs[3*j]; + } + TargetFWIdxs[0]=firstCut+delta; + } +} + +void Alignment::FillFWIdxsState(int* state, int fas, int las, int fat, int lat) { + if (DEBUG) cerr << "FillFWIdxsState ("<< fas <<","<< las<<"," << fat <<"," << lat << ")" << endl; + if (fas==las) las+=1; + if (fat==lat) lat+=1; + for (int idx=0; idx<12; idx++) state[idx]=-1; + if (SourceFWIdxs[0]<=2) { + if (SourceFWIdxs[0]>=1) {state[0]=SourceFWIdxs[1]; state[1]=SourceFWIdxs[2]; state[2]=SourceFWIdxs[3];} + if (SourceFWIdxs[0]==2) {state[3]=SourceFWIdxs[4]; state[4]=SourceFWIdxs[5]; state[5]=SourceFWIdxs[6];} + } else { + if (SourceFWIdxs[1]>fas) { + state[0]=SourceFWIdxs[1]; state[1]=SourceFWIdxs[2]; state[2]=SourceFWIdxs[3]; + } else { + ostringstream issf; ostringstream isse; + for (int idx=1; idx<=SourceFWIdxs[0]; idx++) { + if (SourceFWIdxs[3*idx-2]>las) break; + if (idx>1) { issf << " "; isse << " ";}; + issf << TD::Convert(SourceFWIdxs[3*idx-1]); + isse << TD::Convert(SourceFWIdxs[3*idx]); + state[0]=SourceFWIdxs[3*idx-2]; + if (state[0]>=fas) break; + } + if (state[0]>=0) { + state[1]=TD::Convert(issf.str())*-1; state[2]=TD::Convert(isse.str()); //multiplying source with -1 as marker + } + } + if (SourceFWIdxs[SourceFWIdxs[0]*3-2]==las) { + state[3]=SourceFWIdxs[SourceFWIdxs[0]*3-2]; + state[4]=SourceFWIdxs[SourceFWIdxs[0]*3-1]; + state[5]=SourceFWIdxs[SourceFWIdxs[0]*3]; + } else { + int lastCut = SourceFWIdxs[0]; + for (int j=lastCut-1; j>=state[0]+1; j--) { + if (SourceFWIdxs[3*j-2]==state[0]) break; + if (SourceFWIdxs[3*j-2]<las) break; + lastCut=j; + } + state[3]=SourceFWIdxs[3*lastCut-2]; + ostringstream issf; ostringstream isse; + for (int idx=lastCut; idx<=SourceFWIdxs[0]; idx++) { + if (idx>lastCut) { issf << " "; isse << " ";}; + issf << TD::Convert(SourceFWIdxs[3*idx-1]); + isse << TD::Convert(SourceFWIdxs[3*idx]); + } + if (state[3]>=0) { + //multiplying source with -1 as compound marker + state[4]=TD::Convert(issf.str())*-1; state[5]=TD::Convert(isse.str()); + } + } + } + if (TargetFWIdxs[0]<=2) { + if (TargetFWIdxs[0]>=1) {state[6]=TargetFWIdxs[1]; state[7]=TargetFWIdxs[2]; state[8]=TargetFWIdxs[3];} + if (TargetFWIdxs[0]==2) {state[9]=TargetFWIdxs[4]; state[10]=TargetFWIdxs[5]; state[11]=TargetFWIdxs[6];} + } else { + if (TargetFWIdxs[1]>fat) { //shouldn't come here if SetTargetBorderingFW is invoked + state[6]=TargetFWIdxs[1]; state[7]=TargetFWIdxs[2]; state[8]=TargetFWIdxs[3]; + } else { + ostringstream issf; ostringstream isse; + for (int idx=1; idx<=TargetFWIdxs[0]; idx++) { + if (TargetFWIdxs[3*idx-2]>fat) break; + if (idx>1) { issf << " "; isse << " ";}; + issf << TD::Convert(TargetFWIdxs[3*idx-1]); + isse << TD::Convert(TargetFWIdxs[3*idx]); + state[6]=TargetFWIdxs[3*idx-2]; + } + state[7]=TD::Convert(issf.str()); state[8]=TD::Convert(isse.str())*-1; + //multiplying target with -1 as compound marker + } + if (TargetFWIdxs[TargetFWIdxs[0]*3-2]==lat) { + state[9]=TargetFWIdxs[TargetFWIdxs[0]*3-2]; + state[10]=TargetFWIdxs[TargetFWIdxs[0]*3-1]; + state[11]=TargetFWIdxs[TargetFWIdxs[0]*3]; + } else { + int lastCut = TargetFWIdxs[0]; + for (int j=lastCut-1; j>=1; j--) { + if (TargetFWIdxs[3*j-2]<=state[9]) break; + if (TargetFWIdxs[3*j-2]<lat) break; + lastCut=j; + } + state[9]=TargetFWIdxs[3*lastCut-2]; + ostringstream issf; ostringstream isse; + for (int idx=lastCut; idx<=TargetFWIdxs[0]; idx++) { + if (idx>lastCut) issf << " "; isse << " ";; + issf << TD::Convert(TargetFWIdxs[3*idx-1]); + isse << TD::Convert(TargetFWIdxs[3*idx]); + } + state[10]=TD::Convert(issf.str()); state[11]=TD::Convert(isse.str())*-1; + } + } +} + +void Alignment::simplifyBackward(vector<int *>*blocks, int* block, const vector<int>& danglings) { +// given a *block*, see whether its target span contains any index inside *danglings*. +// if yes, break it; otherwise, keep it. put the result(s) to *blocks* + if (DEBUG) cerr << "simplifyBackward[" << block[0] << "," << block[1] << "," << block[2] << "," << block[3] << "]" << endl; + if (DEBUG) for (int i=0; i<danglings.size(); i++) cerr << "danglings[" << i << "] = " << danglings[i] << endl; + if (danglings.size()==0) { + blocks->push_back(block); + if (DEBUG) cerr << "pushing(0) " << block[0] << "," << block[1] << "," << block[2] << "," << block[3] << endl; + return; + } + int currIdx = block[2]; + int i_dangling = 0; + while (block[2]>danglings[i_dangling]) { + if (i_dangling+1 >= danglings.size()) break; + i_dangling++; + } + while (danglings[i_dangling]==currIdx) { + i_dangling++; + currIdx++; + } + /*if (i_dangling>=danglings.size() && currIdx) { + blocks->push_back(block); + if (DEBUG) cerr << "pushing(1) " << block[0] << "," << block[1] << "," << block[2] << "," << block[3] << endl; + return; + } + if (block[3]<danglings[i_dangling]) { + blocks->push_back(block); + if (DEBUG) cerr << "pushing(2) " << block[0] << "," << block[1] << "," << block[2] << "," << block[3] << endl; + return; + }*/ + if (DEBUG) cerr << "i_dangling = " << i_dangling << endl; + int anchorIdx = danglings[i_dangling]; + if (i_dangling+1>=danglings.size() || anchorIdx>block[3]+1) anchorIdx=block[3]+1; + if (DEBUG) cerr << "anchorIdx = " << anchorIdx << ", currIdx = " << currIdx << endl; + do { + while(currIdx<anchorIdx) { + if (DEBUG) cerr << "currIdx = " << currIdx << ", anchorIdx = " << anchorIdx << endl; + bool isMoved = false; + for (int idx=anchorIdx-1; idx>=currIdx; idx--) { + int *nublock = blockTarget(currIdx,idx); + if (nublock[2]==currIdx && nublock[3]==idx) { + if (nublock[0]!=MINIMUM_INIT) { + blocks->push_back(nublock); + if (DEBUG) cerr << "pushing(3) " << nublock[0] << "," << nublock[1] << "," << nublock[2] << "," << nublock[3] << endl; + } else { + delete nublock; + } + isMoved = true; + currIdx=idx+1; break; + } else { + delete nublock; + } + } + if (DEBUG) cerr << "isMoved=" << isMoved << ", currIdx=" << currIdx << endl; + if (!isMoved) { + int source = sourceOf(currIdx); + while (source>=0) { + if (source >= block[0]) { + int* nublock = new int[4]; + nublock[0]=source; nublock[1]=source; nublock[2]=currIdx; nublock[3]=currIdx; + blocks->push_back(nublock); + if (DEBUG) cerr << "pushing(4) " << nublock[0] << "," << nublock[1] << "," << nublock[2] << "," << nublock[3] << endl; + } + source = sourceOf(currIdx,source+1); + } + currIdx++; + } + } + currIdx=anchorIdx+1; + anchorIdx=block[3]+1; + if (i_dangling+1<danglings.size()) anchorIdx=danglings[++i_dangling]; + } while(currIdx<=block[3]); +} + +void Alignment::simplify(int* ret) { + // the idea is to create blocks of maximal consistent alignment in between a pair of function words + // exceptional cases include: one to non-contiguous many (or vice versa) -> treat this as one alignment each + // record all function word alignments first, important because it may be unaligned + // return true if it's truly simple (no function word alignment involves); false, otherwise + if (DEBUG) cerr << "begin simplify" << endl; + reset(0,0); reset(_J-1,_I-1); // remove the phrase boundary alignments, NEED TO CHECK AGAIN !!! + if (SourceFWIdxs[0]+TargetFWIdxs[0]==0) { // return singleton + if (DEBUG) cerr << "no function words" << endl; + for (int idx=0; idx<12; idx++) ret[idx]=-1; + ret[12]=1; ret[13]=0; ret[14]=0; // 0-0 + FillFWIdxsState(ret,0,0,0,0); + return; + } + curr_al.insert(curr_al.begin(),curr_al.size()); + curr_al.push_back(SourceFWIdxs[0]); + for (int i=1; i<=SourceFWIdxs[0]; i++) curr_al.push_back(SourceFWIdxs[3*i-2]); + curr_al.push_back(TargetFWIdxs[0]); + for (int i=1; i<=TargetFWIdxs[0]; i++) curr_al.push_back(TargetFWIdxs[3*i-2]); + vector<int> el; + if (simplify_hash.find(curr_al)==simplify_hash.end()) { + if (DEBUG) { + cerr << "SourceFWIdxs:" << SourceFWIdxs[0] << endl; + for (int i=1; i<=SourceFWIdxs[0]; i++) + cerr << SourceFWIdxs[3*i-2] << "," << SourceFWIdxs[3*i-1] << "," << SourceFWIdxs[3*i] << endl; + cerr << "TargetFWIdxs:" << TargetFWIdxs[0] << endl; + for (int i=1; i<=TargetFWIdxs[0]; i++) { + cerr << TargetFWIdxs[3*i-2] << "," << TargetFWIdxs[3*i-1] << "," << TargetFWIdxs[3*i] << endl; + } + } + + vector< int* > blocks; // each element contains s1,s2,t1,t2 + int currIdx = 1; // start from 1 to avoid considering phrase start + std::set<int> FWIdxs; + std::vector<int> DanglingTargetFWIdxs; + for (int i=1; i<= SourceFWIdxs[0]; i++) FWIdxs.insert(SourceFWIdxs[3*i-2]); + for (int i=1; i<= TargetFWIdxs[0]; i++) { + int source = sourceOf(TargetFWIdxs[3*i-2]); + if (source>=0) { + do { + FWIdxs.insert(source); + source = sourceOf(TargetFWIdxs[3*i-2],source+1); + } while(source >=0); + } else { + int *block = new int[4]; + block[0]=-1; block[1]=-1; block[2]=TargetFWIdxs[3*i-2]; block[3]=TargetFWIdxs[3*i-2]; + blocks.push_back(block); + if (DEBUG) cerr << "pushing[1] " << block[0] << "," << block[1] << "," << block[2] << "," << block[3] << endl; + DanglingTargetFWIdxs.push_back(TargetFWIdxs[3*i-2]); + } + } + if (DEBUG) + for (std::set<int>::const_iterator iter=FWIdxs.begin(); iter!=FWIdxs.end(); iter++) { + cerr << "FWIdxs=" << *iter << endl; + } + std::set<int>::const_iterator currFWIdx = FWIdxs.begin(); + if (currFWIdx == FWIdxs.end()) { + int* block = new int[4]; + block[0]=1; block[1]=_J-2; block[2]=1; block[3]=_I-2; // no need to consider phrase boundaries + simplifyBackward(&blocks,block,DanglingTargetFWIdxs); + } else { + int anchorIdx = *currFWIdx; // also used to denote _J+1 + do { + // add alignments whose source from currIdx to currFWIdx-1 + while (currIdx<anchorIdx) { + bool isMoved = false; + //cerr << "anchorIdx = " << anchorIdx << ", currIdx = " << currIdx << endl; + for (int idx=anchorIdx-1; idx>=currIdx; idx--) { + int* block = blockSource(currIdx,idx); + if (block[0]==currIdx&&block[1]==idx) { + if (block[2]!=MINIMUM_INIT) { // must be aligned + simplifyBackward(&blocks,block,DanglingTargetFWIdxs); + } else { + delete block; + } + currIdx = idx+1; isMoved = true; + break; + } else { + delete block; + } + } + if (!isMoved) { + int target = targetOf(currIdx); + while (target>=0) { + int* block = new int[4]; + block[0]=currIdx; block[1]=currIdx; block[2]=target; block[3]=target; + blocks.push_back(block); + if (DEBUG) cerr << "pushing[2] " << block[0] << "," << block[1] << "," << block[2] << "," << block[3] << endl; + target = targetOf(currIdx,target+1); + } + currIdx++; + } + } + // add function word alignments (anchorIdx) + if (anchorIdx==getJ()) break; + int target = targetOf(anchorIdx); + do { + int* block = new int[4]; + block[0]=anchorIdx; block[1]=anchorIdx; block[2]=target; block[3]=target; + blocks.push_back(block); + if (DEBUG) cerr << "pushing[3] " << block[0] << "," << block[1] << "," << block[2] << "," << block[3] << endl; + if (target>=0) target = targetOf(anchorIdx,target+1); + } while (target>=0); + // advance indexes + currIdx = anchorIdx+1; + anchorIdx = getJ()-1; // was minus 2 + if (++currFWIdx!=FWIdxs.end()) anchorIdx = *currFWIdx; + } while (currIdx<=getJ()-2); + } + + + vector<int> source_block_mapper(getJ(),-1); + vector<int> target_block_mapper(getI(),-1); + for (int i = 0; i<blocks.size(); i++) { + if (DEBUG) cerr << "blocks[" << i << "]=" << blocks[i][0] << "," << blocks[i][1] << "," << blocks[i][2] << "," << blocks[i][3] << endl; + if (blocks[i][0]>=0) source_block_mapper[blocks[i][0]]=1; + if (blocks[i][2]>=0) target_block_mapper[blocks[i][2]]=1; + } + int curr = 1; + int prev = -1; + for (int idx=0; idx<source_block_mapper.size(); idx++) { + if (source_block_mapper[idx]>0) { + source_block_mapper[idx]=curr++; + prev = curr; + } else { + source_block_mapper[idx]=prev; + } + } + curr = 1; + for (int idx=0; idx<target_block_mapper.size(); idx++) { + if (target_block_mapper[idx]>0) { + target_block_mapper[idx]=curr++; + prev = curr; + } else { + target_block_mapper[idx]=prev; + } + } + + //assert(blocks.size()<=50); + if (DEBUG) cerr << "resulting alignment:" << endl; + for (int i = 0; i<blocks.size(); i++) { + if (blocks[i][2]<0 || blocks[i][0]<0) continue; + int source = source_block_mapper[blocks[i][0]]-1; + int target = target_block_mapper[blocks[i][2]]-1; + el.push_back(link(source,target)); + if (DEBUG) cerr << source << "-" << target << " "; + } + el.insert(el.begin(),el.size()); + if (DEBUG) cerr << endl; + el.push_back(SourceFWIdxs[0]); + for (int idx=1; idx<=SourceFWIdxs[0]; idx++) { + if (DEBUG) cerr << "SourceFWIdxs[" << (3*idx-2) << "] from " << SourceFWIdxs[3*idx-2] << endl; + el.push_back(source_block_mapper[SourceFWIdxs[3*idx-2]]-1); + } + el.push_back(TargetFWIdxs[0]); + for (int idx=1; idx<=TargetFWIdxs[0]; idx++) { + if (DEBUG) cerr << "TargetFWIdxs[" << (3*idx-2) << "] from " << TargetFWIdxs[3*idx-2] << endl; + el.push_back(target_block_mapper[TargetFWIdxs[3*idx-2]]-1); + } + el.push_back(source_block_mapper[fas]-1); + el.push_back(source_block_mapper[las]-1); + el.push_back(target_block_mapper[fat]-1); + el.push_back(target_block_mapper[lat]-1); + if (DEBUG) { + cerr << "insert key:el = "; + for (int ii=0; ii<el.size(); ii++) + cerr << ii << "." << el[ii] << " "; + cerr << " || " << endl; + } + if (DEBUG) cerr << "trying to insertL " << endl; + if (DEBUG) { + cerr << "size=" << curr_al.size() << " "; + for (int ii=0; ii<curr_al.size(); ii++) cerr << "curr_al[" << ii << "]=" << curr_al[ii] << " "; + cerr << endl; + } + simplify_hash.insert(pair<vector<int>, vector<int> > (curr_al,el)); + if (DEBUG) cerr << "inserted" << endl; + } else { + el = simplify_hash[curr_al]; + } + if (DEBUG) { + cerr << "pull key:el = "; + for (int ii=0; ii<el.size(); ii++) + cerr << ii << "." << el[ii] << " "; + cerr << endl; + } + ret[12] = el[0]; + for (int i=1; i<=el[0]; i++) ret[12+i] = el[i]; + int istart = el[0]+1; + assert(el[istart]==SourceFWIdxs[0]); + for (int i=1; i<=el[istart]; i++) SourceFWIdxs[3*i-2]=el[istart+i]; + istart += el[istart]+1; + assert(el[istart]==TargetFWIdxs[0]); + for (int i=1; i<=el[istart]; i++) TargetFWIdxs[3*i-2]=el[istart+i]; + istart += el[istart]+1; + FillFWIdxsState(ret,el[istart],el[istart+1],el[istart+2],el[istart+3]); +} + +void Alignment::simplify_nofw(int* ret) { + for (int i=0; i<12; i++) ret[i]=-1; + ret[12]=1; ret[13]=0; +} + +void Alignment::sort(int* num) { + if (num[0]>1) quickSort(num,1,num[0]); +} + +void Alignment::quickSort(int arr[], int left, int right) { + int i = left, j = right; + int tmp1,tmp2,tmp3; + int mid = (left + right) / 2; + int pivot = arr[3*mid-2]; + + /* partition */ + while (i <= j) { + while (arr[3*i-2] < pivot) i++; + while (arr[3*j-2] > pivot) j--; + if (i <= j) { + tmp1 = arr[3*i-2]; tmp2 = arr[3*i-1]; tmp3 = arr[3*i]; + arr[3*i-2] = arr[3*j-2]; arr[3*i-1] = arr[3*j-1]; arr[3*i] = arr[3*j]; + arr[3*j-2] = tmp1; arr[3*j-1] = tmp2; arr[3*j] = tmp3; + i++; + j--; + } + }; + + /* recursion */ + if (left < j) quickSort(arr, left, j); + if (i < right) quickSort(arr, i, right); +} + +double Alignment::ScoreOrientation(const CountTable& table, int offset, int ori, WordID cond1, WordID cond2) { + string source = TD::Convert(cond1); + string sourceidx; + if (table.mode == 1) { + sourceidx = source; + int slashidx = sourceidx.find_last_of("/"); + source = sourceidx.substr(0,slashidx); + string idx = sourceidx.substr(slashidx+1); + if (DEBUG) cerr << " sourceidx = " << sourceidx << ", idx = " << idx << endl; + if (idx == "X") { + if (DEBUG) cerr << " idx == X, returning 0" << endl; + return 0; + } + } + string target = TD::Convert(cond2); + if (DEBUG) cerr << "sourceidx='" << sourceidx << "', source='" << source << "', target='" << target << "'" << endl; + double count = table.ultimate[offset+ori-1]; + double total = table.ultimate[offset+5]; + double alpha = 0.1; + double prob = count/total; + if (DEBUG) cerr << "level0 " << count << "/" << total << "=" << prob << endl; + + WordID key_id = (table.mode!=1) ? cond1 : TD::Convert(source); + map<WordID,int*>::const_iterator it = table.model.find(key_id); + bool stop = (it==table.model.end()); + if (!stop) { + stop=true; + if (it->second[offset+5]>=0) { + count = it->second[offset+ori-1] + alpha * prob; + total = it->second[offset+5] + alpha; + prob = count/total; + stop = false; + if (DEBUG) cerr << "level1 " << count << "/" << total << "=" << prob << endl; + } + } + if (stop) return prob; + + string key = source + " " + target; + it = table.model.find(TD::Convert(key)); + stop = (it==table.model.end()); + if (!stop) { + stop = true; + if (it->second[offset+5]>=0) { + count = it->second[offset+ori-1] + alpha * prob; + total = it->second[offset+5] + alpha; + prob = count/total; + stop = false; + if (DEBUG) cerr << "level2 " << count << "/" << total << "=" << prob << endl; + } + } + + if (stop || table.mode!=1) return prob; + + key = sourceidx + " " + target; + it = table.model.find(TD::Convert(key)); + if (it!=table.model.end()) { + if (it->second[offset+5]>=0) { + count = it->second[offset+ori-1] + alpha * prob; + total = it->second[offset+5] + alpha; + prob = count/total; + if (DEBUG) cerr << "level3 " << count << "/" << total << "=" << prob << endl; + } + } + + return prob; +} + +void Alignment::ScoreOrientation(const CountTable& table, int offset, int ori, WordID cond1, WordID cond2, + bool isBonus, double *cost, double *bonus, double *bo1, double *bo1_bonus, double *bo2, double *bo2_bonus, + double alpha1, double beta1) { + if (DEBUG) cerr << "ScoreOrientation:" << TD::Convert(cond1) << "," << TD::Convert(cond2) << ", alpha1 = " << alpha1 << ", beta1 = " << beta1 << endl; + double ret = ScoreOrientation(table,offset,ori,cond1,cond2); + if (isBonus) { + if (table.mode == 0) *bonus += log(ret); else *bonus += ret; + } else { + if (table.mode == 0) *cost += log(ret); else *cost += ret; + } +} + +double Alignment::ScoreOrientationLeft(const CountTable& table, int ori, WordID cond1, WordID cond2) { + double ret = ScoreOrientation(table,0,ori,cond1,cond2); + if (table.mode == 0) return log(ret); + return ret; +} + +double Alignment::ScoreOrientationLeftBackward(const CountTable& table, int ori, WordID cond1, WordID cond2) { + double ret = ScoreOrientation(table,12,ori,cond1,cond2); + if (table.mode == 0) return log(ret); + return ret; +} + +double Alignment::ScoreOrientationRight(const CountTable& table, int ori, WordID cond1, WordID cond2) { + double ret = ScoreOrientation(table,6,ori,cond1,cond2); + if (table.mode == 0) return log(ret); + return ret; +} + +double Alignment::ScoreOrientationRightBackward(const CountTable& table, int ori, WordID cond1, WordID cond2) { + double ret = ScoreOrientation(table,18,ori,cond1,cond2); + if (table.mode == 0) return log(ret); + return ret; +} + +void Alignment::ScoreOrientationLeft(const CountTable& table, int ori, WordID cond1, WordID cond2, + bool isBonus, double *cost, double *bonus, double *bo1, double *bo1_bonus, double *bo2, double *bo2_bonus, double alpha1, double beta1) { + if (DEBUG) cerr << "ScoreOrientationLeft(" << isBonus << ")" << endl; + ScoreOrientation(table,0,ori,cond1,cond2,isBonus,cost,bonus,bo1,bo1_bonus,bo2,bo2_bonus,alpha1,beta1); +} + +void Alignment::ScoreOrientationLeftBackward(const CountTable& table, int ori, WordID cond1, WordID cond2, + bool isBonus, double *cost, double *bonus, double *bo1, double *bo1_bonus, double *bo2, double *bo2_bonus, double alpha1, double beta1) { + if (DEBUG) cerr << "ScoreOrientationLeftBackward" << endl; + ScoreOrientation(table,12,ori,cond1,cond2,isBonus,cost,bonus,bo1,bo1_bonus,bo2,bo2_bonus,alpha1,beta1); +} + +void Alignment::ScoreOrientationRight(const CountTable& table, int ori, WordID cond1, WordID cond2, + bool isBonus, double *cost, double *bonus, double *bo1, double *bo1_bonus, double *bo2, double *bo2_bonus, double alpha1, double beta1) { + if (DEBUG) cerr << "ScoreOrientationRight(" << isBonus << ")" << endl; + ScoreOrientation(table,6,ori,cond1,cond2,isBonus,cost,bonus,bo1,bo1_bonus,bo2,bo2_bonus,alpha1,beta1); +} + +void Alignment::ScoreOrientationRightBackward(const CountTable& table, int ori, WordID cond1, WordID cond2, + bool isBonus, double *cost, double *bonus, double *bo1, double *bo1_bonus, double *bo2, double *bo2_bonus, double alpha1, double beta1) { + if (DEBUG) cerr << "ScoreOrientationRightBackward" << endl; + ScoreOrientation(table,18,ori,cond1,cond2,isBonus,cost,bonus,bo1,bo1_bonus,bo2,bo2_bonus,alpha1,beta1); +} + +void Alignment::computeOrientationSourceBackwardPos(const CountTable& table, double *cost, double *bonus, + double *bo1, double *bo1_bonus, double *bo2, double *bo2_bonus, int maxfwidx, int maxdepth1, int maxdepth2) { + if (DEBUG) cerr << "computeOrientationSourceBackward" << endl; + int oril, orir; + for (int idx=1; idx<=SourceFWRuleIdxs[0]; idx++) { + if (DEBUG) cerr << "considering SourceFWRuleIdxs[" << idx << "]: " << SourceFWRuleIdxs[3*idx-2] << endl; + if (!(SourceFWRuleAbsIdxs[idx]<=maxdepth1 || maxfwidx-SourceFWRuleAbsIdxs[idx]+1<=maxdepth2)) continue; + int* fwblock = blockSource(SourceFWRuleIdxs[3*idx-2],SourceFWRuleIdxs[3*idx-2]); + bool aligned = (fwblock[2]!=MINIMUM_INIT); + if (aligned) { + OrientationTarget(fwblock[2],fwblock[3],&oril,&orir); + } else { + OrientationSource(SourceFWRuleIdxs[3*idx-2],&oril,&orir); + } + if (DEBUG) cerr << "oril = " << oril << ", orir = " << orir << endl; + bool isBonus = false; // fas -> first aligned source word, las -> last aligned source word + if ((aligned && fwblock[2]<=fat)|| + (!aligned && SourceFWRuleIdxs[3*idx-2]<=fas)) isBonus=true; + if (SourceFWRuleAbsIdxs[idx]<=maxdepth1) { + ostringstream nusource; + nusource << TD::Convert(SourceFWRuleIdxs[3*idx-1]) << "/" << SourceFWRuleAbsIdxs[idx]; + ScoreOrientationLeftBackward(table,oril,TD::Convert(nusource.str()),SourceFWRuleIdxs[3*idx], + isBonus,cost,bonus,bo1,bo1_bonus,bo2,bo2_bonus,alpha_oris,beta_oris); + } + if (maxfwidx-SourceFWRuleAbsIdxs[idx]+1<=maxdepth2) { + ostringstream nusource; + nusource << TD::Convert(SourceFWRuleIdxs[3*idx-1]) << "/" << ((maxfwidx-SourceFWRuleAbsIdxs[idx]+1)*-1); + ScoreOrientationLeftBackward(table,oril,TD::Convert(nusource.str()),SourceFWRuleIdxs[3*idx], + isBonus,cost,bonus,bo1,bo1_bonus,bo2,bo2_bonus,alpha_oris,beta_oris); + } + isBonus = false; + if ((aligned && lat<=fwblock[3])|| + (!aligned && las<=SourceFWRuleIdxs[3*idx-2])) isBonus=true; + if (SourceFWRuleAbsIdxs[idx]<=maxdepth1) { + ostringstream nusource; + nusource << TD::Convert(SourceFWRuleIdxs[3*idx-1]) << "/" << SourceFWRuleAbsIdxs[idx]; + ScoreOrientationRightBackward(table,orir,SourceFWRuleIdxs[3*idx-1],SourceFWRuleIdxs[3*idx], + isBonus,cost,bonus,bo1,bo1_bonus,bo2,bo2_bonus,alpha_oris,beta_oris); + } + if (maxfwidx-SourceFWRuleAbsIdxs[idx]+1<=maxdepth2) { + ostringstream nusource; + nusource << TD::Convert(SourceFWRuleIdxs[3*idx-1]) << "/" << ((maxfwidx-SourceFWRuleAbsIdxs[idx]+1)*-1); + ScoreOrientationRightBackward(table,orir,SourceFWRuleIdxs[3*idx-1],SourceFWRuleIdxs[3*idx], + isBonus,cost,bonus,bo1,bo1_bonus,bo2,bo2_bonus,alpha_oris,beta_oris); + } + delete fwblock; + } + for (int i_ant=0; i_ant<_Arity; i_ant++) { + // antfas -> first aligned source word antecedent-wise + // antlas -> last aligned source word antecedent-wise + int antfat = firstTargetAligned(TargetAntsIdxs[i_ant][1]); + int antlat = lastTargetAligned(TargetAntsIdxs[i_ant][TargetAntsIdxs[i_ant][0]]); + int antfas = firstSourceAligned(SourceAntsIdxs[i_ant][1]); + int antlas = lastSourceAligned(SourceAntsIdxs[i_ant][SourceAntsIdxs[i_ant][0]]); + assert(antfat <= antlat); + assert(antfas <= antlas); + for (int idx=1; idx<=SourceFWAntsIdxs[i_ant][0]; idx++) { + if (DEBUG) + cerr << "considering SourceFWAntsIdxs[" << i_ant << "][" << idx << "]: " << SourceFWAntsIdxs[i_ant][3*idx-2] << endl; + if (!(SourceFWAntsAbsIdxs[i_ant][idx]<=maxdepth1 || maxfwidx-SourceFWAntsAbsIdxs[i_ant][idx]+1<=maxdepth2)) continue; + int* fwblock = blockSource(SourceFWAntsIdxs[i_ant][3*idx-2],SourceFWAntsIdxs[i_ant][3*idx-2]); + //bool aligned = (minTSpan(SourceFWAntsIdxs[i_ant][3*idx-2])!=MINIMUM_INIT); + bool aligned = (fwblock[2]!=MINIMUM_INIT); + bool Lcompute = true; bool Rcompute = true; + if (DEBUG) { + cerr << " aligned = " << aligned << endl; + cerr << " fwblock = " << fwblock[0] << "," << fwblock[1] << "," << fwblock[2] << "," << fwblock[3] << endl; + cerr << " antfas=" << antfas << ", antlas=" << antlas << ", antfat=" << antfat << ", antlat=" << antlat << endl; + } + if (aligned) { + if (DEBUG) cerr << "laligned" << endl; + if (antfat<fwblock[2]) { + if (DEBUG) cerr << antfat << "<" << fwblock[2] << endl; + Lcompute=false; + } + } else { + if (DEBUG) cerr << "!laligned" << endl; + if (antfas<fwblock[0] && fwblock[1] < antlas) Lcompute=false; + } + if (aligned) { + if (DEBUG) cerr << "raligned" << endl; + if (fwblock[3]<antlat) { + if (DEBUG) cerr << fwblock[3] << "<" << antlat << endl; + Rcompute=false; + } + } else { + if (DEBUG) cerr << "!raligned" << endl; + if (fwblock[1]<antlas && fwblock[1] < antlas) Rcompute=false; + } + if (!Lcompute && !Rcompute) continue; + if (!aligned) { + OrientationSource(SourceFWAntsIdxs[i_ant][3*idx-2],&oril,&orir,Lcompute,Rcompute); + } else { + OrientationTarget(fwblock[2],fwblock[3],&oril,&orir,Lcompute,Rcompute); + } + if (DEBUG) cerr << "oril = " << oril << ", orir = " << orir << endl; + bool isBonus = false; + if (Lcompute) { + if ((aligned && fwblock[3]<=fat) || + (!aligned && SourceFWAntsIdxs[i_ant][3*idx-2]<=fas)) isBonus = true; + if (SourceFWAntsAbsIdxs[i_ant][idx]<=maxdepth1) { + ostringstream nusource; + nusource << TD::Convert(SourceFWAntsIdxs[i_ant][3*idx-1]) << "/" << SourceFWAntsAbsIdxs[i_ant][idx]; + ScoreOrientationLeftBackward(table,oril,TD::Convert(nusource.str()),SourceFWAntsIdxs[i_ant][3*idx], + isBonus,cost,bonus,bo1,bo1_bonus,bo2,bo2_bonus,alpha_oris,beta_oris); + } + if (maxfwidx-SourceFWAntsAbsIdxs[i_ant][idx]+1<=maxdepth2) { + ostringstream nusource; + nusource << TD::Convert(SourceFWAntsIdxs[i_ant][3*idx-1]) << "/" << (-1*(maxfwidx-SourceFWAntsAbsIdxs[i_ant][idx]+1)); + ScoreOrientationLeftBackward(table,oril,TD::Convert(nusource.str()),SourceFWAntsIdxs[i_ant][3*idx], + isBonus,cost,bonus,bo1,bo1_bonus,bo2,bo2_bonus,alpha_oris,beta_oris); + } + } + isBonus = false; + if (Rcompute) { + if ((aligned && lat<=fwblock[2]) || + (!aligned && las<=SourceFWAntsIdxs[i_ant][3*idx-2]))isBonus = true; + if (SourceFWAntsAbsIdxs[i_ant][idx]<=maxdepth1) { + ostringstream nusource; + nusource << TD::Convert(SourceFWAntsIdxs[i_ant][3*idx-1]) << "/" << SourceFWAntsAbsIdxs[i_ant][idx]; + ScoreOrientationRightBackward(table,orir,TD::Convert(nusource.str()),SourceFWAntsIdxs[i_ant][3*idx], + isBonus,cost,bonus,bo1,bo1_bonus,bo2,bo2_bonus,alpha_oris,beta_oris); + } + if (maxfwidx-SourceFWAntsAbsIdxs[i_ant][idx]+1<=maxdepth2) { + ostringstream nusource; + nusource << TD::Convert(SourceFWAntsIdxs[i_ant][3*idx-1]) << "/" << (-1*(maxfwidx-SourceFWAntsAbsIdxs[i_ant][idx]+1)); + ScoreOrientationRightBackward(table,orir,TD::Convert(nusource.str()),SourceFWAntsIdxs[i_ant][3*idx], + isBonus,cost,bonus,bo1,bo1_bonus,bo2,bo2_bonus,alpha_oris,beta_oris); + } + } + delete fwblock; + } + } +} + + +void Alignment::computeOrientationSourceBackward(const CountTable& table, double *cost, double *bonus, + double *bo1, double *bo1_bonus, double *bo2, double *bo2_bonus) { + if (DEBUG) cerr << "computeOrientationSourceBackward" << endl; + int oril, orir; + for (int idx=1; idx<=SourceFWRuleIdxs[0]; idx++) { + if (DEBUG) cerr << "considering SourceFWRuleIdxs[" << idx << "]: " << SourceFWRuleIdxs[3*idx-2] << endl; + int* fwblock = blockSource(SourceFWRuleIdxs[3*idx-2],SourceFWRuleIdxs[3*idx-2]); + bool aligned = (fwblock[2]!=MINIMUM_INIT); + if (aligned) { + OrientationTarget(fwblock[2],fwblock[3],&oril,&orir); + } else { + OrientationSource(SourceFWRuleIdxs[3*idx-2],&oril,&orir); + } + if (DEBUG) cerr << "oril = " << oril << ", orir = " << orir << endl; + bool isBonus = false; // fas -> first aligned source word, las -> last aligned source word + if ((aligned && fwblock[2]<=fat)|| + (!aligned && SourceFWRuleIdxs[3*idx-2]<=fas)) isBonus=true; + ScoreOrientationLeftBackward(table,oril,SourceFWRuleIdxs[3*idx-1],SourceFWRuleIdxs[3*idx], + isBonus,cost,bonus,bo1,bo1_bonus,bo2,bo2_bonus,alpha_oris,beta_oris); + isBonus = false; + if ((aligned && lat<=fwblock[3])|| + (!aligned && las<=SourceFWRuleIdxs[3*idx-2])) isBonus=true; + ScoreOrientationRightBackward(table,orir,SourceFWRuleIdxs[3*idx-1],SourceFWRuleIdxs[3*idx], + isBonus,cost,bonus,bo1,bo1_bonus,bo2,bo2_bonus,alpha_oris,beta_oris); + delete fwblock; + } + for (int i_ant=0; i_ant<_Arity; i_ant++) { + // antfas -> first aligned source word antecedent-wise + // antlas -> last aligned source word antecedent-wise + int antfat = firstTargetAligned(TargetAntsIdxs[i_ant][1]); + int antlat = lastTargetAligned(TargetAntsIdxs[i_ant][TargetAntsIdxs[i_ant][0]]); + int antfas = firstSourceAligned(SourceAntsIdxs[i_ant][1]); + int antlas = lastSourceAligned(SourceAntsIdxs[i_ant][SourceAntsIdxs[i_ant][0]]); + assert(antfat <= antlat); + assert(antfas <= antlas); + for (int idx=1; idx<=SourceFWAntsIdxs[i_ant][0]; idx++) { + if (DEBUG) + cerr << "considering SourceFWAntsIdxs[" << i_ant << "][" << idx << "]: " << SourceFWAntsIdxs[i_ant][3*idx-2] << endl; + int* fwblock = blockSource(SourceFWAntsIdxs[i_ant][3*idx-2],SourceFWAntsIdxs[i_ant][3*idx-2]); + //bool aligned = (minTSpan(SourceFWAntsIdxs[i_ant][3*idx-2])!=MINIMUM_INIT); + bool aligned = (fwblock[2]!=MINIMUM_INIT); + bool Lcompute = true; bool Rcompute = true; + if (DEBUG) { + cerr << " aligned = " << aligned << endl; + cerr << " fwblock = " << fwblock[0] << "," << fwblock[1] << "," << fwblock[2] << "," << fwblock[3] << endl; + cerr << " antfas=" << antfas << ", antlas=" << antlas << ", antfat=" << antfat << ", antlat=" << antlat << endl; + } + if (aligned) { + if (DEBUG) cerr << "laligned" << endl; + if (antfat<fwblock[2]) { + if (DEBUG) cerr << antfat << "<" << fwblock[2] << endl; + Lcompute=false; + } + } else { + if (DEBUG) cerr << "!laligned" << endl; + if (antfas<fwblock[0] && fwblock[1] < antlas) Lcompute=false; + } + if (aligned) { + if (DEBUG) cerr << "raligned" << endl; + if (fwblock[3]<antlat) { + if (DEBUG) cerr << fwblock[3] << "<" << antlat << endl; + Rcompute=false; + } + } else { + if (DEBUG) cerr << "!raligned" << endl; + if (fwblock[1]<antlas && fwblock[1] < antlas) Rcompute=false; + } + if (!Lcompute && !Rcompute) continue; + if (!aligned) { + OrientationSource(SourceFWAntsIdxs[i_ant][3*idx-2],&oril,&orir,Lcompute,Rcompute); + } else { + OrientationTarget(fwblock[2],fwblock[3],&oril,&orir,Lcompute,Rcompute); + } + if (DEBUG) cerr << "oril = " << oril << ", orir = " << orir << endl; + bool isBonus = false; + if (Lcompute) { + if ((aligned && fwblock[3]<=fat) || + (!aligned && SourceFWAntsIdxs[i_ant][3*idx-2]<=fas)) isBonus = true; + ScoreOrientationLeftBackward(table,oril,SourceFWAntsIdxs[i_ant][3*idx-1],SourceFWAntsIdxs[i_ant][3*idx], + isBonus,cost,bonus,bo1,bo1_bonus,bo2,bo2_bonus,alpha_oris,beta_oris); + } + isBonus = false; + if (Rcompute) { + if ((aligned && lat<=fwblock[2]) || + (!aligned && las<=SourceFWAntsIdxs[i_ant][3*idx-2]))isBonus = true; + ScoreOrientationRightBackward(table,orir,SourceFWAntsIdxs[i_ant][3*idx-1],SourceFWAntsIdxs[i_ant][3*idx], + isBonus,cost,bonus,bo1,bo1_bonus,bo2,bo2_bonus,alpha_oris,beta_oris); + } + delete fwblock; + } + } +} + +void Alignment::computeOrientationSourcePos(const CountTable& table, double *cost, double *bonus, + double *bo1, double *bo1_bonus, double *bo2, double *bo2_bonus, int maxfwidx, int maxdepth1, int maxdepth2) { + // This implementation is actually really bad, not reusing codes at all + if (DEBUG) cerr << "computeOrientationSourcePos(maxfwidx=" << maxfwidx << ",maxdepth=" << maxdepth1 << "," << maxdepth2 << ")" << endl; + if (maxdepth1+maxdepth2==0) return; + int oril, orir; + ostringstream oss; + WordID sourceID; + for (int idx=1; idx<=SourceFWRuleIdxs[0]; idx++) { + if (DEBUG) cerr << "considering SourceFWRuleIdxs[" << idx << "]: " << SourceFWRuleIdxs[3*idx-2] << endl; + //if (!((SourceFWRuleAbsIdxs[idx]<=maxdepth1) || (maxfwidx-SourceFWRuleAbsIdxs[idx]+1<=maxdepth2))) continue; + string source = TD::Convert(SourceFWRuleIdxs[3*idx-1]); + OrientationSource(SourceFWRuleIdxs[3*idx-2],&oril,&orir); + bool isBonus = false; // fas -> first aligned source word, las -> last aligned source word + if (SourceFWRuleIdxs[3*idx-2]<=fas) isBonus=true; + if (!isBonus) // this is unnecessary because fas <= las assertion + if (minTSpan(SourceFWRuleIdxs[3*idx-2])==MINIMUM_INIT && las<=SourceFWRuleIdxs[3*idx-2]) isBonus=true; + if (maxdepth1>0) { + oss << source << "/"; + if (SourceFWRuleAbsIdxs[idx]<=maxdepth1) + oss << SourceFWRuleAbsIdxs[idx]; + else + oss << "X"; + sourceID = TD::Convert(oss.str()); + oss.str(""); + ScoreOrientationLeft(table,oril,sourceID,SourceFWRuleIdxs[3*idx], + isBonus,cost,bonus,bo1,bo1_bonus,bo2,bo2_bonus,alpha_oris,beta_oris); + } + if (maxdepth2>0) { + oss << source << "/"; + if (maxfwidx-SourceFWRuleAbsIdxs[idx]+1<=maxdepth2) + oss << ((maxfwidx-SourceFWRuleAbsIdxs[idx]+1)*-1); + else + oss << "X"; + sourceID = TD::Convert(oss.str()); + oss.str(""); + ScoreOrientationLeft(table,oril,sourceID,SourceFWRuleIdxs[3*idx], + isBonus,cost,bonus,bo1,bo1_bonus,bo2,bo2_bonus,alpha_oris,beta_oris); + } + isBonus = false; + if (las<=SourceFWRuleIdxs[3*idx-2]) isBonus=true; + if (!isBonus) // this is unnecessary becuase fas <= las assertion + if (minTSpan(SourceFWRuleIdxs[3*idx-2])==MINIMUM_INIT && SourceFWRuleIdxs[3*idx-2]<=fas) isBonus=true; + if (maxdepth1>0) { + oss << source << "/"; + if (SourceFWRuleAbsIdxs[idx]<=maxdepth1) + oss << SourceFWRuleAbsIdxs[idx]; + else + oss << "X"; + sourceID = TD::Convert(oss.str()); + oss.str(""); + ScoreOrientationRight(table,orir,sourceID,SourceFWRuleIdxs[3*idx], + isBonus,cost,bonus,bo1,bo1_bonus,bo2,bo2_bonus,alpha_oris,beta_oris); + } + if (maxdepth2>0) { + oss << source << "/"; + if (maxfwidx-SourceFWRuleAbsIdxs[idx]+1<=maxdepth2) + oss << ((maxfwidx-SourceFWRuleAbsIdxs[idx]+1)*-1); + else + oss << "X"; + sourceID = TD::Convert(oss.str()); + oss.str(""); + ScoreOrientationRight(table,orir,sourceID,SourceFWRuleIdxs[3*idx], + isBonus,cost,bonus,bo1,bo1_bonus,bo2,bo2_bonus,alpha_oris,beta_oris); + } + + } + for (int i_ant=0; i_ant<_Arity; i_ant++) { + for (int idx=1; idx<=SourceFWAntsIdxs[i_ant][0]; idx++) { + if (DEBUG) + cerr << "considering SourceFWAntsIdxs[" << i_ant << "][" << idx << "]: " << SourceFWAntsIdxs[i_ant][3*idx-2] << endl; + //if (!((SourceFWAntsAbsIdxs[i_ant][idx]<=maxdepth1)||(maxfwidx-SourceFWAntsAbsIdxs[i_ant][idx]+1<=maxdepth2))) continue; + // antfas -> first aligned source word antecedent-wise + // antlas -> last aligned source word antecedent-wise + int antfas = firstSourceAligned(SourceAntsIdxs[i_ant][1]); + int antlas = lastSourceAligned(SourceAntsIdxs[i_ant][SourceAntsIdxs[i_ant][0]]); + if (DEBUG) cerr << " SourceFWAntsAbsIdxs[i_ant][3*idx-1]=" << SourceFWAntsAbsIdxs[i_ant][3*idx-1] << endl; + string source = TD::Convert(SourceFWAntsIdxs[i_ant][3*idx-1]); + assert(antfas <= antlas); + bool aligned = (minTSpan(SourceFWAntsIdxs[i_ant][3*idx-2])!=MINIMUM_INIT); + bool Lcompute = true;bool Rcompute = true; + if ((aligned && antfas<SourceFWAntsIdxs[i_ant][3*idx-2]) || + (!aligned && antfas < SourceFWAntsIdxs[i_ant][3*idx-2] && SourceFWAntsIdxs[i_ant][3*idx-2] < antlas)) + Lcompute=false; + if ((aligned && SourceFWAntsIdxs[i_ant][3*idx-2]<antlas) || + (!aligned && antfas < SourceFWAntsIdxs[i_ant][3*idx-2] && SourceFWAntsIdxs[i_ant][3*idx-2] < antlas)) + Rcompute=false; + if (!Lcompute && !Rcompute) continue; + OrientationSource(SourceFWAntsIdxs[i_ant][3*idx-2],&oril,&orir,Lcompute, Rcompute); + bool isBonus = false; + if (Lcompute) { + if (SourceFWAntsIdxs[i_ant][3*idx-2]<=fas) isBonus = true; + //if (!isBonus) // this is unnecessary + // if (!aligned && las<=SourceFWAntsIdxs[i_ant][3*idx-2]) isBonus=true; + if (maxdepth1>0) { + oss << source << "/"; + if (SourceFWAntsAbsIdxs[i_ant][idx]<=maxdepth1) + oss << SourceFWAntsAbsIdxs[i_ant][idx]; + else + oss << "X"; + sourceID = TD::Convert(oss.str()); + oss.str(""); + ScoreOrientationLeft(table,oril,sourceID,SourceFWAntsIdxs[i_ant][3*idx], + isBonus,cost,bonus,bo1,bo1_bonus,bo2,bo2_bonus,alpha_oris,beta_oris); + } + if (maxdepth2>0) { + oss << source << "/"; + if (maxfwidx-SourceFWAntsAbsIdxs[i_ant][idx]+1<=maxdepth2) + oss << ((maxfwidx-SourceFWAntsAbsIdxs[i_ant][idx]+1)*-1); + else + oss << "X"; + sourceID = TD::Convert(oss.str()); + oss.str(""); + ScoreOrientationLeft(table,oril,sourceID,SourceFWAntsIdxs[i_ant][3*idx], + isBonus,cost,bonus,bo1,bo1_bonus,bo2,bo2_bonus,alpha_oris,beta_oris); + } + } + isBonus = false; + if (Rcompute) { + if (las<=SourceFWAntsIdxs[i_ant][3*idx-2]) isBonus = true; + //if (!isBonus) // this is unnecessary + // if (!aligned && SourceFWAntsIdxs[i_ant][3*idx-2]<=fas) isBonus=true; + if (maxdepth1>0) { + oss << source << "/"; + if (SourceFWAntsAbsIdxs[i_ant][idx]<=maxdepth1) + oss << SourceFWAntsAbsIdxs[i_ant][idx]; + else + oss << "X"; + sourceID = TD::Convert(oss.str()); + oss.str(""); + ScoreOrientationRight(table,orir,sourceID,SourceFWAntsIdxs[i_ant][3*idx], + isBonus,cost,bonus,bo1,bo1_bonus,bo2,bo2_bonus,alpha_oris,beta_oris); + } + if (maxdepth2>0) { + oss << source << "/"; + if (maxfwidx-SourceFWAntsAbsIdxs[i_ant][idx]+1<=maxdepth2) + oss << ((maxfwidx-SourceFWAntsAbsIdxs[i_ant][idx]+1)*-1); + else + oss << "X"; + sourceID = TD::Convert(oss.str()); + oss.str(""); + ScoreOrientationRight(table,orir,sourceID,SourceFWAntsIdxs[i_ant][3*idx], + isBonus,cost,bonus,bo1,bo1_bonus,bo2,bo2_bonus,alpha_oris,beta_oris); + } + } + } + } +} + +void Alignment::computeOrientationSource(const CountTable& table, double *cost, double *bonus, + double *bo1, double *bo1_bonus, double *bo2, double *bo2_bonus) { +// a bit complex due to imperfect state (TO DO!!!) +// 1. there are cases where function word alignments come from antecedents, which orientation +// (either its left or its right) has been computed earlier. +// 2. some orientation will go as bonus + if (DEBUG) cerr << "computeOrientationSource" << endl; + int oril, orir; + for (int idx=1; idx<=SourceFWRuleIdxs[0]; idx++) { + if (DEBUG) cerr << "considering SourceFWRuleIdxs[" << idx << "]: " << SourceFWRuleIdxs[3*idx-2] << endl; + OrientationSource(SourceFWRuleIdxs[3*idx-2],&oril,&orir); + bool isBonus = false; // fas -> first aligned source word, las -> last aligned source word + if (SourceFWRuleIdxs[3*idx-2]<=fas) isBonus=true; + if (!isBonus) // this is unnecessary because fas <= las assertion + if (minTSpan(SourceFWRuleIdxs[3*idx-2])==MINIMUM_INIT && las<=SourceFWRuleIdxs[3*idx-2]) isBonus=true; + ScoreOrientationLeft(table,oril,SourceFWRuleIdxs[3*idx-1],SourceFWRuleIdxs[3*idx], + isBonus,cost,bonus,bo1,bo1_bonus,bo2,bo2_bonus,alpha_oris,beta_oris); + isBonus = false; + if (las<=SourceFWRuleIdxs[3*idx-2]) isBonus=true; + if (!isBonus) // this is unnecessary becuase fas <= las assertion + if (minTSpan(SourceFWRuleIdxs[3*idx-2])==MINIMUM_INIT && SourceFWRuleIdxs[3*idx-2]<=fas) isBonus=true; + ScoreOrientationRight(table,orir,SourceFWRuleIdxs[3*idx-1],SourceFWRuleIdxs[3*idx], + isBonus,cost,bonus,bo1,bo1_bonus,bo2,bo2_bonus,alpha_oris,beta_oris); + } + for (int i_ant=0; i_ant<_Arity; i_ant++) { + for (int idx=1; idx<=SourceFWAntsIdxs[i_ant][0]; idx++) { + if (DEBUG) + cerr << "considering SourceFWAntsIdxs[" << i_ant << "][" << idx << "]: " << SourceFWAntsIdxs[i_ant][3*idx-2] << endl; + // antfas -> first aligned source word antecedent-wise + // antlas -> last aligned source word antecedent-wise + int antfas = firstSourceAligned(SourceAntsIdxs[i_ant][1]); + int antlas = lastSourceAligned(SourceAntsIdxs[i_ant][SourceAntsIdxs[i_ant][0]]); + assert(antfas <= antlas); + bool aligned = (minTSpan(SourceFWAntsIdxs[i_ant][3*idx-2])!=MINIMUM_INIT); + bool Lcompute = true;bool Rcompute = true; + if ((aligned && antfas<SourceFWAntsIdxs[i_ant][3*idx-2]) || + (!aligned && antfas < SourceFWAntsIdxs[i_ant][3*idx-2] && SourceFWAntsIdxs[i_ant][3*idx-2] < antlas)) + Lcompute=false; + if ((aligned && SourceFWAntsIdxs[i_ant][3*idx-2]<antlas) || + (!aligned && antfas < SourceFWAntsIdxs[i_ant][3*idx-2] && SourceFWAntsIdxs[i_ant][3*idx-2] < antlas)) + Rcompute=false; + if (!Lcompute && !Rcompute) continue; + OrientationSource(SourceFWAntsIdxs[i_ant][3*idx-2],&oril,&orir,Lcompute, Rcompute); + bool isBonus = false; + if (Lcompute) { + if (SourceFWAntsIdxs[i_ant][3*idx-2]<=fas) isBonus = true; + //if (!isBonus) // this is unnecessary + // if (!aligned && las<=SourceFWAntsIdxs[i_ant][3*idx-2]) isBonus=true; + ScoreOrientationLeft(table,oril,SourceFWAntsIdxs[i_ant][3*idx-1],SourceFWAntsIdxs[i_ant][3*idx], + isBonus,cost,bonus,bo1,bo1_bonus,bo2,bo2_bonus,alpha_oris,beta_oris); + } + isBonus = false; + if (Rcompute) { + if (las<=SourceFWAntsIdxs[i_ant][3*idx-2]) isBonus = true; + //if (!isBonus) // this is unnecessary + // if (!aligned && SourceFWAntsIdxs[i_ant][3*idx-2]<=fas) isBonus=true; + ScoreOrientationRight(table,orir,SourceFWAntsIdxs[i_ant][3*idx-1],SourceFWAntsIdxs[i_ant][3*idx], + isBonus,cost,bonus,bo1,bo1_bonus,bo2,bo2_bonus,alpha_oris,beta_oris); + } + } + } +} + +void Alignment::computeOrientationSourceGen(const CountTable& table, double *cost, double *bonus, + double *bo1, double *bo1_bonus, double *bo2, double *bo2_bonus, const map<WordID,WordID>& tags) { + if (DEBUG) cerr << "computeOrientationSourceGen" << endl; + int oril, orir; + for (int idx=1; idx<=SourceFWRuleIdxs[0]; idx++) { + if (DEBUG) cerr << "considering SourceFWRuleIdxs[" << idx << "]: " << SourceFWRuleIdxs[3*idx-2] << endl; + OrientationSource(SourceFWRuleIdxs[3*idx-2],&oril,&orir); + bool isBonus = false; // fas -> first aligned source word, las -> last aligned source word + if (SourceFWRuleIdxs[3*idx-2]<=fas) isBonus=true; + if (!isBonus) // this is unnecessary because fas <= las assertion + if (minTSpan(SourceFWRuleIdxs[3*idx-2])==MINIMUM_INIT && las<=SourceFWRuleIdxs[3*idx-2]) isBonus=true; + ScoreOrientationLeft(table,oril,generalize(SourceFWRuleIdxs[3*idx-1],tags),SourceFWRuleIdxs[3*idx], + isBonus,cost,bonus,bo1,bo1_bonus,bo2,bo2_bonus,alpha_oris,beta_oris); + isBonus = false; + if (las<=SourceFWRuleIdxs[3*idx-2]) isBonus=true; + if (!isBonus) // this is unnecessary becuase fas <= las assertion + if (minTSpan(SourceFWRuleIdxs[3*idx-2])==MINIMUM_INIT && SourceFWRuleIdxs[3*idx-2]<=fas) isBonus=true; + ScoreOrientationRight(table,orir,generalize(SourceFWRuleIdxs[3*idx-1],tags),SourceFWRuleIdxs[3*idx], + isBonus,cost,bonus,bo1,bo1_bonus,bo2,bo2_bonus,alpha_oris,beta_oris); + } + for (int i_ant=0; i_ant<_Arity; i_ant++) { + for (int idx=1; idx<=SourceFWAntsIdxs[i_ant][0]; idx++) { + if (DEBUG) + cerr << "considering SourceFWAntsIdxs[" << i_ant << "][" << idx << "]: " << SourceFWAntsIdxs[i_ant][3*idx-2] << endl; + // antfas -> first aligned source word antecedent-wise + // antlas -> last aligned source word antecedent-wise + int antfas = firstSourceAligned(SourceAntsIdxs[i_ant][1]); + int antlas = lastSourceAligned(SourceAntsIdxs[i_ant][SourceAntsIdxs[i_ant][0]]); + assert(antfas <= antlas); + bool aligned = (minTSpan(SourceFWAntsIdxs[i_ant][3*idx-2])!=MINIMUM_INIT); + bool Lcompute = true;bool Rcompute = true; + if ((aligned && antfas<SourceFWAntsIdxs[i_ant][3*idx-2]) || + (!aligned && antfas < SourceFWAntsIdxs[i_ant][3*idx-2] && SourceFWAntsIdxs[i_ant][3*idx-2] < antlas)) + Lcompute=false; + if ((aligned && SourceFWAntsIdxs[i_ant][3*idx-2]<antlas) || + (!aligned && antfas < SourceFWAntsIdxs[i_ant][3*idx-2] && SourceFWAntsIdxs[i_ant][3*idx-2] < antlas)) + Rcompute=false; + if (!Lcompute && !Rcompute) continue; + OrientationSource(SourceFWAntsIdxs[i_ant][3*idx-2],&oril,&orir,Lcompute, Rcompute); + bool isBonus = false; + if (Lcompute) { + if (SourceFWAntsIdxs[i_ant][3*idx-2]<=fas) isBonus = true; + //if (!isBonus) // this is unnecessary + // if (!aligned && las<=SourceFWAntsIdxs[i_ant][3*idx-2]) isBonus=true; + ScoreOrientationLeft(table,oril,generalize(SourceFWAntsIdxs[i_ant][3*idx-1],tags),SourceFWAntsIdxs[i_ant][3*idx], + isBonus,cost,bonus,bo1,bo1_bonus,bo2,bo2_bonus,alpha_oris,beta_oris); + } + isBonus = false; + if (Rcompute) { + if (las<=SourceFWAntsIdxs[i_ant][3*idx-2]) isBonus = true; + //if (!isBonus) // this is unnecessary + // if (!aligned && SourceFWAntsIdxs[i_ant][3*idx-2]<=fas) isBonus=true; + ScoreOrientationRight(table,orir,generalize(SourceFWAntsIdxs[i_ant][3*idx-1],tags),SourceFWAntsIdxs[i_ant][3*idx], + isBonus,cost,bonus,bo1,bo1_bonus,bo2,bo2_bonus,alpha_oris,beta_oris); + } + } + } +} +void Alignment::computeOrientationTarget(const CountTable& table, double *cost, double *bonus, double *bo1, double *bo1_bonus, double *bo2, double *bo2_bonus) { + if (DEBUG) cerr << "computeOrientationTarget" << endl; + int oril, orir; + for (int idx=1; idx<=TargetFWRuleIdxs[0]; idx++) { + if (DEBUG) cerr << "considering TargetFWRuleIdxs[" << idx << "]: " << TargetFWRuleIdxs[3*idx-2] << endl; + OrientationTarget(TargetFWRuleIdxs[3*idx-2],&oril,&orir); + // the second and the third parameters of ScoreOrientationLeft must be e and f (not f and then e) + bool isBonus = false; + if (TargetFWRuleIdxs[3*idx-2]<=fat) isBonus = true; + if (!isBonus) + if (minSSpan(TargetFWRuleIdxs[3*idx-2])==MINIMUM_INIT && lat<=TargetFWRuleIdxs[3*idx-2]) isBonus = true; + ScoreOrientationLeft(table,oril,TargetFWRuleIdxs[3*idx-1],TargetFWRuleIdxs[3*idx], + isBonus,cost,bonus,bo1,bo1_bonus,bo2,bo2_bonus,alpha_orit,beta_orit); + isBonus = false; + if (lat<=TargetFWRuleIdxs[3*idx-2]) isBonus = true; + if (!isBonus) + if (minSSpan(TargetFWRuleIdxs[3*idx-2])==MINIMUM_INIT && TargetFWRuleIdxs[3*idx-2]<=fat) isBonus=true; + ScoreOrientationRight(table,orir,TargetFWRuleIdxs[3*idx-1],TargetFWRuleIdxs[3*idx], + isBonus,cost,bonus,bo1,bo1_bonus,bo2,bo2_bonus,alpha_orit,beta_orit); + } + + for (int i_ant=0; i_ant<_Arity; i_ant++) { + for (int idx=1; idx<=TargetFWAntsIdxs[i_ant][0]; idx++) { + if (DEBUG) cerr << "considering TargetFWAntsIdxs[" << i_ant << "][" << idx << "]: " << TargetFWAntsIdxs[i_ant][3*idx-2] << endl; + int antfat = firstTargetAligned(TargetAntsIdxs[i_ant][1]); + int antlat = lastTargetAligned(TargetAntsIdxs[i_ant][TargetAntsIdxs[i_ant][0]]); + int aligned = (minSSpan( TargetFWAntsIdxs[i_ant][3*idx-2])!=MINIMUM_INIT); + bool Lcompute = true; bool Rcompute = true; + if ((aligned && antfat<TargetFWAntsIdxs[i_ant][3*idx-2]) || + (!aligned && antfat < TargetFWAntsIdxs[i_ant][3*idx-2] && TargetFWAntsIdxs[i_ant][3*idx-2] < antlat)) + Lcompute=false; + if ((aligned && TargetFWAntsIdxs[i_ant][3*idx-2]<antlat) || + (!aligned && antfat < TargetFWAntsIdxs[i_ant][3*idx-2] && TargetFWAntsIdxs[i_ant][3*idx-2] < antlat)) + Rcompute=false; + if (!Lcompute && !Rcompute) continue; + bool isBonus = false; + OrientationTarget(TargetFWAntsIdxs[i_ant][3*idx-2],&oril,&orir, Lcompute, Rcompute); + if (Lcompute) { + if (TargetFWAntsIdxs[i_ant][3*idx-2]<=fat) isBonus=true; + //if (!isBonus) + // if (!aligned && lat<=TargetFWAntsIdxs[i_ant][3*idx-2]) isBonus=true; + ScoreOrientationLeft(table,oril,TargetFWAntsIdxs[i_ant][3*idx-1],TargetFWAntsIdxs[i_ant][3*idx], + isBonus,cost,bonus,bo1,bo1_bonus,bo2,bo2_bonus,alpha_orit,beta_orit); + } + isBonus = false; + if (Rcompute) { + if (lat<=TargetFWAntsIdxs[i_ant][3*idx-2]) isBonus=true; + if (!isBonus) + //if (!aligned && TargetFWAntsIdxs[i_ant][3*idx-2]<=fat) isBonus=true; + ScoreOrientationRight(table,orir,TargetFWAntsIdxs[i_ant][3*idx-1],TargetFWAntsIdxs[i_ant][3*idx], + isBonus,cost,bonus,bo1,bo1_bonus,bo2,bo2_bonus,alpha_orit,beta_orit); + } + } + } +} + +void Alignment::computeOrientationTargetBackward(const CountTable& table, double *cost, double *bonus, double *bo1, double *bo1_bonus, double *bo2, double *bo2_bonus) { + if (DEBUG) cerr << "computeOrientationTargetBackward" << endl; + int oril, orir; + for (int idx=1; idx<=TargetFWRuleIdxs[0]; idx++) { + if (DEBUG) cerr << "considering TargetFWRuleIdxs[" << idx << "]: " << TargetFWRuleIdxs[3*idx-2] << endl; + int* fwblock = blockSource(TargetFWRuleIdxs[3*idx-2],TargetFWRuleIdxs[3*idx-2]); + bool aligned = (fwblock[0] == MINIMUM_INIT); + if (aligned) { + OrientationSource(fwblock[0],fwblock[1],&oril,&orir); + } else { + OrientationTarget(TargetFWRuleIdxs[3*idx-2],&oril,&orir); + } + delete fwblock; + // the second and the third parameters of ScoreOrientationLeft must be e and f (not f and then e) + bool isBonus = false; + if (TargetFWRuleIdxs[3*idx-2]<=fat) isBonus = true; + //if (!isBonus) // unnecessary + //if (minSSpan(TargetFWRuleIdxs[3*idx-2])==MINIMUM_INIT && lat<=TargetFWRuleIdxs[3*idx-2]) isBonus = true; + ScoreOrientationLeftBackward(table,oril,TargetFWRuleIdxs[3*idx-1],TargetFWRuleIdxs[3*idx], + isBonus,cost,bonus,bo1,bo1_bonus,bo2,bo2_bonus,alpha_orit,beta_orit); + isBonus = false; + if (lat<=TargetFWRuleIdxs[3*idx-2]) isBonus = true; + //if (!isBonus) // unnecessary + //if (minSSpan(TargetFWRuleIdxs[3*idx-2])==MINIMUM_INIT && TargetFWRuleIdxs[3*idx-2]<=fat) isBonus=true; + ScoreOrientationRightBackward(table,orir,TargetFWRuleIdxs[3*idx-1],TargetFWRuleIdxs[3*idx], + isBonus,cost,bonus,bo1,bo1_bonus,bo2,bo2_bonus,alpha_orit,beta_orit); + } + + for (int i_ant=0; i_ant<_Arity; i_ant++) { + int antfat = firstTargetAligned(TargetAntsIdxs[i_ant][1]); + int antlat = lastTargetAligned(TargetAntsIdxs[i_ant][TargetAntsIdxs[i_ant][0]]); + int antfas = firstSourceAligned(SourceAntsIdxs[i_ant][1]); + int antlas = lastSourceAligned(SourceAntsIdxs[i_ant][SourceAntsIdxs[i_ant][0]]); + for (int idx=1; idx<=TargetFWAntsIdxs[i_ant][0]; idx++) { + if (DEBUG) cerr << "considering TargetFWAntsIdxs[" << i_ant << "][" << idx << "]: " << TargetFWAntsIdxs[i_ant][3*idx-2] << endl; + int* fwblock = blockTarget(TargetFWAntsIdxs[i_ant][3*idx-2],TargetFWAntsIdxs[i_ant][3*idx-2]); + bool aligned = (fwblock[0]!=MINIMUM_INIT); + //bool aligned = (minSSpan( TargetFWAntsIdxs[i_ant][3*idx-2])!=MINIMUM_INIT); + bool Lcompute = true; bool Rcompute = true; + if ((aligned && antfas<fwblock[0]) || + (!aligned && antfat < fwblock[2])) + Lcompute=false; + if ((aligned && fwblock[0]<antlas) || + (!aligned && fwblock[3] < antlat)) + Rcompute=false; + if (!Lcompute && !Rcompute) continue; + bool isBonus = false; + if (aligned) { + OrientationSource(fwblock[0],fwblock[1],&oril,&orir,Lcompute,Rcompute); + } else { + OrientationTarget(TargetFWAntsIdxs[i_ant][3*idx-2],&oril,&orir, Lcompute, Rcompute); + } + if (Lcompute) { + if ((aligned && fwblock[1]<=fas) || + (!aligned && fwblock[3]<=fat)) + isBonus=true; + //if (!isBonus) + // if (!aligned && lat<=TargetFWAntsIdxs[i_ant][3*idx-2]) isBonus=true; + ScoreOrientationLeftBackward(table,oril,TargetFWAntsIdxs[i_ant][3*idx-1],TargetFWAntsIdxs[i_ant][3*idx], + isBonus,cost,bonus,bo1,bo1_bonus,bo2,bo2_bonus,alpha_orit,beta_orit); + } + isBonus = false; + if (Rcompute) { + if ((aligned && las<=fwblock[0]) || + (!aligned && lat<=fwblock[2])) + isBonus=true; + if (!isBonus) + //if (!aligned && TargetFWAntsIdxs[i_ant][3*idx-2]<=fat) isBonus=true; + ScoreOrientationRightBackward(table,orir,TargetFWAntsIdxs[i_ant][3*idx-1],TargetFWAntsIdxs[i_ant][3*idx], + isBonus,cost,bonus,bo1,bo1_bonus,bo2,bo2_bonus,alpha_orit,beta_orit); + } + delete fwblock; + } + } +} + +bool Alignment::MemberOf(int* FWIdxs, int pos1, int pos2) { + for (int idx=2; idx<=FWIdxs[0]; idx++) { + if (FWIdxs[3*(idx-1)-2]==pos1 && FWIdxs[3*idx-2]==pos2) return true; + } + return false; +} + +void Alignment::computeDominanceSource(const CountTable& table, WordID lfw, WordID rfw, + double *cost, double *bonus, double *bo1, double *bo1_bonus, double *bo2, double *bo2_bonus) { + // no bonus yet + if (DEBUG) cerr << "computeDominanceSource" << endl; + if (DEBUG) cerr << " initial cost=" << *cost << ", initial bonus=" << *bonus << endl; + for (int idx=2; idx<=SourceFWIdxs[0]; idx++) { + if (DEBUG) { + cerr << "PrevSourceFWIdxs :" << SourceFWIdxs[3*(idx-1)-2] << "," << SourceFWIdxs[3*(idx-1)-1] + << "," << SourceFWIdxs[3*(idx-1)] << endl; + cerr << "CurrSourceFWIdxs :" << SourceFWIdxs[3*(idx)-2] << "," << SourceFWIdxs[3*(idx)-1] + << "," << SourceFWIdxs[3*(idx)] << endl; + } + bool compute = true; + for (int i_ant=0; i_ant<_Arity && compute; i_ant++) { + if (MemberOf(SourceFWAntsIdxs[i_ant],SourceFWIdxs[3*(idx-1)-2],SourceFWIdxs[3*(idx)-2])) { + //cerr << "Skipping, they have been calculated in the " << (i_ant+1) << "-th branch" << endl; + compute=false; + } + } + if (compute) { + int dom = DominanceSource(SourceFWIdxs[3*(idx-1)-2],SourceFWIdxs[3*idx-2]); + if (DEBUG) cerr << "dom = " << dom << endl; + ScoreDominance(table,dom,SourceFWIdxs[3*(idx-1)-1],SourceFWIdxs[3*idx-1],SourceFWIdxs[3*(idx-1)],SourceFWIdxs[3*idx], + cost,bo1,bo2,false,alpha_doms,beta_doms); + if (DEBUG) cerr << "cost now is " << *cost << endl; + } + } + if (SourceFWIdxs[0]>0) { + if (lfw>=0) { + int dom = DominanceSource(0,SourceFWIdxs[1]); + if (DEBUG) cerr << " --> lfw = " << lfw << "-" << TD::Convert(lfw) << endl; + if (DEBUG) cerr << " --> rfw = " << rfw << "-" << TD::Convert(rfw) << endl; + ScoreDominance(table,dom,lfw,SourceFWIdxs[2],lfw,SourceFWIdxs[3],bonus,bo1_bonus,bo2_bonus,true,alpha_doms,beta_doms); + } + if (rfw>=0) { + int dom = DominanceSource(SourceFWIdxs[3*SourceFWIdxs[0]-2],_J-1); + ScoreDominance(table,dom,SourceFWIdxs[3*SourceFWIdxs[0]-1],rfw,SourceFWIdxs[3*SourceFWIdxs[0]], + rfw,bonus,bo1_bonus,bo2_bonus,true,alpha_doms,beta_doms); + } + } +} + +void Alignment::computeDominanceSourcePos(const CountTable& table, WordID lfw, WordID rfw, + double *cost, double *bonus, double *bo1, double *bo1_bonus, double *bo2, double *bo2_bonus, int maxfwidx, int maxdepth1, int maxdepth2) { + if (DEBUG) cerr << "computeDominanceSourcePos" << endl; + if (DEBUG) cerr << " initial cost=" << *cost << ", initial bonus=" << *bonus << endl; + ostringstream oss; + for (int idx=2; idx<=SourceFWIdxs[0]; idx++) { + if (DEBUG) { + cerr << "PrevSourceFWIdxs :" << SourceFWIdxs[3*(idx-1)-2] << "," << SourceFWIdxs[3*(idx-1)-1] + << "," << SourceFWIdxs[3*(idx-1)] << endl; + cerr << "CurrSourceFWIdxs :" << SourceFWIdxs[3*(idx)-2] << "," << SourceFWIdxs[3*(idx)-1] + << "," << SourceFWIdxs[3*(idx)] << endl; + } + //if (!((SourceFWAbsIdxs[3*(idx-1)-2]<=maxdepth1 && SourceFWAbsIdxs[3*idx-2]<=maxdepth1) || + // (maxfwidx-SourceFWAbsIdxs[3*(idx-1)-2]+1<=maxdepth2 && maxfwidx-SourceFWAbsIdxs[3*idx-2]+1<=maxdepth2))) continue; + bool compute = true; + for (int i_ant=0; i_ant<_Arity && compute; i_ant++) { + if (MemberOf(SourceFWAntsIdxs[i_ant],SourceFWIdxs[3*(idx-1)-2],SourceFWIdxs[3*(idx)-2])) { + //cerr << "Skipping, they have been calculated in the " << (i_ant+1) << "-th branch" << endl; + compute=false; + } + } + if (compute) { + int dom = DominanceSource(SourceFWIdxs[3*(idx-1)-2],SourceFWIdxs[3*idx-2]); + if (DEBUG) cerr << "dom = " << dom << endl; + if (maxdepth1+maxdepth2>0) { + string source1 = TD::Convert(SourceFWIdxs[3*(idx-1)-1]); + string source2 = TD::Convert(SourceFWIdxs[3*(idx)-1]); + if (maxdepth1>0) { + oss << source1 << "/"; + if (SourceFWAbsIdxs[3*(idx-1)-2]<=maxdepth1) + oss << SourceFWAbsIdxs[3*(idx-1)-2]; + else + oss << "X"; + WordID source1id = TD::Convert(oss.str()); + oss.str(""); + oss << source2 << "/"; + if (SourceFWAbsIdxs[3*idx-2]<=maxdepth1) + oss << SourceFWAbsIdxs[3*idx-2]; + else + oss << "X"; + WordID source2id = TD::Convert(oss.str()); + oss.str(""); + ScoreDominance(table,dom,source1id,source2id,SourceFWIdxs[3*(idx-1)],SourceFWIdxs[3*idx], + cost,bo1,bo2,false,alpha_doms,beta_doms); + } + if (maxdepth2>0) { + oss << source1 << "/"; + if (maxfwidx-SourceFWAbsIdxs[3*(idx-1)-2]+1<=maxdepth2) + oss << ((maxfwidx-SourceFWAbsIdxs[3*(idx-1)-2]+1)*-1); + else + oss << "X"; + WordID source1id = TD::Convert(oss.str()); + oss.str(""); + oss << source2 << "/"; + if (maxfwidx-SourceFWAbsIdxs[3*idx-2]+1<=maxdepth2) + oss << ((maxfwidx-SourceFWAbsIdxs[3*(idx-1)-2]+1)*-1); + else + oss << "X"; + WordID source2id = TD::Convert(oss.str()); + oss.str(""); + ScoreDominance(table,dom,source1id,source2id,SourceFWIdxs[3*(idx-1)],SourceFWIdxs[3*idx], + cost,bo1,bo2,false,alpha_doms,beta_doms); + } + } + } + } + if (SourceFWIdxs[0]>0) { + if (lfw>=0) { + int dom = DominanceSource(0,SourceFWIdxs[1]); + string source1 = TD::Convert(lfw); + string source2 = TD::Convert(SourceFWIdxs[2]); + if (maxdepth1>0) { + oss << source1 << "/"; + if (SourceFWAbsIdxs[1]-1<=maxdepth1) + oss << (SourceFWAbsIdxs[1]-1); + else + oss << "X"; + WordID source1id = TD::Convert(oss.str()); + oss.str(""); + oss << source2 << "/"; + if (SourceFWAbsIdxs[1]<=maxdepth1) + oss << SourceFWAbsIdxs[1]; + else + oss << "X"; + WordID source2id = TD::Convert(oss.str()); + oss.str(""); + ScoreDominance(table,dom,source1id,source2id,lfw,SourceFWIdxs[3],bonus,bo1_bonus,bo2_bonus,true,alpha_doms,beta_doms); + } + if (maxdepth2>0) { + oss << source1 << "/"; + if (maxfwidx-(SourceFWAbsIdxs[1]-1)+1<=maxdepth2) + oss << ((maxfwidx-(SourceFWAbsIdxs[1]-1)+1)*-1); + else + oss << "X"; + WordID source1id = TD::Convert(oss.str()); + oss.str(""); + oss << source2 << "/"; + if (maxfwidx-SourceFWAbsIdxs[1]+1<=maxdepth2) + oss << ((maxfwidx-SourceFWAbsIdxs[1]+1)*-1); + else + oss << "X"; + WordID source2id = TD::Convert(oss.str()); + oss.str(""); + ScoreDominance(table,dom,source1id,source2id,lfw,SourceFWIdxs[3],bonus,bo1_bonus,bo2_bonus,true,alpha_doms,beta_doms); + } + } + if (rfw>=0) { + int dom = DominanceSource(SourceFWIdxs[3*SourceFWIdxs[0]-2],_J-1); + string source1 = TD::Convert(SourceFWIdxs[3*SourceFWIdxs[0]-1]); + string source2 = TD::Convert(rfw); + if (maxdepth1>0) { + oss << source1 << "/"; + if (SourceFWAbsIdxs[3*SourceFWAbsIdxs[0]-2]<=maxdepth1) + oss << SourceFWAbsIdxs[3*SourceFWIdxs[0]-2]; + else + oss << "X"; + WordID source1id = TD::Convert(oss.str()); + oss.str(""); + oss << source2 << "/"; + if (SourceFWAbsIdxs[3*SourceFWAbsIdxs[0]-2]+1<=maxdepth1) + oss << (SourceFWAbsIdxs[3*SourceFWAbsIdxs[0]-2]+1); + else + oss << "X"; + WordID source2id = TD::Convert(oss.str()); + ScoreDominance(table,dom,source1id,source2id,SourceFWIdxs[3*SourceFWIdxs[0]], + rfw,bonus,bo1_bonus,bo2_bonus,true,alpha_doms,beta_doms); + } + if (maxdepth2>0) { + oss << source1 << "/"; + if (maxfwidx-SourceFWAbsIdxs[3*SourceFWAbsIdxs[0]-2]+1<=maxdepth2) + oss << ((maxfwidx-SourceFWAbsIdxs[3*SourceFWAbsIdxs[0]-2]+1)*-1); + else + oss << "X"; + WordID source1id = TD::Convert(oss.str()); + oss.str(""); + oss << source2 << "/"; + if (maxfwidx-(SourceFWAbsIdxs[3*SourceFWAbsIdxs[0]-2]+1)+1<=maxdepth2) + oss << ((maxfwidx-(SourceFWAbsIdxs[3*SourceFWAbsIdxs[0]-2]+1)+1)*-1); + else + oss << "X"; + WordID source2id = TD::Convert(oss.str()); + oss.str(""); + ScoreDominance(table,dom,source1id,source2id,SourceFWIdxs[3*SourceFWIdxs[0]], + rfw,bonus,bo1_bonus,bo2_bonus,true,alpha_doms,beta_doms); + } + } + } +} + + +void Alignment::computeDominanceTarget(const CountTable& table, WordID lfw, WordID rfw, + double *cost, double *bonus, double *bo1, double *bo1_bonus, double *bo2, double *bo2_bonus) { + if (DEBUG) cerr << "computeDominanceTarget" << endl; + for (int idx=2; idx<=TargetFWIdxs[0]; idx++) { + if (DEBUG) cerr << "PrevTargetFWIdxs :" << TargetFWIdxs[3*(idx-1)-2] << "," << TargetFWIdxs[3*(idx-1)-1] << "," <<TargetFWIdxs[3*(idx-1)] << endl; + if (DEBUG) cerr << "CurrTargetFWIdxs :" << TargetFWIdxs[3*(idx)-2] << "," << TargetFWIdxs[3*(idx)-1] << "," <<TargetFWIdxs[3*(idx)] << endl; + bool compute = true; + for (int i_ant=0; i_ant <_Arity && compute; i_ant++) { + if (MemberOf(TargetFWAntsIdxs[i_ant],TargetFWIdxs[3*(idx-1)-2],TargetFWIdxs[3*idx-2])) { + if (DEBUG) cerr << "Skipping, they have been calculated in the " << (i_ant+1) << "-th branch" << endl; + compute = false; + } + } + if (compute) { + int dom = DominanceTarget(TargetFWIdxs[3*(idx-1)-2],TargetFWIdxs[3*idx-2]); + //cerr << (3*(idx-1)) << "," << (3*idx) << "," << (3*(idx-1)-1) << "," << (3*idx-1) << endl; + if (DEBUG) cerr << "dom target = " << dom << endl; + ScoreDominance(table,dom,TargetFWIdxs[3*(idx-1)],TargetFWIdxs[3*idx],TargetFWIdxs[3*(idx-1)-1],TargetFWIdxs[3*idx-1], + cost,bo1,bo2,false,alpha_domt,beta_domt); + } + } + if (TargetFWIdxs[0]>0) { + if (DEBUG) cerr << "backoff dominance " << endl; + if (lfw>=0) { + int dom = DominanceTarget(0,TargetFWIdxs[1]); + if (DEBUG) cerr << "dom target (with left) = " << dom << endl; + ScoreDominance(table,dom,lfw,lfw,TargetFWIdxs[2],TargetFWIdxs[3],bonus,bo1_bonus,bo2_bonus,true,alpha_domt,beta_domt); + } + if (rfw>=0) { + int dom = DominanceTarget(TargetFWIdxs[3*TargetFWIdxs[0]-2],_I-1); + if (DEBUG) cerr << "dom target (with right) = " << dom << endl; + ScoreDominance(table,dom,TargetFWIdxs[3*TargetFWIdxs[0]-1],TargetFWIdxs[3*TargetFWIdxs[0]], + rfw,rfw,bonus,bo1_bonus,bo2_bonus,true,alpha_domt,beta_domt); + } + } + + //cerr << "END of computeDominanceTarget" << endl; +} + +double Alignment::ScoreDominance(const CountTable& table, int dom, WordID source1, WordID source2, WordID target1, WordID target2) { + if (DEBUG) { + cerr << "ScoreDominance(source1=" << TD::Convert(source1) << ",source2=" << TD::Convert(source2) + << ",target1=" << TD::Convert(target1) << ",target2=" << TD::Convert(target2) << ", dom=" << dom << endl; + } + string _source1 = TD::Convert(source1); + string _source2 = TD::Convert(source2); + string _source1idx; string _source2idx; + if (table.mode==1) { + _source1idx = _source1; _source2idx = _source2; + _source1 = _source1idx.substr(0,_source1idx.find_last_of("/")); + _source2 = _source2idx.substr(0,_source2idx.find_last_of("/")); + } + string _target1 = TD::Convert(target1); + string _target2 = TD::Convert(target2); + + double count = table.ultimate[dom]; + double total = table.ultimate[4]; + double prob = count/total; + if (DEBUG) cerr << "level0 " << count << "/" << total << "=" << prob << endl; + double alpha = 0.1; + + string key = _source1 + " " + _source2; + WordID key_id = TD::Convert(key); + map<WordID,int*>::const_iterator it = table.model.find(key_id); + bool stop = (it==table.model.end()); + if (!stop) { + stop = true; + if (it->second[4]>=0) { + count = it->second[dom] + alpha*prob; + total = it->second[4] + alpha; + prob = count/total; + if (DEBUG) cerr << "level1 " << count << "/" << total << "=" << prob << endl; + stop = false; + } + } + if (stop) return prob; + + key = _source1 + " " + _source2 + " " + _target1 + " " + _target2; + key_id = TD::Convert(key); + it = table.model.find(key_id); + stop = (it==table.model.end()); + if (!stop) { + stop = true; + if (it->second[4]>=0) { + count = it->second[dom] + alpha*prob; + total = it->second[4] + alpha; + prob = count/total; + if (DEBUG) cerr << "level2 " << count << "/" << total << "=" << prob << endl; + stop = false; + } + } + + if (table.mode!=1 || stop) return prob; + key = _source1 + " " + _source2 + " " + _target1 + " " + _target2; + key_id = TD::Convert(key); + it = table.model.find(key_id); + if (it!=table.model.end()) { + if (it->second[4]>=0) { + count = it->second[dom] + alpha*prob; + total = it->second[4] + alpha; + if (DEBUG) cerr << "level3 " << count << "/" << total << "=" << prob << endl; + prob = count/total; + } + } + + return prob; +} + +void Alignment::ScoreDominance(const CountTable& table, int dom, WordID source1, WordID source2, WordID target1, WordID target2, double *cost, double *bo1, double *bo2, bool isBonus, double alpha2, double beta2) { + if (DEBUG) + cerr << "ScoreDominance(source1=" << TD::Convert(source1) << ",source2=" << TD::Convert(source2) + << ",target1=" << TD::Convert(target1) << ",target2=" << TD::Convert(target2) << ",isBonus=" << isBonus << ", alpha2 = " << alpha2 << ", beta2 = " << beta2 << endl; + if (DEBUG) cerr << " BEFORE=" << *cost << endl; + *cost += ScoreDominance(table,dom,source1,source2,target1,target2); + if (DEBUG) cerr << " AFTER=" << *cost << endl; +} + +WordID Alignment::F2EProjectionFromExternal(int idx, const vector<AlignmentPoint>& als, const string& delimiter) { + if (DEBUG) { + cerr << "F2EProjectionFromExternal=" << idx << endl; + for (int i=0; i< als.size(); i++) cerr << "als[" << i << "]=" << als[i] << " "; + cerr << endl; + } + vector<int> alignedTo; + for (int i=0; i<als.size(); i++) { + if (DEBUG) cerr << als[i] << " "; + if (als[i].s_==idx) + alignedTo.push_back(als[i].t_); + } + if (DEBUG) { + cerr << endl; + cerr << "alignedTo = "; + for (int i=0; i<alignedTo.size(); i++) cerr << alignedTo[i] << " "; + cerr << endl; + } + if (alignedTo.size()==0) { + if (DEBUG) cerr << "returns [NULL] : " << TD::Convert("NULL") << endl; + return TD::Convert("NULL"); + } else if (alignedTo.size()==1) { + if (DEBUG) cerr << "returns [" << TD::Convert(_e[alignedTo[0]]) << "] : " << _e[alignedTo[0]] << endl; + return _e[alignedTo[0]]; // if not aligned to many, why bother continuing + } else { + ostringstream projection; + for (int i=0; i<alignedTo.size(); i++) { + if (i>0) projection << delimiter; + projection << TD::Convert(_e[alignedTo[i]]); + } + if (DEBUG) { + cerr << "projection = " << projection.str() << endl; + cerr << "returns = " << TD::Convert(projection.str()) << endl; + } + return TD::Convert(projection.str()); + } +} + +WordID Alignment::E2FProjectionFromExternal(int idx, const vector<AlignmentPoint>& als, const string& delimiter) { + vector<int> alignedTo; + for (int i=0; i<als.size(); i++) + if (als[i].t_==idx) alignedTo.push_back(als[i].s_); + if (alignedTo.size()==0) { + return TD::Convert("NULL"); + } else if (alignedTo.size()==1) { + return _f[alignedTo[0]]; // if not aligned to many, why bother continuing + } else { + ostringstream projection; + for (int i=0; i<alignedTo.size(); i++) { + if (i>0) projection << delimiter; + projection << TD::Convert(_f[alignedTo[i]]); + } + return TD::Convert(projection.str()); + } +} + + +WordID Alignment::F2EProjection(int idx, const string& delimiter) { + if (DEBUG) cerr << "F2EProjection(" << idx << ")" << endl; + int e = targetOf(idx); + if (e<0) { + if (DEBUG) cerr << "projection = NULL" << endl; + return TD::Convert("NULL"); + } else { + if (targetOf(idx,e+1)<0) { + if (DEBUG) cerr << "e-1=" << (e-1) << ", size=" << _e.size() << endl; + return getE(e-1); // if not aligned to many, why bother continuing + } + ostringstream projection; + bool firstTime = true; + do { + if (!firstTime) projection << delimiter; + projection << TD::Convert(_e[e-1]); // transform space + firstTime = false; + e = targetOf(idx,e+1); + //if (DEBUG) cerr << "projection = " << projection.str() << endl; + } while(e>=0); + return TD::Convert(projection.str()); + } +} + +WordID Alignment::E2FProjection(int idx, const string& delimiter) { + //cerr << "E2FProjection(" << idx << ")" << endl; + //cerr << "i" << endl; + int f = sourceOf(idx); + //cerr << "j, f=" << f << endl; + if (f<0) { + //cerr << "projection = NULL" << endl; + return TD::Convert("NULL"); + } else { + if (sourceOf(idx,f+1)<0) return getF(f-1); + bool firstTime = true; + ostringstream projection(ostringstream::out); + do { + if (!firstTime) projection << delimiter; + projection << TD::Convert(_f[f-1]); //transform space + firstTime = false; + f = sourceOf(idx,f+1); + //cerr << "projection = " << projection.str() << endl; + } while(f>=0); + return TD::Convert(projection.str()); + } +} +void Alignment::computeBorderDominanceSource(const CountTable& table, double *cost, double *bonus, double *state_mono, + double *state_nonmono, TRule &rule, const std::vector<const void*>& ant_contexts, const map<WordID,int>& sfw) { + // HACK: GOAL is assumed to always be "S" + if (DEBUG) cerr << "computeBorderDominanceSource" << endl; + std::vector<WordID> f = rule.f(); + std::vector<WordID> e = rule.e(); + int nt_index[f.size()]; + int nt_count=0; + for (int i=0; i<f.size(); i++) nt_index[i] = (f[i]<0)? ++nt_count : 0; + if (DEBUG) { + cerr << "f = "; + for (int i=0; i<f.size(); i++) cerr << i << "." << "[" << f[i] << "] "; + cerr << endl; + cerr << "e = "; + for (int i=0; i<e.size(); i++) cerr << i << "." << "[" << e[i] << "] "; + cerr << endl; + } + bool flag[f.size()]; + for (int idx=0; idx<f.size(); idx++) flag[idx]=false; + //collect alignments + vector<int> als; + for (std::vector<AlignmentPoint>::const_iterator i = rule.als().begin(); i != rule.als().end(); ++i) { + int s = i->s_; int t = i->t_; + als.push_back(link(t,s)); + } + if (DEBUG) cerr << "rule.Arity=" << rule.Arity() << endl; + if (rule.Arity()>0) { + int ntc=0; + for (int s=0; s<f.size(); s++) { + if (f[s]<=0) { + if (DEBUG) cerr << "f[s]=" << f[s] << "+" << s << " - "; + for (int t=0; t<e.size(); t++) { + if (e[t]==ntc) { + if (DEBUG) cerr << "e[t]=" << e[t] << "+" << t <<endl; + als.push_back(link(t,s)); + ntc--; break; + } + } + } + } + } + if (DEBUG) { + cerr << "unsorted alignments (nonterminals and terminals)" << endl; + for (int i=0; i<als.size(); i++) + cerr << source(als[i]) << "-" << target(als[i]) << " "; + cerr << endl; + } + // sort alignments according to target + std::sort(als.begin(),als.end()); + if (DEBUG) { + cerr << "sorted alignments (nonterminals and terminals)" << endl; + for (int i=0; i<als.size(); i++) + cerr << source(als[i]) << "-" << target(als[i]) << " "; + cerr << endl; + } + // 0 -> neither, 1 -> leftFirst, 2 -> rightFirst, 3 -> dontCare + // ScoreDominance(const CountTable& table, int dom, WordID source1, WordID source2, WordID target1, WordID target2) + int prevs = 0; + for (int i=0; i<als.size(); i++) { + int currs = target(als[i]); //int currt = source(als[i]); + if (DEBUG) cerr << "prevs=" << prevs << ", currs=" << currs << endl << endl; + if (currs<prevs) { + if (DEBUG) cerr << "currs<prevs" << endl; + for (int s = currs; s <= prevs; s++) { + if (sfw.find(f[s])!=sfw.end()) { + WordID target = F2EProjectionFromExternal(s,rule.a_,"_SEP_"); + if (DEBUG) cerr<<" f[s]="<<TD::Convert(f[s])<<" is a function word, target="<<TD::Convert(target)<<endl; + //*cost += ScoreDominance(table,1,kSOS,f[s],kSOS,target) + ScoreDominance(table,2,f[s],kEOS,target,kEOS); + *cost += ScoreDominance(table,1,kSOS,f[s],kUNK,kUNK) + ScoreDominance(table,2,f[s],kEOS,kUNK,kUNK); + if (DEBUG) cerr << " resulting cost="<< *cost << endl; + } else if (f[s]<=0) { + if (DEBUG) cerr << " f[s]= is a nonterminal" << endl; + const int* ants = reinterpret_cast<const int *>(ant_contexts[nt_index[s]-1]); + *cost += Dwarf::IntegerToDouble(ants[51]); // 50->mono, 51->non-mono + if (DEBUG) cerr << " adding "<< Dwarf::IntegerToDouble(ants[51]) << " into cost, resulting = " << *cost << endl; + } + flag[s] = true; + } + } + prevs = currs; + } + if (DEBUG) cerr << "bonus and state matter" << endl; + for (int s=0; s<rule.f().size(); s++) { + if (!flag[s]) { + if (sfw.find(f[s])!=sfw.end()) { + WordID target = F2EProjectionFromExternal(s,rule.a_,"_SEP_"); + if (DEBUG) cerr<<" f[s]="<<TD::Convert(f[s])<<" is a function word, target="<<TD::Convert(target)<<endl; + //double indbonus = ScoreDominance(table,3,kSOS,f[s],kSOS,target) + ScoreDominance(table,3,f[s],kEOS,target,kEOS); + double indbonus = ScoreDominance(table,3,kSOS,f[s],kUNK,kUNK) + ScoreDominance(table,3,f[s],kEOS,kUNK,kUNK); + *bonus += indbonus; + *state_mono += indbonus; + //*state_nonmono += ScoreDominance(table,1,kSOS,f[s],kSOS,target) + ScoreDominance(table,2,f[s],kEOS,target,kEOS); + *state_nonmono += ScoreDominance(table,1,kSOS,f[s],kUNK,kUNK) + ScoreDominance(table,2,f[s],kEOS,kUNK,kUNK); + if (DEBUG) cerr<<" new bonus="<<*bonus<<", new state="<<*state_mono<<","<<*state_nonmono<<endl; + } else if (f[s]<=0) { + if (DEBUG) cerr << " f[s]="<< f[s] <<" is a nonterminal" << endl; + const int* ants = reinterpret_cast<const int *>(ant_contexts[nt_index[s]-1]); + double indbonus = Dwarf::IntegerToDouble(ants[50]); + *bonus += indbonus; + *state_mono += indbonus; + *state_nonmono += Dwarf::IntegerToDouble(ants[51]); + if (DEBUG) cerr << " propagating state=" << *state_mono <<","<< *state_nonmono<< endl; + } + } + } + if (DEBUG) cerr << "LHS:" << rule.GetLHS() << ":" << TD::Convert(rule.GetLHS()*-1) <<endl; + if (rule.GetLHS()*-1==TD::Convert("S")) { + *state_mono = 0; + *state_nonmono = 0; + for (int i=0; i<rule.Arity(); i++) { + const int* ants = reinterpret_cast<const int *>(ant_contexts[i]); + *cost += Dwarf::IntegerToDouble(ants[50]); + } + *bonus = 0; + } + if (DEBUG) cerr << "-->>>> cost="<<*cost<<", bonus="<<*bonus<<", state_mono="<<*state_mono<<", state_nonmono="<<*state_nonmono<<endl; +} + +bool Alignment::prepare(TRule& rule, const std::vector<const void*>& ant_contexts, const map<WordID,int>& sfw, const map<WordID,int>& tfw,const Lattice& sourcelattice, int spanstart, int spanend) { + if (DEBUG) cerr << "===Rule===" << rule.AsString() << endl; + _f = rule.f(); + _e = rule.e(); + _Arity = rule.Arity(); + if (DEBUG) { + cerr << "F: "; + for (int idx=0; idx<_f.size(); idx++) cerr << _f[idx] << " "; + cerr << endl; + cerr << "F': "; + for (int idx=0; idx<_f.size(); idx++) + if (_f[idx]>=0) { + cerr << TD::Convert(_f[idx]) << " "; + } else { + cerr << TD::Convert(_f[idx]*-1); + } + cerr << endl; + cerr << "E: "; + for (int idx=0; idx<_e.size(); idx++) + cerr << _e[idx] << " "; + cerr << endl; + cerr << "E': "; + for (int idx=0; idx<_e.size(); idx++) + if (_e[idx]>0) { + cerr << TD::Convert(_e[idx]) << " "; + } else { + cerr << "[NT]" << " "; + } + cerr << endl; + } + + SourceFWRuleIdxs[0]=0; + SourceFWRuleAbsIdxs[0]=0; + for (int idx=1; idx<=_f.size(); idx++) { // in transformed space + if (sfw.find(_f[idx-1])!=sfw.end()) { + SourceFWRuleIdxs[0]++; + SourceFWRuleAbsIdxs[++SourceFWRuleAbsIdxs[0]]=GetFWGlobalIdx(idx,sourcelattice,_f,spanstart,spanend,ant_contexts,sfw); + SourceFWRuleIdxs[3*SourceFWRuleIdxs[0]-2]=idx; + SourceFWRuleIdxs[3*SourceFWRuleIdxs[0]-1]=_f[idx-1]; + SourceFWRuleIdxs[3*SourceFWRuleIdxs[0]] =F2EProjectionFromExternal(idx-1,rule.a_,"_SEP_"); + } + } + TargetFWRuleIdxs[0]=0; + for (int idx=1; idx<=_e.size(); idx++) { // in transformed space + if (tfw.find(_e[idx-1])!=tfw.end()) { + TargetFWRuleIdxs[0]++; + TargetFWRuleIdxs[3*TargetFWRuleIdxs[0]-2]=idx; + TargetFWRuleIdxs[3*TargetFWRuleIdxs[0]-1]=E2FProjectionFromExternal(idx-1,rule.a_,"_SEP_"); + TargetFWRuleIdxs[3*TargetFWRuleIdxs[0]] =_e[idx-1]; + } + } + + if (DEBUG) { + cerr << "SourceFWRuleIdxs[" << SourceFWRuleIdxs[0] << "]:"; + for (int idx=1; idx<=SourceFWRuleIdxs[0]; idx++) { + cerr << " idx:" << SourceFWRuleIdxs[3*idx-2]; + cerr << " absidx:" << SourceFWRuleAbsIdxs[idx]; + cerr << " F:" << SourceFWRuleIdxs[3*idx-1]; + cerr << " E:" << SourceFWRuleIdxs[3*idx]; + cerr << "; "; + } + cerr << endl; + cerr << "TargetFWRuleIdxs[" << TargetFWRuleIdxs[0] << "]:"; + for (int idx=1; idx<=TargetFWRuleIdxs[0]; idx++) { + cerr << " idx:" << TargetFWRuleIdxs[3*idx-2]; + cerr << " F:" << TargetFWRuleIdxs[3*idx-1]; + cerr << " E:" << TargetFWRuleIdxs[3*idx]; + } + cerr << endl; + } + if (SourceFWRuleIdxs[0]+TargetFWRuleIdxs[0]==0) { + bool nofw = true; + for (int i_ant=0; i_ant<_Arity && nofw; i_ant++) { + const int* ants = reinterpret_cast<const int *>(ant_contexts[i_ant]); + if (ants[0]>=0||ants[3]>=0||ants[6]>=0||ants[9]>=0) nofw=false; + } + if (nofw) return true; + } + //cerr << "clearing als first" << endl; + clearAls(_J,_I); + + if (DEBUG) cerr << "A["<< rule.a_.size() << "]: " ; + RuleAl[0]=0; + // add phrase start boundary + RuleAl[0]++; RuleAl[RuleAl[0]*2-1]=0; RuleAl[RuleAl[0]*2]=0; + if (DEBUG) cerr << RuleAl[RuleAl[0]*2-1] << "-" << RuleAl[RuleAl[0]*2] << " "; + for (int idx=0; idx<rule.a_.size(); idx++) { + RuleAl[0]++; + RuleAl[RuleAl[0]*2-1]=rule.a_[idx].s_+1; + RuleAl[RuleAl[0]*2] =rule.a_[idx].t_+1; + if (DEBUG) cerr << RuleAl[RuleAl[0]*2-1] << "-" << RuleAl[RuleAl[0]*2] << " "; + } + // add phrase end boundary + RuleAl[0]++; RuleAl[RuleAl[0]*2-1]=_f.size()+1; RuleAl[RuleAl[0]*2]=_e.size()+1; + if (DEBUG) cerr << RuleAl[RuleAl[0]*2-1] << "-" << RuleAl[RuleAl[0]*2] << " "; + if (DEBUG) cerr << endl; + + SourceRuleIdxs[0] = _f.size()+2; // +2 (phrase boundaries) + TargetRuleIdxs[0] = _e.size()+2; + int ntidx=-1; + for (int idx=0; idx<_f.size()+2; idx++) { // idx in transformed space + SourceRuleIdxs[idx+1]=idx; + if (0<idx && idx<=_f.size()) if (_f[idx-1]<0) SourceRuleIdxs[idx+1]=ntidx--; + } + for (int idx=0; idx<_e.size()+2; idx++) { + TargetRuleIdxs[idx+1]=idx; + if (0<idx && idx<=_e.size()) { + //cerr << "_e[" <<(idx-1)<< "]=" << _e[idx-1] << endl; + if (_e[idx-1]<=0) TargetRuleIdxs[idx+1]=_e[idx-1]-1; + } + } + if (DEBUG) { + cerr << "SourceRuleIdxs:"; + for (int idx=0; idx<SourceRuleIdxs[0]+1; idx++) + cerr << " " << SourceRuleIdxs[idx]; + cerr << endl; + cerr << "TargetRuleIdxs:"; + for (int idx=0; idx<TargetRuleIdxs[0]+1; idx++) + cerr << " " << TargetRuleIdxs[idx]; + cerr << endl; + } + + // sloppy, the integrity of anstates is assumed + // total = 50 bytes + // first 3 ints for leftmost source function words (1 for index, 4 for source WordID and 4 for target WordI + // second 3 for rightmost source function words + // third 3 for leftmost target function words + // fourth 3 for rightmost target function words + // the next 1 int for the number of alignments + // the remaining 37 ints for alignments (source then target) + for (int i_ant=0; i_ant<_Arity; i_ant++) { + const int* ants = reinterpret_cast<const int *>(ant_contexts[i_ant]); + int span = ants[Dwarf::STATE_SIZE-1]; + if (DEBUG) { + cerr << "antcontexts[" << i_ant << "] "; + for (int idx=0; idx<Dwarf::STATE_SIZE; idx++) cerr << idx << "." << ants[idx] << " "; + cerr << endl; + cerr << "i,j = " << source(ants[Dwarf::STATE_SIZE-1]) << "," << target(ants[Dwarf::STATE_SIZE-1]) << endl; + } + SourceFWAntsIdxs[i_ant][0]=0; + SourceFWAntsAbsIdxs[i_ant][0]=0; + if (ants[0]>=0) { + // Given a span, give the index of the first function word + int firstfwidx = GetFirstFWIdx(source(span),target(span),sourcelattice,sfw); + if (DEBUG) cerr << " firstfwidx = " << firstfwidx << endl; + int fwcount = 0; + if (ants[1]>=0) { // one function word + SourceFWAntsIdxs[i_ant][0]++; SourceFWAntsIdxs[i_ant][1]=ants[0]; + SourceFWAntsIdxs[i_ant][2]=ants[1]; SourceFWAntsIdxs[i_ant][3]=ants[2]; + fwcount++; + } else { // if ants[1] < 0 then compound fws + //cerr << "ants[1]<0" << endl; + istringstream ossf(TD::Convert(ants[1]*-1)); string ffw; + istringstream osse(TD::Convert(ants[2])); string efw; //projection would be mostly NULL + int delta=ants[0]; + while (osse >> efw && ossf >> ffw) { + SourceFWAntsIdxs[i_ant][0]++; + SourceFWAntsIdxs[i_ant][SourceFWAntsIdxs[i_ant][0]*3-2]=ants[0]-(delta--); + SourceFWAntsIdxs[i_ant][SourceFWAntsIdxs[i_ant][0]*3-1]=TD::Convert(ffw); + SourceFWAntsIdxs[i_ant][SourceFWAntsIdxs[i_ant][0]*3] =TD::Convert(efw); + fwcount++; + } + } + if (DEBUG) cerr << " fwcount=" << fwcount << endl; + SourceFWAntsAbsIdxs[i_ant][0]=fwcount; + for (int i=1; i<=fwcount; i++) SourceFWAntsAbsIdxs[i_ant][i]=firstfwidx++; + } + if (ants[3]>=0) { + int lastfwidx = GetLastFWIdx(source(span),target(span),sourcelattice,sfw); + if (DEBUG) cerr << " lastfwidx = " << lastfwidx << endl; + int fwcount=0; + if (ants[4]>=0) { + fwcount++; + SourceFWAntsIdxs[i_ant][0]++; + SourceFWAntsIdxs[i_ant][SourceFWAntsIdxs[i_ant][0]*3-2]=ants[3]; + SourceFWAntsIdxs[i_ant][SourceFWAntsIdxs[i_ant][0]*3-1]=ants[4]; + SourceFWAntsIdxs[i_ant][SourceFWAntsIdxs[i_ant][0]*3] =ants[5]; + } else { // if ants[4] < 0 then compound fws + //cerr << "ants[4]<0" << endl; + istringstream ossf(TD::Convert(ants[4]*-1)); string ffw; + istringstream osse(TD::Convert(ants[5])); string efw; + int delta=0; + while (osse >> efw && ossf >> ffw) { + fwcount++; + SourceFWAntsIdxs[i_ant][0]++; + SourceFWAntsIdxs[i_ant][SourceFWAntsIdxs[i_ant][0]*3-2]=ants[3]+(delta++); + SourceFWAntsIdxs[i_ant][SourceFWAntsIdxs[i_ant][0]*3-1]=TD::Convert(ffw); + SourceFWAntsIdxs[i_ant][SourceFWAntsIdxs[i_ant][0]*3] =TD::Convert(efw); + } + } + if (DEBUG) cerr << " fwcount=" << fwcount << endl; + for (int i=1; i<=fwcount; i++) SourceFWAntsAbsIdxs[i_ant][SourceFWAntsAbsIdxs[i_ant][0]+i]=lastfwidx-fwcount+i; + SourceFWAntsAbsIdxs[i_ant][0]+=fwcount; + } + TargetFWAntsIdxs[i_ant][0]=0; + if (ants[6]>=0) { + if (ants[8]>=0) { // check the e part + TargetFWAntsIdxs[i_ant][0]++; + TargetFWAntsIdxs[i_ant][1]=ants[6]; + TargetFWAntsIdxs[i_ant][2]=ants[7]; + TargetFWAntsIdxs[i_ant][3]=ants[8]; + } else { // if ants[8] < 0 then compound fws + //cerr << "ants[8]<0" << endl; + //cerr << "ants[7]=" << TD::Convert(ants[7]) << endl; + //cerr << "ants[8]=" << TD::Convert(ants[8]*-1) << endl; + istringstream ossf(TD::Convert(ants[7])); string ffw; + istringstream osse(TD::Convert(ants[8]*-1)); string efw; + int delta=ants[6]; + while (osse >> efw && ossf >> ffw) { + //cerr << "efw="<< efw << ",ffw=" << ffw << endl; + TargetFWAntsIdxs[i_ant][0]++; + TargetFWAntsIdxs[i_ant][TargetFWAntsIdxs[i_ant][0]*3-2]=ants[6]-(delta--); + TargetFWAntsIdxs[i_ant][TargetFWAntsIdxs[i_ant][0]*3-1]=TD::Convert(ffw); + TargetFWAntsIdxs[i_ant][TargetFWAntsIdxs[i_ant][0]*3] =TD::Convert(efw); + } + } + } + if (ants[9]>=0) { + if (ants[11]>=0) { + TargetFWAntsIdxs[i_ant][0]++; + TargetFWAntsIdxs[i_ant][TargetFWAntsIdxs[i_ant][0]*3-2]=ants[9]; + TargetFWAntsIdxs[i_ant][TargetFWAntsIdxs[i_ant][0]*3-1]=ants[10]; + TargetFWAntsIdxs[i_ant][TargetFWAntsIdxs[i_ant][0]*3] =ants[11]; + } else { + //cerr << "ants[11]<0" << endl; + //cerr << "ants[10]=" << TD::Convert(ants[10]) << endl; + //cerr << "ants[11]=" << TD::Convert(ants[11]*-1) << endl; + istringstream ossf(TD::Convert(ants[10])); string ffw; + istringstream osse(TD::Convert(ants[11]*-1)); string efw; + int delta = 0; + while (osse >> efw && ossf >> ffw) { + //cerr << "efw="<< efw << ",ffw=" << ffw << endl; + TargetFWAntsIdxs[i_ant][0]++; + TargetFWAntsIdxs[i_ant][TargetFWAntsIdxs[i_ant][0]*3-2]=ants[9]+(delta++); + TargetFWAntsIdxs[i_ant][TargetFWAntsIdxs[i_ant][0]*3-1]=TD::Convert(ffw); + TargetFWAntsIdxs[i_ant][TargetFWAntsIdxs[i_ant][0]*3] =TD::Convert(efw); + } + } + } + AntsAl[i_ant][0]=ants[12];//number of alignments + for (int idx=1; idx<=AntsAl[i_ant][0]; idx++) { + AntsAl[i_ant][idx*2-1] = source(ants[12+idx]); + AntsAl[i_ant][idx*2] = target(ants[12+idx]); + } + } + + for (int i_ant=0; i_ant<_Arity; i_ant++) { + int length = AntsAl[i_ant][0]; + int maxs = -1000; + int maxt = -1000; + for (int idx=0; idx<length; idx++) { + if (maxs<AntsAl[i_ant][2*idx+1]) maxs = AntsAl[i_ant][2*idx+1]; + if (maxt<AntsAl[i_ant][2*idx+2]) maxt = AntsAl[i_ant][2*idx+2]; + } + if (DEBUG) cerr << "SourceFWAntsIdxs[" <<i_ant<<"][0]=" << SourceFWAntsIdxs[i_ant][0] << endl; + for (int idx=1; idx<=SourceFWAntsIdxs[i_ant][0]; idx++) { + if (DEBUG) { + cerr << "SourceFWAntsIdxs["<<i_ant<<"]["<<(3*idx-2)<<"]="<<SourceFWAntsIdxs[i_ant][3*idx-2]; + cerr << ","<<SourceFWAntsIdxs[i_ant][3*idx-1]<<","<<SourceFWAntsIdxs[i_ant][3*idx]<<endl; + cerr << "SourceFWAntsAbsIdxs["<<i_ant<<"]["<<idx<<"]="<<SourceFWAntsAbsIdxs[i_ant][idx] << endl; + } + if (maxs<SourceFWAntsIdxs[i_ant][3*idx-2]) maxs=SourceFWAntsIdxs[i_ant][3*idx-2]; + } + if (DEBUG) cerr << "TargetFWAntsIdxs[" <<i_ant<<"][0]=" << TargetFWAntsIdxs[i_ant][0] << endl; + for (int idx=1; idx<=TargetFWAntsIdxs[i_ant][0]; idx++) { + if (DEBUG) { + cerr << "TargetFWAntsIdxs["<<i_ant<<"]["<<(3*idx-2)<<"]="<<TargetFWAntsIdxs[i_ant][3*idx-2]; + cerr << ","<<TargetFWAntsIdxs[i_ant][3*idx-1]<<","<<TargetFWAntsIdxs[i_ant][3*idx]<<endl; + } + if (maxt<TargetFWAntsIdxs[i_ant][3*idx-2]) maxt=TargetFWAntsIdxs[i_ant][3*idx-2]; + } + SourceAntsIdxs[i_ant][0] = maxs+1; + if (DEBUG) cerr << "SourceAntsIdxs[" << i_ant << "][0]=" <<SourceAntsIdxs[i_ant][0] << endl; + for (int idx=0; idx<SourceAntsIdxs[i_ant][0]; idx++) SourceAntsIdxs[i_ant][idx+1]=idx; + TargetAntsIdxs[i_ant][0] = maxt+1; + if (DEBUG) cerr << "TargetAntsIdxs[" << i_ant << "][0]=" <<TargetAntsIdxs[i_ant][0] << endl; + for (int idx=0; idx<TargetAntsIdxs[i_ant][0]; idx++) TargetAntsIdxs[i_ant][idx+1]=idx; + } + int TotalSource = SourceRuleIdxs[0] - _Arity; + for (int idx=0; idx<_Arity; idx++) TotalSource += SourceAntsIdxs[idx][0]; + int TotalTarget = TargetRuleIdxs[0] - _Arity; + for (int idx=0; idx<_Arity; idx++) TotalTarget += TargetAntsIdxs[idx][0]; + if (DEBUG) cerr << "TotalSource = "<< TotalSource << ", TotalTarget = "<< TotalTarget << endl; + int curr = 0; + for (int idx=1; idx<=SourceRuleIdxs[0]; idx++) { + if (SourceRuleIdxs[idx]>=0) { + SourceRuleIdxs[idx]=curr++; + } else { + int i_ant = SourceRuleIdxs[idx]*-1-1; + if (DEBUG) cerr << "SourceAntsIdxs[" << i_ant << "]" << endl; + for (int idx2=1; idx2<=SourceAntsIdxs[i_ant][0]; idx2++) { + SourceAntsIdxs[i_ant][idx2]=curr++; + if (DEBUG) cerr << SourceAntsIdxs[i_ant][idx2] << " "; + } + if (DEBUG) cerr << endl; + } + } + if (DEBUG) { + cerr << "SourceRuleIdxs" << endl; + for (int idx=1; idx<=SourceRuleIdxs[0]; idx++) cerr << SourceRuleIdxs[idx] << " "; + cerr << endl; + } + curr = 0; + for (int idx=1; idx<=TargetRuleIdxs[0]; idx++) { + if (TargetRuleIdxs[idx]>=0) { + TargetRuleIdxs[idx]=curr++; + } else { + int i_ant = TargetRuleIdxs[idx]*-1-1; + if (DEBUG) cerr << "TargetRuleIdxs[" << i_ant << "]" << endl; + for (int idx2=1; idx2<=TargetAntsIdxs[i_ant][0]; idx2++) { + TargetAntsIdxs[i_ant][idx2]=curr++; + if (DEBUG) cerr << TargetAntsIdxs[i_ant][idx2] << " "; + } + if (DEBUG) cerr << endl; + } + } + if (DEBUG) { + cerr << "TargetRuleIdxs" << endl; + for (int idx=1; idx<=TargetRuleIdxs[0]; idx++) cerr << TargetRuleIdxs[idx] << " "; + cerr << endl; + } + for (int idx=1; idx<=RuleAl[0]; idx++) { + if (DEBUG) { + cerr << RuleAl[idx*2-1] << " - " << RuleAl[idx*2] << " to "; + cerr << SourceRuleIdxs[RuleAl[idx*2-1]+1] << " - " << TargetRuleIdxs[RuleAl[idx*2]+1] << endl; + } + set(SourceRuleIdxs[RuleAl[idx*2-1]+1],TargetRuleIdxs[RuleAl[idx*2]+1]); + } + for (int i_ant=0; i_ant<_Arity; i_ant++) { + for (int idx=1; idx<=AntsAl[i_ant][0]; idx++) { + if (DEBUG) { + cerr << AntsAl[i_ant][2*idx-1] << " - " << AntsAl[i_ant][2*idx] << " to "; + cerr << SourceAntsIdxs[i_ant][AntsAl[i_ant][2*idx-1]+1] << " - "; + cerr << TargetAntsIdxs[i_ant][AntsAl[i_ant][2*idx]+1] << endl; + } + set(SourceAntsIdxs[i_ant][AntsAl[i_ant][2*idx-1]+1],TargetAntsIdxs[i_ant][AntsAl[i_ant][2*idx]+1]); + } + } + SourceFWIdxs[0]=0; + SourceFWAbsIdxs[0]=0; + if (DEBUG) cerr << "SourceFWRuleIdxs:" << endl; + for (int idx=1; idx<=SourceFWRuleIdxs[0]; idx++) { + if (DEBUG) cerr << SourceFWRuleIdxs[3*idx-2] << " to " << SourceRuleIdxs[SourceFWRuleIdxs[3*idx-2]+1] << endl; + SourceFWRuleIdxs[3*idx-2] = SourceRuleIdxs[SourceFWRuleIdxs[3*idx-2]+1]; + SourceFWAbsIdxs[0]++; + SourceFWAbsIdxs[3*SourceFWAbsIdxs[0]-2]=SourceFWRuleAbsIdxs[idx]; + SourceFWIdxs[0]++; + SourceFWIdxs[3*SourceFWIdxs[0]-2]=SourceFWRuleIdxs[3*idx-2]; + SourceFWIdxs[3*SourceFWIdxs[0]-1]=SourceFWRuleIdxs[3*idx-1]; + SourceFWIdxs[3*SourceFWIdxs[0]] =SourceFWRuleIdxs[3*idx]; + } + for (int i_ant=0; i_ant<_Arity; i_ant++) { + if (DEBUG) cerr << "SourceFWAntsIdxs[" << i_ant << "]" << endl; + for (int idx=1; idx<=SourceFWAntsIdxs[i_ant][0]; idx++) { + if (DEBUG) + cerr << SourceFWAntsIdxs[i_ant][3*idx-2] << " to " << SourceAntsIdxs[i_ant][SourceFWAntsIdxs[i_ant][3*idx-2]+1] << endl; + SourceFWAntsIdxs[i_ant][3*idx-2] = SourceAntsIdxs[i_ant][SourceFWAntsIdxs[i_ant][3*idx-2]+1]; + SourceFWAbsIdxs[0]++; + SourceFWAbsIdxs[3*SourceFWAbsIdxs[0]-2]=SourceFWAntsAbsIdxs[i_ant][idx]; + SourceFWIdxs[0]++; + SourceFWIdxs[3*SourceFWIdxs[0]-2]=SourceFWAntsIdxs[i_ant][3*idx-2]; + SourceFWIdxs[3*SourceFWIdxs[0]-1]=SourceFWAntsIdxs[i_ant][3*idx-1]; + SourceFWIdxs[3*SourceFWIdxs[0]] =SourceFWAntsIdxs[i_ant][3*idx]; + } + } + sort(SourceFWIdxs); + sort(SourceFWAbsIdxs); + if (DEBUG) { + cerr << "SourceFWIdxs : "; + for (int idx=1; idx<=SourceFWIdxs[0]; idx++) { + cerr << "idx:" << SourceFWIdxs[3*idx-2] << ","; + cerr << "F:" << SourceFWIdxs[3*idx-1] << ","; + cerr << "E:" << SourceFWIdxs[3*idx] << " "; + } + cerr << endl; + } + TargetFWIdxs[0]=0; + if (DEBUG) cerr << "TargetFWRuleIdxs:" << endl; + for (int idx=1; idx<=TargetFWRuleIdxs[0]; idx++) { + if (DEBUG) cerr << TargetFWRuleIdxs[3*idx-2] << " to " << TargetRuleIdxs[TargetFWRuleIdxs[3*idx-2]+1] << endl; + TargetFWRuleIdxs[3*idx-2] = TargetRuleIdxs[TargetFWRuleIdxs[3*idx-2]+1]; + TargetFWIdxs[0]++; + TargetFWIdxs[3*TargetFWIdxs[0]-2]=TargetFWRuleIdxs[3*idx-2]; + TargetFWIdxs[3*TargetFWIdxs[0]-1]=TargetFWRuleIdxs[3*idx-1]; + TargetFWIdxs[3*TargetFWIdxs[0]] =TargetFWRuleIdxs[3*idx]; + } + for (int i_ant=0; i_ant<_Arity; i_ant++) { + if (DEBUG) cerr << "TargetFWAntsIdxs[" << i_ant << "]" << endl; + for (int idx=1; idx<=TargetFWAntsIdxs[i_ant][0]; idx++) { + if (DEBUG) cerr << TargetFWAntsIdxs[i_ant][3*idx-2] << " to " << TargetAntsIdxs[i_ant][TargetFWAntsIdxs[i_ant][3*idx-2]+1] << endl; + TargetFWAntsIdxs[i_ant][3*idx-2] = TargetAntsIdxs[i_ant][TargetFWAntsIdxs[i_ant][3*idx-2]+1]; + TargetFWIdxs[0]++; + TargetFWIdxs[3*TargetFWIdxs[0]-2]=TargetFWAntsIdxs[i_ant][3*idx-2]; + TargetFWIdxs[3*TargetFWIdxs[0]-1]=TargetFWAntsIdxs[i_ant][3*idx-1]; + TargetFWIdxs[3*TargetFWIdxs[0]] =TargetFWAntsIdxs[i_ant][3*idx]; + } + } + sort(TargetFWIdxs); + if (DEBUG) { + cerr << "TargetFWIdxs : "; + for (int idx=1; idx<=TargetFWIdxs[0]; idx++) { + cerr << "idx:" << TargetFWIdxs[3*idx-2]<< ","; + cerr << "E:" << TargetFWIdxs[3*idx-1]<< ","; + cerr << "F:" << TargetFWIdxs[3*idx]<< " "; + } + cerr << endl; + cerr << AsString() << endl; + } + fas = firstSourceAligned(1); las = lastSourceAligned(_J-2); + fat = firstTargetAligned(1); lat = lastTargetAligned(_I-2); + if (DEBUG) cerr << "fas=" << fas << ", las=" << las << ", fat=" << fat << ", lat=" << lat << endl; + assert(fas<=las); + assert(fat<=lat); + SetCurrAlVector(); + if (DEBUG) cerr << "end prepare" << endl; + return false; +} + +string Alignment::AsStringSimple() { + ostringstream stream; + for (int j=0; j<getJ(); j++) { + int t = targetOf(j,minTSpan(j)); + while (t>=0) { + stream << " " << j << "-" << t; + t = targetOf(j,t+1); + } + } + return stream.str(); +}; + + +string Alignment::AsString() { + ostringstream stream; + stream << "J:" << getJ() << " I:" << getI(); + for (int j=0; j<getJ(); j++) { + int t = targetOf(j,minTSpan(j)); + while (t>=0) { + stream << " " << j << "-" << t; + t = targetOf(j,t+1); + } + } + stream << " TargetSpan:"; + for (int j=0; j<getJ(); j++) + if (minTSpan(j)!=MINIMUM_INIT) + stream << " " << j << "[" << minTSpan(j) << "," << maxTSpan(j) << "]"; + else + stream << " " << j << "[-,-]"; + stream << " SourceSpan:"; + for (int i=0; i<getI(); i++) + if (minSSpan(i)!=MINIMUM_INIT) + stream << " " << i << "[" << minSSpan(i) << "," << maxSSpan(i) << "]"; + else + stream << " " << i << "[-,-]"; + return stream.str(); +}; + +void Alignment::SetCurrAlVector() { + curr_al.clear(); + for (int j=0; j<_J; j++) { + int i = targetOf(j); + while (i>=0) { + curr_al.push_back(link(j,i)); + i = targetOf(j,i+1); + } + } +} + +const void CountTable::print() { + cerr << "+++ Model +++" << endl; + for (map<WordID,int*>::const_iterator iter=model.begin(); iter!=model.end(); iter++) { + cerr << TD::Convert(iter->first) << " "; + for (int i=0; i<numColumn; i++) cerr << iter->second[i] << " "; + cerr << endl; + } + cerr << "+++ Ultimate +++" << endl; + for (int i=0; i<numColumn; i++) cerr << ultimate[i] << " "; + cerr << endl; +} + +void Alignment::ToArrayInt(vector<int>* ret) { + ret->clear(); + for (int i=0; i<_J; i++) { + int t = targetOf(i); + while (t>=0) { + ret->push_back(link(i,t)); + t = targetOf(i,t+1); + } + } +} + +int Alignment::GetFWGlobalIdx(int idx, const Lattice& sourcelattice, vector<WordID>& sources, int spanstart, int spanend, const std::vector<const void*>& ant_contexts, const map<WordID,int>& sfw) { + // get the index of the function word in the lattice + if (DEBUG) cerr << " GetFWGlobalIdx(" << idx << "," << spanstart << "," << spanend << ")" << endl; + int curr = spanstart; int i_ant = 0; + for (int i=1; i<sources.size() && i<idx; i++) { // sources contain <s> and </s> + if (sources[i]<0) { + const int* ants = reinterpret_cast<const int *>(ant_contexts[i_ant++]); + int antstate = ants[Dwarf::STATE_SIZE-1]; + if (DEBUG) cerr << " found NT[" << target(antstate) << "," << source(antstate) << "]" << endl; + curr += target(antstate)-source(antstate); + } else { + curr++; + } + } + if (DEBUG) cerr << " curr = " << curr << endl; + //compute the fw index + int ret = 1; + for (int i=0; i<curr; i++) { + if (sfw.find(sourcelattice[i][0].label)!=sfw.end()) ret++; + } + if (DEBUG) cerr << " ret = " << ret << endl; + return ret; +} + +int Alignment::GetFirstFWIdx(int spanstart,int spanend, const Lattice& sourcelattice, const map<WordID,int>& sfw) { + if (DEBUG) cerr << " GetFirstFWIdx(" << spanstart << "," << spanend << ")" << endl; + int curr=0; + for (int i=0; i<spanend; i++) { + if (sfw.find(sourcelattice[i][0].label)!=sfw.end()) { + curr++; + if (i>=spanstart) return curr; + } + } +// assert(0); + return curr; +} + +int Alignment::GetLastFWIdx(int spanstart,int spanend, const Lattice& sourcelattice, const map<WordID,int>& sfw) { + if (DEBUG) cerr << " GetLastFWIdx(" << spanstart << "," << spanend << ")" << endl; + int curr=0; + for (int i=0; i<spanend; i++) { + if (sfw.find(sourcelattice[i][0].label)!=sfw.end()) { + curr++; + } + } + return curr; +} + +WordID Alignment::generalize(WordID original, const map<WordID,WordID>& tags, bool pos) { + if (!pos) { + map<WordID,WordID>::const_iterator it = tags.find(original); + if (it!=tags.end()) { + return it->second; + } + } else { + string key,idx; + Dwarf::stripIndex(TD::Convert(original),&key,&idx); + map<WordID,WordID>::const_iterator it = tags.find(TD::Convert(key)); + if (it!=tags.end()) { + ostringstream oss; + oss << TD::Convert(it->second) << "/" << idx; + return TD::Convert(oss.str()); + } + } + return original; +} + +int* Alignment::SOS() { + int* neighbor = new int[4]; + neighbor[0]=0; neighbor[1]=0; + neighbor[2]=0; neighbor[3]=0; + return neighbor; +} + +int* Alignment::EOS() { + int* neighbor = new int[4]; + neighbor[0]=getJ()-1; neighbor[1]=neighbor[0]; + neighbor[2]=getI()-1; neighbor[3]=neighbor[2]; + return neighbor; +} + +int* Alignment::neighborLeft(int startidx, int endidx, bool* getit) { + if (DEBUG) cerr << " neighborLeft("<<startidx<<","<<endidx<<")"<<endl; + int lborder = startidx; + int* ret; + while(lborder<=endidx) { + ret = blockSource(lborder,endidx); + if (ret[0]==lborder && ret[1]==endidx && ret[2]!=MINIMUM_INIT) { + *getit = true; + return ret; + } else { + delete[] ret; + lborder++; + } + } + ret = new int[4]; + ret[0]=-1; ret[1]=-1; ret[2]=-1; ret[3]=-1; + *getit = false; + return ret; +} + +int* Alignment:: neighborRight(int startidx, int endidx, bool* getit) { + if (DEBUG) cerr << " neighborRight("<<startidx<<","<<endidx<<")"<<endl; + int rborder = endidx; + int* ret; + while(startidx<=rborder) { + ret = blockSource(startidx,rborder); + if (ret[0]==startidx && ret[1]==rborder && ret[2]!=MINIMUM_INIT) { + *getit = true; + return ret; + } else { + delete[] ret; + rborder--; + } + } + ret = new int[4]; + ret[0]=-1; ret[1]=-1; ret[2]=-1; ret[3]=-1; + *getit = false; + return ret; +} diff --git a/decoder/dwarf.h b/decoder/dwarf.h new file mode 100644 index 00000000..83a0cae9 --- /dev/null +++ b/decoder/dwarf.h @@ -0,0 +1,286 @@ +#ifndef DWARF_H +#define DWARF_H + +#include <cstdlib> +#include <vector> +#include <map> +#include <string> +#include <ostream> +#include "wordid.h" +#include "lattice.h" +#include "trule.h" +#include "tdict.h" +#include <boost/functional/hash.hpp> +#include <tr1/unordered_map> +#include <boost/tuple/tuple.hpp> + +using namespace std; +using namespace std::tr1; +using namespace boost::tuples; +using namespace boost; + +const static bool DEBUG = false; + +class CountTable { +public: + int* ultimate; + map<WordID,int*> model; + int mode; + int numColumn; + const void print(); + void setup(int _numcolumn, int _mode) { + mode = _mode; numColumn = _numcolumn; + } +}; + +class Alignment { +/* Alignment represents an alignment object in a 2D format to support function word-based models calculation + + A note about model's parameter estimation: + ========================================== + The model is estimated as a two-level Dirichlet process. + For orientation model, the first tier estimation is: + P(o|f,e) where *o* is the orientation value to estimate, *f* is the source function word aligned to *e* + its second tier is: P(o|f), while its third tier is P(o) + For dominance model, the first tier estimation is: + P(d|f1,f2,e1,e2) where *d* is a dominance value to estimate, *f1,f2* are the neighboring function words on the source + aligned to *e1,e2* on the target side + its second tier is: P(d|f1,f2) while its third tier is P(d) + + Taking orientation model as a case in point, a two level estimation proceeds as follow: + P(o|f,e) = c(o,f,e) + alpha { c(o,f) + beta [ c (o) / c(.) ] } + ------------------------------ + c(f) + beta + ------------------------------------------------- + c(f,e) + alpha + where c() is a count function, alpha and beta are the concentration parameter + of the first and second Dirichlet process respectively + To encourage or penalize the use of second and third tier statistics, bo1 and bo2 binary features are introduced +*/ +public: + const static int MAX_WORDS = 200; + const static int MINIMUM_INIT = 1000; + const static int MAXIMUM_INIT = -1000; + const static int MAX_ARITY = 2; + WordID kSOS; + WordID kEOS; + WordID kUNK; + double alpha_oris; // 1st concentration parameter for orientation model + double beta_oris; // 2nd concentration parameter for orientation model + double alpha_orit; // 1st concentration parameter for orientation model + double beta_orit; // 2nd concentration parameter for orientation model + double alpha_doms; // idem as above but for dominance model + double beta_doms; + double alpha_domt; // idem as above but for dominance model + double beta_domt; + + // ACCESS to alignment + void set(int j,int i); // j is the source index, while i is the target index + void reset(int j,int i); // idem as above + inline bool at(int j, int i) { return _matrix[j][i]; }; + inline int getJ() {return _J;}; // max source of the current alignment + inline int getI() {return _I;}; // max target of the current alignment + inline void setI(int I) { _I = I; }; + inline void setJ(int J) { _J = J; }; + inline void setF(vector<WordID> f) { _f=f;}; + inline void setE(vector<WordID> e) { _e=e;}; + inline WordID getF(int id) { if (id<0) return TD::Convert("<s>"); if (id>=_f.size()) return TD::Convert("</s>"); return _f[id];}; + inline WordID getE(int id) { if (id<0) return TD::Convert("<s>"); if (id>=_e.size()) return TD::Convert("</s>"); return _e[id];}; + void clearAls(int prevJ=200, int prevI=200); + int sourceOf(int i, int start = -1); + int targetOf(int j, int start = -1); + inline int minSSpan(int i) { return _sSpan[i][0];} + inline int maxSSpan(int i) { return _sSpan[i][1];} + inline int minTSpan(int j) { return _tSpan[j][0];} + inline int maxTSpan(int j) { return _tSpan[j][1];} + static inline int link(int s, int t) { return (s << 16) | t; } + static inline int source(int st) {return st >> 16; } + static inline int target(int st) {return st & 0xffff; } + inline void setAlphaOris(double val) { alpha_oris=val; } + inline void setAlphaOrit(double val) { alpha_orit=val; } + inline void setAlphaDoms(double val) { alpha_doms=val; } + inline void setAlphaDomt(double val) { alpha_domt=val; } + inline void setBetaOris(double val) { beta_oris=val; } + inline void setBetaOrit(double val) { beta_orit=val; } + inline void setBetaDoms(double val) { beta_doms=val; } + inline void setBetaDomt(double val) { beta_domt=val; } + inline void setFreqCutoff(int val) { cout << _freq_cutoff << " to " << val << endl; _freq_cutoff=val; } + string AsString(); + string AsStringSimple(); + int* SOS(); + int* EOS(); + + // Model related function + Alignment(); + // Given the current *rule* and its antecedents, construct an alignment space and mark the function word alignments + // according *sfw* and *tfw* + bool prepare(TRule& rule, const std::vector<const void*>& ant_contexts, + const map<WordID,int>& sfw, const map<WordID,int>& tfw, const Lattice& sourcelattice, int spanstart, int spanend); + + // Compute orientation model score which parameters are stored in *table* and pass the values accordingly + // will call Orientation(Source|Target) and ScoreOrientation(Source|Target) + void computeOrientationSource(const CountTable& table, double *cost, double *bonus, double *bo1, + double *bo1_bonus, double *bo2, double *bo2_bonus); + void computeOrientationSourcePos(const CountTable& table, double *cost, double *bonus, + double *bo1, double *bo1_bonus, double *bo2, double *bo2_bonus, int maxfwidx, int maxdepth1, int maxdepth2); + void computeOrientationSourceGen(const CountTable& table, double *cost, double *bonus, double *bo1, + double *bo1_bonus, double *bo2, double *bo2_bonus, const map<WordID,WordID>& tags); + void computeOrientationSourceBackward(const CountTable& table, double *cost, double *bonus, double *bo1, + double *bo1_bonus, double *bo2, double *bo2_bonus); + void computeOrientationSourceBackwardPos(const CountTable& table, double *cost, double *bonus, double *bo1, + double *bo1_bonus, double *bo2, double *bo2_bonus, int maxfwidx, int maxdepth1, int maxdepth2); + void computeOrientationTarget(const CountTable& table, double *cost, double *bonus, double *bo1, + double *bo1_bonus, double *bo2, double *bo2_bonus); + void computeOrientationTargetBackward(const CountTable& table, double *cost, double *bonus, double *bo1, + double *bo1_bonus, double *bo2, double *bo2_bonus); + // Get the orientation value of a function word at a particular index *fw* + // assign the value to either *oril* or *orir* accoring to *Lcompute* and *Rcompute* + void OrientationSource(int fw, int*oril, int* orir, bool Lcompute=true, bool Rcompute=true); + void OrientationSource(int fw0, int fw1, int*oril, int* orir, bool Lcompute=true, bool Rcompute=true); + int OrientationSource(int* left, int* right); + void OrientationTarget(int fw, int*oril, int* orir, bool Lcompute=true, bool Rcompute=true); + void OrientationTarget(int fw0, int fw1, int*oril, int* orir, bool Lcompute=true, bool Rcompute=true); + + vector<int> OrientationSourceLeft4Sampler(int fw0, int fw1); + vector<int> OrientationSourceLeft4Sampler(int fw); + vector<int> OrientationSourceRight4Sampler(int fw0, int fw1); + vector<int> OrientationSourceRight4Sampler(int fw); + vector<int> OrientationTargetLeft4Sampler(int fw0, int fw1); + vector<int> OrientationTargetLeft4Sampler(int fw); + vector<int> OrientationTargetRight4Sampler(int fw0, int fw1); + vector<int> OrientationTargetRight4Sampler(int fw); + + // Given an orientation value *ori*, estimate the score accoding to *cond1*, *cond2* + // and assign the value accordingly according to *isBonus* and whether the first or the second tier estimation + // is used or not + void ScoreOrientationRight(const CountTable& table, int ori, WordID cond1, WordID cond2, + bool isBonus, double *cost, double *bonus, double *bo1, double *bo1_bonus, + double *bo2, double *bo2_bonus, double alpha1, double beta1); + void ScoreOrientationLeft(const CountTable& table, int ori, WordID cond1, WordID cond, + bool isBonus, double *cost, double *bonus, double *bo1, double *bo1_bonus, + double *bo2, double *bo2_bonus, double alpha1, double beta1); + double ScoreOrientationRight(const CountTable& table, int ori, WordID cond1, WordID cond2); + double ScoreOrientationLeft(const CountTable& table, int ori, WordID cond1, WordID cond); + void ScoreOrientationRightBackward(const CountTable& table, int ori, WordID cond1, WordID cond2, + bool isBonus, double *cost, double *bonus, double *bo1, double *bo1_bonus, + double *bo2, double *bo2_bonus, double alpha1, double beta1); + void ScoreOrientationLeftBackward(const CountTable& table, int ori, WordID cond1, WordID cond, + bool isBonus, double *cost, double *bonus, double *bo1, double *bo1_bonus, + double *bo2, double *bo2_bonus, double alpha1, double beta1); + double ScoreOrientationRightBackward(const CountTable& table, int ori, WordID cond1, WordID cond2); + double ScoreOrientationLeftBackward(const CountTable& table, int ori, WordID cond1, WordID cond); + void ScoreOrientation(const CountTable& table, int offset, int ori, WordID cond1, WordID cond2, + bool isBonus, double *cost, double *bonus, double *bo1, double *bo1_bonus, + double *bo2, double *bo2_bonus, double alpha1, double beta1); + double ScoreOrientation(const CountTable& table, int offset, int ori, WordID cond1, WordID cond2); + + // idem as above except these are for dominance model + void computeDominanceSource(const CountTable& table, WordID lfw, WordID rfw, double *cost, double *bonus, + double *bo1, double *bo1_bonus, double *bo2, double *bo2_bonus); + void computeDominanceSourcePos(const CountTable& table, WordID lfw, WordID rfw, double *cost, double *bonus, + double *bo1, double *bo1_bonus, double *bo2, double *bo2_bonus, int maxfwidx, int maxdepth1, int maxdepth2); + void computeDominanceTarget(const CountTable& table, WordID lfw, WordID rfw, double *cost, double *bonus, + double *bo1, double *bo1_bonus, double *bo2, double *bo2_bonus); + void computeBorderDominanceSource(const CountTable& table, double *cost, double *bonus, + double *state_mono, double *state_nonmono, + TRule &rule, const std::vector<const void*>& ant_contexts, const map<WordID,int>& sfw); + int DominanceSource(int fw1, int fw2); + int DominanceTarget(int fw1, int fw2); + vector<int> DominanceSource4Sampler(int fw1, int fw2); + vector<int> DominanceTarget4Sampler(int fw1, int fw2); + void ScoreDominance(const CountTable& table, int dom, WordID s1, WordID s2, WordID t1, WordID t2, + double *cost, double *bo1, double *bo2, bool isBonus, double alpha2, double beta2); + double ScoreDominance(const CountTable& table, int dom, WordID s1, WordID s2, WordID t1, WordID t2); + + // Remove all function word alignments except those at the borders + // May result in more than two function word alignments at each side, because this function + // will continue keeping function word alignments until the first aligned word at each side + void BorderingSFWsOnly(); + void BorderingTFWsOnly(); + void simplify(int *ret); // preparing the next state + void simplify_nofw(int *ret); // preparing the next state when no function word appears + // set the first part of the next state, which concerns with function word + // fas, las, fat, lat is the (f)irst or (l)ast function word alignments either on the (s)ource or (t)arget + // these parameters to anticipate cases where there are more than two function word alignments + void FillFWIdxsState(int *state, int fas, int las, int fat, int lat); + + // Helper function to obtain the aligned words on the other side + // WARNING!!! Only to be used if the als are in sync with either source or target sentences + WordID F2EProjectionFromExternal(int idx, const vector<AlignmentPoint>& als, const string& delimiter=" "); + WordID E2FProjectionFromExternal(int idx, const vector<AlignmentPoint>& als, const string& delimiter=" "); + // WARNING!!! Only to be used in dwarf_main.cc + // These two function words assume that the alignment contains phrase boundary + // but the source and target sentences do not + WordID F2EProjection(int idx, const string& delimiter=" "); + WordID E2FProjection(int idx, const string& delimiter=" "); + void SetCurrAlVector(); + int* blockSource(int fw1, int fw2); + int* blockTarget(int fw1, int fw2); + void ToArrayInt(vector<int>* arr); + int* neighborLeft(int startidx, int endidx, bool* found); + int* neighborRight(int startidx, int endidx, bool* found); +private: + // Hash to avoid redundancy + unordered_map<vector<int>, int, boost::hash<vector<int> > > oris_hash; + unordered_map<vector<int>, int, boost::hash<vector<int> > > orit_hash; + unordered_map<vector<int>, int, boost::hash<vector<int> > > doms_hash; + unordered_map<vector<int>, int, boost::hash<vector<int> > > domt_hash; + unordered_map<vector<int>, vector<int>, boost::hash<vector<int> > > simplify_hash; + unordered_map<vector<int>, vector<int>, boost::hash<vector<int> > > prepare_hash; + + int _J; // effective source length; + int _I; // effective target length; + bool _matrix[MAX_WORDS][MAX_WORDS]; // true if aligned + short _sSpan[MAX_WORDS][2]; //the source span of a target index; 0->min, 1->max + short _tSpan[MAX_WORDS][2]; //the target span of a source index; 0->min, 2->max + int _freq_cutoff; + int SourceFWRuleIdxs[40]; //the indexes of function words in the rule; + // The following applies to all *FW*Idxs + // *FW*Idxs[0] = size + // *FW*Idxs[idx*3-2] = index in the alignment, where idx starts from 1 to size + // *FW*Idxs[idx*3-1] = source WordID + // *FW*Idxs[idx*3] = target WordID + int SourceFWRuleAbsIdxs[40]; + int TargetFWRuleIdxs[40]; //the indexes of function words in the rule; zeroth element is the count + int ** SourceFWAntsIdxs; //the indexes of function words in antecedents + int ** SourceFWAntsAbsIdxs; + int ** TargetFWAntsIdxs; //the indexes of function words in antecedents + int SourceRuleIdxs[40]; //the indexes of SOURCE tokens (zeroth element is the number of source tokens) + //>0 means terminal, -i means the i-th Xs + int TargetRuleIdxs[40]; //the indexes of TARGET tokens (zeroth element is the number of target tokens) + int ** SourceAntsIdxs; //the array of indexes of a particular antecedent's SOURCE tokens + int ** TargetAntsIdxs; //the array of indexes of a particular antecedent's TARGET tokens + int SourceFWIdxs[40]; + int SourceFWAbsIdxs[40]; + int TargetFWIdxs[40]; + // *sort* and *quickSort* are used to sort *FW*Idxs + void sort(int* num); + void quickSort(int arr[], int top, int bottom); + + // *block(Source|Target)* finds the minimum block that containts two indexes (fw1 and fw2) + inline int least(int i1, int i2) { return (i1<i2)?i1:i2; } + inline int most(int i1, int i2) { return (i1>i2)?i1:i2; } + void simplifyBackward(vector<int *>*blocks, int* block, const vector<int>& danglings); + // used in simplify to check whether an atomic block according to source function words is also atomic according + // to target function words as well, otherwise break it + // the resulting blocks are added into *blocks* + int _Arity; + std::vector<WordID> _f; // the source sentence of the **current** rule (may not consistent with the current alignment) + std::vector<WordID> _e; // the target sentence of the **current** rule + int RuleAl[40]; + int **AntsAl; + int firstSourceAligned(int start); + int firstTargetAligned(int start); + int lastSourceAligned(int end); + int lastTargetAligned(int end); + int fas, las, fat, lat; // first aligned source, last aligned source, first aligned target, last aligned target + bool MemberOf(int* FWIdxs, int pos1, int pos2); // whether FWIdxs contains pos1 and pos2 consecutively + // Convert the alignment to vector form, will be used for hashing purposes + vector<int> curr_al; + int GetFWGlobalIdx(int idx, const Lattice& sourcelattice, vector<WordID>& sources, int spanstart, int spanend, const std::vector<const void*>& ant_contexts, const map<WordID,int>& sfw); + int GetFirstFWIdx(int spanstart,int spanend, const Lattice& sourcelattice, const map<WordID,int>& sfw); + int GetLastFWIdx(int spanstart,int spanend, const Lattice& sourcelattice, const map<WordID,int>& sfw); + WordID generalize(WordID original, const map<WordID,WordID>& tags, bool pos=false); +}; + +#endif diff --git a/decoder/ff_dwarf.cc b/decoder/ff_dwarf.cc new file mode 100644 index 00000000..3daa85ac --- /dev/null +++ b/decoder/ff_dwarf.cc @@ -0,0 +1,893 @@ +#include <vector> +#include <sstream> +#include <fstream> +#include <string> +#include <iostream> +#include <map> +#include "ff_dwarf.h" +#include "dwarf.h" +#include "wordid.h" +#include "tdict.h" +#include "filelib.h" +#include "sentence_metadata.h" +#include "stringlib.h" + +using namespace std; + +Dwarf::Dwarf(const std::string& param) { +/* Param is a space separated string which contains any or all of the following: + oris|orit|doms|domt=filename + e.g. oris=/fs/clip-galep3eval/hendra/z2e/oris128.gz +*/ + sSOS="<s>"; + sEOS="</s>"; + kSOS=TD::Convert(sSOS); + kEOS=TD::Convert(sEOS); + kGOAL=TD::Convert("S")*-1; + _sent_id = (int *)malloc(sizeof(int)); + *_sent_id = -1; + if (DEBUG) cerr << "here = " << *_sent_id << endl; + _fwcount = (int *)malloc(sizeof(int)); + *_fwcount = -1; + cerr << "initializing dwarf" << endl; + flag_oris=false; flag_orit=false; flag_doms=false; flag_domt=false; flag_tfw_count=false; + flag_bdoms=false; flag_porislr=false, flag_porisrl=false, flag_goris=false; flag_pgorislr=false, flag_pgorisrl=false; + flag_pdomslr=false; flag_pdomsrl=false; flag_pgdomslr=false; flag_pgdomsrl=false; flag_gdoms=false; + flag_oris_backward=false; flag_orit_backward=false; + explicit_soseos=false; + SetStateSize(STATE_SIZE*sizeof(int)); + als = new Alignment(); + als->clearAls(Alignment::MAX_WORDS,Alignment::MAX_WORDS); + istringstream iss(param); string w; + while(iss >> w) { + int equal = w.find_first_of("="); + if (equal!=string::npos) { + string model = w.substr(0,equal); + vector<string> params; + Tokenize(w.substr(equal+1),',',¶ms); + string fn = params[0]; + if (model == "minfreq") { + cerr << "model minfreq " << fn << endl; + als->setFreqCutoff(atoi(fn.c_str())); + } else if (model == "oris") { + flag_oris = readOrientation(&toris,fn,&sfw); + if (flag_oris) { + oris_ = FD::Convert("OrientationSource"); + //oris_bo1_ = FD::Convert("OrientationSource_BO1"); + //oris_bo2_ = FD::Convert("OrientationSource_BO2"); + } + if (params.size()>1) als->setAlphaOris(atof(params[1].c_str())); + if (params.size()>2) als->setBetaOris(atof(params[2].c_str())); + } else if (model == "porislr") { + flag_porislr = readOrientation(&tporislr,fn,&sfw,true); + poris_nlr = 0; + if (flag_porislr) { + porislr_ = FD::Convert("OrientationSourcePositionfulLeftRight"); + } + if (params.size()>1) poris_nlr = atoi(params[1].c_str()); + if (DEBUG) cerr << " maximum poris depth=" << poris_nlr << endl; + } else if (model == "porisrl") { + flag_porisrl = readOrientation(&tporisrl,fn,&sfw,true); + poris_nrl = 0; + if (flag_porisrl) { + porisrl_ = FD::Convert("OrientationSourcePositionfulRightLeft"); + } + if (params.size()>1) poris_nrl = atoi(params[1].c_str()); + if (DEBUG) cerr << " maximum poris depth=" << poris_nrl << endl; + } else if (model=="goris") { + flag_goris = readOrientation(&tgoris,fn,&sfw); + if (flag_goris) { + goris_ = FD::Convert("OrientationSourceGeneralized"); + } + if (params.size()>1) { + readTags(params[1],&tags); + generalizeOrientation(&tgoris,tags); + } + } else if (model=="pgorislr") { + flag_pgorislr = readOrientation(&tpgorislr,fn,&sfw,true); + pgoris_nlr = 0; + if (flag_pgorislr) { + pgorislr_ = FD::Convert("OrientationSourceGeneralizedPositionfulLeftRight"); + } + if (DEBUG) { + cerr << "BEFORE GENERALIZATION" << endl; + tpgorislr.print(); + } + if (params.size()>1) pgoris_nlr = atoi(params[1].c_str()); + if (params.size()>2) { + readTags(params[2],&tags); + generalizeOrientation(&tpgorislr,tags,true); + } + if (DEBUG) { + cerr << "AFTER GENERALIZATION" << endl; + tpgorislr.print(); + } + } else if (model=="pgorisrl") { + flag_pgorisrl = readOrientation(&tpgorisrl,fn,&sfw,true); + pgoris_nrl = 0; + if (flag_pgorisrl) { + pgorisrl_ = FD::Convert("OrientationSourceGeneralizedPositionfulLeftRight"); + } + if (params.size()>1) pgoris_nrl = atoi(params[1].c_str()); + if (params.size()>2) { + readTags(params[2],&tags); + generalizeOrientation(&tpgorisrl,tags,true); + } + } else if (model == "oris_backward") { + flag_oris_backward = true; + if (!flag_oris) readOrientation(&toris,fn,&sfw); + oris_backward_ = FD::Convert("OrientationSourceBackward"); + if (params.size()>1) als->setAlphaOris(atof(params[1].c_str())); + if (params.size()>2) als->setBetaOris(atof(params[2].c_str())); + } else if (model == "orit") { + flag_orit = readOrientation(&torit,fn,&tfw); + if (flag_orit) { + orit_ = FD::Convert("OrientationTarget"); + //orit_bo1_ = FD::Convert("OrientationTarget_BO1"); + //orit_bo2_ = FD::Convert("OrientationTarget_BO2"); + } + if (params.size()>1) als->setAlphaOrit(atof(params[1].c_str())); + if (params.size()>2) als->setBetaOrit(atof(params[2].c_str())); + } else if (model == "orit_backward") { + flag_orit_backward = true; + if (!flag_orit) readOrientation(&torit,fn,&tfw); + orit_backward_ = FD::Convert("OrientationTargetBackward"); + if (params.size()>1) als->setAlphaOrit(atof(params[1].c_str())); + if (params.size()>2) als->setBetaOrit(atof(params[2].c_str())); + } else if (model == "doms") { + flag_doms = readDominance(&tdoms,fn,&sfw); + if (flag_doms) { + doms_ = FD::Convert("DominanceSource"); + //doms_bo1_ = FD::Convert("DominanceSource_BO1"); + //doms_bo2_ = FD::Convert("DominanceSource_BO2"); + } + if (params.size()>1) als->setAlphaDoms(atof(params[1].c_str())); + if (params.size()>2) als->setBetaDoms(atof(params[2].c_str())); + } else if (model == "pdomsrl") { + flag_pdomsrl = readDominance(&tpdomsrl,fn,&sfw,true); + if (flag_pdomsrl) { + pdomsrl_ = FD::Convert("DominanceSourcePositionfulRightLeft"); + } + if (params.size()>1) pdoms_nrl = atoi(params[1].c_str()); + } else if (model == "pdomslr") { + flag_pdomslr = readDominance(&tpdomslr,fn,&sfw,true); + tpdomslr.print(); + if (flag_pdomslr) { + pdomslr_ = FD::Convert("DominanceSourcePositionfulLeftRight"); + } + if (params.size()>1) pdoms_nlr = atoi(params[1].c_str()); + } else if (model == "pgdomsrl") { + flag_pgdomsrl = readDominance(&tpgdomsrl,fn,&sfw,true); + if (flag_pgdomsrl) { + pgdomsrl_ = FD::Convert("DominanceSourceGeneralizedPositionfulRightLeft"); + } + if (params.size()>1) pgdoms_nrl = atoi(params[1].c_str()); + if (params.size()>2) { + readTags(params[2],&tags); + generalizeDominance(&tpgdomsrl,tags,true); + } + } else if (model == "pgdomslr") { + flag_pgdomslr = readDominance(&tpgdomslr,fn,&sfw,true); + if (flag_pgdomslr) { + pgdomslr_ = FD::Convert("DominanceSourceGeneralizedPositionfulLeftRight"); + } + if (params.size()>1) pgdoms_nlr = atoi(params[1].c_str()); + if (params.size()>2) { + readTags(params[2],&tags); + if (DEBUG) { + for (map<WordID,WordID>::const_iterator it=tags.begin(); it!=tags.end(); it++) { + cerr << "tags = " << TD::Convert(it->first) << ", " << TD::Convert(it->second) << endl; + } + } + generalizeDominance(&tpgdomslr,tags,true); + } + if (DEBUG) tpgdomslr.print(); + } else if (model == "bdoms") { + flag_bdoms = readDominance(&tbdoms,fn,&sfw); + if (flag_bdoms) { + bdoms_ = FD::Convert("BorderDominanceSource"); + } + } else if (model == "domt") { + flag_domt = readDominance(&tdomt,fn,&tfw); + if (flag_domt) { + domt_ = FD::Convert("DominanceTarget"); + //domt_bo1_ = FD::Convert("DominanceTarget_BO1"); + //domt_bo2_ = FD::Convert("DominanceTarget_BO2"); + } + if (params.size()>1) als->setAlphaDomt(atof(params[1].c_str())); + if (params.size()>2) als->setBetaDomt(atof(params[2].c_str())); + } else if (model== "tfw_count") { + flag_tfw_count = readList(fn,&tfw); + tfw_count_ = FD::Convert("TargetFunctionWordsCount"); + } else { + cerr << "DWARF doesn't understand this model: " << model << endl; + } + } else { + if (w=="tfw_count") { + flag_tfw_count = true; + tfw_count_ = FD::Convert("TargetFunctionWordsCount"); + } else if (w=="oris_backward") { + flag_oris_backward = true; + oris_backward_ = FD::Convert("OrientationSourceBackward"); + } else if (w=="orit_backward") { + flag_orit_backward = true; + orit_backward_ = FD::Convert("OrientationTargetBackward"); + } else if (w=="explicit_soseos") { + explicit_soseos=true; + } else { + cerr << "DWARF doesn't need this param: " << param << endl; + } + } + } + for (map<WordID,int>::const_iterator it=sfw.begin(); it!=sfw.end() && DEBUG; it++) { + cerr << " FW:" << TD::Convert(it->first) << endl; + } +} + +void Dwarf::TraversalFeaturesImpl(const SentenceMetadata& smeta, + const Hypergraph::Edge& edge, + const std::vector<const void*>& ant_contexts, + SparseVector<double>* features, + SparseVector<double>* estimated_features, + void* context) const { + if (DEBUG) cerr << "TraversalFeaturesImpl" << endl; + double cost, bonus, bo1, bo2, bo1_bonus, bo2_bonus; + double bdoms_state_mono= 0; double bdoms_state_nonmono = 0; + TRule r = *edge.rule_; + if (DEBUG) cerr << " sent_id=" << *_sent_id << ", " << smeta.GetSentenceID() << endl; + if (DEBUG) cerr << "rule = " << r.AsString() << endl; + if (DEBUG) cerr << "rule[i,j] = " << edge.i_ << "," << edge.j_ << endl; + if (*_sent_id != smeta.GetSentenceID()) { //new sentence + *_sent_id = smeta.GetSentenceID(); + const Lattice l = smeta.GetSourceLattice(); + *_fwcount=0; + for (int i=0; i<smeta.GetSourceLength(); i++) { + if (sfw.find(l[i][0].label)!=sfw.end()) { + *_fwcount+=1; + } + } + if (DEBUG) cerr << "new sentence[" << *_sent_id << "]="<<*_fwcount<<endl; + } + bool nofw = als->prepare(*edge.rule_, ant_contexts, sfw, tfw,smeta.GetSourceLattice(),edge.i_,edge.j_); + bool isFinal = (edge.i_==0 && edge.j_==smeta.GetSourceLength() && r.GetLHS()==kGOAL); + // prepare *nofw* outputs whether the resulting alignment, contains function words or not + // if not, the models do not have to be calcualted and *simplify* is very simple + if (DEBUG) cerr << "nofw = " << nofw << endl; + if (flag_tfw_count) { + double count = 0; + for (int i=0; i<r.e_.size(); i++) { + if (tfw.find(r.e_[i])!=tfw.end()) count++; + } + features->set_value(tfw_count_,count); + } + if (flag_oris) { + cost=0; bonus=0; bo1=0; bo2=0; bo1_bonus=0; bo2_bonus=0; + if (!nofw) als->computeOrientationSource(toris,&cost,&bonus,&bo1,&bo1_bonus,&bo2,&bo2_bonus); + if (isFinal&&!explicit_soseos) { + cost += bonus; + bonus = 0; + } + features->set_value(oris_,cost); + //features->set_value(oris_bo1_,bo1); + //features->set_value(oris_bo2_,bo2); + estimated_features->set_value(oris_,bonus); + //estimated_features->set_value(oris_bo1_,bo1_bonus); + //estimated_features->set_value(oris_bo2_,bo2_bonus); + } + if (flag_porislr) { + cost=0; bonus=0; bo1=0; bo2=0; bo1_bonus=0; bo2_bonus=0; + if (!nofw) + als->computeOrientationSourcePos(tporislr,&cost,&bonus,&bo1,&bo1_bonus,&bo2,&bo2_bonus,*_fwcount,poris_nlr,0); + if (isFinal&&!explicit_soseos) { + cost += bonus; + bonus = 0; + } + features->set_value(porislr_,cost); + estimated_features->set_value(porislr_,bonus); + } + if (flag_porisrl) { + cost=0; bonus=0; bo1=0; bo2=0; bo1_bonus=0; bo2_bonus=0; + if (!nofw) + als->computeOrientationSourcePos(tporisrl,&cost,&bonus,&bo1,&bo1_bonus,&bo2,&bo2_bonus,*_fwcount,0,poris_nrl); + if (isFinal&&!explicit_soseos) { + cost += bonus; + bonus = 0; + } + features->set_value(porisrl_,cost); + estimated_features->set_value(porisrl_,bonus); + } + if (flag_pgorislr) { + cost=0; bonus=0; bo1=0; bo2=0; bo1_bonus=0; bo2_bonus=0; + if (!nofw) + als->computeOrientationSourcePos(tpgorislr,&cost,&bonus,&bo1,&bo1_bonus,&bo2,&bo2_bonus,*_fwcount,pgoris_nlr,0); + if (isFinal&&!explicit_soseos) { + cost += bonus; + bonus = 0; + } + features->set_value(pgorislr_,cost); + estimated_features->set_value(pgorislr_,bonus); + } + if (flag_pgorisrl) { + cost=0; bonus=0; bo1=0; bo2=0; bo1_bonus=0; bo2_bonus=0; + if (!nofw) + als->computeOrientationSourcePos(tpgorisrl,&cost,&bonus,&bo1,&bo1_bonus,&bo2,&bo2_bonus,*_fwcount,0,pgoris_nrl); + if (isFinal&&!explicit_soseos) { + cost += bonus; + bonus = 0; + } + features->set_value(pgorisrl_,cost); + estimated_features->set_value(pgorisrl_,bonus); + } + if (flag_goris) { + cost=0; bonus=0; + if (!nofw) als->computeOrientationSource(tgoris,&cost,&bonus,&bo1,&bo1_bonus,&bo2,&bo2_bonus); + if (isFinal&&!explicit_soseos) { + cost += bonus; + bonus = 0; + } + features->set_value(goris_,cost); + estimated_features->set_value(goris_,bonus); + } + if (flag_oris_backward) { + cost=0; bonus=0; + if (!nofw) + als->computeOrientationSourceBackward(toris,&cost,&bonus,&bo1,&bo1_bonus,&bo2,&bo2_bonus); + if (isFinal&&!explicit_soseos) { + cost += bonus; + bonus = 0; + } + features->set_value(oris_backward_,cost); + estimated_features->set_value(oris_backward_,bonus); + } + WordID _lfw = kSOS; + WordID _rfw = kEOS; + if (flag_doms || flag_pdomslr || flag_pdomsrl || flag_pgdomslr || flag_pgdomsrl) { + if (DEBUG) cerr << " seeking lfw and rfw" << endl; + int start = edge.i_; + int end = edge.j_; + if (DEBUG) cerr << " start=" << start << ", end=" << end << endl; + const Lattice l = smeta.GetSourceLattice(); + for (int idx=start-1; idx>=0; idx--) { + if (DEBUG) cerr << " checking idx=" << idx << ", label=" << l[idx][0].label << "-" << TD::Convert(l[idx][0].label) << endl; + if (sfw.find(l[idx][0].label) !=sfw.end()) { + if (DEBUG) cerr << "+"; + _lfw=l[idx][0].label; break; + } + } + for (int idx=end; idx<l.size(); idx++) { // end or end+1 + if (DEBUG) cerr << " checking idx=" << idx << ", label=" << l[idx][0].label << "-" << TD::Convert(l[idx][0].label) << endl; + if (sfw.find(l[idx][0].label)!=sfw.end()) { + if (DEBUG) cerr << "."; + _rfw=l[idx][0].label; break; + } + } + if (isFinal&&!explicit_soseos) { + _lfw=kSOS; _rfw=kEOS; + } + } + if (flag_doms) { + cost=0; bonus=0; bo1=0; bo2=0; bo1_bonus=0; bo2_bonus=0; + if (!nofw) als->computeDominanceSource(tdoms,_lfw,_rfw,&cost,&bonus, + &bo1,&bo1_bonus,&bo2,&bo2_bonus); + if (DEBUG) cerr << " COST=" << cost << ", BONUS=" << bonus << endl; + if (isFinal&&!explicit_soseos) { + cost += bonus; + if (DEBUG) cerr << " final and !explicit_soseos, thus cost = " << cost << endl; + bonus = 0; + } + features->set_value(doms_,cost); + estimated_features->set_value(doms_,bonus); + } + if (flag_pdomslr) { + if (DEBUG) cerr << " flag_pdomslr true, nofw=" << nofw << endl; + if (DEBUG) cerr << " lfw=" << _lfw << ", rfw=" << _rfw << endl; + if (DEBUG) cerr << " kSOS=" << kSOS << ", kEOS=" << kEOS << endl; + cost=0; bonus=0; bo1=0; bo2=0; bo1_bonus=0; bo2_bonus=0; + if (!nofw) als->computeDominanceSourcePos(tpdomslr,_lfw,_rfw,&cost,&bonus, + &bo1,&bo1_bonus,&bo2,&bo2_bonus,*_fwcount,pdoms_nlr,0); + if (isFinal&&!explicit_soseos) { + cost += bonus; + bonus = 0; + } + features->set_value(pdomslr_,cost); + estimated_features->set_value(pdomslr_,bonus); + } + if (flag_pdomsrl) { + cost=0; bonus=0; bo1=0; bo2=0; bo1_bonus=0; bo2_bonus=0; + if (!nofw) als->computeDominanceSourcePos(tpdomsrl,_lfw,_rfw,&cost,&bonus, + &bo1,&bo1_bonus,&bo2,&bo2_bonus,*_fwcount,0,pdoms_nrl); + if (isFinal&&!explicit_soseos) { + cost += bonus; + bonus = 0; + } + features->set_value(pdomsrl_,cost); + estimated_features->set_value(pdomsrl_,bonus); + } + if (flag_pgdomslr) { + cost=0; bonus=0; bo1=0; bo2=0; bo1_bonus=0; bo2_bonus=0; + if (!nofw) als->computeDominanceSourcePos(tpgdomslr,_lfw,_rfw,&cost,&bonus, + &bo1,&bo1_bonus,&bo2,&bo2_bonus,*_fwcount,pgdoms_nlr,0); + if (isFinal&&!explicit_soseos) { + cost += bonus; + bonus = 0; + } + features->set_value(pgdomslr_,cost); + estimated_features->set_value(pgdomslr_,bonus); + } + if (flag_pgdomsrl) { cost=0; bonus=0; bo1=0; bo2=0; bo1_bonus=0; bo2_bonus=0; + if (!nofw) als->computeDominanceSourcePos(tpgdomsrl,_lfw,_rfw,&cost,&bonus, + &bo1,&bo1_bonus,&bo2,&bo2_bonus,*_fwcount,0,pgdoms_nrl); + if (isFinal&&!explicit_soseos) { + cost += bonus; + bonus = 0; + } + features->set_value(pgdomsrl_,cost); + estimated_features->set_value(pgdomsrl_,bonus); + } + + + if (flag_bdoms) { + cost=0; bonus=0; bdoms_state_mono=0; bdoms_state_nonmono=0; + if (!nofw) + als->computeBorderDominanceSource(tbdoms,&cost,&bonus, + &bdoms_state_mono, &bdoms_state_nonmono,*edge.rule_, ant_contexts, sfw); + features->set_value(bdoms_,cost); + estimated_features->set_value(bdoms_,bonus); + } + if (flag_orit) { + cost=0; bonus=0; bo1=0; bo2=0; bo1_bonus=0; bo2_bonus=0; + if (!nofw) als->computeOrientationTarget(torit,&cost,&bonus,&bo1,&bo1_bonus,&bo2,&bo2_bonus); + if (DEBUG) cerr << "cost=" << cost << ", bonus=" << bonus << ", bo1=" << bo1 << ", bo1_bonus=" << bo1_bonus << ", bo2=" << bo2 << ", bo2_bonus=" << bo2_bonus << endl; + features->set_value(orit_,cost); + //features->set_value(orit_bo1_,bo1); + //features->set_value(orit_bo2_,bo2); + estimated_features->set_value(orit_,bonus); + //estimated_features->set_value(orit_bo1_,bo1_bonus); + //estimated_features->set_value(orit_bo2_,bo2_bonus); + } + if (flag_orit_backward) { + cost=0; bonus=0; + if (!nofw) als->computeOrientationTargetBackward(torit,&cost,&bonus,&bo1,&bo1_bonus,&bo2,&bo2_bonus); + features->set_value(orit_backward_,cost); + estimated_features->set_value(orit_backward_,bonus); + } + if (flag_domt) { + cost=0; bonus=0; bo1=0; bo2=0; bo1_bonus=0; bo2_bonus=0; + WordID _lfw=-1; int start = edge.i_; + WordID _rfw=-1; int end = edge.j_; + if (smeta.HasReference()) { + const Lattice l = smeta.GetReference(); + for (int idx=start-1; idx>=0; idx--) { + if (l.size()>0) + if (tfw.find(l[idx][0].label) !=tfw.end()) { + _lfw=l[idx][0].label; break; + } + } + for (int idx=end; idx<l.size(); idx++) { // end or end+1 + if (l[idx].size()>0) + if (tfw.find(l[idx][0].label)!=tfw.end()) { + _rfw=l[idx][0].label; break; + } + } + } + //neighboringFWs(smeta.GetReference(),edge.i_,edge.j_,tfw,&_lfw,&_rfw); + if (!nofw) als->computeDominanceTarget(tdomt,_lfw,_rfw,&cost,&bonus, + &bo1,&bo1_bonus,&bo2,&bo2_bonus); + features->set_value(domt_,cost); + //features->set_value(domt_bo1_,bo1); + //features->set_value(domt_bo2_,bo2); + estimated_features->set_value(domt_,bonus); + //estimated_features->set_value(domt_bo1_,bo1_bonus); + //estimated_features->set_value(domt_bo2_,bo2_bonus); + } + int* vcontext = reinterpret_cast<int *>(context); + if (!nofw) { + als->BorderingSFWsOnly(); + als->BorderingTFWsOnly(); + als->simplify(vcontext); + } else { + als->simplify_nofw(vcontext); + } + vcontext[50] = DoubleToInteger(bdoms_state_mono); + vcontext[51] = DoubleToInteger(bdoms_state_nonmono); + vcontext[STATE_SIZE-1] = Alignment::link(edge.i_,edge.j_); + if (DEBUG) { + cerr << "state@traverse = "; + for (int idx=0; idx<STATE_SIZE; idx++) cerr << idx << "." << vcontext[idx] << " "; + cerr << endl; + cerr << "bdoms_state_mono=" << bdoms_state_mono << ", state[50]=" << IntegerToDouble(vcontext[50]) << endl; + cerr << "bdoms_state_nonmono=" << bdoms_state_nonmono << ", state[51]=" << IntegerToDouble(vcontext[51]) << endl; + } +} + +int Dwarf::DoubleToInteger(double val) { + float x = (float)val; + float* px = &x; + int* pix = reinterpret_cast<int *>(px); + return *pix; +} + +double Dwarf::IntegerToDouble(int val) { + int *py = &val; + float* pd = reinterpret_cast<float *>(py); + return (double)*pd; +} + +void Dwarf::neighboringFWs(const Lattice& l, const int& i, const int& j, const map<WordID,int>& fw_hash, int* lfw, int* rfw) { + *lfw=0; *rfw=0; + int idx=i-l[i][0].dist2next; + while (idx>=0) { + if (l[idx].size()>0) { + if (fw_hash.find(l[idx][0].label)!=fw_hash.end()) { + *lfw++; + } + } + idx-=l[idx][0].dist2next; + } + idx=j+l[j][0].dist2next; + while (idx<l.size()) { + if (l[idx].size()>0) { + if (fw_hash.find(l[idx][0].label)!=fw_hash.end()) { + *rfw++; + } + } + idx+=l[idx][0].dist2next; + } +} + +bool Dwarf::readOrientation(CountTable* table, const std::string& filename, std::map<WordID,int> *fw, bool pos) { + // the input format is + // source target 0 1 2 3 4 0 1 2 3 4 + // 0 -> MA, 1 -> RA, 2 -> MG, 3 -> RG, 4 -> NO_NEIGHBOR + // first 01234 corresponds to the left neighbor, the second 01234 corresponds to the right neighbor + // append 2 more at the end as precomputed total + + // TONS of hack here. CountTable should be wrapped as a class + // TODO: check whether the file exists or not, return false if not + if (DEBUG) cerr << " readOrientation(" << filename << ", pos=" << pos << ")" << endl; + ReadFile rf(filename); + istream& in = *rf.stream(); + table->setup(24,pos); + table->ultimate = new int[24]; + for (int i=0; i<24; i++) table->ultimate[i]=0; + ostringstream oss; + while (in) { + string line; + getline(in,line); + if (line=="") break; + istringstream tokenizer(line); + string sourceidx, source, target, word; + tokenizer >> source >> target; + if (pos) { + sourceidx = source; + source = sourceidx.substr(0,sourceidx.find_last_of("/")); + } + if (fw->find(TD::Convert(source))==fw->end()) fw->insert(pair<WordID,int>(TD::Convert(source),1)); + + + int* element = new int[24]; + element[5] = 0; + for (int i=0; i<5; i++) { + element[i] = 0; + if (tokenizer >> word) element[i] = atoi(word.c_str()); + element[5] += element[i]; + } + element[11] = 0; + for (int i=6; i<11; i++) { + element[i] = 0; + if (tokenizer >> word) element[i] = atoi(word.c_str()); + element[11] += element[i]; + } + element[17] = 0; + for (int i=12; i<17; i++) { + element[i] = 0; + if (tokenizer >> word) element[i] = atoi(word.c_str()); + element[17] += element[i]; + } + element[23] = 0; + for (int i=18; i<23; i++) { + element[i] = 0; + if (tokenizer >> word) element[i] = atoi(word.c_str()); + element[23] += element[i]; + } + for (int i=0; i<24; i++) table->ultimate[i] += element[i]; + oss << source << " " << target; + WordID key_id = TD::Convert(oss.str()); + oss.str(""); + if (table->model.find(key_id)!=table->model.end()) { + for (int i=0; i<24; i++) table->model[key_id][i]+=element[i]; + } else { + int* el2 = new int[24]; + for (int i=0; i<24; i++) el2[i] = element[i]; + table->model.insert(pair<WordID,int*>(key_id,el2)); + } + + oss << source; + key_id = TD::Convert(oss.str()); + oss.str(""); + if (table->model.find(key_id)!=table->model.end()) { + for (int i=0; i<24; i++) table->model[key_id][i]+=element[i]; + } else { + int* el2 = new int[24]; + for (int i=0; i<24; i++) el2[i] = element[i]; + table->model.insert(pair<WordID,int*>(key_id,el2)); + } + + if (pos) { + oss << sourceidx << " " << target; + key_id = TD::Convert(oss.str()); + oss.str(""); + if (table->model.find(key_id)!=table->model.end()) { + for (int i=0; i<24; i++) table->model[key_id][i]+=element[i]; + } else { + int* el2 = new int[24]; + for (int i=0; i<24; i++) el2[i] = element[i]; + table->model.insert(pair<WordID,int*>(key_id,el2)); + } + } + delete[] element; + } + return true; +} + +bool Dwarf::readList(const std::string& filename, std::map<WordID,int>* fw) { + ReadFile rf(filename); + istream& in = *rf.stream(); + while (in) { + string word; + getline(in,word); + if (fw->find(TD::Convert(word))==fw->end()) fw->insert(pair<WordID,int>(TD::Convert(word),1)); + } + return true; +} + +bool Dwarf::readDominance(CountTable* table, const std::string& filename, std::map<WordID,int>* fw, bool pos) { + // the input format is + // source1 source2 target1 target2 0 1 2 3 + // 0 -> dontcase 1->leftfirst 2->rightfirst 3->neither + if (DEBUG) cerr << "readDominance(" << filename << ",pos="<< pos << ")" << endl; + ReadFile rf(filename); + istream& in = *rf.stream(); + table->ultimate = new int[5]; + table->setup(5,pos); + for (int i=0; i<5; i++) table->ultimate[i]=0; + while (in) { + string line, word; + getline(in,line); + if (line=="") break; + string source1idx, source2idx, target1, target2, source1, source2; + ostringstream oss; + WordID key_id; + istringstream tokenizer(line); + tokenizer >> source1 >> source2 >> target1 >> target2; + if (pos) { + source1idx = source1; + source2idx = source2; + source1 = source1idx.substr(0,source1idx.find_last_of("/")); + source2 = source2idx.substr(0,source2idx.find_last_of("/")); + } + if (fw->find(TD::Convert(source1))==fw->end()) fw->insert(pair<WordID,int>(TD::Convert(source1),1)); + if (fw->find(TD::Convert(source2))==fw->end()) fw->insert(pair<WordID,int>(TD::Convert(source2),1)); + + int* element = new int[5]; + element[4]=0; + for (int i=0; i<4; i++) { + element[i] = 0; + if (tokenizer >> word) element[i] = atoi(word.c_str()); + element[4]+=element[i]; + } + for (int i=0; i<5; i++) table->ultimate[i] += element[i]; + + oss << source1 << " " << source2 << " " << target1 << " " << target2; + key_id = TD::Convert(oss.str()); + oss.str(""); + if (table->model.find(key_id)!=table->model.end()) { + for (int i=0; i<5; i++) table->model[key_id][i]+=element[i]; + } else { + int* el2 = new int[5]; + for (int i=0; i<5; i++) el2[i]=element[i]; + table->model.insert(pair<WordID,int*>(key_id,el2)); + } + + oss << source1 << " " << source2; + key_id = TD::Convert(oss.str()); + oss.str(""); + if (table->model.find(key_id)!=table->model.end()) { + for (int i=0; i<5; i++) table->model[key_id][i]+=element[i]; + } else { + int* el2 = new int[5]; + for (int i=0; i<5; i++) el2[i]=element[i]; + table->model.insert(pair<WordID,int*>(key_id,el2)); + } + + if (pos) { + oss << source1idx << " " << source2idx << " " << target1 << " " << target2; + key_id = TD::Convert(oss.str()); + oss.str(""); + if (table->model.find(key_id)!=table->model.end()) { + for (int i=0; i<5; i++) table->model[key_id][i]+=element[i]; + } else { + int* el2 = new int[5]; + for (int i=0; i<5; i++) el2[i]=element[i]; + table->model.insert(pair<WordID,int*>(key_id,el2)); + } + } + delete element; + } + + return true; +} + +bool Dwarf::readTags(const std::string& filename, std::map<WordID,WordID>* tags) { + ReadFile rf(filename); + istream& in = *rf.stream(); + while(in) { + string line, word, tag; + getline(in,line); + if (line=="") break; + istringstream tokenizer(line); + tokenizer >> tag >> word; + tags->insert(pair<WordID,WordID>(TD::Convert(word),TD::Convert(tag))); + } + return true; +} + +bool Dwarf::generalizeOrientation(CountTable* table, const std::map<WordID,WordID>& tags, bool pos) { + map<string,int*> generalized; + for (map<WordID,int*>::iterator it=table->model.begin(); it!=table->model.end(); it++) { + string source, target; + istringstream tokenizer(TD::Convert(it->first)); + tokenizer >> source >> target; + string idx = ""; + if (pos) { + int found = source.find_last_of("/"); + if (found!=string::npos && found>0) { + idx = source.substr(found+1); + source = source.substr(0,found); + } + } + map<WordID,WordID>::const_iterator tags_iter = tags.find(TD::Convert(source)); + if (tags_iter!=tags.end()) { + ostringstream genkey; + genkey << TD::Convert(tags_iter->second); + if (idx!="") genkey << "/" << idx; + if (target!="") genkey << " " << target; + int* model; + if (generalized.find(genkey.str())!=generalized.end()) { + model = generalized[genkey.str()]; + for (int i=0; i<24; i++) model[i] += it->second[i]; + } else { + int* el = new int[24]; + for (int i=0; i<24; i++) el[i] = it->second[i]; + generalized.insert(pair<string,int*>(genkey.str(),el)); + } + } + } + for (map<WordID,int*>::iterator it=table->model.begin(); it!=table->model.end(); it++) { + string source, target; + istringstream tokenizer(TD::Convert(it->first)); + tokenizer >> source >> target; + string idx = ""; + if (pos) { + int found = source.find_last_of("/"); + if (found!=string::npos && found>0) { + idx = source.substr(found+1); + source = source.substr(0,found); + } + } + map<WordID,WordID>::const_iterator tags_iter = tags.find(TD::Convert(source)); + if (tags_iter!=tags.end()) { + ostringstream genkey; + genkey << TD::Convert(tags_iter->second); + if (idx!="") genkey << "/" << idx; + if (target!="") genkey << " " << target; + if (generalized.find(genkey.str())!=generalized.end()) { + delete it->second; + it->second = generalized[genkey.str()]; + } + } + } + +} + + + +bool Dwarf::generalizeDominance(CountTable* table, const std::map<WordID,WordID>& tags, bool pos) { + map<string,int*> generalized; + ostringstream oss; + for (map<WordID,int*>::iterator it=table->model.begin(); it!=table->model.end(); it++) { + string source1, source2, target1, target2; + string idx1 = ""; string idx2 = ""; + istringstream tokenizer(TD::Convert(it->first)); + tokenizer >> source1 >> source2 >> target1 >> target2; + if (DEBUG) cerr << "source1=|" << source1 << "|, source2=|" << source2 << "|, target1=|" << target1 << "|, target2=|" << target2 << "|" << endl; + if (pos) { + int found1 = source1.find_last_of("/"); + int found2 = source2.find_last_of("/"); + if (found1!=string::npos && found2!=string::npos && found1>0 && found2>0) { + idx1 = source1.substr(found1+1); + source1 = source1.substr(0,found1); + idx2 = source2.substr(found2+1); + source2 = source2.substr(0,found2); + } + } + if (DEBUG) + cerr << "[U]source1='" << source1 << "', idx1='"<< idx1 << "', source2='" << source2 << "', idx2='"<< idx2 << "', target1='" << target1 << "', target2='" << target2 << "'" << endl; + map<WordID,WordID>::const_iterator tags_iter1 = tags.find(TD::Convert(source1)); + map<WordID,WordID>::const_iterator tags_iter2 = tags.find(TD::Convert(source2)); + if (tags_iter1!=tags.end()) + source1 = TD::Convert(tags_iter1->second); + oss << source1; + if (idx1!="") oss << "/" << idx1; + if (tags_iter2!=tags.end()) + source2 = TD::Convert(tags_iter2->second); + oss << " " << source2; + if (idx2!="") oss << "/" << idx2; + if (target1!="" && target2!="") oss << " " << target1 << " " << target2; + + if (DEBUG) cerr << "generalized key = '" << oss.str() << "'" << endl; + if (generalized.find(oss.str())!=generalized.end()) { + int* model = generalized[oss.str()]; + for (int i=0; i<5; i++) model[i] += it->second[i]; + } else { + int* model = new int[5]; + for (int i=0; i<5; i++) model[i] = it->second[i]; + generalized.insert(pair<string,int*>(oss.str(),model)); + } + oss.str(""); + } + + if (DEBUG) { + for (map<string,int*>::const_iterator it=generalized.begin(); it!=generalized.end(); it++) { + cerr << "GENERALIZED = " << it->first << ", "; + for (int i=0; i<5; i++) cerr << it->second[i] << " "; + cerr << endl; + } + } + + for (map<WordID,int*>::iterator it=table->model.begin(); it!=table->model.end(); it++) { + string source1, source2, target1, target2; + string idx1 = ""; string idx2 = ""; + istringstream tokenizer(TD::Convert(it->first)); + tokenizer >> source1 >> source2 >> target1 >> target2; + if (pos) { + int found1 = source1.find_last_of("/"); + int found2 = source2.find_last_of("/"); + if (found1!=string::npos && found2!=string::npos && found1>0 && found2>0) { + idx1 = source1.substr(found1+1); + source1 = source1.substr(0,found1); + idx2 = source2.substr(found2+1); + source2 = source2.substr(0,found2); + } + } + map<WordID,WordID>::const_iterator tags_iter1 = tags.find(TD::Convert(source1)); + map<WordID,WordID>::const_iterator tags_iter2 = tags.find(TD::Convert(source2)); + if (tags_iter1!=tags.end()) + source1 = TD::Convert(tags_iter1->second); + oss << source1; + if (idx1!="") oss << "/" << idx1; + if (tags_iter2!=tags.end()) + source2 = TD::Convert(tags_iter2->second); + oss << " " << source2; + if (idx2!="") oss << "/" << idx2; + if (target1!="" && target2!="") oss << " " << target1 << " " << target2; + + if (generalized.find(oss.str())!=generalized.end()) { + if (DEBUG) cerr << " generalizing "<< TD::Convert(it->first) << " into " << oss.str() << endl; + if (DEBUG) { + cerr << " model from "; + for (int i=0; i<5; i++) cerr << it->second[i] << " "; + cerr << endl; + } + delete it->second; + it->second = generalized[oss.str()]; + if (DEBUG) { + cerr << " into "; + for (int i=0; i<5; i++) cerr << it->second[i] << " "; + cerr << endl; + } + } + oss.str(""); + } + +} diff --git a/decoder/ff_dwarf.h b/decoder/ff_dwarf.h new file mode 100644 index 00000000..083fcc7c --- /dev/null +++ b/decoder/ff_dwarf.h @@ -0,0 +1,100 @@ +#include <vector> +#include <map> +#include <string> +#include "ff.h" +#include "dwarf.h" +#include "lattice.h" + +using namespace std; + +class Dwarf : public FeatureFunction { + public: + Dwarf(const std::string& param); + /* State-related param + STATE_SIZE: the number of ints + MAXIMUM_ALIGNMENTS: the maximum number of alignments in the states, + each alignment point is encoded in one int + (the first two bytes for source, and the remaining one for target) + */ + static const int STATE_SIZE=53; + static const int IMPOSSIBLY_LARGE_POS = 9999999; + static const int MAXIMUM_ALIGNMENTS=37; + /* Read from file the Orientation(Source|Target model parameter. */ + static bool readOrientation(CountTable* table, const std::string& filename, std::map<WordID,int> *fw, bool pos=false); + /* Read from file the Dominance(Source|Target) model parameter. */ + static bool readDominance(CountTable* table, const std::string& filename, std::map<WordID,int> *fw, bool pos=false); + static bool readList(const std::string& filename, std::map<WordID,int>* fw); + static double IntegerToDouble(int val); + static int DoubleToInteger(double val); + bool readTags(const std::string& filename, std::map<WordID,WordID>* tags); + bool generalizeOrientation(CountTable* table, const std::map<WordID,WordID>& tags, bool pos=false); + bool generalizeDominance(CountTable* table, const std::map<WordID,WordID>& tags, bool pos=false); + static void stripIndex(const string& source, string* pkey, string* pidx) { + if (DEBUG) cerr << " stripIndex(" << source << ")" << endl; + int found = source.find_last_of("/"); + string idx = source.substr(found+1); + string key = source.substr(0,found); + if (DEBUG) cerr << " found=" << found << "," << key << "," << idx << endl; + pkey = &key; + pidx = &idx; + } + + + protected: + /* The high-level workflow is as follow: + 1. call *als->prepare*, which constructs the full alignment of the edge while taking into account the antecedents + also in this call, function words are identified. Most of the work in this call is to make sure the indexes + of the alignments (including the function words) are consistent with the newly created alignment + 2. call *als->computeOrientationSource*, *als->computeOrientationTarget*, + *als->computeDominanceSource*, or *als->computeDominanceTarget* + and pass the resulting score to either *features* or to *estimated_features* + 3. call *als->BorderingSFWsOnly()* and *als->BorderingTFWsOnly()*, which removes records of all function word + alignments except those at the borders. Note that fw alignments kept may be more than two on each side + for examples if there are a number of unaligned fw alignments before the leftmost alignment or the rightmost one + 4. call *als->simplify()*, which assigns the state of this edge (*context*). It simplifies the alignment space to + its most compact representation, enough to compute the unscored models. This is done by observing the surviving + function word alignments set by 3. + */ + void TraversalFeaturesImpl(const SentenceMetadata& smeta, + const Hypergraph::Edge& edge, + const std::vector<const void*>& ant_contexts, + SparseVector<double>* features, + SparseVector<double>* estimated_features, + void* context) const; + private: + Alignment* als; + /* Feature IDs set by calling FD::Convert(model's string) */ + int oris_, oris_bo1_, oris_bo2_, orit_, orit_bo1_, orit_bo2_; + int oris_backward_, orit_backward_, porislr_, porisrl_, goris_, pgorislr_, pgorisrl_; + int pdomslr_, pdomsrl_, pgdomslr_, pgdomsrl_; + int doms_, doms_bo1_, doms_bo2_, domt_, domt_bo1_, domt_bo2_; + int tfw_count_; + int bdoms_; + int poris_count; + int pgoris_count; + int poris_nlr, poris_nrl; // maximum depth (1->from the beginning of the sentence, 2-> from the end of the sentence) + int pgoris_nlr, pgoris_nrl; + int pdoms_nlr, pdoms_nrl; + int pgdoms_nlr, pgdoms_nrl; + int* _sent_id; + int* _fwcount; + WordID kSOS; + WordID kEOS; + string sSOS; + string sEOS; + WordID kGOAL; + /* model's flag, if set true will invoke the model scoring */ + bool flag_oris, flag_orit, flag_doms, flag_domt, flag_tfw_count, flag_oris_backward, flag_orit_backward, flag_bdoms; + bool flag_porislr, flag_porisrl, flag_goris, flag_pgorislr, flag_pgorisrl; + bool explicit_soseos; + bool flag_pdomslr, flag_pdomsrl, flag_pgdomslr, flag_pgdomsrl, flag_gdoms; + /* a collection of Source function words (sfw) and Target function words (tfw) */ + std::map<WordID,int> sfw; + std::map<WordID,int> tfw; + std::map<WordID,WordID> tags; + /* a collection of model's parameter */ + CountTable toris, torit, tdoms, tbdoms, tdomt, tporislr, tporisrl, tgoris, tpgorislr, tpgorisrl; + CountTable tpdomslr, tpdomsrl, tpgdomslr, tpgdomsrl; + void neighboringFWs(const Lattice& l, const int& i, const int& j, const map<WordID,int>& fw_hash, int* lfw, int* rfw); +}; + diff --git a/decoder/grammar.cc b/decoder/grammar.cc index 7e6bbc66..9e4065a6 100644 --- a/decoder/grammar.cc +++ b/decoder/grammar.cc @@ -157,6 +157,7 @@ PassThroughGrammar::PassThroughGrammar(const Lattice& input, const string& cat, has_rule_[i].insert(j); const string& src = TD::Convert(alts[k].label); TRulePtr pt(new TRule("[" + cat + "] ||| " + src + " ||| " + src + " ||| PassThrough=1")); + pt->a_.push_back(AlignmentPoint(0,0)); AddRule(pt); RefineRule(pt, ctf_level); } diff --git a/decoder/rule_lexer.l b/decoder/rule_lexer.l index 2e5d3bf5..383dd336 100644 --- a/decoder/rule_lexer.l +++ b/decoder/rule_lexer.l @@ -46,11 +46,15 @@ int scfglex_num_feats; #define MAX_ARITY 20 int scfglex_nt_sanity[MAX_ARITY]; int scfglex_src_nts[MAX_ARITY]; -float scfglex_nt_size_means[MAX_ARITY]; -float scfglex_nt_size_vars[MAX_ARITY]; +// float scfglex_nt_size_means[MAX_ARITY]; +// float scfglex_nt_size_vars[MAX_ARITY]; std::stack<TRulePtr> ctf_rule_stack; unsigned int ctf_level = 0; +#define MAX_ALS 50 +AlignmentPoint scfglex_als[MAX_ALS]; +int scfglex_num_als; + void sanity_check_trg_symbol(WordID nt, int index) { if (scfglex_src_nts[index-1] != nt) { std::cerr << "Target symbol with index " << index << " is of type " << TD::Convert(nt*-1) @@ -79,6 +83,7 @@ void scfglex_reset() { scfglex_num_feats = 0; scfglex_src_rhs_size = 0; scfglex_trg_rhs_size = 0; + scfglex_num_als = 0; } void check_and_update_ctf_stack(const TRulePtr& rp) { @@ -215,7 +220,7 @@ NT [^\t \[\],]+ std::cerr << "Line " << lex_line << ": LHS and RHS arity mismatch!\n"; abort(); } - TRulePtr rp(new TRule(scfglex_lhs, scfglex_src_rhs, scfglex_src_rhs_size, scfglex_trg_rhs, scfglex_trg_rhs_size, scfglex_feat_ids, scfglex_feat_vals, scfglex_num_feats, scfglex_src_arity)); + TRulePtr rp(new TRule(scfglex_lhs, scfglex_src_rhs, scfglex_src_rhs_size, scfglex_trg_rhs, scfglex_trg_rhs_size, scfglex_feat_ids, scfglex_feat_vals, scfglex_num_feats, scfglex_src_arity, scfglex_als, scfglex_num_als)); check_and_update_ctf_stack(rp); TRulePtr coarse_rp = ((ctf_level == 0) ? TRulePtr() : ctf_rule_stack.top()); rule_callback(rp, ctf_level, coarse_rp, rule_callback_extra); @@ -280,7 +285,7 @@ NT [^\t \[\],]+ b += yytext[i] - '0'; ++i; } - // TODO store alignment points somewhere + scfglex_als[scfglex_num_als++]=AlignmentPoint(a,b); } <ALIGNS>[ \t] ; <ALIGNS>. { diff --git a/decoder/trule.h b/decoder/trule.h index acdbc5cf..6bffc277 100644 --- a/decoder/trule.h +++ b/decoder/trule.h @@ -4,6 +4,7 @@ #include <algorithm> #include <vector> #include <cassert> +#include <iostream> #include <boost/shared_ptr.hpp> #include "sparse_vector.h" @@ -12,18 +13,27 @@ class TRule; typedef boost::shared_ptr<TRule> TRulePtr; -struct NTSizeSummaryStatistics { - NTSizeSummaryStatistics(int arity) : means(arity), vars(arity) {} - std::vector<float> means; - std::vector<float> vars; +struct AlignmentPoint { + AlignmentPoint() : s_(), t_() {} + AlignmentPoint(int s, int t) : s_(s), t_(t) {} + AlignmentPoint Inverted() const { + return AlignmentPoint(t_, s_); + } + short s_; + short t_; }; +inline std::ostream& operator<<(std::ostream& os, const AlignmentPoint& p) { + return os << '(' << static_cast<int>(p.s_) << '-' << static_cast<int>(p.t_) << ')'; +} + // Translation rule class TRule { public: TRule() : lhs_(0), prev_i(-1), prev_j(-1) { } - TRule(WordID lhs, const WordID* src, int src_size, const WordID* trg, int trg_size, const int* feat_ids, const double* feat_vals, int feat_size, int arity) : - e_(trg, trg + trg_size), f_(src, src + src_size), lhs_(lhs), arity_(arity), prev_i(-1), prev_j(-1) { + TRule(WordID lhs, const WordID* src, int src_size, const WordID* trg, int trg_size, const int* feat_ids, const double* feat_vals, int feat_size, int arity, const AlignmentPoint* als, int alsnum) : + e_(trg, trg + trg_size), f_(src, src + src_size), lhs_(lhs), arity_(arity), prev_i(-1), prev_j(-1), + a_(als, als + alsnum) { for (int i = 0; i < feat_size; ++i) scores_.set_value(feat_ids[i], feat_vals[i]); } @@ -113,6 +123,7 @@ class TRule { const std::vector<WordID>& f() const { return f_; } const std::vector<WordID>& e() const { return e_; } + const std::vector<AlignmentPoint>& als() const { return a_; } int EWords() const { return ELength() - Arity(); } int FWords() const { return FLength() - Arity(); } @@ -141,8 +152,7 @@ class TRule { short int prev_i; short int prev_j; - // may be null - boost::shared_ptr<NTSizeSummaryStatistics> nt_size_summary_; + std::vector<AlignmentPoint> a_; // alignment points, may be empty // only for coarse-to-fine decoding boost::shared_ptr<std::vector<TRulePtr> > fine_rules_; |