seek to a frame

1 files changed, 6 insertions, 0 deletions

diff --git a/live-mogrify.py b/live-mogrify.py
index b511af1..fedb246 100644
--- a/live-mogrify.py
+++ b/live-mogrify.py
@@ -199,6 +199,8 @@ class Listener():
       print('load sequence: {}'.format(payload))
       self.data_opt.sequence_name = payload
       self.data_opt.load_sequence = True
+    if cmd == 'seek':
+      self.data_opt.seek_to = payload
     if cmd == 'get_status':
       return {
         'processing': self.data_opt.processing,
@@ -259,6 +261,10 @@ def process_live_input(opt, data_opt, rpc_client, model):
         print("Got sequence {}, {} images, first: {}".format(data_opt.sequence_name, len(sequence), sequence[0]))
         sequence = new_sequence
         sequence_i = 1
+    if data_opt.seek_to != 1:
+      if data_opt.seek_to > 0 and data_opt.seek_to < len(sequence_i):
+        sequence_i = data_opt.seek_to
+      data_opt.seek_to = 1
 
     model.set_input(data)
     model.test()