83 lines
2.6 KiB
Python
83 lines
2.6 KiB
Python
# -*- coding: utf-8 -*-
|
||
"""
|
||
批量转换：把 catalog\step\ 下所有 STEP 转成 glTF/GLB（网页 3D 轻量化用）。
|
||
|
||
用法：
|
||
python batch-convert-glb.py <step目录> <gltf输出目录> [deflection]
|
||
|
||
说明：
|
||
- 走本地 OCCT 服务的 /convert 接口（需先启动 occt_service.py）
|
||
- 同名 .step → .glb，断点续传（已存在且非空的 .glb 跳过）
|
||
- 单线程、逐文件，避免给服务造成压力
|
||
- 默认 deflection=0.5（屏幕预览够用）
|
||
"""
|
||
import os
|
||
import sys
|
||
import time
|
||
import urllib.request
|
||
|
||
OCCT_URL = os.environ.get("OCCT_URL", "http://localhost:8090")
|
||
|
||
|
||
def convert_one(step_path, glb_path, deflection):
|
||
with open(step_path, "rb") as f:
|
||
data = f.read()
|
||
req = urllib.request.Request(
|
||
"%s/convert?deflection=%s" % (OCCT_URL, deflection),
|
||
data=data, method="POST",
|
||
headers={"Content-Type": "application/octet-stream"},
|
||
)
|
||
resp = urllib.request.urlopen(req, timeout=120)
|
||
out = resp.read()
|
||
if not out or out[:4] != b"glTF":
|
||
raise RuntimeError("转换失败(非法 glb): %s" % step_path)
|
||
# 原子写：先写临时再改名
|
||
tmp = glb_path + ".tmp"
|
||
with open(tmp, "wb") as f:
|
||
f.write(out)
|
||
os.replace(tmp, glb_path)
|
||
|
||
|
||
def main():
|
||
if len(sys.argv) < 3:
|
||
print("用法: python batch-convert-glb.py <step目录> <gltf输出目录> [deflection]")
|
||
sys.exit(1)
|
||
step_dir = sys.argv[1]
|
||
out_dir = sys.argv[2]
|
||
deflection = float(sys.argv[3]) if len(sys.argv) > 3 else 0.5
|
||
os.makedirs(out_dir, exist_ok=True)
|
||
|
||
steps = []
|
||
for root, _, files in os.walk(step_dir):
|
||
for fn in files:
|
||
if fn.lower().endswith((".step", ".stp")):
|
||
steps.append(os.path.join(root, fn))
|
||
steps.sort()
|
||
|
||
ok = 0
|
||
skip = 0
|
||
fail = 0
|
||
t0 = time.time()
|
||
for i, sp in enumerate(steps):
|
||
base = os.path.splitext(os.path.basename(sp))[0]
|
||
gp = os.path.join(out_dir, base + ".glb")
|
||
if os.path.exists(gp) and os.path.getsize(gp) > 0:
|
||
skip += 1
|
||
continue
|
||
try:
|
||
convert_one(sp, gp, deflection)
|
||
ok += 1
|
||
if (ok + skip) % 50 == 0:
|
||
print("[%d/%d] ok=%d skip=%d fail=%d %.1fs" % (i + 1, len(steps), ok, skip, fail, time.time() - t0))
|
||
except Exception as e:
|
||
fail += 1
|
||
print("[FAIL] %s: %s" % (os.path.basename(sp), e))
|
||
|
||
print("完成: 总 %d, 转换 %d, 跳过 %d, 失败 %d, 耗时 %.1fs" % (len(steps), ok, skip, fail, time.time() - t0))
|
||
if fail:
|
||
sys.exit(2)
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|