Files
RIT-nexus/generate_pwa_icons.py

113 lines
3.3 KiB
Python

import zlib
import struct
import math
def make_png(width, height, draw_func):
# PNG signature
png = bytearray(b'\x89PNG\r\n\x1a\n')
# IHDR chunk
ihdr_data = struct.pack('!IIBBBBB', width, height, 8, 6, 0, 0, 0)
ihdr_crc = zlib.crc32(b'IHDR' + ihdr_data)
png.extend(struct.pack('!I', len(ihdr_data)))
png.extend(b'IHDR')
png.extend(ihdr_data)
png.extend(struct.pack('!I', ihdr_crc))
# IDAT chunk
raw_data = bytearray()
for y in range(height):
raw_data.append(0) # Filter type 0 (None)
for x in range(width):
r, g, b, a = draw_func(x, y, width, height)
raw_data.extend([r, g, b, a])
compressed = zlib.compress(raw_data, 9)
idat_crc = zlib.crc32(b'IDAT' + compressed)
png.extend(struct.pack('!I', len(compressed)))
png.extend(b'IDAT')
png.extend(compressed)
png.extend(struct.pack('!I', idat_crc))
# IEND chunk
iend_crc = zlib.crc32(b'IEND')
png.extend(struct.pack('!I', 0))
png.extend(b'IEND')
png.extend(struct.pack('!I', iend_crc))
return bytes(png)
def icon_drawer(x, y, w, h):
# Orange gradient background (#F97316 to #EA580C)
t = y / h
r_bg = int(249 * (1 - t) + 234 * t)
g_bg = int(115 * (1 - t) + 88 * t)
b_bg = int(22 * (1 - t) + 12 * t)
cx, cy = w / 2, h / 2
dx = x - cx
dy = y - cy
dist = math.sqrt(dx*dx + dy*dy)
# Outer rounded rectangle mask
corner_radius = w * 0.22
in_box = True
if abs(dx) > (w/2 - corner_radius) and abs(dy) > (h/2 - corner_radius):
cdx = abs(dx) - (w/2 - corner_radius)
cdy = abs(dy) - (h/2 - corner_radius)
if cdx*cdx + cdy*cdy > corner_radius*corner_radius:
in_box = False
if not in_box:
return 0, 0, 0, 0
# Draw White Cap / Rocket 'R' symbol in center
# Top diamond (Cap)
cap_y = cy - h * 0.1
if abs(dy + h*0.08) < h*0.18 and abs(dx) < (h*0.18 - abs(dy + h*0.08))*1.8:
return 255, 255, 255, 255
# Cap base pillar
if abs(dx) < w*0.08 and cy < y < cy + h*0.18:
return 255, 255, 255, 255
# Cap tassel line
if w*0.18 < dx < w*0.25 and cy - h*0.05 < y < cy + h*0.12:
return 255, 255, 255, 255
return r_bg, g_bg, b_bg, 255
def maskable_drawer(x, y, w, h):
# Full bleed background for maskable icon
t = y / h
r_bg = int(249 * (1 - t) + 234 * t)
g_bg = int(115 * (1 - t) + 88 * t)
b_bg = int(22 * (1 - t) + 12 * t)
cx, cy = w / 2, h / 2
dx = x - cx
dy = y - cy
# Draw White Cap in center (slightly scaled inside safe zone)
if abs(dy + h*0.06) < h*0.14 and abs(dx) < (h*0.14 - abs(dy + h*0.06))*1.6:
return 255, 255, 255, 255
if abs(dx) < w*0.06 and cy < y < cy + h*0.14:
return 255, 255, 255, 255
return r_bg, g_bg, b_bg, 255
with open('public/pwa-192x192.png', 'wb') as f:
f.write(make_png(192, 192, icon_drawer))
with open('public/pwa-512x512.png', 'wb') as f:
f.write(make_png(512, 512, icon_drawer))
with open('public/apple-touch-icon.png', 'wb') as f:
f.write(make_png(180, 180, icon_drawer))
with open('public/pwa-maskable-512x512.png', 'wb') as f:
f.write(make_png(512, 512, maskable_drawer))
print("PWA icons generated successfully!")