vars..

1 files changed, 14 insertions, 6 deletions

diff --git a/live-mogrify.py b/live-mogrify.py
index f1e83e6..fc54571 100644
--- a/live-mogrify.py
+++ b/live-mogrify.py
@@ -31,8 +31,10 @@ def lerp(n,a,b):
   return (b-a)*n+a
 
 def load_opt():
-  opt = TestOptions().parse()
-  data_opt = DatasetOptions().parse(opt.unknown)
+  opt_parser = TestOptions()
+  opt = opt_parser.parse()
+  data_opt_parser = DatasetOptions()
+  data_opt = data_opt_parser.parse(opt.unknown)
   opt.nThreads = 1   # test code only supports nThreads = 1
   opt.batchSize = 1  # test code only supports batchSize = 1
   opt.serial_batches = True  # no shuffle
@@ -40,7 +42,7 @@ def load_opt():
 
   data_opt.tag = get_tag(opt, data_opt)
   opt.render_dir = opt.results_dir + opt.name + "/" + data_opt.tag + "/"
-  return opt, data_opt
+  return opt, data_opt, data_opt_parser
 
 def get_tag(opt, data_opt):
   if data_opt.tag == '':
@@ -124,14 +126,20 @@ def process_image(opt, data_opt, im):
 
 class Listener():
   def __init__(self):
-    opt, data_opt = load_opt()
+    opt, data_opt, data_opt_parser = load_opt()
     self.opt = opt
     self.data_opt = data_opt
+    self.data_opt_parser = data_opt_parser
     self.working = False
   def _set_fn(self, key, value):
     if hasattr(self.data_opt, key):
-      print('set {} {}: {}'.format(type(value), key, value))
-      setattr(self.data_opt, key, value)
+      try:
+        new_opt = parser.parse_known_args([ key, value ])
+        new_value = getattr(new_opt, key)
+        print('set {} {}: {}'.format(type(value), key, new_value))
+        setattr(self.data_opt, key, new_value)
+      catch(e):
+        print('error setting value {}: {}'.format(key, value))
   def _get_fn(self):
     return vars(self.data_opt)
   def _ready_fn(self, rpc_client):