diff --git a/ref/vk/tools/rendertest.py b/ref/vk/tools/rendertest.py index ccc01385..4564483b 100755 --- a/ref/vk/tools/rendertest.py +++ b/ref/vk/tools/rendertest.py @@ -1,5 +1,13 @@ #!/usr/bin/env python3 +saves = [ + 'brush2_01', + 'brush_01', + 'brush_02', + 'c0a0d_emissive', + 'light_01', +] + displays = { 'full': '', 'basecolor': 'basecolor', @@ -13,29 +21,32 @@ displays = { 'indirect_diff': 'indirect_diff', } -saves = [ - 'brush2_01', - 'brush_01', - 'brush_02', - 'c0a0d_emissive', - 'light_01', -] +import argparse -header = '''m_ignore 1 +parser = argparse.ArgumentParser(description='Generate scripts and makefiles for rendertest') +parser.add_argument('--script', '-s', type=argparse.FileType('w'), help='Console script for generating images') +args = parser.parse_args() + +def make_script(file): + header = '''m_ignore 1 scr_conspeed 100000 con_notifytime 0 hud_draw 0 r_speeds 0 rt_debug_fixed_random_seed 31337 -developer 0''' +developer 0 +''' -print(header) + file.write(header) -for save in saves: - screenshot_base = 'rendertest/' - print('') - print(f'load rendertest_{save}; wait 20') - for name, display in displays.items(): - print(f'rt_debug_display_only "{display}"; screenshot {screenshot_base}{save}_{name}.png; wait 1') + for save in saves: + screenshot_base = 'rendertest/' + file.write(f'\nload rendertest_{save}; wait 20') + for name, display in displays.items(): + file.write(f'rt_debug_display_only "{display}"; screenshot {screenshot_base}{save}_{name}.png; wait 1\n') -print('quit') + file.write('\nquit\n') + +if args.script: + print(f'Generating script {args.script.name}') + make_script(args.script)