diff --git a/tools/sim_recorder.py b/tools/sim_recorder.py index 0539ca6a79a947e5f8f798713beafaba7e11fde3..b96f376c336e533074e795248313e6aa8680a833 100644 --- a/tools/sim_recorder.py +++ b/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') diff --git a/tools/sim_video_3d_cap_ext.py b/tools/sim_video_3d_cap_ext.py index da74c32e83a6d365fd4f5ac9c8e13896c10d946a..b0c267423be6f136ef548511c05335106cf9ee17 100644 --- a/tools/sim_video_3d_cap_ext.py +++ b/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 diff --git a/tools/universe_sim.bat b/tools/universe_sim.bat new file mode 100644 index 0000000000000000000000000000000000000000..6fe0d32bb636e05e18fe9b3e1be195f9f76132da --- /dev/null +++ b/tools/universe_sim.bat @@ -0,0 +1,25 @@ +@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 + + + diff --git a/tools/video_cap.bat b/tools/video_cap.bat new file mode 100644 index 0000000000000000000000000000000000000000..55a7801ce766d549bcbb47d4a5ab4a9310d4899d --- /dev/null +++ b/tools/video_cap.bat @@ -0,0 +1,18 @@ +@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 + +