diff options
Diffstat (limited to 'mgizapp/src/snt2cooc-reduce-mem-preprocess.cpp')
-rw-r--r-- | mgizapp/src/snt2cooc-reduce-mem-preprocess.cpp | 79 |
1 files changed, 79 insertions, 0 deletions
diff --git a/mgizapp/src/snt2cooc-reduce-mem-preprocess.cpp b/mgizapp/src/snt2cooc-reduce-mem-preprocess.cpp new file mode 100644 index 0000000..3309a7e --- /dev/null +++ b/mgizapp/src/snt2cooc-reduce-mem-preprocess.cpp @@ -0,0 +1,79 @@ +#include <iostream> +#include <string> +#include <strstream> +#include <fstream> +#include <map> +#include <vector> +#include <set> +#include <stdio.h> +#include <stdlib.h> + +using namespace std; + +void readVoc(istream&in,map<string,string>&voc) +{ + string line,s1,s2; + voc["1"]="UNK"; + if( !in )cerr <<"Vocabulary does not exist.\n"; + while(getline(in,line)) + { + istrstream eingabe(line.c_str()); + if( !(eingabe>>s1>>s2)) + cerr << "ERROR in vocabulary '" << line << "'\n"; + voc[s1]=s2; + } +} + +int maxElems=0; +int main(int argc,char **argv) +{ + if( argc!=4&&argc!=5 ) + { + cerr << "Usage: " << argv[0] << " output vcb1 vcb2 snt12 \n"; + cerr << "Converts GIZA++ snt-format into plain text.\n"; + exit(1); + } + bool counts=0; + if( argc==6 ) + { + if(string(argv[4])!="-counts") + cerr << "ERROR: wrong option " << argv[5] << endl; + counts=1; + maxElems=10000000; + } + ifstream v1(argv[1]),v2(argv[2]),t(argv[3]); + map<string,string>voc1,voc2; + readVoc(v1,voc1); + readVoc(v2,voc2); + string line1,line2,line3; + int nLine=0; + int totalElems=0; + while(getline(t,line1)&&getline(t,line2)&&getline(t,line3)) + { + istrstream eingabe1(line1.c_str()),eingabe2(line2.c_str()),eingabe3(line3.c_str()); + double count; + string word; + eingabe1>>count; + vector<int>l1,l2; + while(eingabe2>>word) + l1.push_back(atoi(word.c_str())); + while(eingabe3>>word) + l2.push_back(atoi(word.c_str())); + if( ((++nLine)%1000)==0 ) + cerr << "line " << nLine << '\n'; + for(unsigned int j=0; j<l2.size(); ++j) + { + cout << 0 << " " << l2[j] << endl; + } + for(unsigned int i=0; i<l1.size(); ++i) + { + for(unsigned int j=0; j<l2.size(); ++j) + { + cout << l1[i] << " " << l2[j] << endl; + } + } + } + cerr << "END.\n"; +} + + |