1 files changed, 12 insertions, 9 deletions
diff --git a/live-mogrify.py b/live-mogrify.py
index b077508..6b14eb8 100644
--- a/live-mogrify.py
+++ b/live-mogrify.py
@@ -122,15 +122,18 @@ def process_image(opt, data_opt, im):
   img = np.add(array_a, array_b).astype('uint8')
   return img
 
-def listen():
-  opt, data_opt = load_opt()
-  def set_data_opt(key, value):
-    data_opt[key] = value
-  def get_opts():
-    return vars(data_opt)
-  def activate():
-    process_live_input(opt, data_opt, rpc_client)
-  rpc_client = CortexRPC(get_opts, set_data_opt, activate)
+class Listener(CortexRPC):
+  def __init__:
+    opt, data_opt = load_opt()
+    self.opt = opt
+    self.data_opt = data_opt
+    self.rpc_client = CortexRPC(self.get_opts, self.set_data_opt, self.activate)
+  def _set_fn(key, value):
+    self.data_opt[key] = value
+  def _get_fn(self):
+    return vars(self.data_opt)
+  def _ready_fn():
+    process_live_input(self.opt, self.data_opt, self.rpc_client)
 
 def process_live_input(opt, data_opt, rpc_client):
   A_offset, A_im, A_dir = load_first_frame(opt, data_opt)