From c770b03d4a89e6f8d3b9b2571336df59bcb62b5b Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Fl=C3=A1vio=20Ribeiro?= Date: Wed, 18 Nov 2015 01:07:57 -0500 Subject: [PATCH] bugfix on the order of the frames --- generator | 12 +++++------- 1 file changed, 5 insertions(+), 7 deletions(-) diff --git a/generator b/generator index 3801f45..a6451c8 100755 --- a/generator +++ b/generator @@ -31,7 +31,6 @@ def generate_video_thumbnail(args): interval = int(args['']) size = (int(args['']), int(args[''])) outputPrefix = get_output_prefix() - generate_frames(videoFileClip, interval, outputPrefix, size) columns = int(args['']) @@ -40,15 +39,16 @@ def generate_video_thumbnail(args): def generate_frames(videoFileClip, interval, outputPrefix, size): print "Extracting", int(videoFileClip.duration / interval), "frames" + frameCount = 0 for i in range(0, int(videoFileClip.duration), interval): - extract_frame(videoFileClip, i, outputPrefix, size) + extract_frame(videoFileClip, i, outputPrefix, size, frameCount) + frameCount += 1 print " Frames extracted." - -def extract_frame(videoFileClip, moment, outputPrefix, size): +def extract_frame(videoFileClip, moment, outputPrefix, size, frameCount): sys.stdout.write(".") sys.stdout.flush() - output = outputPrefix + str(moment) + ".png" + output = outputPrefix + ("%05d.png" % frameCount) videoFileClip.save_frame(output, t=int(moment)) resize_image(output, size) @@ -78,13 +78,11 @@ def merge_frames(images, finalImage, columns, size, output): shutil.rmtree(TMP_FRAMES_PATH, ignore_errors=True) print "Saved!" - def get_output_prefix(): if not os.path.exists(TMP_FRAMES_PATH): os.makedirs(TMP_FRAMES_PATH) return TMP_FRAMES_PATH + ("%032x_" % random.getrandbits(128)) - if __name__ == "__main__": arguments = docopt(__doc__, version='Naval Fate 2.0') generate_video_thumbnail(arguments)