def generate_teebob(self, text: str) -> BytesIO:
base = Image.open(f'{DIR}/memes/teebob.png')
canv = ImageDraw.Draw(base)
font = ImageFont.truetype(f'{DIR}/fonts/normal.ttf', 40)
box = ((100, 110), (360, 370))
wrap_new(canv, box, text, font=font)
return save(base)
def generate(self, name: str, text1: str, text2: str=None) -> BytesIO:
base = Image.open(f'{DIR}/memes/{name}.png')
canv = ImageDraw.Draw(base)
font = ImageFont.truetype(f'{DIR}/fonts/normal.ttf', 46)
canv.text((600, 100), wrap(font, text1, 400), fill='black', font=font)
if text2 is not None:
canv.text((600, 500), wrap(font, text2, 400), fill='black', font=font)
return save(base)
def generate_clown(self, text1: str, text2: str, text3: str, text4: str) -> BytesIO:
base = Image.open(f'{DIR}/memes/clown.png')
canv = ImageDraw.Draw(base)
font = ImageFont.truetype(f'{DIR}/fonts/normal.ttf', 30)
canv.text((10, 10), wrap(font, text1, 310), fill='black', font=font)
canv.text((10, 180), wrap(font, text2, 310), fill='black', font=font)
canv.text((10, 360), wrap(font, text3, 310), fill='black', font=font)
canv.text((10, 530), wrap(font, text4, 310), fill='black', font=font)
return save(base)
loss.backward(retain_graph=True)
loss_val = optimizer.step(lambda: loss)
# compute loss for output
print_loss = loss_val.item() / arguments.style_weight
print_loss /= arguments.map_channel_weight
if content_loss is not None:
print_loss /= arguments.content_weight
t1 = time.time() - t0
logging_info(
"Phase {:d}: Loss = {:f}, Escaped Time = {:.0f}m {:.0f}s".format(
phase, print_loss, t1 // 60 % 60, t1 % 60))
# save output
if arguments.save_interval is not None and phase % arguments.save_interval == 0:
filepath = os.path.join(run.output_path,
"output-{}.png".format(phase))
image.save(filepath, target)
logging_info("Output image saved to '{}'.".format(filepath))
# update live plot
if arguments.plot_interval is not None and phase % arguments.plot_interval == 0:
live_plot.update(target)
# write output to disk
filepath = os.path.join(run.output_path, "result.png")
image.save(filepath, target)
logging_info("Result image saved to '{}'.".format(filepath))
def test_graph(self, savedir, loaddir):
generated_image = self._generate_image(loaddir=loaddir, num_image=100)
image.save(savedir, generated_image, row=10, column=10)
def generate_hours_image(self,
data: Dict[str, List[asyncpg.Record]]) -> BytesIO:
font_small = ImageFont.truetype(f'{DIR}/fonts/normal.ttf', 16)
color_light = (100, 100, 100)
colors = (
'orange',
'red',
'forestgreen',
'dodgerblue',
'orangered',
'orchid',
'burlywood',
'darkcyan',
'royalblue',
'olive',
)
base = Image.open(f'{DIR}/hours_background.png')
canv = ImageDraw.Draw(base)
width, height = base.size
margin = 50
plot_width = width - margin * 2
plot_height = height - margin * 2
# draw area bg
bg = Image.new('RGBA', (plot_width, plot_height), color=(0, 0, 0, 100))
base.alpha_composite(bg, dest=(margin, margin))
# draw hours
x = margin
y = height - margin
hour_width = plot_width / 24
now = datetime.utcnow()
for hour in range(25):
xy = ((x, margin), (x, y - 1)) # fix overflow
canv.line(xy, fill=color_light, width=1)
if 0 <= hour <= 23:
text = str(hour)
w, h = font_small.getsize(text)
xy = (x + center(w, hour_width), y + h)
color = 'green' if hour == now.hour else color_light
canv.text(xy, text, fill=color, font=font_small)
x += hour_width
# draw players
extra = 2
size = (plot_width * 2, (plot_height + extra * 2) * 2)
plot = Image.new('RGBA', size, color=(0, 0, 0, 0))
plot_canv = ImageDraw.Draw(plot)
for hours, color in reversed(list(zip(data.values(), colors))):
hours = [
next((h['finishes'] for h in hours if h['hour'] == i), 0)
for i in range(24)
]
mult = lambda f: plot_height * 2 * (1 - f / max(hours)) + extra
x = -hour_width
xy = [(x, mult(hours[-1]))]
for finishes in hours:
x += hour_width * 2
y = mult(finishes)
rect_xy = ((x - 5, y - 5), (x + 5, y + 5))
plot_canv.rectangle(rect_xy, fill=color)
xy.append((x, y))
xy.append((x + hour_width * 2, mult(hours[0])))
plot_canv.line(xy, fill=color, width=6)
size = (plot_width, plot_height + extra * 2)
plot = plot.resize(size, resample=Image.LANCZOS,
reducing_gap=1.0) # antialiasing
base.alpha_composite(plot, dest=(margin, margin - extra))
# draw header
def check(w: int, size: int) -> int:
return w + (size / 3) * (4 * len(data) - 2)
font = auto_font((f'{DIR}/fonts/normal.ttf', 24),
''.join(data),
plot_width,
check=check)
space = font.size / 3
x = margin
_, h = font.getsize(
'yA') # max name height, needs to be hardcoded to align names
for player, color in zip(data, colors):
y = center(space, margin)
xy = ((x, y), (x + space, y + space))
canv.rectangle(xy, fill=color)
x += space * 2
w, _ = font.getsize(player)
xy = (x, center(h, margin))
canv.text(xy, player, fill='white', font=font)
x += w + space * 2
return save(base.convert('RGB'))
def generate_map_image(self, data: asyncpg.Record) -> BytesIO:
font_48 = ImageFont.truetype(f'{DIR}/fonts/normal.ttf', 46)
font_36 = ImageFont.truetype(f'{DIR}/fonts/normal.ttf', 36)
font_32 = ImageFont.truetype(f'{DIR}/fonts/normal.ttf', 32)
font_26 = ImageFont.truetype(f'{DIR}/fonts/normal.ttf', 26)
font_24 = ImageFont.truetype(f'{DIR}/fonts/normal.ttf', 24)
font_22 = ImageFont.truetype(f'{DIR}/fonts/normal.ttf', 22)
font_20 = ImageFont.truetype(f'{DIR}/fonts/normal.ttf', 20)
font_16 = ImageFont.truetype(f'{DIR}/fonts/normal.ttf', 16)
name = data['name']
color = data['color']
color = clamp_luminance(color, 0.7)
base = Image.open(f'{DIR}/map_backgrounds/{name}.png')
base = base.filter(ImageFilter.GaussianBlur(radius=3))
canv = ImageDraw.Draw(base)
width, height = base.size
outer = 32
inner = int(outer / 2)
margin = outer + inner
# draw bg
size = (width - outer * 2, height - outer * 2)
bg = round_rectangle(size, 12, color=(0, 0, 0, 175))
base.alpha_composite(bg, dest=(outer, outer))
# draw header
mappers = data['mappers']
name_height = 50
radius = int(name_height / 2)
text = name if mappers is None else f'{name} by {mappers}'
font = auto_font(font_36, text, width - margin * 2 - radius * 2)
w, _ = font.getsize(text)
_, h = font.getsize('yA')
size = (w + radius * 2, name_height)
name_bg = round_rectangle(size, radius, color=(150, 150, 150, 75))
base.alpha_composite(name_bg, dest=(margin, margin))
xy = (margin + radius, margin + center(h, name_height))
canv.text(xy, text, fill='white', font=font)
# draw info
server = data['server']
points = data['points']
finishers = data['finishers']
timestamp = data['timestamp']
info_width = (width - margin * 2) / 2.5
x = margin + info_width + inner
y = margin + name_height + inner
xy = ((x, margin + name_height + inner), (x, height - margin))
canv.line(xy, fill='white', width=3) # border
y += inner
servers = {
'Novice': (1, 0),
'Moderate': (2, 5),
'Brutal': (3, 15),
'Insane': (4, 30),
'Dummy': (5, 5),
'DDmaX': (4, 0),
'Oldschool': (6, 0),
'Solo': (4, 0),
'Race': (2, 0),
}
mult, offset = servers[server]
stars = int((points - offset) / mult)
lines = (((server.upper(), 'white', font_32), ),
(('★' * stars + '☆' * (5 - stars), 'white',
font_48), ), ((str(points), color, font_26),
(plural(points,
' point').upper(), 'white', font_20)),
((str(finishers), color, font_26),
(plural(finishers, ' finisher').upper(), 'white', font_20)),
(('RELEASED ', 'white', font_16),
(timestamp.strftime('%b %d %Y').upper(), color, font_22)))
for line in lines:
sizes = [f.getsize(t) for t, _, f in line]
x = margin + center(sum(w for w, _ in sizes), info_width)
y += max(h for _, h in sizes)
for (text, color_, font), (w, h) in zip(line, sizes):
canv.text((x, y - h), text, fill=color_, font=font)
x += w
y += inner
xy = ((margin, y), (margin + info_width, y))
canv.line(xy, fill='white', width=3) # border
y += inner
# draw tiles
tiles = data['tiles']
if tiles:
# TODO: wrap tiles over multiple rows
size = 40
while size * len(tiles) > info_width:
size -= 1
x = margin + center(size * len(tiles), info_width)
y += center(size, height - margin - y)
for tile in tiles:
tile = Image.open(f'{DIR}/tiles/{tile}.png').resize(
(size, size))
base.alpha_composite(tile, dest=(x, y))
x += size
# draw ranks
ranks = data['ranks']
if ranks:
font = font_24
def humanize_time(time):
return '%02d:%05.2f' % divmod(abs(time), 60)
time_w, _ = font.getsize(
humanize_time(max(r['time'] for r in ranks)))
rank_w, _ = font.getsize(f'#{max(r["rank"] for r in ranks)}')
_, h = font.getsize('yA')
y = margin + name_height + inner
space = (height - margin - y - h * 10) / 11
for player, rank, time in ranks:
y += space
x = margin + info_width + inner * 2
canv.text((x, y), f'#{rank}', fill='white', font=font)
x += rank_w + inner
x += time_w
text = humanize_time(time)
w, _ = font.getsize(text)
canv.text((x - w, y), text, fill=color, font=font)
x += inner
_, h_org = font.getsize(player)
font_player = auto_font(font, player, width - margin - x)
_, h_new = font_player.getsize(player)
canv.text((x, y - center(h_org - h_new)),
player,
fill='white',
font=font_player)
y += h
return save(base.convert('RGB'))
def generate_profile_image(self, data: asyncpg.Record) -> BytesIO:
font_normal = ImageFont.truetype(f'{DIR}/fonts/normal.ttf', 24)
font_bold = ImageFont.truetype(f'{DIR}/fonts/bold.ttf', 34)
font_big = ImageFont.truetype(f'{DIR}/fonts/bold.ttf', 48)
now = datetime.utcnow()
if data['day'] == now.day and data['month'] == now.month:
img = 'birthday'
color = (54, 70, 137)
else:
thresholds = {
18000: ('justice_2', (184, 81, 50)),
16000: ('back_in_the_days_3', (156, 162, 142)),
14000: ('heartcore', (86, 79, 81)),
12000: ('aurora', (55, 103, 156)),
10000: ('narcissistic', (122, 32, 43)),
9000: ('aim_10', (93, 128, 144)),
8000: ('barren', (196, 172, 140)),
7000: ('back_in_time', (148, 156, 161)),
6000: ('nostalgia', (161, 140, 148)),
5000: ('sweet_shot', (229, 148, 166)),
4000: ('chained', (183, 188, 198)),
3000: ('intothenight', (60, 76, 89)),
2000: ('darkvine', (145, 148, 177)),
1000: ('crimson_woods', (108, 12, 12)),
1: ('kobra_4', (148, 167, 75)),
0: ('stronghold', (156, 188, 220)),
}
img, color = next(e for t, e in thresholds.items()
if data['total_points'] >= t)
base = Image.open(f'{DIR}/profile_backgrounds/{img}.png')
canv = ImageDraw.Draw(base)
width, height = base.size
outer = 32
inner = int(outer / 2)
margin = outer + inner
# draw bg
size = (width - outer * 2, height - outer * 2)
bg = round_rectangle(size, 12, color=(0, 0, 0, 150))
base.alpha_composite(bg, dest=(outer, outer))
# draw name
try:
flag = Image.open(f'{DIR}/flags/{data["country"]}.png')
except FileNotFoundError:
flag = Image.open(f'{DIR}/flags/UNK.png')
flag_w, flag_h = flag.size
name = ' ' + data['name']
w, _ = font_bold.getsize(name)
_, h = font_bold.getsize('yA') # hardcoded to align names
name_height = 50
radius = int(name_height / 2)
size = (flag_w + w + radius * 2, name_height)
name_bg = round_rectangle(size, radius, color=(150, 150, 150, 75))
base.alpha_composite(name_bg, dest=(margin, margin))
x = margin + radius
dest = (x, margin + center(flag_h, name_height))
base.alpha_composite(flag, dest=dest)
xy = (x + flag_w, margin + center(h, name_height))
canv.text(xy, name, fill='white', font=font_bold)
# draw points
points_width = (width - margin * 2) / 3
x = margin + points_width + inner
y = margin + name_height + inner
xy = ((x, y), (x, height - margin))
canv.line(xy, fill='white', width=3)
text = f'#{data["total_rank"]}'
w, h = font_big.getsize(text)
xy = (margin + center(w, points_width), y)
canv.text(xy, text, fill='white', font=font_big)
offset = h * 0.25 # true drawn height is only 3 / 4
text = str(data['total_points'])
w, h = font_bold.getsize(text)
suffix = plural(data['total_points'], ' point').upper()
w2, h2 = font_normal.getsize(suffix)
x = margin + center(w + w2, points_width)
y = height - margin - offset
canv.text((x, y - h), text, fill=color, font=font_bold)
canv.text((x + w, y - h2), suffix, fill=color, font=font_normal)
# draw ranks
types = {
'TEAM RANK ': (data['team_rank'], data['team_points']),
'RANK ': (data['solo_rank'], data['solo_points'])
}
_, h = font_bold.getsize('A')
yy = (margin + name_height + inner + h * 1.25,
height - margin - h * 0.5)
for (type_, (rank, points)), y in zip(types.items(), yy):
line = [(type_, 'white', font_normal)]
if rank is None:
line.append(('UNRANKED', (150, 150, 150), font_bold))
else:
line.extend((
(f'#{rank}', 'white', font_bold),
(' ', 'white', font_bold), # border placeholder
(str(points), color, font_bold),
(plural(points, ' point').upper(), color, font_normal),
))
x = width - margin
for text, color_, font in line[::-1]:
w, h = font.getsize(text)
x -= w # adjust x before drawing since we're drawing reverse
if text == ' ':
xy = ((x + w / 2, y - h * 0.75), (x + w / 2, y - 1)
) # fix line width overflow
canv.line(xy, fill=color_, width=1)
else:
canv.text((x, y - h), text, fill=color_, font=font)
return save(base.convert('RGB'))
def generate_points_image(
self, data: Dict[str, List[asyncpg.Record]]) -> BytesIO:
font_small = ImageFont.truetype(f'{DIR}/fonts/normal.ttf', 16)
color_light = (100, 100, 100)
color_dark = (50, 50, 50)
colors = (
'orange',
'red',
'forestgreen',
'dodgerblue',
'orangered',
'orchid',
'burlywood',
'darkcyan',
'royalblue',
'olive',
)
base = Image.open(f'{DIR}/points_background.png')
canv = ImageDraw.Draw(base)
width, height = base.size
margin = 50
plot_width = width - margin * 2
plot_height = height - margin * 2
end_date = datetime.utcnow().date()
is_leap = end_date.month == 2 and end_date.month == 29
start_date = min(r['timestamp'] for d in data.values() for r in d)
start_date = min(
start_date,
end_date.replace(year=end_date.year - 1,
day=end_date.day - is_leap))
total_points = max(sum(r['points'] for r in d) for d in data.values())
total_points = max(total_points, 1000)
days_mult = plot_width / (end_date - start_date).days
points_mult = plot_height / total_points
# draw area bg
bg = Image.new('RGBA', (plot_width, plot_height), color=(0, 0, 0, 100))
base.alpha_composite(bg, dest=(margin, margin))
# draw years
prev_x = margin
for year in range(start_date.year, end_date.year + 2):
date = datetime(year=year, month=1, day=1).date()
if date < start_date:
continue
if date > end_date:
x = width - margin
else:
x = margin + (date - start_date).days * days_mult
xy = ((x, margin), (x, height - margin))
canv.line(xy, fill=color_dark, width=1)
text = str(year - 1)
w, h = font_small.getsize(text)
area_width = x - prev_x
if w <= area_width:
xy = (prev_x + center(w, area_width), height - margin + h)
canv.text(xy, text, fill=color_light, font=font_small)
prev_x = x
# draw points
thresholds = {
15000: 5000,
10000: 2500,
5000: 2000,
3000: 1000,
1000: 500,
0: 250,
}
steps = next(s for t, s in thresholds.items() if total_points > t)
w, _ = font_small.getsize('00.0K') # max points label width
points_margin = center(w, margin)
for points in range(0, total_points + 1, int(steps / 5)):
y = height - margin - points * points_mult
xy = ((margin, y), (width - margin - 1, y))
if points % steps == 0:
canv.line(xy, fill=color_light, width=2)
text = humanize_points(points)
w, h = font_small.getsize(text)
xy = (margin - points_margin - w, y + center(h))
canv.text(xy, text, fill=color_light, font=font_small)
else:
canv.line(xy, fill=color_dark, width=1)
# draw players
extra = 2
size = (plot_width * 2, (plot_height + extra * 2) * 2)
plot = Image.new('RGBA', size, color=(0, 0, 0, 0))
plot_canv = ImageDraw.Draw(plot)
labels = []
for dates, color in reversed(list(zip(data.values(), colors))):
x = 0
y = (plot_height + extra) * 2
xy = [(x, y)]
prev_date = start_date
for date, points in dates:
delta = (date - prev_date).days * days_mult * 2
x += delta
if delta / (plot_width * 2) > 0.1:
xy.append((x, y))
y -= points * points_mult * 2
xy.append((x, y))
prev_date = date
if prev_date != end_date:
xy.append((plot_width * 2, y))
plot_canv.line(xy, fill=color, width=6)
labels.append((margin - extra + y / 2, color))
size = (plot_width, plot_height + extra * 2)
plot = plot.resize(size, resample=Image.LANCZOS,
reducing_gap=1.0) # antialiasing
base.alpha_composite(plot, dest=(margin, margin - extra))
# remove overlapping labels TODO: optimize
_, h = font_small.getsize('0')
offset = center(h)
for _ in range(len(labels)):
labels.sort()
for i, (y1, _) in enumerate(labels):
if i == len(labels) - 1:
break
y2 = labels[i + 1][0]
if y1 - offset >= y2 + offset and y2 - offset >= y1 + offset:
labels[i] = ((y1 + y2) / 2, 'white')
del labels[i + 1]
# draw player points
for y, color in labels:
points = int((height - margin - y) / points_mult)
text = humanize_points(points)
xy = (width - margin + points_margin, y + offset)
canv.text(xy, text, fill=color, font=font_small)
# draw header
def check(w: int, size: int) -> float:
return w + (size / 3) * (4 * len(data) - 2)
font = auto_font((f'{DIR}/fonts/normal.ttf', 24),
''.join(data),
plot_width,
check=check)
space = font.size / 3
x = margin
for player, color in zip(data, colors):
y = center(space, margin)
xy = ((x, y), (x + space, y + space))
canv.rectangle(xy, fill=color)
x += space * 2
w, _ = font.getsize(player)
_, h = font.getsize(
'yA') # max name height, needs to be hardcoded to align names
xy = (x, center(h, margin))
canv.text(xy, player, fill='white', font=font)
x += w + space * 2
return save(base.convert('RGB'))