wtf

Signed-off-by: Balazs Horvath <[email protected]>

.zshrc

@@ -913,80 +913,7 @@ filePath = '$filePath'
 print(json.loads(safetensors.safe_open(filePath, 'np').metadata().get('ss_seed', 'Not found')))"
 }
 
-# Function: png2mp4
-# Description:
-#   This function converts a series of PNG images into an MP4 video file using ffmpeg.
-#
-# Usage:
-#   png2mp4 [--max <number>]
-#
-# Parameters:
-#   --max <number> (optional): Stops the video at the specified number of images.
-#
-# Actions:
-#   1. Sets the frame rate to 8 fps
-#   2. Uses glob pattern to include all PNG files in the current directory
-#   3. Optionally limits the number of input images based on the --max argument
-#   4. Scales the output video to 1024x1024 resolution
-#   5. Sets the Constant Rate Factor (CRF) to 28 for good compression
-#   6. Uses the libx264 codec for H.264 encoding
-#   7. Sets the pixel format to yuv420p for compatibility
-#   8. Outputs the result as 'out.mp4' in the current directory
-#
-# Notes:
-#   - Requires ffmpeg to be installed and accessible in the system path
-#   - PNG files should be in the current directory
-#   - Output video will be named 'out.mp4' and placed in the current directory
-#   - Adjust the framerate, scale, or CRF value as needed for different results
-#   - Use the --max argument to limit the number of input images
-png2mp4() {
-    conda deactivate
-    local max_images=""
-    local step_multiplier=1
-    local prefix=""
-    local repeat=1
-    local temp_dir="/home/kade/.local/tmp"
-
-    while [[ "$#" -gt 0 ]]; do
-        case $1 in
-            --max) max_images="$2"; shift ;;
-            --step) step_multiplier="$2"; shift ;;
-            --repeat) repeat="$2"; shift ;;
-            *) echo "Unknown parameter passed: $1"; return 1 ;;
-        esac
-        shift
-    done
-
-    mkdir -p "$temp_dir"
-
-    # Extract the prefix from the first image filename
-    prefix=$(/usr/bin/env ls *.png | head -n 1 | cut -d'-' -f1 | tr -d '[:cntrl:]' | tr -d '[]')
-
-    # Create repeated images
-    for img in *.png; do
-        for i in $(seq 1 $repeat); do
-            cp "$img" "$temp_dir/${img%.*}_${i}.png"
-        done
-    done
-
-    if [[ -n "$max_images" ]]; then
-        ffmpeg -framerate 60 -pattern_type glob -i "$temp_dir/*.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)}':x=10:y=h-th-10:fontsize=24:fontcolor=white" -crf 28 \
-        -c:v libx264 -pix_fmt yuv420p -y "$temp_dir/temp.mp4"
-    else
-        ffmpeg -framerate 60 -pattern_type glob -i "$temp_dir/*.png" -vf "scale=1024x1024,drawtext=fontfile=/usr/share/fonts/TTF/Inconsolata-Light.ttf:text='Steps\: %{expr\:trunc(n*$step_multiplier/$repeat)}':x=10:y=h-th-10:fontsize=24:fontcolor=white" -crf 28 \
-        -c:v libx264 -pix_fmt yuv420p -y "$temp_dir/temp.mp4"
-    fi
-
-    # Add 3 seconds of the last frame, then 5 seconds delay, and then fade out smoothly for 5 seconds
-    duration=$(ffmpeg -i "$temp_dir/temp.mp4" 2>&1 | grep 'Duration' | awk '{print $2}' | tr -d , | awk -F: '{print ($1 * 3600) + ($2 * 60) + $3}')
-    fade_start=$(echo "$duration + 3" | bc)
-    ffmpeg -i "$temp_dir/temp.mp4" -vf "tpad=stop_mode=clone:stop_duration=8,fade=t=out:st=$fade_start:d=5" -c:v libx264 -pix_fmt yuv420p -y "${prefix}.mp4"
-
-    # Clean up temporary files
-    rm -rf "$temp_dir"
-
-    conda activate
-}
+source ~/toolkit/zsh/png2mp4.zsh
 
 # Function: c
 # Description:
@@ -1022,10 +949,6 @@ c() {
     conda activate comfyui
     python main.py --listen 0.0.0.0 --preview-method taesd --use-pytorch-cross-attention --disable-xformers --front-end-version Comfy-Org/ComfyUI_frontend@latest --fast
 }
-
-# Function: conda_prompt_info
-# Description:
-#   This function displays information about the currently active Conda environment.
 #
 # Usage:
 #   conda_prompt_info

zsh/png2mp4.zsh

@@ -1,8 +1,126 @@
+# 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. Deactivates conda environment
+# 2. Creates a temporary directory for processing
+# 3. Finds all PNG files in the current directory
+# 4. Uses the current directory name as the output filename prefix
+# 5. Copies and optionally repeats images to the temp directory
+# 6. 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 to remove 6 zeros)
+#    - High-quality encoding settings
+# 7. Adds padding and fade-out effect to the final video
+# 8. Cleans up temporary files
+# 9. Reactivates conda environment
+#
+# Requirements:
+#   - ffmpeg with libx265 support
+#   - bc (basic calculator)
+#   - Inconsolata-Light.ttf font in /usr/share/fonts/TTF/
+png2mp4_x265() {
+    conda deactivate
+    local max_images=""
+    local step_multiplier=1
+    local repeat=1
+    local temp_dir="/home/kade/.local/tmp"
+    while [[ "$#" -gt 0 ]]; do
+        case $1 in
+            --max) max_images="$2"; shift ;;
+            --step) step_multiplier="$2"; shift ;;
+            --repeat) repeat="$2"; shift ;;
+            *) echo "Unknown parameter passed: $1"; return 1 ;;
+        esac
+        shift
+    done
+    echo "Creating temporary directory..."
+    mkdir -p "$temp_dir"
+    echo "Checking for PNG files..."
+    png_files=($(/usr/bin/env ls *.png 2>/dev/null))
+    if [ ${#png_files[@]} -eq 0 ]; then
+        echo "Error: No PNG files found in the current directory."
+        return 1
+    fi
+    echo "Setting output filename..."
+    output_filename="$(basename "$(pwd)")_x265.mp4"
+    echo "Output filename: $output_filename"
+    echo "Creating repeated images..."
+    for img in "${png_files[@]}"; do
+        for i in $(seq 1 $repeat); do
+            cp "$img" "$temp_dir/${img%.*}_${i}.png"
+        done
+    done
+    echo "Running ffmpeg with x265 encoding..."
+    if [[ -n "$max_images" ]]; then
+        ffmpeg -framerate 60 -pattern_type glob -i "$temp_dir/*.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" \
+        -c:v libx265 -preset slower -x265-params "crf=18:qcomp=0.8:aq-mode=3:aq-strength=0.8:deblock=-1,-1" -pix_fmt yuv420p -y "$temp_dir/temp.mp4"
+    else
+        ffmpeg -framerate 60 -pattern_type glob -i "$temp_dir/*.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" \
+        -c:v libx265 -preset slower -x265-params "crf=18:qcomp=0.8:aq-mode=3:aq-strength=0.8:deblock=-1,-1" -pix_fmt yuv420p -y "$temp_dir/temp.mp4"
+    fi
+    if [ $? -ne 0 ]; then
+        echo "Error: ffmpeg command failed."
+        rm -rf "$temp_dir"
+        return 1
+    fi
+    echo "Processing final video..."
+    duration=$(ffmpeg -i "$temp_dir/temp.mp4" 2>&1 | grep 'Duration' | awk '{print $2}' | tr -d , | awk -F: '{print ($1 * 3600) + ($2 * 60) + $3}')
+    fade_start=$(echo "$duration + 3" | bc)
+    ffmpeg -i "$temp_dir/temp.mp4" -vf "tpad=stop_mode=clone:stop_duration=8,fade=t=out:st=$fade_start:d=5" \
+    -c:v libx265 -preset slower -x265-params "crf=18:qcomp=0.8:aq-mode=3:aq-strength=0.8:deblock=-1,-1" -pix_fmt yuv420p -y "$output_filename"
+    if [ $? -ne 0 ]; then
+        echo "Error: Final ffmpeg processing failed."
+        rm -rf "$temp_dir"
+        return 1
+    fi
+    echo "Cleaning up temporary files..."
+    rm -rf "$temp_dir"
+    conda activate
+    echo "Process completed successfully."
+}
+
+# png2mp4()
+# Converts a series of PNG images to an MP4 video using x264 encoding
+#
+# Usage: png2mp4 [--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. Deactivates conda environment
+# 2. Creates a temporary directory for processing
+# 3. Finds all PNG files in the current directory
+# 4. Uses the current directory name as the output filename prefix
+# 5. Copies and optionally repeats images to the temp directory
+# 6. Uses ffmpeg to create an MP4 with x264 encoding, including:
+#    - Frame rate of 60 fps
+#    - Image scaling to 1024x1024
+#    - Step count overlay text (divided by 1,000,000 to remove 6 zeros)
+#    - CRF value of 28 for compression
+# 7. Adds padding and fade-out effect to the final video
+# 8. Cleans up temporary files
+# 9. Reactivates conda environment
+#
+# Requirements:
+#   - ffmpeg with libx264 support
+#   - bc (basic calculator)
+#   - Inconsolata-Light.ttf font in /usr/share/fonts/TTF/
 png2mp4() {
     conda deactivate
     local max_images=""
     local step_multiplier=1
-    local prefix=""
     local repeat=1
     local temp_dir="/home/kade/.local/tmp"
 
@@ -26,9 +144,9 @@ png2mp4() {
         return 1
     fi
 
-    echo "Extracting prefix..."
-    prefix=$(echo "${png_files[0]}" | cut -d'-' -f1 | tr -d '[:cntrl:]' | tr -d '[]')
-    echo "Prefix: $prefix"
+    echo "Setting output filename..."
+    output_filename="$(basename "$(pwd)").mp4"
+    echo "Output filename: $output_filename"
 
     echo "Creating repeated images..."
     for img in "${png_files[@]}"; do
@@ -39,10 +157,10 @@ png2mp4() {
 
     echo "Running ffmpeg..."
     if [[ -n "$max_images" ]]; then
-        ffmpeg -framerate 60 -pattern_type glob -i "$temp_dir/*.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)}':x=10:y=h-th-10:fontsize=24:fontcolor=white" -crf 28 \
+        ffmpeg -framerate 60 -pattern_type glob -i "$temp_dir/*.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.mp4"
     else
-        ffmpeg -framerate 60 -pattern_type glob -i "$temp_dir/*.png" -vf "scale=1024x1024,drawtext=fontfile=/usr/share/fonts/TTF/Inconsolata-Light.ttf:text='Steps\: %{expr\:trunc(n*$step_multiplier/$repeat)}':x=10:y=h-th-10:fontsize=24:fontcolor=white" -crf 28 \
+        ffmpeg -framerate 60 -pattern_type glob -i "$temp_dir/*.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.mp4"
     fi
 
@@ -55,7 +173,7 @@ png2mp4() {
     echo "Processing final video..."
     duration=$(ffmpeg -i "$temp_dir/temp.mp4" 2>&1 | grep 'Duration' | awk '{print $2}' | tr -d , | awk -F: '{print ($1 * 3600) + ($2 * 60) + $3}')
     fade_start=$(echo "$duration + 3" | bc)
-    ffmpeg -i "$temp_dir/temp.mp4" -vf "tpad=stop_mode=clone:stop_duration=8,fade=t=out:st=$fade_start:d=5" -c:v libx264 -pix_fmt yuv420p -y "${prefix}.mp4"
+    ffmpeg -i "$temp_dir/temp.mp4" -vf "tpad=stop_mode=clone:stop_duration=8,fade=t=out:st=$fade_start:d=5" -c:v libx264 -pix_fmt yuv420p -y "$output_filename"
 
     if [ $? -ne 0 ]; then
         echo "Error: Final ffmpeg processing failed."