89 lines
4.6 KiB
Python
89 lines
4.6 KiB
Python
#!/usr/bin/env python3
|
|
"""Top-down vehicle renders -> 16-facing rotation sheets.
|
|
|
|
flux renders each vehicle facing NORTH from directly above; PIL rotates into 16
|
|
frames (OpenRA facing order: north first, counter-clockwise), packed into one
|
|
horizontal strip with FrameSize metadata. Geometrically correct, unlike rotating
|
|
a 3/4-view sprite. Also: terrain tile + wax crystal density sheet.
|
|
"""
|
|
import os, sys
|
|
sys.path.insert(0, os.path.dirname(__file__))
|
|
from mbgen import run, upload # noqa: E402
|
|
from PIL import Image # noqa: E402
|
|
from PIL.PngImagePlugin import PngInfo # noqa: E402
|
|
|
|
ROOT = os.path.join(os.path.dirname(__file__), '..')
|
|
RENDERS = os.path.join(ROOT, 'assets_src', 'renders_topdown')
|
|
ASSETS = os.path.join(ROOT, 'mods', 'vinylgod', 'sequences', 'assets')
|
|
os.makedirs(RENDERS, exist_ok=True)
|
|
|
|
MRP = 'rusty junkyard robot aesthetic, rust metal with hot pink trim, '
|
|
STR = 'glossy seamless white tech aesthetic, soft blue ring lights, '
|
|
SUF = ', viewed from directly above, top-down aerial view, vehicle pointing straight up toward the top of the image, single object centered, video game sprite, plain flat grey background, no text, no shadow'
|
|
|
|
VEHICLES = [
|
|
('digger', 26, MRP + 'forklift harvester truck carrying crates of vinyl records'),
|
|
('popupvan', 26, MRP + 'delivery truck covered in band stickers with roof loudspeaker'),
|
|
('spinner', 20, MRP + 'small round robot riding a single vinyl record wheel with blade arms'),
|
|
('stacktank', 30, MRP + 'heavy tracked tank made of stacked speaker cabinets with subwoofer turret'),
|
|
('basscannon', 30, MRP + 'long six-wheeled flatbed truck with one colossal subwoofer cannon'),
|
|
('streamvan', 26, STR + 'sleek white autonomous delivery van with blue play-button roof logo'),
|
|
('scraper', 24, STR + 'white autonomous street sweeper vehicle with glowing blue front intake'),
|
|
('drmwalker', 30, STR + 'white quad-legged mech with one blue ring light, seen from above'),
|
|
]
|
|
|
|
|
|
def sheet16(im, size, out):
|
|
im = im.crop(im.getbbox() or (0, 0, im.width, im.height))
|
|
im.thumbnail((size * 4, size * 4), Image.LANCZOS) # rotate at 4x, downscale after
|
|
strip = Image.new('RGBA', (size * 16, size), (0, 0, 0, 0))
|
|
for i in range(16):
|
|
fr = im.rotate(i * 22.5, expand=False, resample=Image.BICUBIC)
|
|
fr.thumbnail((size, size), Image.LANCZOS)
|
|
strip.paste(fr, (i * size + (size - fr.width) // 2, (size - fr.height) // 2), fr)
|
|
meta = PngInfo()
|
|
meta.add_text('FrameSize', f'{size},{size}')
|
|
strip.save(out, pnginfo=meta)
|
|
print(f'[facings] {out}', flush=True)
|
|
|
|
|
|
for name, size, prompt in VEHICLES:
|
|
raw = os.path.join(RENDERS, f'{name}_raw.png')
|
|
cut = os.path.join(RENDERS, f'{name}.png')
|
|
if not os.path.exists(cut):
|
|
run('flux_local', {'prompt': prompt + SUF}, raw)
|
|
run('bg_remove_local', {}, cut, asset_id=upload(raw))
|
|
sheet16(Image.open(cut).convert('RGBA'), size, os.path.join(ASSETS, f'{name}.png'))
|
|
|
|
# terrain tile: seamless ground texture -> 32x32
|
|
traw = os.path.join(RENDERS, 'terrain_raw.png')
|
|
if not os.path.exists(traw):
|
|
run('flux_local', {'prompt': 'seamless tileable texture, cracked dark asphalt with faint dust and tiny speckles, muted grey brown, flat even lighting, top down, no objects'}, traw)
|
|
t = Image.open(traw).convert('RGBA').resize((64, 64), Image.LANCZOS).crop((16, 16, 48, 48))
|
|
t.save(os.path.join(ROOT, 'mods', 'vinylgod', 'tilesets', 'vinylgod', 'vinylgod.png'))
|
|
print('[terrain] tile written', flush=True)
|
|
|
|
# wax crystals: one cluster render -> 12 density frames
|
|
wraw = os.path.join(RENDERS, 'wax_raw.png')
|
|
wcut = os.path.join(RENDERS, 'wax.png')
|
|
if not os.path.exists(wcut):
|
|
run('flux_local', {'prompt': 'cluster of jagged black vinyl crystal shards glowing hot pink from the cracks, viewed from directly above, top down, video game resource sprite, single cluster centered, plain flat grey background'}, wraw)
|
|
run('bg_remove_local', {}, wcut, asset_id=upload(wraw))
|
|
w = Image.open(wcut).convert('RGBA')
|
|
w = w.crop(w.getbbox() or (0, 0, w.width, w.height))
|
|
strip = Image.new('RGBA', (32 * 12, 32), (0, 0, 0, 0))
|
|
spots = [(16, 16), (7, 9), (24, 10), (9, 24), (24, 24), (16, 6), (5, 17), (27, 18), (16, 26), (11, 13), (21, 15), (16, 20)]
|
|
for f in range(12):
|
|
for i in range(f + 1):
|
|
s = 10 + (6 if i == 0 else 0)
|
|
c = w.copy()
|
|
c.thumbnail((s, s), Image.LANCZOS)
|
|
c = c.rotate(i * 47, expand=False)
|
|
x, y = spots[i]
|
|
strip.paste(c, (f * 32 + x - c.width // 2, y - c.height // 2), c)
|
|
meta = PngInfo()
|
|
meta.add_text('FrameSize', '32,32')
|
|
strip.save(os.path.join(ASSETS, 'wax.png'), pnginfo=meta)
|
|
print('[wax] density sheet written', flush=True)
|
|
print('ALL DONE')
|