ffmpeg-builder/lib/pgo.sh

#!/usr/bin/env bash

PGO_DIR="${IGN_DIR}/pgo"
PGO_PROFDATA="${PGO_DIR}/prof.profdata"
gen_profdata() {
	recreate_dir "${PGO_DIR}" || return 1
	cd "${PGO_DIR}" || return 1
	setup_pgo_clips || return 1
	for vid in *.mkv; do
		local args=()
		# add precalculated grain amount based off of filename
		line_contains "${vid}" 'grain' && args+=(-g 16)
		# make fhd preset 2
		line_contains "${vid}" 'fhd' && args+=(-P 2)

		LLVM_PROFILE_FILE="${PGO_DIR}/default_%p.profraw" \
			encode -i "${vid}" "${args[@]}" "encoded-${vid}" || return 1
	done

	# merge profraw into profdata
	local mergeCmd=()
	# darwin needs special invoke
	if is_darwin; then
		mergeCmd+=(xcrun)
	fi

	mergeCmd+=(
		llvm-profdata
		merge
		"--output=${PGO_PROFDATA}"
	)
	"${mergeCmd[@]}" default*.profraw || return 1

	return 0
}

setup_pgo_clips() {
	local clips=(
		"fhd-grainy.mkv 1080p,grain=yes"
		"uhd.mkv 2160p"
		"uhd-hdr.mkv 2160p,hdr=yes"
	)
	for clip in "${clips[@]}"; do
		local genVid genVidArgs pgoFile genVidArgsArr
		IFS=' ' read -r genVid genVidArgs <<<"${clip}"
		# pgo path is separate
		pgoFile="${PGO_DIR}/${genVid}"
		genVid="${TMP_DIR}/${genVid}"
		# create array of args split with ,
		genVidArgsArr=(${genVidArgs//,/ })
		# create generated vid without any profiling if needed
		test -f "${genVid}" ||
			LLVM_PROFILE_FILE='/dev/null' gen_video "${genVid}" "${genVidArgsArr[@]}" || return 1
		# and move to the pgo directory
		test -f "${pgoFile}" ||
			cp "${genVid}" "${pgoFile}" || return 1

	done
}