From 895dfd64ea5599ab16981cbfb538ec5f4073c8c1 Mon Sep 17 00:00:00 2001
From: Michael Denkowski <mdenkows@cs.cmu.edu>
Date: Sun, 15 Sep 2013 20:32:59 -0700
Subject: Move to using named commands

---
 realtime/rt/rt.py | 8 ++++++++
 1 file changed, 8 insertions(+)

(limited to 'realtime/rt/rt.py')

diff --git a/realtime/rt/rt.py b/realtime/rt/rt.py
index 1b8ac58c..0ce05a56 100644
--- a/realtime/rt/rt.py
+++ b/realtime/rt/rt.py
@@ -19,6 +19,8 @@ class RealtimeDecoder:
 
     def __init__(self, configdir, tmpdir='/tmp', cache_size=5, norm=False):
 
+        self.commands = {'LEARN': self.learn}
+
         cdec_root = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 
         # Temporary work dir
@@ -126,6 +128,12 @@ class RealtimeDecoder:
         self.detokenizer.stdin.write('{}\n'.format(line))
         return self.detokenizer.stdout.readline().strip()
 
+    def command(self, args):
+        try:
+            self.commands[args[0]](*args[1:])
+        except:
+            logging.info('Command error: {}'.format(' ||| '.join(args)))
+        
     def learn(self, source, target):
         if '' in (source.strip(), target.strip()):
             logging.info('Error empty source or target: {} ||| {}'.format(source, target))
-- 
cgit v1.2.3