eKeerar
97 lines
5.1 KiB
Python
97 lines
5.1 KiB
Python
#!/usr/bin/env python
|
|
"""
|
|
reactions_layout_render.py — 반응 시퀀서용 레이아웃 사전계산 + 오프라인 합성 검증(캐릭터 무관).
|
|
사용: python reactions_layout_render.py <Profile_dir> <char_prefix>
|
|
출력: <Profile>/06_Reactions/_layout.json, _reaction_preview.png
|
|
브라우저 런타임(reactions.html)이 _layout.json 을 로드해 동일 합성.
|
|
|
|
스케일 기준: 바디는 '상단 25% 최대너비(=어깨/소매 span) -> TARGET_SHOULDER'(캐릭터별 네이티브 크기 차 흡수),
|
|
목 부착점(상단 opaque 중앙)을 무대 NECK에 고정. 머리는 'bbox 너비 -> HEAD_TARGET_W', 목(하단중앙)을 NECK에 정합·회전.
|
|
"""
|
|
import sys, os, json, math
|
|
import numpy as np
|
|
from PIL import Image
|
|
|
|
TH = 32
|
|
NECK_X, NECK_Y = 260, 250
|
|
TARGET_SHOULDER = 230 # 바디 상단 25% 최대너비를 이 값으로
|
|
HEAD_TARGET_W = 150 # 머리 bbox 너비를 이 값으로
|
|
OVERLAP = 6
|
|
|
|
def bbox(a):
|
|
ys, xs = np.nonzero(a > TH); return xs.min(), ys.min(), xs.max(), ys.max()
|
|
|
|
def main(profile, cpfx):
|
|
lib = os.path.join(profile, "03_Assets", "Library"); rdir = os.path.join(profile, "06_Reactions")
|
|
STAGE_W, STAGE_H = 520, 900
|
|
layout = {"stage": {"w": STAGE_W, "h": STAGE_H}, "neck": [NECK_X, NECK_Y], "overlap": OVERLAP,
|
|
"headTargetW": HEAD_TARGET_W, "bodies": {}, "heads": {}}
|
|
|
|
for root, _, files in os.walk(os.path.join(lib, "BakedPoses")):
|
|
for fn in files:
|
|
if not fn.endswith(".png"): continue
|
|
a = np.asarray(Image.open(os.path.join(root, fn)).convert("RGBA"))[:, :, 3]
|
|
x0, y0, x1, y1 = bbox(a); bh = y1 - y0 + 1
|
|
row0 = np.nonzero(a[y0] > TH)[0]; tcx = float((row0.min() + row0.max()) / 2)
|
|
shoulderW = 0 # 상단 25% 최대 행너비 = 어깨/소매 span
|
|
for yy in range(y0, y0 + max(3, int(bh * 0.25))):
|
|
cols = np.nonzero(a[yy] > TH)[0]
|
|
if len(cols): shoulderW = max(shoulderW, int(cols.max() - cols.min() + 1))
|
|
if shoulderW == 0: continue
|
|
scale = TARGET_SHOULDER / shoulderW
|
|
if not (0.1 <= scale <= 3.0): continue
|
|
layout["bodies"][fn[:-4]] = {"scale": round(scale, 4),
|
|
"ox": round(NECK_X - tcx * scale, 1), "oy": round(NECK_Y - y0 * scale, 1)}
|
|
|
|
for fn in os.listdir(os.path.join(lib, "Heads")):
|
|
if not fn.endswith(".png"): continue
|
|
im = Image.open(os.path.join(lib, "Heads", fn)).convert("RGBA"); a = np.asarray(im)[:, :, 3]
|
|
x0, y0, x1, y1 = bbox(a)
|
|
layout["heads"][fn[:-4]] = {"w": int(x1 - x0 + 1),
|
|
"neckNorm": [round(((x0 + x1) / 2) / im.width, 4), round(y1 / im.height, 4)]}
|
|
|
|
os.makedirs(rdir, exist_ok=True)
|
|
json.dump(layout, open(os.path.join(rdir, "_layout.json"), "w"), ensure_ascii=False, indent=1)
|
|
print(f"layout: {len(layout['bodies'])} bodies, {len(layout['heads'])} heads -> _layout.json")
|
|
|
|
# ---- offline verify (character-agnostic: pick by suffix) ----
|
|
T = lambda x, y: np.array([[1,0,x],[0,1,y],[0,0,1]], float)
|
|
def R(d):
|
|
r=math.radians(d);c,s=math.cos(r),math.sin(r);return np.array([[c,-s,0],[s,c,0],[0,0,1]],float)
|
|
Sc = lambda k: np.array([[k,0,0],[0,k,0],[0,0,1]], float)
|
|
def draw(base, img, M):
|
|
Mi=np.linalg.inv(M); a,c,e=Mi[0]; b,d,f=Mi[1]
|
|
return Image.alpha_composite(base, img.transform((STAGE_W,STAGE_H),Image.AFFINE,(a,c,e,b,d,f),resample=Image.BICUBIC))
|
|
def pick_body(suf):
|
|
for k in layout["bodies"]:
|
|
if k.endswith("_"+suf): return k
|
|
return None
|
|
def pick_head(expr):
|
|
for k in layout["heads"]:
|
|
if k.endswith("_"+expr): return k
|
|
return None
|
|
def bpath(k):
|
|
for root,_,files in os.walk(os.path.join(lib,"BakedPoses")):
|
|
if k+".png" in files: return os.path.join(root, k+".png")
|
|
def compose(bk, hk, rot):
|
|
bl=layout["bodies"][bk]; hl=layout["heads"][hk]
|
|
body=Image.open(bpath(bk)).convert("RGBA"); head=Image.open(os.path.join(lib,"Heads",hk+".png")).convert("RGBA")
|
|
fr=Image.new("RGBA",(STAGE_W,STAGE_H),(0,0,0,0))
|
|
fr=draw(fr, body, T(bl["ox"],bl["oy"]) @ Sc(bl["scale"]))
|
|
hs=HEAD_TARGET_W/hl["w"]; ax=hl["neckNorm"][0]*head.width; ay=hl["neckNorm"][1]*head.height
|
|
fr=draw(fr, head, T(NECK_X,NECK_Y+OVERLAP) @ R(rot) @ Sc(hs) @ T(-ax,-ay))
|
|
return fr
|
|
reactions=[("no","armscross","negative",7),("heart","heart","love",-4),("focus","listen","sleepy",4)]
|
|
sc=0.46; fw,fh=int(STAGE_W*sc),int(STAGE_H*sc)
|
|
mont=Image.new("RGBA",(fw*len(reactions),fh),(28,32,40,255))
|
|
for i,(nm,bsuf,expr,rot) in enumerate(reactions):
|
|
bk=pick_body(bsuf); hk=pick_head(expr)
|
|
if not bk or not hk: print(f" skip {nm}: body={bk} head={hk}"); continue
|
|
try:
|
|
mont.alpha_composite(compose(bk,hk,rot).resize((fw,fh),Image.BICUBIC),(fw*i,0)); print(f" {nm}: {bk} + {hk} rot={rot}")
|
|
except Exception as ex: print(f" FAIL {nm}: {ex}")
|
|
out=os.path.join(rdir,"_reaction_preview.png"); mont.convert("RGB").save(out); print("saved:", out)
|
|
|
|
if __name__ == "__main__":
|
|
main(sys.argv[1], sys.argv[2])
|