From 4e7d5f190afcbcfa44cc0d019cbf7d3b4496522b Mon Sep 17 00:00:00 2001
From: Jay Berkenbilt <ejb@ql.org>
Date: Thu, 1 Sep 2022 17:11:56 -0400
Subject: [PATCH] Include memory usage in performance test output

---
 manual/release-notes.rst |  8 +++++++
 performance_check        | 51 ++++++++++++++++++++++++++++++++++++----
 2 files changed, 55 insertions(+), 4 deletions(-)

diff --git a/manual/release-notes.rst b/manual/release-notes.rst
index 54e85b71..49298836 100644
--- a/manual/release-notes.rst
+++ b/manual/release-notes.rst
@@ -253,6 +253,14 @@ For a detailed list of changes, please see the file
       some additional caching to reduce the overhead of repeatedly
       reading environment variables at runtime.
 
+    - The test files used by the ``performance_check`` script at the
+      top of the repository are now available in the
+      `qpdf/performance-test-files github repository
+      <https://github.com/qpdf/performance-test-files>`__. In addition
+      to running time, memory usage is also included in performance
+      test results. The ``performance_check`` tool has only been
+      tested on Linux.
+
     - Lots of code cleanup and refactoring work was contributed in
       multiple pull requests by M. Holger.
 
diff --git a/performance_check b/performance_check
index 8bb7b69d..60ba58a6 100755
--- a/performance_check
+++ b/performance_check
@@ -5,6 +5,8 @@ use strict;
 use File::Basename;
 use Time::HiRes qw(gettimeofday tv_interval);
 use File::Path qw(make_path);
+use IPC::Open3;
+use IO::Pipe;
 
 my $whoami = basename($0);
 $| = 1;
@@ -29,6 +31,7 @@ my %arg_compat = (
     '--remove-unreferenced-resources=no' => '--preserve-unreferenced-resources',
     '--remove-unreferenced-resources=yes' => '',
     '--remove-unreferenced-resources=auto' => undef,
+    '--report-memory-usage' => '',
     );
 
 my $executable = undef;
@@ -172,6 +175,20 @@ Repository URL: https://github.com/qpdf/performance-test-files
     }
 }
 
+my $report_mem = filter_args(["--report-memory-usage"]);
+{
+    my ($r, $mem) = run_cmd($executable, @$report_mem,
+                            "--empty", File::Spec->devnull());
+    if ($r != 0)
+    {
+        die "$whoami: $executable doesn't seem to work\n";
+    }
+    if ($mem == 0)
+    {
+        print "** Note: memory information is not available **\n";
+    }
+}
+
 run_tests();
 print "\n";
 
@@ -211,6 +228,7 @@ sub run_tests
 
     chomp(my $commit = `git describe @`);
     print "commit: $commit\n";
+    print "Format: time-in-seconds  RAM-in-MiB  filename\n";
     make_path($workdir);
     foreach my $test (@tests)
     {
@@ -259,16 +277,17 @@ sub run_test
             last;
         }
     }
-    my @cmd = ($executable, @$args, $file, "$workdir/$outfile");
+    my @cmd = ($executable, @$args, @$report_mem, $file, "$workdir/$outfile");
     # Run once and discard to update caches
     system("sync");
-    system(@cmd);
+    run_cmd(@cmd);
     my $i = 0;
     my $total = 0;
+    my $max_mem = 0;
     while ($i < $iterations)
     {
         my $start = [gettimeofday];
-        my $r = system(@cmd);
+        my ($r, $mem) = run_cmd(@cmd);
         if ($r == 2)
         {
             # interrupt
@@ -280,6 +299,7 @@ sub run_test
             print "  command failed; ignoring results\n";
             return undef;
         }
+        $max_mem = $mem > $max_mem ? $mem : $max_mem;
         my $elapsed = tv_interval($start, $end);
         $total += $elapsed;
         ++$i;
@@ -289,5 +309,28 @@ sub run_test
             last;
         }
     }
-    return sprintf("%0.4f", $total / $i);
+    return sprintf("%8.4f  %8.4f", $total / $i, $max_mem / 1048576);
+}
+
+sub run_cmd
+{
+    my @cmd = @_;
+    my $pipe = IO::Pipe->new();
+    my $pid = open3(my $child_in, '>&STDOUT', $pipe->writer(), @cmd);
+    $child_in->close();
+    waitpid($pid, 0);
+    my $r = $?;
+    my $mem = 0;
+    while (<$pipe>)
+    {
+        if (m/qpdf-max-memory-usage (\d+)/)
+        {
+            $mem = $1;
+        }
+        else
+        {
+            warn $_;
+        }
+    }
+    ($r, $mem);
 }