diff --git a/pptk/viewer/viewer.py b/pptk/viewer/viewer.py index c1f92bd..1106266 100644 --- a/pptk/viewer/viewer.py +++ b/pptk/viewer/viewer.py @@ -5,6 +5,7 @@ import os import inspect import warnings +import time _viewer_dir = os.path.dirname(inspect.getfile(inspect.currentframe())) if ~os.path.isabs(_viewer_dir): @@ -354,6 +355,7 @@ def record(self, folder, poses, ts=[], tlim=[-numpy.inf, numpy.inf], ts: Same as in :meth:`pptk.viewer.play` tlim: Same as in :meth:`pptk.viewer.play` interp: Same as in :meth:`pptk.viewer.play` + shutter_speed: Time before capturing fps: Frames per second prefix: Resulting image file names are prefixed with this string ext: Image format @@ -398,6 +400,9 @@ def record(self, folder, poses, ts=[], tlim=[-numpy.inf, numpy.inf], struct.pack('2f', t, t) + \ struct.pack('?', False) self.__send(msg) + # give viewer time to finish render + if shutter_speed != numpy.inf: + time.sleep(shutter_speed) filename = prefix \ + ('%0' + str(num_digits) + 'd') % (i + 1) + '.' + ext filename = os.path.join(folder, filename)