1 files changed, 79 insertions, 0 deletions
diff --git a/decoder/minimal_decoder.cc b/decoder/minimal_decoder.cc
new file mode 100644
index 00000000..25922674
--- /dev/null
+++ b/decoder/minimal_decoder.cc
@@ -0,0 +1,79 @@
+#include <fstream>
+#include <iostream>
+#include <sstream>
+
+#include "fdict.h"
+#include "filelib.h"
+#include "hg.h"
+#include "hg_io.h"
+#include "sparse_vector.h"
+#include "viterbi.h"
+
+using namespace std;
+
+/*
+ * Reads hypergraph from JSON file argv[1],
+ * reweights it using weights from argv[2],
+ * and outputs viterbi translation.
+ *
+ */
+int
+main(int argc, char** argv)
+{
+  clock_t begin_total = clock();
+
+  // read hg
+  clock_t begin_read = clock();
+  ReadFile rf(argv[1]);
+  Hypergraph hg;
+  HypergraphIO::ReadFromJSON(rf.stream(), &hg);
+  clock_t end_read = clock();
+  double elapsed_secs_read = double(end_read - begin_read) / CLOCKS_PER_SEC;
+  cerr << "read hg " << elapsed_secs_read << " s" << endl;
+
+  // read weights
+  clock_t begin_weights = clock();
+  SparseVector<double> v;
+  ifstream f(argv[2]);
+  string line;
+  while (getline(f, line)) {
+    istringstream ss(line);
+    string k; weight_t w;
+    ss >> k >> w;
+    v.add_value(FD::Convert(k), w);
+  }
+  clock_t end_weights = clock();
+  double elapsed_secs_weights = double(end_weights - begin_weights) / CLOCKS_PER_SEC;
+  cerr << "read weights " << elapsed_secs_weights << " s" << endl;
+
+  // reweight hg
+  clock_t begin_reweight = clock();
+  hg.Reweight(v);
+  clock_t end_reweight = clock();
+  double elapsed_secs_reweight = double(end_reweight - begin_reweight) / CLOCKS_PER_SEC;
+  cerr << "reweight " << elapsed_secs_reweight << " s" << endl;
+
+  // topsort
+  clock_t begin_top = clock();
+  hg.TopologicallySortNodesAndEdges(hg.NumberOfNodes()-1);
+  clock_t end_top = clock();
+  double elapsed_secs_top = double(end_top - begin_top) / CLOCKS_PER_SEC;
+  cerr << "topsort " << elapsed_secs_top << " s" << endl;
+
+  // viterbi
+  clock_t begin_viterbi = clock();
+  vector<WordID> trans;
+  ViterbiESentence(hg, &trans);
+  cout << TD::GetString(trans) << endl << flush;
+  clock_t end_viterbi = clock();
+  double elapsed_secs_viterbi = double(end_viterbi - begin_viterbi) / CLOCKS_PER_SEC;
+  cerr << "viterbi " << elapsed_secs_viterbi << " s" << endl;
+
+  // total
+  clock_t end_total = clock();
+  double elapsed_secs = double(end_total - begin_total) / CLOCKS_PER_SEC;
+  cerr << "total " << elapsed_secs << " s" << endl;
+  
+  return 0;
+}
+