92 lines
2.9 KiB
Python
Executable File
92 lines
2.9 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
"""Video Thumbnail Generator
|
|
|
|
Usage:
|
|
./generator <video> <interval> <width> <height> <columns> <output>
|
|
./generator (-h | --help)
|
|
./generator --version
|
|
|
|
Options:
|
|
-h --help Show this screen.
|
|
--version Show version.
|
|
<video> Video filepath.
|
|
<interval> Interval em seconds between frames.
|
|
<width> Width of each thumbnail.
|
|
<height> Height of each thumbnail.
|
|
<columns> Total number of thumbnails per line.
|
|
<output> Output.
|
|
"""
|
|
|
|
from docopt import docopt
|
|
from moviepy.editor import VideoFileClip
|
|
from PIL import Image
|
|
import glob, os, random, sys, shutil, math
|
|
|
|
|
|
TMP_FRAMES_PATH="/tmp/frames/"
|
|
|
|
def generate_video_thumbnail(args):
|
|
videoFileClip = VideoFileClip(args['<video>'])
|
|
interval = int(args['<interval>'])
|
|
size = (int(args['<width>']), int(args['<height>']))
|
|
outputPrefix = get_output_prefix()
|
|
|
|
generate_frames(videoFileClip, interval, outputPrefix, size)
|
|
|
|
columns = int(args['<columns>'])
|
|
output = args['<output>']
|
|
generate_sprite_from_images(outputPrefix, columns, size, output)
|
|
|
|
def generate_frames(videoFileClip, interval, outputPrefix, size):
|
|
print "Extracting", int(videoFileClip.duration / interval), "frames"
|
|
for i in range(0, int(videoFileClip.duration), interval):
|
|
extract_frame(videoFileClip, i, outputPrefix, size)
|
|
print " Frames extracted."
|
|
|
|
|
|
def extract_frame(videoFileClip, moment, outputPrefix, size):
|
|
sys.stdout.write(".")
|
|
sys.stdout.flush()
|
|
output = outputPrefix + str(moment) + ".png"
|
|
videoFileClip.save_frame(output, t=int(moment))
|
|
resize_image(output, size)
|
|
|
|
def resize_image(filename, size):
|
|
image = Image.open(filename)
|
|
image = image.resize(size, Image.ANTIALIAS)
|
|
image.save(filename)
|
|
|
|
def generate_sprite_from_images(framesPath, columns, size, output):
|
|
framesMap = sorted(glob.glob(framesPath + "*.png"))
|
|
images = [Image.open(filename) for filename in framesMap]
|
|
print "%d frames will be combined." % len(images)
|
|
masterWidth = size[0] * columns
|
|
masterHeight = size[1] * int(math.ceil(float(len(images)) / columns))
|
|
finalImage = Image.new(mode='RGBA', size=(masterWidth, masterHeight), color=(0,0,0,0))
|
|
merge_frames(images, finalImage, columns, size, output)
|
|
|
|
def merge_frames(images, finalImage, columns, size, output):
|
|
line = 0
|
|
for i, image in enumerate(images):
|
|
locationX = size[0] * i
|
|
locationY = size[1] * line
|
|
finalImage.paste(image, (locationX, locationY))
|
|
if i != 0 and i % columns == 0: line += 1
|
|
|
|
finalImage.save(output, transparency=0)
|
|
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)
|
|
|