From 1bb6ddb299b3bd38bd1e7fbed2dff0ea768f1682 Mon Sep 17 00:00:00 2001
From: SuspiciousDuck <99849016+SuspiciousDuck@users.noreply.github.com>
Date: Sat, 7 Dec 2024 13:13:18 -0600
Subject: [PATCH] Configurable temporary directory (#5)

* added --temp option

* specify temp directory path
---
 auto-boost_2.5.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/auto-boost_2.5.py b/auto-boost_2.5.py
index b2f4d5b..316b14e 100755
--- a/auto-boost_2.5.py
+++ b/auto-boost_2.5.py
@@ -32,6 +32,7 @@ else:
 parser = argparse.ArgumentParser()
 parser.add_argument("-s", "--stage", help = "Select stage: 1 = encode, 2 = calculate metrics, 3 = generate zones | Default: all", default=0)
 parser.add_argument("-i", "--input", required=True, help = "Video input filepath (original source file)")
+parser.add_argument("-t", "--temp", help = "The temporary directory for av1an to store files in | Default: video input filename")
 parser.add_argument("-q", "--quality", help = "Base quality (CRF) | Default: 30", default=30)
 parser.add_argument("-d", "--deviation", help = "Maximum CRF change from original | Default: 10", default=10)
 parser.add_argument("-p", "--preset", help = "Fast encode preset | Default: 9", default=9)
@@ -44,7 +45,7 @@ args = parser.parse_args()
 stage = int(args.stage)
 src_file = Path(args.input).resolve()
 output_dir = src_file.parent
-tmp_dir = output_dir / "temp"
+tmp_dir = Path(args.temp).resolve() if args.temp is not None else output_dir / src_file.stem
 output_file = output_dir / f"{src_file.stem}_fastpass.mkv"
 scenes_file = tmp_dir / "scenes.json"
 br = float(args.deviation)