85 lines
3.9 KiB
Python
85 lines
3.9 KiB
Python
"""LittleBear: Hunyuan high -> Studio low/UV -> bake -> UV-preserving reduction.
|
|
|
|
Run with the LLM3DGraphics virtualenv. Only this character's output is written.
|
|
"""
|
|
from __future__ import annotations
|
|
import json
|
|
import shutil
|
|
import subprocess
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
ROOT = Path(__file__).resolve().parents[1]
|
|
STUDIO = Path(r"D:\UD\AI\XModel\LLM3DGraphics")
|
|
BLENDER = Path(r"D:\Tools\Blender\blender-5.0.0-windows-x64\blender.exe")
|
|
IMAGE = ROOT / "Doc/Design/Character/LittleBear/littlebear_tpose.png"
|
|
HIGH = ROOT / "Tools/output/littlebear_3d/littlebear.glb"
|
|
OUT = ROOT / "Tools/output/littlebear_studio"
|
|
NAME, TARGET, SIZE = "LittleBear", 3000, 1024
|
|
|
|
|
|
def update(**data):
|
|
print(json.dumps(data, ensure_ascii=False), flush=True)
|
|
|
|
|
|
def blender(script, args, logfile):
|
|
with (OUT / logfile).open("w", encoding="utf-8") as log:
|
|
subprocess.run([str(BLENDER), "-b", "-t", "2", "--factory-startup", "--python",
|
|
str(ROOT / "Tools/model_tools" / script), "--", *map(str, args)],
|
|
stdout=log, stderr=subprocess.STDOUT, check=True, timeout=1800)
|
|
|
|
|
|
def main():
|
|
sys.path.insert(0, str(ROOT / "Tools"))
|
|
sys.path.insert(0, str(STUDIO))
|
|
from hunyuan_3d import auth
|
|
from hunyuan_3d.glb_check import assert_glb_textures_ok
|
|
from backend import hunyuan_studio as studio
|
|
from backend.retopology import _run_blender_step, compress_bake_maps, limit_studio_faces
|
|
from model_tools.png_compress import compress_dir_pngs
|
|
|
|
if not HIGH.is_file():
|
|
raise SystemExit("Generate the Hunyuan 50k high mesh first: " + str(HIGH))
|
|
assert_glb_textures_ok(HIGH)
|
|
OUT.mkdir(parents=True, exist_ok=True)
|
|
shutil.copy2(HIGH, OUT / "model.glb")
|
|
studio.is_logged_in = lambda client_id=None: auth.is_logged_in()
|
|
studio.load_storage_state = lambda client_id=None: (
|
|
json.loads(auth.storage_state_path().read_text(encoding="utf-8")), None)
|
|
uv = OUT / "model_lowpoly_uv.glb"
|
|
if uv.is_file():
|
|
studio.validate_uv_glb(uv)
|
|
else:
|
|
path = OUT / "hunyuan_studio.json"
|
|
meta = json.loads(path.read_text(encoding="utf-8")) if path.is_file() else {}
|
|
studio.process_existing(HIGH, OUT, update, target_faces=TARGET,
|
|
resume_low_id=meta.get("lowpoly_works_id"),
|
|
resume_uv_id=meta.get("uv_works_id"))
|
|
aligned = OUT / "model_lowpoly_uv_aligned.glb"
|
|
if not aligned.is_file():
|
|
blender("bl_align_studio_uv.py", [OUT], "alignment.log")
|
|
baked = OUT / "model_baked.glb"
|
|
if not baked.is_file():
|
|
_run_blender_step(OUT, str(BLENDER), "blender_lowpoly_bake.py",
|
|
(OUT / "model.glb", baked, 1, SIZE, aligned),
|
|
"bake.log", marker="Lowpoly bake saved to")
|
|
compress_bake_maps(OUT)
|
|
shutil.copy2(baked, OUT / "model_ex.glb")
|
|
mesh, decimation = limit_studio_faces(OUT, TARGET, str(BLENDER), update)
|
|
blender("bl_prepare_studio_trial.py", [OUT, NAME], "static_export.log")
|
|
blender("bl_convert.py", [OUT / "static" / (NAME + ".fbx"), OUT / "xpbr", SIZE], "xpbr.log")
|
|
compress_dir_pngs(str(OUT / "xpbr"), suffixes=("_base.png", "_mix.png"))
|
|
report = dict(character=NAME, image=str(IMAGE.relative_to(ROOT)),
|
|
source=str(HIGH.relative_to(ROOT)), targetTriangles=TARGET,
|
|
triangles=len(mesh.faces), vertices=len(mesh.vertices),
|
|
textureSize=SIZE, decimation=decimation,
|
|
staticFbx=str((OUT / "static" / (NAME + ".fbx")).relative_to(ROOT)),
|
|
rig=False, skinningPolicy="fixed Mixamo skeleton + TokenRig weights only",
|
|
fbxAxes=json.loads((OUT / "static_export.json").read_text(encoding="utf-8"))["fbxAxes"])
|
|
(OUT / "mesh_receipt.json").write_text(json.dumps(report, ensure_ascii=False, indent=2), encoding="utf-8")
|
|
update(**report)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|