1 files changed, 98 insertions, 0 deletions
diff --git a/media/libjxl/src/tools/benchmark/hm/decode.sh b/media/libjxl/src/tools/benchmark/hm/decode.sh
new file mode 100755
index 0000000000..624c8ba729
--- /dev/null
+++ b/media/libjxl/src/tools/benchmark/hm/decode.sh
@@ -0,0 +1,98 @@
+#!/usr/bin/env bash
+
+# Copyright (c) the JPEG XL Project Authors. All rights reserved.
+#
+# Use of this source code is governed by a BSD-style
+# license that can be found in the LICENSE file.
+
+set -euo pipefail
+
+decoder="$(dirname "$0")"/TAppDecoderHighBitDepthStatic
+
+usage() {
+  echo "$0 [-v] <input.bin> <output.png>" >&2
+  exit 1
+}
+
+verbose=0
+
+while getopts ':hv' arg; do
+  case "$arg" in
+    h)
+      usage
+      ;;
+
+    v)
+      verbose=1
+      ;;
+
+    \?)
+      echo "Unrecognized option -$OPTARG" >&2
+      exit 1
+      ;;
+  esac
+done
+shift $((OPTIND-1))
+
+if [ $# -lt 2 ]; then
+  usage
+fi
+
+run() {
+  if [ "$verbose" -eq 1 ]; then
+    "$@"
+  else
+    "$@" > /dev/null 2>&1
+  fi
+}
+
+input="$1"
+output="$2"
+
+bin="$(mktemp)"
+yuv="$(mktemp)"
+width_file="$(mktemp)"
+height_file="$(mktemp)"
+icc_file="$(mktemp --suffix=.icc)"
+
+cleanup() {
+  rm -- "$bin" "$yuv" "$width_file" "$height_file" "$icc_file"
+}
+trap cleanup EXIT
+
+unpack_program="$(cat <<'END'
+  use File::Copy;
+  my ($input, $bin, $width_file, $height_file, $icc_file) = @ARGV;
+  open my $input_fh, '<:raw', $input;
+  sysread($input_fh, my $size, 8) == 8 or die;
+  my ($width, $height) = unpack 'NN', $size;
+  open my $width_fh, '>', $width_file;
+  print {$width_fh} "$width\n";
+  open my $height_fh, '>', $height_file;
+  print {$height_fh} "$height\n";
+  sysread($input_fh, my $icc_size, 4) == 4 or die;
+  $icc_size = unpack 'N', $icc_size;
+  sysread($input_fh, my $icc_data, $icc_size) == $icc_size or die;
+  open my $icc_fh, '>', $icc_file;
+  print {$icc_fh} $icc_data;
+  copy $input_fh, $bin;
+END
+)"
+run perl -Mstrict -Mwarnings -Mautodie -e "$unpack_program" -- "$input" "$bin" "$width_file" "$height_file" "$icc_file"
+
+width="$(cat "$width_file")"
+height="$(cat "$height_file")"
+
+start="$EPOCHREALTIME"
+run "$decoder" --OutputBitDepth=10 -b "$bin" -o "$yuv"
+end="$EPOCHREALTIME"
+
+elapsed="$(echo "$end - $start" | bc)"
+run echo "Completed in $elapsed seconds"
+
+echo "$elapsed" > "${output%.png}".time
+
+run ffmpeg -hide_banner -f rawvideo -vcodec rawvideo -s "${width}x$height" -r 25 -pix_fmt yuv444p10le -i "$yuv" -pix_fmt rgb24 -vf scale=in_color_matrix=bt709 -y "$output"
+if [ -s "$icc_file" ]; then
+  mogrify -profile "$icc_file" "$output"
+fi