From 3acdf1e4b37637d6df86a7b54fb0f1b0464c172b Mon Sep 17 00:00:00 2001
From: Victor Chahuneau <vchahune@cs.cmu.edu>
Date: Wed, 6 Jun 2012 01:10:49 -0400
Subject: Python - Added hypergraph functionnality

- k-best translations
- 1-best, k-best structures
- sample derivations from the forest
- intersect hypergraph and lattice
- lattice
---
 python/src/hypergraph.pxd | 35 +++++++++++++++++++++++++++++++++--
 1 file changed, 33 insertions(+), 2 deletions(-)

(limited to 'python/src/hypergraph.pxd')

diff --git a/python/src/hypergraph.pxd b/python/src/hypergraph.pxd
index 05977fa8..4aef6955 100644
--- a/python/src/hypergraph.pxd
+++ b/python/src/hypergraph.pxd
@@ -1,10 +1,41 @@
 from libcpp.string cimport string
 from libcpp.vector cimport vector
-from utils cimport WordID
+from utils cimport *
+cimport lattice
 
 cdef extern from "decoder/hg.h":
     cdef cppclass Hypergraph:
+        cppclass Node:
+            int id_
+            WordID cat_
+            WordID NT()
+            #EdgesVector in_edges_
+            #EdgesVector out_edges_
         Hypergraph(Hypergraph)
+        vector[Node] nodes_
 
 cdef extern from "decoder/viterbi.h":
-    cdef string ViterbiESentence(Hypergraph hg, vector[WordID]* trans)
+    cdef prob_t ViterbiESentence(Hypergraph hg, vector[WordID]* trans)
+    cdef string ViterbiETree(Hypergraph hg)
+
+cdef extern from "decoder/hg_io.h" namespace "HypergraphIO":
+    bint ReadFromJSON(istream* inp, Hypergraph* out)
+    bint WriteToJSON(Hypergraph hg, bint remove_rules, ostream* out)
+
+    #void WriteAsCFG(Hypergraph hg)
+    #void WriteTarget(string base, unsigned sent_id, Hypergraph hg)
+
+    void ReadFromPLF(string inp, Hypergraph* out, int line=*)
+    string AsPLF(Hypergraph hg, bint include_global_parentheses=*)
+    string AsPLF(lattice.Lattice lat, bint include_global_parentheses=*)
+    void PLFtoLattice(string plf, lattice.Lattice* pl)
+
+cdef extern from "decoder/hg_intersect.h" namespace "HG":
+    bint Intersect(lattice.Lattice target, Hypergraph* hg)
+
+cdef extern from "decoder/hg_sampler.h" namespace "HypergraphSampler":
+    cdef cppclass Hypothesis:
+        vector[WordID] words
+        SparseVector[double] fmap
+        prob_t model_score
+    void sample_hypotheses(Hypergraph hg, unsigned n, MT19937* rng, vector[Hypothesis]* hypos)
-- 
cgit v1.2.3