feat: add CaptureWorker to automate CS2 cubemap generation and screenshot stitching

Author: dertwist

src/forms/cubemap_maker/worker.py

@@ -30,7 +30,15 @@ def load_local_faces(self, ss_dir, ev_suffix, session_id):
 
         for name in face_names:
             pattern = os.path.join(ss_dir, f"{session_id}_cube_{ev_suffix}_{name}*.png")
-            files = glob.glob(pattern)
+            
+            # Retry mechanism for file appearance (Source 2 can be slow to flush to disk)
+            files = []
+            for _ in range(15):  # Try for 3 seconds
+                files = glob.glob(pattern)
+                if files:
+                    break
+                time.sleep(0.2)
+
             if not files:
                 raise FileNotFoundError(f"Missing face: {name} for session {session_id}")
 
@@ -132,7 +140,7 @@ def run(self):
             time.sleep(0.5)
 
             # Build command list to avoid semicolon issues in netcon
-            t = 0.1 # Tick interval
+            t = 0.2 # Increased tick interval for stability
             cmds = [
                 "sv_cheats 1", "noclip 1", "r_drawviewmodel 0", "cl_drawhud 0", "r_drawpanorama 0", "cl_firstperson_legs 0",
                 "fov_cs_debug 106.260205", "ent_fire cmd kill", "ent_create point_servercommand {targetname cmd}",
@@ -141,34 +149,34 @@ def run(self):
                 # Forward
                 f'ent_fire worldent addoutput "OnUser1>cmd>command>setpos_exact {x} {y} {z}>{1*t}>1"',
                 f'ent_fire worldent addoutput "OnUser1>cmd>command>setang_exact 1 0 0>{1*t}>1"',
-                f'ent_fire worldent addoutput "OnUser1>cmd>command>screenshot_prefix {session_id}_cube_{ev_suffix}_forward>{1*t + 0.01}>1"',
-                f'ent_fire worldent addoutput "OnUser1>cmd>command>png_screenshot>{1*t + 0.02}>1"',
+                f'ent_fire worldent addoutput "OnUser1>cmd>command>screenshot_prefix {session_id}_cube_{ev_suffix}_forward>{1*t + 0.05}>1"',
+                f'ent_fire worldent addoutput "OnUser1>cmd>command>png_screenshot>{1*t + 0.1}>1"',
                 # Right
                 f'ent_fire worldent addoutput "OnUser1>cmd>command>setpos_exact {x} {y} {z}>{2*t}>1"',
                 f'ent_fire worldent addoutput "OnUser1>cmd>command>setang_exact 0 270 -1>{2*t}>1"',
-                f'ent_fire worldent addoutput "OnUser1>cmd>command>screenshot_prefix {session_id}_cube_{ev_suffix}_right>{2*t + 0.01}>1"',
-                f'ent_fire worldent addoutput "OnUser1>cmd>command>png_screenshot>{2*t + 0.02}>1"',
+                f'ent_fire worldent addoutput "OnUser1>cmd>command>screenshot_prefix {session_id}_cube_{ev_suffix}_right>{2*t + 0.05}>1"',
+                f'ent_fire worldent addoutput "OnUser1>cmd>command>png_screenshot>{2*t + 0.1}>1"',
                 # Back
                 f'ent_fire worldent addoutput "OnUser1>cmd>command>setpos_exact {x} {y} {z}>{3*t}>1"',
                 f'ent_fire worldent addoutput "OnUser1>cmd>command>setang_exact -1 180 0>{3*t}>1"',
-                f'ent_fire worldent addoutput "OnUser1>cmd>command>screenshot_prefix {session_id}_cube_{ev_suffix}_back>{3*t + 0.01}>1"',
-                f'ent_fire worldent addoutput "OnUser1>cmd>command>png_screenshot>{3*t + 0.02}>1"',
+                f'ent_fire worldent addoutput "OnUser1>cmd>command>screenshot_prefix {session_id}_cube_{ev_suffix}_back>{3*t + 0.05}>1"',
+                f'ent_fire worldent addoutput "OnUser1>cmd>command>png_screenshot>{3*t + 0.1}>1"',
                 # Left
                 f'ent_fire worldent addoutput "OnUser1>cmd>command>setpos_exact {x} {y} {z}>{4*t}>1"',
                 f'ent_fire worldent addoutput "OnUser1>cmd>command>setang_exact 0 90 1>{4*t}>1"',
-                f'ent_fire worldent addoutput "OnUser1>cmd>command>screenshot_prefix {session_id}_cube_{ev_suffix}_left>{4*t + 0.01}>1"',
-                f'ent_fire worldent addoutput "OnUser1>cmd>command>png_screenshot>{4*t + 0.02}>1"',
+                f'ent_fire worldent addoutput "OnUser1>cmd>command>screenshot_prefix {session_id}_cube_{ev_suffix}_left>{4*t + 0.05}>1"',
+                f'ent_fire worldent addoutput "OnUser1>cmd>command>png_screenshot>{4*t + 0.1}>1"',
                 # Up
                 f'ent_fire worldent addoutput "OnUser1>cmd>command>fov_cs_debug 106.260205>{5*t}>1"',
                 f'ent_fire worldent addoutput "OnUser1>cmd>command>setpos_exact {x} {y} {z}>{5*t}>1"',
                 f'ent_fire worldent addoutput "OnUser1>cmd>command>setang_exact -89 0 0>{5*t}>1"',
-                f'ent_fire worldent addoutput "OnUser1>cmd>command>screenshot_prefix {session_id}_cube_{ev_suffix}_up>{5*t + 0.01}>1"',
-                f'ent_fire worldent addoutput "OnUser1>cmd>command>png_screenshot>{5*t + 0.02}>1"',
+                f'ent_fire worldent addoutput "OnUser1>cmd>command>screenshot_prefix {session_id}_cube_{ev_suffix}_up>{5*t + 0.05}>1"',
+                f'ent_fire worldent addoutput "OnUser1>cmd>command>png_screenshot>{5*t + 0.1}>1"',
                 # Down
                 f'ent_fire worldent addoutput "OnUser1>cmd>command>setpos_exact {x} {y} {z}>{6*t}>1"',
-                f'ent_fire worldent addoutput "OnUser1>cmd>command>setang_exact 89 180 180>{6*t}>1"',
-                f'ent_fire worldent addoutput "OnUser1>cmd>command>screenshot_prefix {session_id}_cube_{ev_suffix}_down>{6*t + 0.01}>1"',
-                f'ent_fire worldent addoutput "OnUser1>cmd>command>png_screenshot>{6*t + 0.02}>1"',
+                f'ent_fire worldent addoutput "OnUser1>cmd>command>setang_exact 89 180 0>{6*t}>1"',
+                f'ent_fire worldent addoutput "OnUser1>cmd>command>screenshot_prefix {session_id}_cube_{ev_suffix}_down>{6*t + 0.05}>1"',
+                f'ent_fire worldent addoutput "OnUser1>cmd>command>png_screenshot>{6*t + 0.1}>1"',
                 # Cleanup
                 f'ent_fire worldent addoutput "OnUser1>cmd>command>cl_drawhud 1;r_drawviewmodel 1;r_drawpanorama 1;cl_firstperson_legs 1;fov_cs_debug 0;noclip 1>{7*t}>1"',
                 f'ent_fire worldent addoutput "OnUser1>cmd>command>r_always_render_all_windows {original_render_all}>{7*t + 0.1}>1"',
@@ -184,6 +192,8 @@ def run(self):
                 sentinel="[Cubemap Done]",
                 timeout=15.0
             )
+            # Small delay after each EV to ensure engine stability
+            time.sleep(1.0)
 
         # 3. Stitching
         self.progress.emit("Stitching images...")