From 979ce211c872b79106eae331ba14da0d128b94c9 Mon Sep 17 00:00:00 2001
From: yosh <yosh@unix.dog>
Date: Sun, 14 Apr 2024 00:29:29 -0400
Subject: [PATCH] coalesce ffmpeg and imgtools into one script

---
 ffmoovcheck |  2 --
 ffvmaf      | 19 --------------
 ffwrap.sh   | 41 +++++++++++++++++++++++++++++
 imgtools.sh | 75 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 4 files changed, 116 insertions(+), 21 deletions(-)
 delete mode 100755 ffmoovcheck
 delete mode 100755 ffvmaf
 create mode 100755 ffwrap.sh
 create mode 100755 imgtools.sh

diff --git a/ffmoovcheck b/ffmoovcheck
deleted file mode 100755
index 8d9b840..0000000
--- a/ffmoovcheck
+++ /dev/null
@@ -1,2 +0,0 @@
-#!/bin/sh
-ffprobe -v trace -i "$1" 2>&1 | rg -e type:\'mdat\' -e type:\'moov\'
diff --git a/ffvmaf b/ffvmaf
deleted file mode 100755
index 863e0e4..0000000
--- a/ffvmaf
+++ /dev/null
@@ -1,19 +0,0 @@
-#!/bin/sh
-set -ef
-
-while getopts :s:t:T: OPT; do
-	case "$OPT" in
-		s) start="$OPTARG" ;;
-		t) time="$OPTARG" ;;
-		T) to="$OPTARG" ;;
-		*) : ;
-	esac
-done
-shift $((OPTIND - 1))
-
-exec ffmpeg ${start:+-ss "$start"} ${to:+-to "$to"} \
-	-i "$1" -i "$2" ${time:+-t "$time"} \
-	-lavfi "[0:v] setpts=PTS-STARTPTS [ref]; \
-			[1:v] setpts=PTS-STARTPTS [dis]; \
-			[dis][ref] libvmaf=log_fmt=xml:log_path=/dev/stdout:n_threads=$(nproc)" \
-	-f null - \
diff --git a/ffwrap.sh b/ffwrap.sh
new file mode 100755
index 0000000..c44eae4
--- /dev/null
+++ b/ffwrap.sh
@@ -0,0 +1,41 @@
+#!/bin/sh
+set -euf
+
+errecho() { echo "$*" >&2 ; }
+
+vmaf() {
+	while getopts :s:t:T: OPT; do
+		case "$OPT" in
+			s) start="$OPTARG" ;;
+			t) time="$OPTARG" ;;
+			T) to="$OPTARG" ;;
+			*) : ;
+		esac
+	done
+	shift $((OPTIND - 1))
+
+	exec ffmpeg ${start:+-ss "$start"} ${to:+-to "$to"} \
+		-i "$1" -i "$2" ${time:+-t "$time"} \
+		-lavfi "[0:v] setpts=PTS-STARTPTS [ref]; \
+				[1:v] setpts=PTS-STARTPTS [dis]; \
+				[dis][ref] libvmaf=log_fmt=xml:log_path=/dev/stdout:n_threads=$(nproc)" \
+		-f null - \
+}
+
+moovcheck() {
+	ffprobe -v trace -i "$1" 2>&1 | grep -F -e "type:'mdat'" -e "type:'moov'"
+}
+
+cmd=$1; shift
+case "$cmd" in
+	vmaf) vmaf "$@" ;;
+	moovcheck) moovcheck "$@" ;;
+	help|-h) cat >&2 <<-EOF
+		available commands:
+		* vmaf $1 $2 - get vmaf of $1 (ref) to $2 (distorted)
+		  options: -s <start> -t <time> -T <to>
+		* moovcheck $1 - check if moov is before mdat in $1
+	EOF
+	;;
+	*) errecho "wrong command. use help or -h for help" ;;
+esac
diff --git a/imgtools.sh b/imgtools.sh
new file mode 100755
index 0000000..537bba4
--- /dev/null
+++ b/imgtools.sh
@@ -0,0 +1,75 @@
+#!/bin/sh
+set -euf
+
+dirminsize () {
+	cleanup() {
+		trap 'exit' INT HUP QUIT TERM EXIT
+		rm -f "$tmpjxl7" "$tmpjxl8" "$tmpjxl9" "$tmpwebp"
+		kill 0
+	}
+	tmpwebp=$(mktemp -u -t webp2jxl_XXXX.webp)
+	tmpjxl7=$(mktemp -u -t webp2jxl_XXXX.jxl)
+	tmpjxl8=$(mktemp -u -t webp2jxl_XXXX.jxl)
+	tmpjxl9=$(mktemp -u -t webp2jxl_XXXX.jxl)
+
+	trap 'cleanup' INT HUP QUIT TERM EXIT
+
+	ext="webp,png,jxl" depth="-maxdepth 1"
+	while getopts ":e:r" OPT; do
+		case "$OPT" in
+			e) ext="$OPTARG" ;;
+			r) depth="" ;;
+			*) exit 1 ;;
+		esac
+	done
+	shift $((OPTIND - 1))
+
+	exts=$(awk -F ',' -v OFS=' -o ' '{ for(i=1; i<=NF; i++) $i="-iname *." $i; print}' <<-EOF
+			$ext
+		EOF
+	)
+
+	while IFS= read -r f; do
+		[ -z "$f" ] && continue
+		echo "$f"
+		vips webpsave "$f" "$tmpwebp" --min-size --lossless &
+		vips jxlsave "$f" "$tmpjxl7" --lossless --effort 7 &
+		vips jxlsave "$f" "$tmpjxl8" --lossless --effort 8 &
+		vips jxlsave "$f" "$tmpjxl9" --lossless --effort 9 &
+		wait
+		read -r _ outfile <<-EOF
+			$(wc -c "$f" "$tmpwebp" "$tmpjxl7" "$tmpjxl8" "$tmpjxl9" | sort -n)
+		EOF
+		mv "$outfile" "${f%.*}.${outfile##*.}" || continue
+		[ "${outfile##*.}" != "${f##*.}" ] && rm "$f" # remove if there are 2
+	done <<-EOF
+		$(set -f; find "${1:-.}" -type f $depth $exts)
+	EOF
+}
+
+jxld2q() {
+	for n; do
+		case "$n" in
+			*.*) i=d; d="$n" ;;
+			*) i=q; q="$n" ;;
+		esac
+		[ "$i" = q ] && d="$(printf '0.1 + (100 - %s) *  0.09\n' "$q" | bc)"
+		[ "$i" = d ] && q="$(printf '100 - ((%s - 0.1) / 0.09)\n' "$d" | bc)"
+		echo "distance = $d | quality = $q"
+	done
+}
+
+cmd=$1; shift
+case "$cmd" in
+	dirminsize) dirminsize "$@" ;;
+	jxld2q) jxld2q "$@" ;;
+	help|-h) cat >&2 <<-EOF
+		available commands:
+		* dirminsize - tries to losslessly squeeze out as much size as possible from every file
+		  in the directory by converting to webp/jxl with various efforts
+		  options: -r (recursive) -e <ext1,ext2,...>
+		* jxld2q $@ - convert to/from jxl distance/quality for each arg
+	EOF
+	;;
+	*) errecho "wrong command. use help or -h for help" ;;
+esac