提交 e4f1cd27 编写于 作者: 三月三net's avatar 三月三net

Python超人-宇宙模拟器

上级 32d89561
...@@ -228,7 +228,7 @@ class JupiterMoonProtectsEarthSim(UniverseSimScenes): ...@@ -228,7 +228,7 @@ class JupiterMoonProtectsEarthSim(UniverseSimScenes):
print("满足条件退出") print("满足条件退出")
exit() exit()
if ctime - self.run_begin_time > 600: if ctime - self.run_begin_time > 700:
print("没有满足条件退出") print("没有满足条件退出")
exit() exit()
......
...@@ -20,18 +20,29 @@ def get_window_handle(window_name="宇宙模拟器(universe sim)"): ...@@ -20,18 +20,29 @@ def get_window_handle(window_name="宇宙模拟器(universe sim)"):
handle = win32gui.FindWindow(None, window_name) handle = win32gui.FindWindow(None, window_name)
return handle 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(): def get_args():
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument('--fps', type=int, default=30, help='frame per second') 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('--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('--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') # parser.add_argument('--screen_type', default=0, type=int, choices=[0, 1], help='1: full screen, 0: region screen')
args = parser.parse_args() args = parser.parse_args()
print("total_time:", args.total_time) print("total_time:", args.total_time)
print("fps:", args.fps) print("fps:", args.fps)
args.save_name = rename_exists(args.save_name)
print("save_name:", args.save_name) print("save_name:", args.save_name)
print("wait:", args.wait) print("wait:", args.wait)
if args.save_name == '': if args.save_name == '':
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册