diff --git a/sim_scenes/featured/jupiter_moon_protects_earth.py b/sim_scenes/featured/jupiter_moon_protects_earth.py index 3a3a4069dfe904a14c61015d2e032c7b7fecc8f7..e4345c7b7dc8c098aafb06d29f1049eddf3305e2 100644 --- a/sim_scenes/featured/jupiter_moon_protects_earth.py +++ b/sim_scenes/featured/jupiter_moon_protects_earth.py @@ -228,7 +228,7 @@ class JupiterMoonProtectsEarthSim(UniverseSimScenes): print("满足条件退出") exit() - if ctime - self.run_begin_time > 600: + if ctime - self.run_begin_time > 700: print("没有满足条件退出") exit() diff --git a/tools/sim_video_cap.py b/tools/sim_video_cap.py index 672d6cefb5785e4ee11ffac25a879ee0973991e1..0dc85a341cc20a7470261566bd5f7628a637094e 100644 --- a/tools/sim_video_cap.py +++ b/tools/sim_video_cap.py @@ -20,18 +20,29 @@ def get_window_handle(window_name="宇宙模拟器(universe sim)"): handle = win32gui.FindWindow(None, window_name) return handle +def rename_exists(save_name): + file_no = 0 + rename_save_name = save_name + while os.path.exists(os.path.abspath(rename_save_name)): + rename_save_name = f"{save_name[:-4]}_{file_no}{save_name[-4:]}" + file_no += 1 + + return rename_save_name def get_args(): parser = argparse.ArgumentParser() parser.add_argument('--fps', type=int, default=30, help='frame per second') 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('--save_name', type=str, default='jupiter_moon_protects_earth7.mp4', help='save file name') parser.add_argument('--wait', type=int, default=0, help='wait for recoding') # parser.add_argument('--screen_type', default=0, type=int, choices=[0, 1], help='1: full screen, 0: region screen') args = parser.parse_args() print("total_time:", args.total_time) print("fps:", args.fps) + + args.save_name = rename_exists(args.save_name) + print("save_name:", args.save_name) print("wait:", args.wait) if args.save_name == '':