From d5bb1263d6f94f6c7ea5ad9ce37e71f124ee873e Mon Sep 17 00:00:00 2001
From: Natalia Portillo <claunia@claunia.com>
Date: Sat, 3 Oct 2026 15:15:45 +0100
Subject: [PATCH] Skip benchmark algorithms that are not compiled in.

---
 benchmark/benchmark.c   | 42 ++++++++++++++++++++++++-----------------
 benchmark/compression.c | 28 +++++++++++++++++++++++++++
 benchmark/compression.h |  4 ++++
 3 files changed, 57 insertions(+), 17 deletions(-)

diff --git a/benchmark/benchmark.c b/benchmark/benchmark.c
index f5d6908..f79bfcc 100644
--- a/benchmark/benchmark.c
+++ b/benchmark/benchmark.c
@@ -406,6 +406,7 @@ int main(int argc, char *argv[])
     const size_t algorithm_count  = sizeof(algorithms) / sizeof(algorithms[0]);
 
     benchmark_result results[sizeof(algorithms) / sizeof(algorithms[0])];
+    int              ran[sizeof(algorithms) / sizeof(algorithms[0])] = {0};
 
     print_section_header("Compression Benchmarks");
 
@@ -413,6 +414,13 @@ int main(int argc, char *argv[])
     {
         print_subsection_header(algorithm_names[i]);
 
+        // Skip algorithms whose library was not compiled in
+        if(!is_algorithm_available(algorithms[i]))
+        {
+            printf("  %s⚠ Skipped: not compiled in%s\n\n", clr(ANSI_YELLOW), clr(ANSI_RESET));
+            continue;
+        }
+
         char output_path[512];
         snprintf(output_path, sizeof(output_path), "%s.%s.aaruformat", input_path, algorithm_names[i]);
 
@@ -434,6 +442,7 @@ int main(int argc, char *argv[])
         }
 
         results[i].elapsed_ns = get_time_ns() - start_time;
+        ran[i]                = 1;
 
         // Print results for this algorithm
         char compressed_str[64];
@@ -462,6 +471,8 @@ int main(int argc, char *argv[])
 
     for(size_t i = 0; i < algorithm_count; i++)
     {
+        if(!ran[i]) continue;
+
         const double ratio  = (double)results[i].compressed_size / (double)info.total_uncompressed_size * 100.0;
         const double time_s = ns_to_seconds(results[i].elapsed_ns);
 
@@ -471,28 +482,25 @@ int main(int argc, char *argv[])
                clr(ANSI_BOLD), clr(get_ratio_color(ratio)), ratio, clr(ANSI_RESET), time_s);
     }
 
-    // Find best compression
-    size_t   best_compression_idx = 0;
-    uint64_t best_compressed_size = results[0].compressed_size;
-    for(size_t i = 1; i < algorithm_count; i++)
+    // Find best compression and fastest among the algorithms that ran
+    size_t best_compression_idx = algorithm_count;
+    size_t fastest_idx          = algorithm_count;
+    for(size_t i = 0; i < algorithm_count; i++)
     {
-        if(results[i].compressed_size < best_compressed_size)
-        {
-            best_compressed_size = results[i].compressed_size;
+        if(!ran[i]) continue;
+
+        if(best_compression_idx == algorithm_count ||
+           results[i].compressed_size < results[best_compression_idx].compressed_size)
             best_compression_idx = i;
-        }
+
+        if(fastest_idx == algorithm_count || results[i].elapsed_ns < results[fastest_idx].elapsed_ns) fastest_idx = i;
     }
 
-    // Find fastest
-    size_t   fastest_idx  = 0;
-    uint64_t fastest_time = results[0].elapsed_ns;
-    for(size_t i = 1; i < algorithm_count; i++)
+    if(best_compression_idx == algorithm_count)
     {
-        if(results[i].elapsed_ns < fastest_time)
-        {
-            fastest_time = results[i].elapsed_ns;
-            fastest_idx  = i;
-        }
+        printf("\n  %s⚠ No algorithms were benchmarked%s\n\n", clr(ANSI_YELLOW), clr(ANSI_RESET));
+        close_image(&info);
+        return 1;
     }
 
     printf("\n  %s────────────────────────────────────────────────────────────────────────────────%s\n",
diff --git a/benchmark/compression.c b/benchmark/compression.c
index e5f0826..070bb54 100644
--- a/benchmark/compression.c
+++ b/benchmark/compression.c
@@ -277,6 +277,34 @@ int compress_data(const compression_algorithm algorithm, const uint8_t *input, c
     }
 }
 
+// Check if an algorithm's library was compiled in
+int is_algorithm_available(const compression_algorithm algorithm)
+{
+    switch(algorithm)
+    {
+        case COMP_LZMA:
+            return 1;
+#ifdef HAVE_BZ3
+        case COMP_BZIP3:
+            return 1;
+#endif
+#ifdef HAVE_ZSTD
+        case COMP_ZSTD:
+            return 1;
+#endif
+#ifdef HAVE_BROTLI
+        case COMP_BROTLI:
+            return 1;
+#endif
+#ifdef HAVE_OPENZL
+        case COMP_OPENZL:
+            return 1;
+#endif
+        default:
+            return 0;
+    }
+}
+
 // Get compression type for header
 int get_compression_type(const compression_algorithm algorithm)
 {
diff --git a/benchmark/compression.h b/benchmark/compression.h
index 52aee1e..92bd87e 100644
--- a/benchmark/compression.h
+++ b/benchmark/compression.h
@@ -45,6 +45,10 @@ zstd_dict_context *train_zstd_dictionary(const uint8_t *sample_data, size_t samp
 // Free dictionary context
 void free_zstd_dictionary(zstd_dict_context *dict_ctx);
 
+// Check if an algorithm's library was compiled in
+// Returns 1 if available, 0 otherwise
+int is_algorithm_available(compression_algorithm algorithm);
+
 // Get compression type identifier for block header
 int get_compression_type(compression_algorithm algorithm);