diff --git a/scripts/helpers/video.py b/scripts/helpers/video.py
index f90b82d..f148cd5 100755
--- a/scripts/helpers/video.py
+++ b/scripts/helpers/video.py
@@ -6,7 +6,7 @@ from datetime import datetime
if __name__ == '__main__':
args = sys.argv[1:]
if len(args) < 2:
- print('Usage: compress [rename/cpu/gpu] [crf=20]')
+ print('Usage: compress [rename/cpu/gpu/cpucmd/gpucmd] [crf=24] [additional args]')
# Command to rename all screen recordings
if args[0] == 'rename':
@@ -27,11 +27,20 @@ if __name__ == '__main__':
i = args[1]
crf = args[2] if len(args) > 2 else '24'
out = i[:i.rindex('.')] + f'.{processor[0]}265'
+ cmd = 'cmd' in processor
+ if cmd:
+ processor = processor.replace('cmd', '')
if processor == 'cpu':
out += f'-{crf}'
out += '.mp4'
+ additional_args = ' '.join(args[3:] if len(args) > 3 else [])
if processor == 'cpu':
- os.system(f'ffmpeg -i "{i}" -vcodec libx265 -crf {crf} "{out}"')
+ c = f'ffmpeg -i "{i}" -vcodec libx265 -crf {crf} {additional_args} "{out}"'
if processor == 'gpu':
- os.system(f'ffmpeg -i "{i}" -c:v hevc_videotoolbox -b:v 4000k "{out}"')
+ c = f'ffmpeg -i "{i}" -c:v hevc_videotoolbox -b:v 1000k {additional_args} "{out}"'
+
+ if cmd:
+ print(c)
+ else:
+ os.system()
\ No newline at end of file