png2mp4: only draw text when needed, remove x265 as it has been electrified too much by an engineer with access to a large language model 🐺

zsh/png2mp4.zsh

@@ -1,82 +1,6 @@
-# png2mp4_x265()
-# Converts a series of PNG images to an MP4 video using x265 encoding
-#
-# Usage: png2mp4_x265 [--max <number>] [--step <number>] [--repeat <number>]
-#
-# Options:
-#   --max <number>    : Maximum number of images to process
-#   --step <number>   : Multiplier for step count in overlay text (default: 1)
-#   --repeat <number> : Number of times to repeat each image (default: 1)
-#
-# This function:
-# 1. Finds all PNG files in the current directory
-# 2. Uses the current directory name as the output filename prefix
-# 3. Uses ffmpeg to create an MP4 with x265 encoding, including:
-#    - Frame rate of 60 fps
-#    - Image scaling to 1024x1024
-#    - Step count overlay text (divided by 1,000,000 and truncated to remove 6 zeros and decimal places)
-#    - High-quality encoding settings
-# 4. Adds padding and fade-out effect to the final video
-#
-# Requirements:
-#   - ffmpeg with libx265 support
-#   - Inconsolata-Light.ttf font in /usr/share/fonts/TTF/
-png2mp4_x265() {
-    local step_multiplier=1
-    local repeat=1
-    local frame_rate=60
-    local resolution="1024x1024"
-
-    while [[ "$#" -gt 0 ]]; do
-        case $1 in
-            --step) step_multiplier="$2"; shift ;;
-            --repeat) repeat="$2"; shift ;;
-            *) echo "Unknown parameter passed: $1"; return 1 ;;
-        esac
-        shift
-    done
-
-    # Find all unique sample numbers
-    local samples=($(ls *.png | sed -n 's/.*_\([0-9][0-9]\)_.*/\1/p' | sort -u))
-
-    for sample in "${samples[@]}"; do
-        local output_filename="$(basename "$(pwd)")_sample${sample}_x265.mp4"
-        echo "Processing sample ${sample}. Output filename: $output_filename"
-
-        local nframes=$(ls *_${sample}_*.png | wc -l)
-        local duration=$(($nframes * $repeat / ${frame_rate}.))
-        local fade_start=$((duration + 3))
-        echo "Found $nframes frames for sample ${sample}, duration of $duration seconds"
-
-        echo "Running ffmpeg with x265 encoding for sample ${sample}..."
-        local font=/usr/share/fonts/TTF/Inconsolata-Light.ttf
-        local drawtext="drawtext=fontfile=${font}:text='Steps\: %{eif\\:trunc(n*$step_multiplier)\\:u\\:3}':x=10:y=h-th-10:fontsize=24:fontcolor=white"
-        local fadeout="tpad=stop_mode=clone:stop_duration=8,fade=t=out:st=${fade_start}:d=5"
-        local encoder=(
-            -pix_fmt yuv420p
-            -c:v libx265
-            -preset slower
-            -tune animation
-            -crf 22
-            -x265-params "keyint=${repeat}:min-keyint=$((repeat-1)):scenecut=0:ref=5:bframes=8:b-adapt=2:rc-lookahead=$((2*repeat)):lookahead-slices=4:aq-mode=3:aq-strength=0.8:deblock=-1,-1:sao=0"
-        )
-
-        ffmpeg -framerate "$frame_rate/$repeat" -pattern_type glob -i "*_${sample}_*.png" \
-            -vf "scale=${resolution},${drawtext},fps=${frame_rate},${fadeout}" \
-            "${encoder[@]}" -y "$output_filename"
-        if [ $? -ne 0 ]; then
-            echo "Error: ffmpeg command failed for sample ${sample}."
-            return 1
-        fi
-    done
-
-    echo "All samples processed successfully."
-}
-
-# Same thing but x264
 png2mp4() {
     local max_images=""
-    local step_multiplier=1
+    local step_multiplier=""
     local repeat=1
     local temp_dir="/home/kade/.local/tmp"
 
@@ -115,14 +39,18 @@ png2mp4() {
         done
 
         echo "Running ffmpeg for sample ${sample}..."
+        local vf_options="scale=1024x1024"
+        if [[ -n "$step_multiplier" ]]; then
+            vf_options+=",drawtext=fontfile=/usr/share/fonts/TTF/Inconsolata-Light.ttf:text='Steps\: %{expr\:trunc(n*$step_multiplier/$repeat/1000000)}':x=10:y=h-th-10:fontsize=24:fontcolor=white"
+        fi
+        
         if [[ -n "$max_images" ]]; then
-            ffmpeg -framerate 60 -pattern_type glob -i "$temp_dir/*_${sample}_*.png" -vf "scale=1024x1024,select='not(mod(n\,$max_images))',drawtext=fontfile=/usr/share/fonts/TTF/Inconsolata-Light.ttf:text='Steps\: %{expr\:trunc(n*$step_multiplier/$repeat/1000000)}':x=10:y=h-th-10:fontsize=24:fontcolor=white" -crf 28 \
-            -c:v libx264 -pix_fmt yuv420p -y "$temp_dir/temp_${sample}.mp4"
-        else
-            ffmpeg -framerate 60 -pattern_type glob -i "$temp_dir/*_${sample}_*.png" -vf "scale=1024x1024,drawtext=fontfile=/usr/share/fonts/TTF/Inconsolata-Light.ttf:text='Steps\: %{expr\:trunc(n*$step_multiplier/$repeat/1000000)}':x=10:y=h-th-10:fontsize=24:fontcolor=white" -crf 28 \
-            -c:v libx264 -pix_fmt yuv420p -y "$temp_dir/temp_${sample}.mp4"
+            vf_options="select='not(mod(n\,$max_images))',$vf_options"
         fi
 
+        ffmpeg -framerate 60 -pattern_type glob -i "$temp_dir/*_${sample}_*.png" -vf "$vf_options" -crf 28 \
+        -c:v libx264 -pix_fmt yuv420p -y "$temp_dir/temp_${sample}.mp4"
+
         if [ $? -ne 0 ]; then
             echo "Error: ffmpeg command failed for sample ${sample}."
             rm -rf "$temp_dir"