Files
Robot_Arm/src/capture_p07.py
T

32 lines
1.4 KiB
Python
Raw Normal View History

import os
import subprocess
import glob
CAND = [r"C:\Program Files (x86)\Microsoft\Edge\Application\msedge.exe",
r"C:\Program Files\Microsoft\Edge\Application\msedge.exe"]
EDGE = next((c for c in CAND if os.path.exists(c)), None)
if not EDGE:
EDGE = glob.glob(r"C:\Users\*\AppData\Local\Microsoft\Edge\*\msedge.exe")[0]
HTML = r"G:/Robot_v2/机器人手臂/web/机器人手臂3D查看器.html"
ART = r"C:/Users/cbb/.gemini/antigravity/brain/5b3fdbf8-960a-428d-b114-f1ec3e9a8af3"
shots = [
("shot_p07_az0", "?az=0&po=1.5708&rd=220&tx=18.5&ty=250&tz=-140&xray=0"),
("shot_p07_az90", "?az=1.5708&po=1.5708&rd=220&tx=18.5&ty=250&tz=-140&xray=0"),
("shot_p07_az180", "?az=3.14159&po=1.5708&rd=220&tx=18.5&ty=250&tz=-140&xray=0"),
("shot_p07_azneg90", "?az=-1.5708&po=1.5708&rd=220&tx=18.5&ty=250&tz=-140&xray=0"),
]
for nm, q in shots:
out = os.path.join(ART, f"{nm}.png")
target = f"file:///{HTML}{q}&still=1"
cmd = [EDGE, "--headless=new", "--use-gl=angle", "--use-angle=d3d11",
"--disable-cache", "--disable-application-cache", "--disk-cache-size=0",
"--enable-gpu-rasterization", "--ignore-gpu-blocklist",
"--hide-scrollbars", "--force-device-scale-factor=1",
"--window-size=1200,900", "--virtual-time-budget=5000",
f"--screenshot={out}", target]
subprocess.run(cmd, capture_output=True, timeout=30)
print(f"{nm}: size={os.path.getsize(out) if os.path.exists(out) else 0}")