0x2620/oxtimelines
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

multiline support

Browse source
This commit is contained in:
j 2019-11-16 16:45:31 +01:00
parent 846f0ceb5b
commit 82df6efee4
1 changed files with 41 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

46
oxtimelines/timeline.py
View file

@ -4,6 +4,8 @@ from __future__ import division, print_function, absolute_import
from glob import glob from glob import glob
from time import time from time import time
import multiprocessing
from multiprocessing.pool import ThreadPool
try: try:
from PIL import Image from PIL import Image
except: except:
@ -40,6 +42,15 @@ class Timelines():
self.modes = modes self.modes = modes
self.render_antialias = 'antialias' in modes self.render_antialias = 'antialias' in modes
self.render_slitscan = 'slitscan' in modes self.render_slitscan = 'slitscan' in modes
if 'multiline' in modes:
info = ox.avinfo(video_files[0])
self.render_multiline = info['video'][0]['width']
self.output_h = sizes[0]
sizes = [info['video'][0]['height']] + sizes[1:]
self.render_multiline = True
self._pool = ThreadPool(multiprocessing.cpu_count())
else:
self.render_multiline = False
self.render_keyframes = 'keyframes' in modes self.render_keyframes = 'keyframes' in modes
self.render_audio = 'audio' in modes self.render_audio = 'audio' in modes
self.render_cuts = 'cuts' in modes self.render_cuts = 'cuts' in modes
@ -191,6 +202,13 @@ class Timelines():
w = self.large_tile_last_w w = self.large_tile_last_w
for mode in filter(lambda x: x in ['antialias', 'slitscan', 'cuts'], self.modes): for mode in filter(lambda x: x in ['antialias', 'slitscan', 'cuts'], self.modes):
self.large_tile_image[mode] = Image.new('RGB', (w, self.large_tile_h)) self.large_tile_image[mode] = Image.new('RGB', (w, self.large_tile_h))
if self.render_multiline:
self.large_tile_image['multiline'] = {}
t0 = time()
#print('create images')
for line in range(self.render_multiline):
self.large_tile_image['multiline'][line] = Image.new('RGB', (w, self.large_tile_h))
#print('create images done', time()-t0)
if self.render_data: if self.render_data:
self.large_tile_image['data'] = Image.new('RGB', (w * 8, 8)) self.large_tile_image['data'] = Image.new('RGB', (w * 8, 8))
paste = (large_tile_x, 0) paste = (large_tile_x, 0)
@ -208,6 +226,11 @@ class Timelines():
if self.render_slitscan: if self.render_slitscan:
crop = (self.frame_center, 0, self.frame_center + 1, self.large_tile_h) crop = (self.frame_center, 0, self.frame_center + 1, self.large_tile_h)
self.large_tile_image['slitscan'].paste(frame_image.crop(crop), paste) self.large_tile_image['slitscan'].paste(frame_image.crop(crop), paste)
if self.render_multiline:
def chrop_line(line):
crop = (line, 0, line + 1, self.large_tile_h)
self.large_tile_image['multiline'][line].paste(frame_image.crop(crop), paste)
self._pool.map(chrop_line, list(range(self.render_multiline)))
self.log and self.profiler.unset_task() self.log and self.profiler.unset_task()
# render data tile # render data tile
@ -266,7 +289,7 @@ class Timelines():
# save large tile # save large tile
self.log and self.profiler.set_task('_video_callback()') self.log and self.profiler.set_task('_video_callback()')
if large_tile_x == self.large_tile_w - 1 or self.is_last_frame: if large_tile_x == self.large_tile_w - 1 or self.is_last_frame:
for mode in filter(lambda x: x in ['antialias', 'slitscan', 'cuts', 'data'], self.modes): for mode in filter(lambda x: x in ['antialias', 'slitscan', 'cuts', 'data', 'multiline'], self.modes):
self._save_tile(mode, self.large_tile_i) self._save_tile(mode, self.large_tile_i)
if self.render_antialias: if self.render_antialias:
self._save_full_tile_image('antialias') self._save_full_tile_image('antialias')
@ -431,10 +454,23 @@ class Timelines():
if mode != 'keyframeswide' or self.render_wide_tiles: if mode != 'keyframeswide' or self.render_wide_tiles:
height = 8 if mode == 'data' else self.large_tile_h height = 8 if mode == 'data' else self.large_tile_h
type = 'png' if mode == 'data' else 'jpg' type = 'png' if mode == 'data' else 'jpg'
tile_file = '%stimeline%s%dp%d.%s' % ( if mode == 'multiline':
self.tile_path, mode, height, index, type def save_line(line):
) tile_file = '%s%s/timeline%s%dp%d.%s' % (
self.large_tile_image[mode].save(tile_file) self.tile_path, line, 'slitscan', self.output_h, index, type
)
ox.makedirs(os.path.dirname(tile_file))
width = self.large_tile_image[mode][line].size[1]
img = self.large_tile_image[mode][line]
if self.output_h != self.large_tile_image[mode][line].size[0]:
img = img.resize((width, self.output_h), Image.BICUBIC)
img.save(tile_file)
self._pool.map(save_line, list(range(self.render_multiline)))
else:
tile_file = '%stimeline%s%dp%d.%s' % (
self.tile_path, mode, height, index, type
)
self.large_tile_image[mode].save(tile_file)
if self.log: if self.log:
print(tile_file) print(tile_file)
if self.render_small_tiles and mode in ['antialias', 'slitscan', 'keyframeswide', 'audio']: if self.render_small_tiles and mode in ['antialias', 'slitscan', 'keyframeswide', 'audio']: