fix: limit maxThreads to 64 for AOM encodes

Fixes #23 Co-authored-by: Frankie Dintino <[email protected]> --------- Co-authored-by: Wyatt Borsos <[email protected]> Co-authored-by: Wyatt <[email protected]> Co-authored-by: Frankie Dintino <[email protected]>
fdintino · Feb 7, 2024 · 6f0ecba · 6f0ecba
1 parent 53732f0
commit 6f0ecba
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/pillow_avif/_avif.c b/src/pillow_avif/_avif.c
@@ -448,7 +448,11 @@ AvifEncoderNew(PyObject *self_, PyObject *args) {
             init_max_threads();
         }
 
-        encoder->maxThreads = max_threads;
+        int is_aom_encode = strcmp(codec, "aom") == 0 ||
+                            (strcmp(codec, "auto") == 0 &&
+                             _codec_available("aom", AVIF_CODEC_FLAG_CAN_ENCODE));
+
+        encoder->maxThreads = is_aom_encode && max_threads > 64 ? 64 : max_threads;
 #if AVIF_VERSION >= 1000000
         if (enc_options.qmin != -1 && enc_options.qmax != -1) {
             encoder->minQuantizer = enc_options.qmin;