diff python/cvutils.py @ 981:c3e690c5536e

corrected bug in display trajectories
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Tue, 27 Feb 2018 15:24:57 -0500
parents 184f1dd307f9
children 132d84ce9f0c
line wrap: on
line diff
--- a/python/cvutils.py	Mon Feb 19 16:32:59 2018 -0500
+++ b/python/cvutils.py	Tue Feb 27 15:24:57 2018 -0500
@@ -28,7 +28,7 @@
 
 videoFilenameExtensions = ['mov', 'avi', 'mp4', 'MOV', 'AVI', 'MP4']
 
-#import aggdraw # agg on top of PIL (antialiased drawing)
+#importaggdraw # agg on top of PIL (antialiased drawing)
 
 
 cvRed = {'default': (0,0,255),
@@ -318,10 +318,14 @@
                 lastFrameNum = float("inf")
             else:
                 lastFrameNum = lastFrameNumArg
-            if nZerosFilenameArg is None:
-                nZerosFilename = int(ceil(log10(lastFrameNum)))
-            else:
-                nZerosFilename = nZerosFilenameArg
+            if saveAllImages:
+                if nZerosFilenameArg is None:
+                    if lastFrameNumArg is None:
+                        nZerosFilename = int(ceil(log10(objects[-1].getLastInstant())))
+                    else:
+                        nZerosFilename = int(ceil(log10(lastFrameNum)))
+                else:
+                    nZerosFilename = nZerosFilenameArg
             while ret and not quitKey(key) and frameNum <= lastFrameNum:
                 ret, img = capture.read()
                 if ret: