1 files changed, 14 insertions, 0 deletions
diff --git a/sa-extract/cstrmap.pyx b/sa-extract/cstrmap.pyx
new file mode 100644
index 00000000..d3883ea5
--- /dev/null
+++ b/sa-extract/cstrmap.pyx
@@ -0,0 +1,14 @@
+
+cdef class StringMap:
+    def __cinit__(self):
+        self.vocab = stringmap_new()
+
+    def __dealloc__(self):
+        stringmap_delete(self.vocab)
+        
+    cdef char *word(self, int i):
+        return stringmap_word(self.vocab, i)
+
+    cdef int index(self, char *s):
+        return stringmap_index(self.vocab, s)
+