diff options
| -rw-r--r-- | live-mogrify.py | 16 | ||||
| -rw-r--r-- | options/dataset_options.py | 11 |
2 files changed, 25 insertions, 2 deletions
diff --git a/live-mogrify.py b/live-mogrify.py index bc687c0..d3c8f5d 100644 --- a/live-mogrify.py +++ b/live-mogrify.py @@ -132,6 +132,8 @@ class Listener(): self.data_opt_parser = data_opt_parser.parser self.working = False def _set_fn(self, key, value): + if key == 'pause' and self.data_opt.processing is False: + process_live_input(self.opt, self.data_opt, self.rpc_client) if hasattr(self.data_opt, key): try: if str(value) == 'True': @@ -149,6 +151,9 @@ class Listener(): print('error {} - cant set value {}: {}'.format(e, key, value)) def _get_fn(self): return vars(self.data_opt) + def _cmd_fn(self, cmd, payload): + print("got command {}".format(cmd)) + return 'yes' def _ready_fn(self, rpc_client): # self.connect() process_live_input(self.opt, self.data_opt, rpc_client) @@ -156,13 +161,16 @@ class Listener(): self.rpc_client = CortexRPC(self._get_fn, self._set_fn, self._ready_fn) def process_live_input(opt, data_opt, rpc_client): - A_offset, A_im, A_dir = load_first_frame(opt, data_opt) - + if data_opt.processing: + print("Already processing...") + data_opt.processing = True data_loader = CreateRecursiveDataLoader(opt) dataset = data_loader.load_data() + model = create_model(opt) print("generating...") + A_offset, A_im, A_dir = load_first_frame(opt, data_opt) last_im = None for i, data in enumerate(data_loader): if i >= opt.how_many: @@ -248,6 +256,10 @@ def process_live_input(opt, data_opt, rpc_client): os.rename(tmp_path, next_path) if data_opt.exit: sys.exit(0) + if data_opt.pause: + data_opt.pause = False + break + data_opt.processing = False if __name__ == '__main__': listener = Listener() diff --git a/options/dataset_options.py b/options/dataset_options.py index 5457c22..92ea2bc 100644 --- a/options/dataset_options.py +++ b/options/dataset_options.py @@ -104,6 +104,17 @@ class DatasetOptions(BaseOptions): ### RECURSION self.parser.add_argument( + '--processing', + action='store_true', + help='internally stores whether or not we should be processing..' + ) + self.parser.add_argument( + '--pause', + action='store_true', + help='internally tells the loop to pause..' + ) + + self.parser.add_argument( '--transition', action='store_true', help=' sine wave transition' |