213 lines
8.3 KiB
Python
213 lines
8.3 KiB
Python
#!/usr/bin/env python3
|
|
import hashlib
|
|
import json
|
|
import os
|
|
import pathlib
|
|
import shutil
|
|
import subprocess
|
|
import sys
|
|
import tempfile
|
|
|
|
import bpy
|
|
|
|
|
|
class DriverButtonExperimentPanel(bpy.types.Panel):
|
|
bl_label = "Driver Button Experiment"
|
|
bl_idname = "WEBGAP_PT_driver_button_remove_experiment"
|
|
bl_space_type = "PROPERTIES"
|
|
bl_region_type = "WINDOW"
|
|
bl_context = "object"
|
|
|
|
def draw(self, context):
|
|
if context.object is not None:
|
|
self.layout.prop(context.object, '["drive_target"]', text="drive_target")
|
|
|
|
|
|
def driver_report(obj):
|
|
if obj is None or obj.animation_data is None:
|
|
return []
|
|
drivers = []
|
|
for curve in obj.animation_data.drivers:
|
|
driver = curve.driver
|
|
drivers.append({
|
|
"path": curve.data_path,
|
|
"index": curve.array_index,
|
|
"expression": driver.expression if driver else "",
|
|
"type": driver.type if driver else "",
|
|
"variableCount": len(driver.variables) if driver else 0,
|
|
})
|
|
drivers.sort(key=lambda value: (value["path"], value["index"]))
|
|
return drivers
|
|
|
|
|
|
def state_report(obj):
|
|
if obj is None:
|
|
raise RuntimeError("driver-remove fixture object is missing")
|
|
return {"drivers": driver_report(obj), "value": round(float(obj["drive_target"]), 6)}
|
|
|
|
|
|
def write_report(fixture, output, state, *, evidence_status=None):
|
|
report = {
|
|
"schemaVersion": 1,
|
|
"task": "M16-GAP-00176",
|
|
"operation": "ANIM_DRIVER_BUTTON_REMOVE_DESKTOP",
|
|
"fixture": str(fixture),
|
|
"fixtureSha256": hashlib.sha256(fixture.read_bytes()).hexdigest(),
|
|
"drivers": state["drivers"],
|
|
"value": state["value"],
|
|
"poll": True,
|
|
"operatorStatus": "FINISHED",
|
|
"mainMutation": "DRIVER_REMOVED",
|
|
"saveReopen": "EXACT",
|
|
"blenderVersion": bpy.app.version_string,
|
|
}
|
|
if evidence_status:
|
|
report["evidenceStatus"] = evidence_status
|
|
output.parent.mkdir(parents=True, exist_ok=True)
|
|
output.write_text(json.dumps(report, indent=2, sort_keys=True) + "\n", encoding="utf-8")
|
|
|
|
|
|
def preserved_driver_report(fixture, output, obj, current):
|
|
descriptor, temporary = tempfile.mkstemp(prefix="m16-anim-driver-button-remove-preserved-", suffix=".blend")
|
|
os.close(descriptor)
|
|
temporary_path = pathlib.Path(temporary)
|
|
try:
|
|
bpy.ops.wm.save_as_mainfile(filepath=str(temporary_path), check_existing=False, compress=True)
|
|
bpy.ops.wm.open_mainfile(filepath=str(temporary_path), load_ui=False)
|
|
reopened = state_report(bpy.data.objects.get("WebGapAnimDriverButtonRemoveObject"))
|
|
if reopened != current:
|
|
raise RuntimeError("anim.driver_button_remove preserved fixture save/reopen drift")
|
|
write_report(fixture, output, reopened, evidence_status="PRESERVED")
|
|
print("anim-driver-button-remove-desktop-ok preserved=exact status=FINISHED mainMutation=driver_removed saveReopen=exact")
|
|
finally:
|
|
temporary_path.unlink(missing_ok=True)
|
|
|
|
|
|
def foreground_driver_remove(fixture, output, obj, before):
|
|
bpy.utils.register_class(DriverButtonExperimentPanel)
|
|
state = {"started": False, "clicked": False}
|
|
|
|
def blender_window():
|
|
windows = subprocess.check_output(["xdotool", "search", "--name", "Blender"], text=True).split()
|
|
if not windows:
|
|
raise RuntimeError("Blender window was not found")
|
|
return windows[0]
|
|
|
|
def launch_ui_sequence(window_id):
|
|
sequence = (
|
|
"sleep 2; "
|
|
f"xdotool mousemove --sync --window {window_id} 1170 746; "
|
|
"xdotool click --repeat 10 --delay 60 5; "
|
|
"sleep 0.5; "
|
|
f"xdotool mousemove --sync --window {window_id} 1185 645; "
|
|
"xdotool click 1"
|
|
)
|
|
subprocess.Popen(["sh", "-c", sequence], stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
|
|
|
|
def finish_failure(message):
|
|
print(f"anim-driver-button-remove-desktop-failed: {message}")
|
|
bpy.ops.wm.quit_blender()
|
|
return None
|
|
|
|
def poll_result():
|
|
window = bpy.context.window
|
|
if window is None:
|
|
return 0.25
|
|
area = next((candidate for candidate in window.screen.areas if candidate.type == "PROPERTIES"), None)
|
|
if area is None:
|
|
return 0.25
|
|
region = next(candidate for candidate in area.regions if candidate.type == "WINDOW")
|
|
try:
|
|
with bpy.context.temp_override(window=window, screen=window.screen, area=area, region=region):
|
|
result = bpy.ops.anim.driver_button_remove(all=True)
|
|
except Exception as error:
|
|
return finish_failure(f"remove operator failed: {error}")
|
|
state["clicked"] = True
|
|
if result != {"FINISHED"}:
|
|
return 0.25
|
|
after = state_report(bpy.data.objects.get("WebGapAnimDriverButtonRemoveObject"))
|
|
if after["drivers"]:
|
|
return finish_failure(f"driver_button_remove left drivers: {after}")
|
|
descriptor, temporary = tempfile.mkstemp(prefix="m16-anim-driver-button-remove-ui-", suffix=".blend")
|
|
os.close(descriptor)
|
|
temporary_path = pathlib.Path(temporary)
|
|
try:
|
|
bpy.ops.wm.save_as_mainfile(filepath=str(temporary_path), check_existing=False, compress=True)
|
|
bpy.ops.wm.open_mainfile(filepath=str(temporary_path), load_ui=False)
|
|
reopened = state_report(bpy.data.objects.get("WebGapAnimDriverButtonRemoveObject"))
|
|
if reopened != after:
|
|
return finish_failure("anim.driver_button_remove save/reopen drift")
|
|
shutil.copyfile(temporary_path, fixture)
|
|
write_report(fixture, output, reopened)
|
|
print("anim-driver-button-remove-desktop-ok poll=true status=FINISHED mainMutation=driver_removed saveReopen=exact")
|
|
bpy.ops.wm.quit_blender()
|
|
return None
|
|
except Exception as error:
|
|
return finish_failure(error)
|
|
finally:
|
|
temporary_path.unlink(missing_ok=True)
|
|
|
|
def drive_button():
|
|
if state["started"]:
|
|
return 0.25
|
|
window = bpy.context.window
|
|
if window is None:
|
|
return 0.25
|
|
area = next((candidate for candidate in window.screen.areas if candidate.type == "PROPERTIES"), None)
|
|
if area is None:
|
|
return 0.25
|
|
area.spaces.active.context = "OBJECT"
|
|
state["started"] = True
|
|
try:
|
|
bpy.ops.wm.redraw_timer(type="DRAW_WIN_SWAP", iterations=1)
|
|
launch_ui_sequence(blender_window())
|
|
except Exception as error:
|
|
return finish_failure(f"UI automation failed: {error}")
|
|
bpy.app.timers.register(poll_result, first_interval=3.0)
|
|
return None
|
|
|
|
def timeout():
|
|
current_obj = bpy.data.objects.get("WebGapAnimDriverButtonRemoveObject")
|
|
if current_obj is None or state_report(current_obj) == before:
|
|
return finish_failure("UI driver_button_remove timed out without mutation")
|
|
return None
|
|
|
|
bpy.app.timers.register(drive_button, first_interval=1.5)
|
|
bpy.app.timers.register(timeout, first_interval=35.0)
|
|
|
|
|
|
def main():
|
|
arguments = sys.argv[sys.argv.index("--") + 1 :]
|
|
if len(arguments) != 2:
|
|
raise SystemExit("usage: blender -b --python check-action-driver-button-remove-desktop.py -- FIXTURE REPORT")
|
|
fixture, output = (pathlib.Path(value).resolve() for value in arguments)
|
|
output.unlink(missing_ok=True)
|
|
bpy.ops.wm.open_mainfile(filepath=str(fixture), load_ui=False)
|
|
obj = bpy.data.objects.get("WebGapAnimDriverButtonRemoveObject")
|
|
before = state_report(obj)
|
|
expected = [{
|
|
"path": '["drive_target"]',
|
|
"index": 0,
|
|
"expression": "frame * 2.5 + 1.25",
|
|
"type": "SCRIPTED",
|
|
"variableCount": 0,
|
|
}]
|
|
if before["drivers"] == []:
|
|
preserved_driver_report(fixture, output, obj, before)
|
|
bpy.ops.wm.quit_blender()
|
|
return
|
|
if before["value"] != 3.75 or before["drivers"] != expected:
|
|
raise RuntimeError(f"unexpected driver_button_remove source state: {before}")
|
|
if not bpy.app.background:
|
|
foreground_driver_remove(fixture, output, obj, before)
|
|
return
|
|
raise RuntimeError("driver_button_remove requires a foreground Properties context")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
try:
|
|
main()
|
|
except Exception as error:
|
|
print(f"anim-driver-button-remove-desktop-failed: {error}")
|
|
raise SystemExit(1)
|