189 lines
7.2 KiB
Python
Executable file
189 lines
7.2 KiB
Python
Executable file
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
import os
|
|
|
|
from PIL import Image
|
|
from PIL import ImageDraw
|
|
import json
|
|
from optparse import OptionParser
|
|
import ox
|
|
from ox.image import drawText, wrapText
|
|
import sys
|
|
|
|
root_dir = os.path.normpath(os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
|
|
static_root = os.path.join(os.path.dirname(__file__), 'data')
|
|
|
|
def render_poster(data, poster):
|
|
|
|
title = ox.decode_html(data.get('title', ''))
|
|
director = ox.decode_html(', '.join(data.get('director', [])))
|
|
if not director and "type" in data:
|
|
director = ", ".join(data["type"])
|
|
if director == "Voice Over" and data.get("batch"):
|
|
director = "%s (vo)" % (", ".join(data["batch"])).replace('Text-', '')
|
|
vo = True
|
|
else:
|
|
vo = False
|
|
year = str(data.get('year', ''))
|
|
series = data.get('isSeries', False)
|
|
oxdb_id = data['oxdbId']
|
|
imdb_id = data['id']
|
|
frame = data.get('frame')
|
|
timeline = data.get('timeline')
|
|
|
|
def get_oxdb_color(oxdb_id, series=False):
|
|
i = int(round((int(oxdb_id[2:10], 16) * 762 / pow(2, 32))))
|
|
type = data.get('type')
|
|
if type:
|
|
type = data.get('type')[0].lower()
|
|
if type == "original":
|
|
i = 0
|
|
elif type == "background":
|
|
i = 200
|
|
elif type == "foreground":
|
|
i = 400
|
|
elif type == "foreground2":
|
|
i = 500
|
|
elif type == "annimation":
|
|
i = 600
|
|
elif type == "voice over":
|
|
i = 300
|
|
if '1-' in director:
|
|
i = 300
|
|
if '2-' in director:
|
|
i = 320
|
|
if '3-' in director:
|
|
i = 340
|
|
if '4-' in director:
|
|
i = 360
|
|
if '5-' in director:
|
|
i = 380
|
|
if i < 127:
|
|
color = (127, i, 0)
|
|
elif i < 254:
|
|
color = (254 - i, 127, 0)
|
|
elif i < 381:
|
|
color = (0, 127, i - 254)
|
|
elif i < 508:
|
|
color = (0, 508 - i, 127)
|
|
elif i < 635:
|
|
color = (i - 508, 0, 127)
|
|
else:
|
|
color = (127, 0, 762 - i)
|
|
if series:
|
|
color = tuple(map(lambda x: x + 128, color))
|
|
return color
|
|
|
|
poster_width = 640
|
|
poster_height = 1024
|
|
poster_ratio = poster_width / poster_height
|
|
poster_image = Image.new('RGB', (poster_width, poster_height))
|
|
draw = ImageDraw.Draw(poster_image)
|
|
font_file = os.path.join(static_root, 'DejaVuSansCondensedBold.ttf')
|
|
|
|
font_size = {
|
|
'small': 28,
|
|
'large': 42,
|
|
}
|
|
|
|
# frame
|
|
frame_width = poster_width
|
|
frame_ratio = 4 / 3
|
|
frame_height = int(round(frame_width / frame_ratio))
|
|
if frame:
|
|
frame_image = Image.open(frame)
|
|
frame_image_ratio = frame_image.size[0] / frame_image.size[1]
|
|
if frame_ratio < frame_image_ratio:
|
|
frame_image = frame_image.resize((int(frame_height * frame_image_ratio), frame_height), Image.LANCZOS)
|
|
left = int((frame_image.size[0] - frame_width) / 2)
|
|
frame_image = frame_image.crop((left, 0, left + frame_width, frame_height))
|
|
else:
|
|
frame_image = frame_image.resize((frame_width, int(frame_width / frame_image_ratio)), Image.LANCZOS)
|
|
top = int((frame_image.size[1] - frame_height) / 2)
|
|
frame_image = frame_image.crop((0, top, frame_width, top + frame_height))
|
|
poster_image.paste(frame_image, (0, 0))
|
|
|
|
|
|
# timeline
|
|
timeline_width = poster_width
|
|
timeline_height = 64
|
|
if timeline:
|
|
timeline_image = Image.open(timeline)
|
|
timeline_image = timeline_image.resize((timeline_width, timeline_height), Image.LANCZOS)
|
|
poster_image.paste(timeline_image, (0, frame_height))
|
|
|
|
# text
|
|
text_width = poster_width
|
|
text_height = poster_height - frame_height - timeline_height
|
|
text_top = frame_height + timeline_height
|
|
text_bottom = text_top + text_height
|
|
text_margin = 16
|
|
text_color = get_oxdb_color(oxdb_id, series)
|
|
font_color = tuple(map(lambda x: x - 128 if series else x + 128, text_color))
|
|
draw.rectangle([(0, text_top), (text_width, text_bottom)], fill=text_color)
|
|
offset_top = text_top + text_margin
|
|
if vo:
|
|
vo_text = (" ".join(data["batch"])).replace("Text-", "")
|
|
drawText(poster_image, (0, 0), vo_text, font_file, 100, font_color)
|
|
if not director:
|
|
title_max_lines = 7
|
|
else:
|
|
title_max_lines = min(len(wrapText(title, text_width - 2 * text_margin, 0, font_file, font_size['large'])), 6)
|
|
director_max_lines = 9 - int((title_max_lines * 3 - 1) / 2)
|
|
if director:
|
|
lines = wrapText(director, text_width - 2 * text_margin, director_max_lines, font_file, font_size['small'])
|
|
for i, line in enumerate(lines):
|
|
size = drawText(poster_image, (text_margin, offset_top), line, font_file, font_size['small'], font_color)
|
|
offset_top += font_size['small'] + 2
|
|
offset_top += size[1] - font_size['small'] + text_margin / 2
|
|
lines = wrapText(title, text_width - 2 * text_margin, title_max_lines, font_file, font_size['large'])
|
|
for i, line in enumerate(lines):
|
|
size = drawText(poster_image, (text_margin, offset_top + 5), line, font_file, font_size['large'], font_color)
|
|
offset_top += font_size['large'] + 3
|
|
offset_top += size[1] - font_size['small'] + text_margin / 2
|
|
if year:
|
|
drawText(poster_image, (text_margin, offset_top), year, font_file, font_size['small'], font_color)
|
|
item_id = imdb_id or oxdb_id
|
|
drawText(poster_image, (text_margin, text_bottom - text_margin - font_size['large'] + 2), item_id, font_file, font_size['large'], font_color)
|
|
|
|
'''
|
|
# logo
|
|
logo_height = 32
|
|
logo_image = Image.open(os.path.join(static_root, '..', '..', 'static', 'png', 'logo.png'))
|
|
logo_width = int(round(logo_height * logo_image.size[0] / logo_image.size[1]))
|
|
logo_image = logo_image.resize((logo_width, logo_height), Image.LANCZOS)
|
|
logo_left = text_width - text_margin - logo_width
|
|
logo_top = text_bottom - text_margin - logo_height
|
|
for y in range(logo_height):
|
|
for x in range(logo_width):
|
|
poster_color = poster_image.getpixel((logo_left + x, logo_top + y))
|
|
logo_color = logo_image.getpixel((x, y))[0]
|
|
alpha = logo_image.getpixel((x, y))[3]
|
|
if series:
|
|
poster_color = tuple(map(lambda x: int(x - (logo_color - 16) * alpha / 255), poster_color))
|
|
else:
|
|
poster_color = tuple(map(lambda x: int(x + (logo_color - 16) * alpha / 255), poster_color))
|
|
poster_image.putpixel((logo_left + x, logo_top + y), poster_color)
|
|
i '''
|
|
poster_image.save(poster)
|
|
|
|
def main():
|
|
parser = OptionParser()
|
|
parser.add_option('-d', '--data', dest='data', help='json file with metadata', default=None)
|
|
parser.add_option('-p', '--poster', dest='poster', help='Poster (image file to be written)')
|
|
(options, args) = parser.parse_args()
|
|
|
|
if None in (options.data, options.poster):
|
|
parser.print_help()
|
|
sys.exit()
|
|
|
|
if options.data == '-':
|
|
data = json.load(sys.stdin)
|
|
else:
|
|
with open(options.data) as f:
|
|
data = json.load(f)
|
|
render_poster(data, options.poster)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|