Python超人-宇宙模拟器

tools/sim_recorder.py

@@ -23,7 +23,7 @@ def crop(mp4_file):
         print("裁剪视频失败")
 
 
-def get_window_img_dc(window_name="universe_sim"):
+def get_window_img_dc(window_name="宇宙模拟器(universe sim)"):
     # 获取桌面
     # hdesktop = win32gui.GetDesktopWindow()
     handle = win32gui.FindWindow(None, window_name)
@@ -33,6 +33,7 @@ def get_window_img_dc(window_name="universe_sim"):
 def record():
     parser = argparse.ArgumentParser()
     parser.add_argument('--fps', type=int, default=30, help='frame per second')
+    parser.add_argument('--window_name', type=str, default='宇宙模拟器(universe sim)', help='window_name')
     parser.add_argument('--total_time', type=int, default=10000000, help='video total time')
     parser.add_argument('--savename', type=str, default='video_right.mp4', help='save file name')
     parser.add_argument('--screen_type', default=0, type=int, choices=[0, 1], help='1: full screen, 0: region screen')

tools/sim_video_3d_cap_ext.py

@@ -29,6 +29,7 @@ def get_window_handle(window_name="宇宙模拟器(universe sim)"):
 def get_args():
     parser = argparse.ArgumentParser()
     parser.add_argument('--fps', type=int, default=30, help='frame per second')
+    parser.add_argument('--window_name', type=str, default='宇宙模拟器(universe sim)', help='window_name')
     parser.add_argument('--total_time', type=int, default=10000000, help='video total time')
     parser.add_argument('--save_name', type=str, default='video.mp4', help='save file name')
     parser.add_argument('--start_index', type=int, default=-1, help='start_index')
@@ -97,6 +98,7 @@ def sim_window_screen_shot(wait_ses=-1):
             try:
                 img = screen_shot(img_dc)
             except Exception as e:
+                press_pause_key()
                 print("ERROR:", str(e))
                 traceback.print_exc()
                 return None

tools/universe_sim.bat

+@REM @echo off
+@REM 设置环境和参数
+SET Anaconda3=D:/Anaconda3
+SET env=pythoncr
+SET DISK=D:
+SET SimDir=%DISK%/gitee/universe_sim
+
+SET SimFileDir=%1
+SET SimFileName=%2
+SET param3=%3
+
+SET PYTHONPATH=%SimDir%;
+SET SimFilePath=%SimDir%/sim_scenes/%SimFileDir%/
+
+CALL %Anaconda3%/Scripts/activate.bat %Anaconda3%
+CALL conda activate %env%
+%DISK%
+
+cd %SimFilePath%
+@REM  universe_sim.bat science speed_of_light_3d
+python -m %SimFileName%
+cd %SimDir%\tools
+
+
+

tools/video_cap.bat

+@REM @echo off
+@REM 设置环境和参数
+SET Anaconda3=D:/Anaconda3
+SET env=pythoncr
+SET DISK=D:
+SET SimDir=%DISK%/gitee/universe_sim
+
+SET SimFileName=%1
+
+CALL %Anaconda3%/Scripts/activate.bat %Anaconda3%
+CALL conda activate %env%
+%DISK%
+
+cd %SimDir%\tools
+@REM video_cap.bat speed_of_light_3d
+python -m sim_video_3d_cap_ext --save_name=%SimFileName%_2.mp4
+
+