fuck zsh

png2mp4

@@ -0,0 +1,144 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+#!/usr/bin/env python3
+
+import argparse
+import os
+import shutil
+import subprocess
+import glob
+import re
+from pathlib import Path
+
+def create_video(input_dir, sample, temp_dir, step_multiplier, repeat, max_images):
+    output_filename = f"{os.path.basename(os.getcwd())}_sample{sample}.mp4"
+    print(f"Processing sample {sample}. Output filename: {output_filename}")
+    
+    # Create repeated images
+    print(f"Creating repeated images for sample {sample}...")
+    for img in glob.glob(f"{input_dir}/*_{sample}_*.png"):
+        for i in range(repeat):
+            base = os.path.splitext(os.path.basename(img))[0]
+            shutil.copy(img, f"{temp_dir}/{base}_{i+1}.png")
+    
+    # Prepare ffmpeg options
+    vf_options = "scale=1024x1024"
+    if step_multiplier:
+        vf_options += f",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"
+    
+    if max_images:
+        vf_options = f"select='not(mod(n\\,{max_images}))',{vf_options}"
+    
+    # Run first ffmpeg command
+    temp_output = f"{temp_dir}/temp_{sample}.mp4"
+    ffmpeg_cmd = [
+        "ffmpeg", "-framerate", "60",
+        "-pattern_type", "glob", "-i", f"{temp_dir}/*_{sample}_*.png",
+        "-vf", vf_options,
+        "-crf", "18", "-c:v", "libx264", "-b:v", "12M",
+        "-pix_fmt", "yuv420p", "-y", temp_output
+    ]
+    
+    try:
+        subprocess.run(ffmpeg_cmd, check=True)
+    except subprocess.CalledProcessError:
+        print(f"Error: ffmpeg command failed for sample {sample}.")
+        return False
+    
+    # Get duration and process final video
+    try:
+        duration_cmd = ["ffmpeg", "-i", temp_output]
+        result = subprocess.run(duration_cmd, capture_output=True, text=True, stderr=subprocess.PIPE)
+        duration_match = re.search(r'Duration: (\d{2}):(\d{2}):(\d{2})', result.stderr)
+        if duration_match:
+            hours, minutes, seconds = map(float, duration_match.groups())
+            duration = hours * 3600 + minutes * 60 + seconds
+            fade_start = duration + 3
+            
+            final_cmd = [
+                "ffmpeg", "-i", temp_output,
+                "-vf", f"tpad=stop_mode=clone:stop_duration=8,fade=t=out:st={fade_start}:d=5",
+                "-c:v", "libx264", "-b:v", "12M", "-crf", "18",
+                "-pix_fmt", "yuv420p", "-y", output_filename
+            ]
+            subprocess.run(final_cmd, check=True)
+        else:
+            print("Error: Could not determine video duration.")
+            return False
+            
+    except subprocess.CalledProcessError:
+        print(f"Error: Final ffmpeg processing failed for sample {sample}.")
+        return False
+    
+    # Clean up temporary files for this sample
+    for f in glob.glob(f"{temp_dir}/*_{sample}_*.png"):
+        os.remove(f)
+    os.remove(temp_output)
+    
+    return True
+
+def get_step_size_from_filenames(sample):
+    files = sorted(glob.glob(f"*_{sample}_*.png"))
+    if len(files) < 2:
+        return None
+    
+    # Extract step numbers from first two files
+    pattern = r'_(\d{6})_'
+    first_match = re.search(pattern, files[0])
+    second_match = re.search(pattern, files[1])
+    
+    if first_match and second_match:
+        first_step = int(first_match.group(1))
+        second_step = int(second_match.group(1))
+        return second_step - first_step
+    return None
+
+def main():
+    parser = argparse.ArgumentParser(description='Convert PNG sequence to MP4')
+    parser.add_argument('--max', type=int, help='Maximum number of images')
+    parser.add_argument('--step', type=int, help='Step multiplier')
+    parser.add_argument('--repeat', type=int, default=1, help='Repeat count')
+    parser.add_argument('--steps-from-filename', action='store_true', help='Calculate steps from filename')
+    
+    args = parser.parse_args()
+    
+    # Create temporary directory
+    temp_dir = os.path.expanduser("~/.local/tmp")
+    os.makedirs(temp_dir, exist_ok=True)
+    print("Created temporary directory...")
+    
+    # Check for PNG files
+    png_files = glob.glob("*.png")
+    if not png_files:
+        print("Error: No PNG files found in the current directory.")
+        return 1
+    
+    # Find all unique sample numbers
+    sample_pattern = r'_(\d{2})_'
+    samples = sorted(set(re.findall(sample_pattern, ' '.join(png_files))))
+    
+    for sample in samples:
+        if args.steps_from_filename:
+            step_multiplier = get_step_size_from_filenames(sample)
+            if step_multiplier:
+                print(f"Detected step size: {step_multiplier}")
+            else:
+                print("Error: Could not determine step size from filenames")
+                continue
+        else:
+            step_multiplier = args.step
+        
+        success = create_video(".", sample, temp_dir, step_multiplier, args.repeat, args.max)
+        if not success:
+            shutil.rmtree(temp_dir)
+            return 1
+    
+    # Clean up
+    print("Cleaning up temporary directory...")
+    shutil.rmtree(temp_dir)
+    print("All samples processed successfully.")
+    return 0
+
+if __name__ == "__main__":
+    exit(main())

zsh/png2mp4.zsh

@@ -1,97 +0,0 @@
-png2mp4() {
-    local max_images=""
-    local step_multiplier=""
-    local repeat=1
-    local temp_dir="/home/kade/.local/tmp"
-    local steps_from_filename=0
-
-    while [[ "$#" -gt 0 ]]; do
-        case $1 in
-            --max) max_images="$2"; shift ;;
-            --step) step_multiplier="$2"; shift ;;
-            --repeat) repeat="$2"; shift ;;
-            --steps-from-filename) steps_from_filename=1; 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
-
-    # 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}.mp4"
-        echo "Processing sample ${sample}. Output filename: $output_filename"
-
-        # Calculate step size from filenames if requested
-        if [[ $steps_from_filename -eq 1 ]]; then
-            # Get first two files for this sample
-            local first_file=$(ls *_${sample}_*.png | sort | head -n 1)
-            local second_file=$(ls *_${sample}_*.png | sort | head -n 2 | tail -n 1)
-            
-            # Extract step numbers from filenames
-            local first_step=$(echo $first_file | grep -o '_[0-9]\{6\}_' | tr -d '_')
-            local second_step=$(echo $second_file | grep -o '_[0-9]\{6\}_' | tr -d '_')
-            
-            # Calculate step difference using bc
-            step_multiplier=$(echo "$second_step - $first_step" | bc)
-            echo "Detected step size: $step_multiplier"
-        fi
-
-        echo "Creating repeated images for sample ${sample}..."
-        for img in *_${sample}_*.png; do
-            for i in $(seq 1 $repeat); do
-                cp "$img" "$temp_dir/${img%.*}_${i}.png"
-            done
-        done
-
-        echo "Running ffmpeg for sample ${sample}..."
-        local vf_options="scale=1024x1024"
-        if [[ -n "$step_multiplier" ]]; then
-            # Modified to use printf for integer formatting
-            vf_options+=",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"
-        fi
-        
-        if [[ -n "$max_images" ]]; then
-            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 18 \
-        -c:v libx264 -b:v 12M -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"
-            return 1
-        fi
-
-        echo "Processing final video for sample ${sample}..."
-        duration=$(ffmpeg -i "$temp_dir/temp_${sample}.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_${sample}.mp4" -vf "tpad=stop_mode=clone:stop_duration=8,fade=t=out:st=$fade_start:d=5" -c:v libx264 -b:v 12M -crf 18 -pix_fmt yuv420p -y "$output_filename"
-
-        if [ $? -ne 0 ]; then
-            echo "Error: Final ffmpeg processing failed for sample ${sample}."
-            rm -rf "$temp_dir"
-            return 1
-        fi
-
-        # Clean up temporary files for this sample
-        rm "$temp_dir"/*_${sample}_*.png "$temp_dir/temp_${sample}.mp4"
-    done
-
-    echo "Cleaning up temporary directory..."
-    rm -rf "$temp_dir"
-
-    echo "All samples processed successfully."
-}