Repository navigation
Expand file tree
/
Copy pathcapture_frame.py
More file actions
74 lines (57 loc) · 2.09 KB
/
Copy pathcapture_frame.py
File metadata and controls
74 lines (57 loc) · 2.09 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
"""Capture a single GPU frame using RenderDoc Python API.
Lighter version: shorter init pump, no refAllResources.
"""
import sys, os, time, shutil
sys.path.insert(0, 'D:/renderdoc/module')
import renderdoc as rd
cap_path = 'D:/renderdoc/captures/test_sphere_shadows.rdc'
exe = r'D:\shaderlang\playground_cpp\build\Release\lux-playground.exe'
args = '--scene sphere --pipeline D:/shaderlang/shadercache/pbr_basic --demo-lights --width 512 --height 512 --interactive'
workdir = r'D:\shaderlang'
opts = rd.CaptureOptions()
opts.apiValidation = False
opts.captureCallstacks = False
opts.refAllResources = False
env = []
print('Executing and injecting...')
result = rd.ExecuteAndInject(exe, workdir, args, env, cap_path, opts, False)
print(f'Ident: {result.ident}')
if result.ident == 0:
print('Failed to inject')
sys.exit(1)
print('Connecting to target...')
target = rd.CreateTargetControl('', result.ident, 'claude-code', True)
if target is None or not target.Connected():
print('Failed to connect')
sys.exit(1)
print('Connected. Brief init pump (3s)...')
for _ in range(30):
target.ReceiveMessage(None)
time.sleep(0.1)
print('Triggering capture...')
target.TriggerCapture(1)
print('Waiting for capture...')
cap_received = False
for i in range(100): # 10 seconds max
msg = target.ReceiveMessage(None)
t = msg.type
if t == rd.TargetControlMessageType.NewCapture:
cap = msg.newCapture
print(f' Capture received: path={cap.path}')
if cap.local and cap.path and cap.path != cap_path:
os.makedirs(os.path.dirname(cap_path), exist_ok=True)
shutil.copy2(cap.path, cap_path)
elif not cap.local:
target.CopyCapture(cap.captureId, cap_path)
cap_received = True
break
elif t == rd.TargetControlMessageType.CaptureProgress:
print(f' Progress: {msg.capProgress:.0%}')
time.sleep(0.1)
if not cap_received:
print('No capture after 10s')
target.Shutdown()
if os.path.exists(cap_path):
print(f'OK: {cap_path} ({os.path.getsize(cap_path):,} bytes)')
else:
print(f'FAILED: no capture file')