avifenc: Change default value of --jobs to "all"

Fix AOMediaCodec#1522.
wantehchang · Nov 7, 2023 · 6db3f05 · 6db3f05
1 parent 9157c74
commit 6db3f05
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 4 deletions.
diff --git a/apps/avifenc.c b/apps/avifenc.c
@@ -209,7 +209,7 @@ static void syntaxLong(void)
            AVIF_SPEED_FASTEST);
     printf("\n");
     printf("Advanced options:\n");
-    printf("    -j,--jobs J                       : Number of jobs (worker threads, default: 1. Use \"all\" to use all available cores)\n");
+    printf("    -j,--jobs J                       : Number of jobs (worker threads). Use \"all\" to use all available cores (default: all)\n");
     printf("    --no-overwrite                    : Never overwrite existing output file\n");
     printf("    -o,--output FILENAME              : Instead of using the last filename given as output, use this filename\n");
 #if defined(AVIF_ENABLE_EXPERIMENTAL_AVIR)
@@ -1422,7 +1422,7 @@ MAIN()
     avifSettings settings;
     memset(&settings, 0, sizeof(settings));
     settings.codecChoice = AVIF_CODEC_CHOICE_AUTO;
-    settings.jobs = 1;
+    settings.jobs = -1;
     settings.targetSize = -1;
     settings.qualityIsConstrained = AVIF_FALSE;
     settings.qualityAlphaIsConstrained = AVIF_FALSE;
@@ -1940,6 +1940,10 @@ MAIN()
         ++argIndex;
     }
 
+    if (settings.jobs == -1) {
+        settings.jobs = avifQueryCPUCount();
+    }
+
     // Check global lossless parameters and set to default if needed.
     if (lossless) {
         // Pixel format.

diff --git a/src/codec_aom.c b/src/codec_aom.c
@@ -748,7 +748,9 @@ static avifResult aomCodecEncodeImage(avifCodec * codec,
             cfg->g_lag_in_frames = 0;
         }
         if (encoder->maxThreads > 1) {
-            cfg->g_threads = encoder->maxThreads;
+            // libaom fails if cfg->g_threads is greater than 64 threads. See MAX_NUM_THREADS in
+            // aom/aom_util/aom_thread.h.
+            cfg->g_threads = AVIF_MIN(encoder->maxThreads, 64);
         }
 
         codec->internal->monochromeEnabled = AVIF_FALSE;

diff --git a/src/codec_avm.c b/src/codec_avm.c
@@ -641,7 +641,9 @@ static avifResult avmCodecEncodeImage(avifCodec * codec,
             cfg->g_lag_in_frames = 0;
         }
         if (encoder->maxThreads > 1) {
-            cfg->g_threads = encoder->maxThreads;
+            // libavm fails if cfg->g_threads is greater than 64 threads. See MAX_NUM_THREADS in
+            // avm/aom_util/aom_thread.h.
+            cfg->g_threads = AVIF_MIN(encoder->maxThreads, 64);
         }
 
         // avm does not handle monochrome as of research-v4.0.0.