Ziggle Wump Media Compressor 0.2-beta.09.10.2024

#!/data/data/com.termux/files/usr/bin/bash

# ATTENTION:

# Copying and pasting the script can introduce formatting that is improper to bash.  If the script doesn't run, you may need to use the dos2unix command to fix it.

# ex.  dos2unix ziggle_wump.sh

# --------------------
#     Licence
# --------------------

# This program is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, either version 3 of the License, or (at your option) any later version.

# This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details.

# You should have received a copy of the GNU General Public License along with  this program. If not, see <https://www.gnu.org/licenses/>.

# -----------------
#     About
# -----------------

# This script is for Termux on Android and is not associated with the apps it uses.

# Copyright Joshua Hansen and contributors: Microsoft Co-Pilot, OpenAI ChatGPT and Google Gemini.

# Shout out to Webernets https://youtu.be/0aeCfDKLfbs?si=58y58eWSiXurcHpj who gave me the inspiration for this, the one who did the heavy lifting figuring out the command line options. And this video helping me figure out how to do it in Handbrake, but I needed it all in a one-line command line option and this was it, I had to write it myself.

# -----------------------
#   Instructions
# -----------------------

# Usage: ./ziggle_wump.sh [-r resolution] [-f "ffmpeg_options"] [-h|--help]
#   -r resolution        Custom resolution height preserving aspect ratio, Downscaling your videos will significantly reduce the file size, but may introduce artifacts and shimmering.  Upscaling will produce larger file sizes and not increase quality.
#   -f "ffmpeg_options" Override default FFmpeg options with custom command line options.  ex. bash ./ziggle_wump.sh -f "-c:v libx264 -c:a libmp3lame"
#   -h, --help           Display this help message

# 1. Download or Copy this script to a file, and rename it if you want.  e.g., ziggle_wump.sh.  Putting it in your Movies folder will make it easy to find on both Android and Termux, although it's possible to put this in your /usr/bin to use it as a system app.  Your mileage may vary.
# 2. Make it executable: chmod +x ziggle_wump.sh
# 3. Run the script: bash ./ziggle_wump.sh [options]

# -------------------
#    README
# -------------------

# Ziggle Wump: The Simple FFmpeg Command Line Companion Script for Termux on Android

# DISCLAIMER:  THE SCRIPT DOES NOT ALLOW YOU TO CONVERT ENCRYPTED FILES, COPYRIGHTED CONTENT IS GENERALLY ENCRYPTED. THE SCRIPT IN ITSELF DOES NOT VIOLATE COPYRIGHT BUT THE USER MIGHT.  CODE TO PREVENT CIRCUMVENTION IS BEYOND MY CAPABILITY.  NOR SHOULD IT BE MY RESPONSIBILITY, IT'S THE COPYRIGHT HOLDERS RESPONSIBILITY TO ENCRYPT THEIR MEDIA.  IT IS NOT RECOMMENDED TO USE THE SCRIPT TO VIOLATE COPYRIGHT LAW.  US COPYRIGHT LAW ALLOWS FOR SPACESHIFTING, AND FAIR USE OF COPYRIGHTED MATERIAL

# NOTE:  Some phones have a battery saving feature such as the Galaxy S24, that can impact the encoding process and leave you with partially encoded files.  Please make sure Termux is in focus otherwise, either full screen or split screen if you want to do other things, and keep the screen on while encoding.

# Check out https://dontkillmyapp.com/ for more information and perhaps find a fix for your particular phone.

# Supported File Types:
# Video:  mp4 mkv avi mov flv wmv webm mts
# Audio: mp3 wav flac aac ogg m4a

# -----------------------------
#        Variables
# -----------------------------

# Values for video output settings.  Either the user defined value or the input video original value will be selected, whichever is lower.  (empty implies original value)
resolution=600 # Output resolution by height.
output_fps=30
max_video_bitrate=1500

# Initialize ffmpeg_custom_options
ffmpeg_custom_options="-c:v libx265 -crf 23 -preset slow -x265-params \"deblock=-1:no-sao=1:keyint=250:aq-mode=3:psy-rd=0.75:psy-rdoq=2.0:rd=4:rdoq-level=1:rect=0:strong-intra-smoothing=0\" -c:a libopus -b:a 96k -vbr on -ac 2 -af \"loudnorm=I=-23:LRA=7:TP=-2\" -c:s srt -c:s ass -c:s ssa -c:s mov_text -c:s webvtt -c:s copy"

# Full Log file path
log_file="$HOME/storage/shared/Movies/VideoDrop/convert.log"

# Shortened log file path for terminal output
log_file_echo=$(echo "$log_file" | sed 's|/data/data/com.termux/files/home|.../home|g')

# Full directory paths
video_drop_dir="$HOME/storage/shared/Movies/VideoDrop"
video_processing_dir="$HOME/storage/shared/Movies/VideoProcessing"
output_dir_base="$HOME/storage/shared/Movies/VideoConverted"

# Shortened directory paths for terminal output
video_drop_dir_echo=$(echo "$video_drop_dir" | sed 's|/data/data/com.termux/files/home|.../home|g')
video_processing_dir_echo=$(echo "$video_processing_dir" | sed 's|/data/data/com.termux/files/home|.../home|g')
output_dir_base_echo=$(echo "$output_dir_base" | sed 's|/data/data/com.termux/files/home|.../home|g')

# Add maxrate and bufsize options if max_video_bitrate is set
if [ -n "$max_video_bitrate" ]; then
    ffmpeg_custom_options="$ffmpeg_custom_options -maxrate ${max_video_bitrate}k -bufsize 15000k"
fi

echo "FFmpeg options: $ffmpeg_custom_options" | tee -a "$log_file"
sleep 1

# -----------------------------
#    Command Flags
# -----------------------------

# Function to display help message
show_help() {
    width=$(tput cols)
    echo "Place your media files in $video_drop_dir_echo." | tee -a "$log_file" | fmt -w $width
    echo "Usage: $0 [-r resolution] [-f \"ffmpeg_options\"] [-h|--help]" | fmt -w $width
    echo "  -r resolution ex. bash $0 -r 720               Custom resolution height preserving aspect ratio, Downscaling your videos will significantly reduce the file size, but may introduce artifacts and shimmering.  Upscaling will produce larger file sizes and not increase quality." | fmt -w $width
    echo "  -f \"ffmpeg_options\"  Override default FFmpeg options with custom command line options.  ex. bash $0 -f \""-c:v libx264 -c:a libmp3lame\" | fmt -w $width
    echo "  -h, --help           Display this help message" | fmt -w $width
    exit 0
}

# Parse command-line options
while getopts "r:f:h" opt; do
    case $opt in
        r) resolution="$OPTARG" ;;  # Set custom resolution by height
        f) ffmpeg_custom_options="$OPTARG" ;;  # Set custom FFmpeg options
        h) show_help ;;  # Display help
        *) show_help ;;  # Display help for invalid options
    esac
done

shift $((OPTIND -1))

# ---------------------------------
#       Dependencies
# ---------------------------------

# Ensure dependencies are installed and all packages are up to date.
    echo "Updating, please wait..."
    sleep 1
    pkg update && pkg upgrade -y
    if ! command -v bc &> /dev/null; then
        pkg install bc -y
    fi
    if ! command -v ffmpeg &> /dev/null; then
        pkg install ffmpeg -y
    fi
    if ! command -v ncurses-utils &> /dev/null; then
        pkg install ncurses-utils -y
    fi
echo "Update complete."
    sleep 1

# -----------------------------------------------------
#     Create Directories and log file
# -----------------------------------------------------

# Create necessary directories if they don't exist
mkdir -p "$video_drop_dir" "$video_processing_dir" "$output_dir_base"

# Change to the VideoDrop directory
cd "$video_drop_dir" || { echo "Directory change failed"; exit 1; }

# Check if log file exists
if [ -f "$log_file" ]; then
    echo "Old log file found.  Deleting..."
    rm "$log_file"
    echo "File deleted."
else
    echo "Log File does not exist.  The script will create $log_file_echo"
fi

# ------------------------
#     Start script
# ------------------------

width=$(tput cols)

echo "Running... Press Ctrl+C to stop."
sleep 1

echo ""
echo "NOTE:  Some phones have a battery saving feature such as the Galaxy S24, that can impact the encoding process and leave you with partially encoded files.  Please make sure Termux is in focus, either full screen or split screen if you want to do other things, and keep the screen on while encoding." | tee -a "$log_file" | fmt -w $width

echo ""
echo "Check out https://dontkillmyapp.com/ for more information and perhaps find a fix for your particular phone." | tee -a "$log_file" | fmt -w $width

echo ""
# Function to prompt the user to continue or quit
prompt_continue_or_quit() {
    while true; do
        # Recursively find and process videos and audio
        video_count=$(find "$video_drop_dir" -type f \( -iname '*.mp3' -o -iname '*.wav' -o -iname '*.flac' -o -iname '*.aac' -o -iname '*.ogg' -o -iname '*.m4a' -o -iname '*.mp4' -o -iname '*.mkv' -o -iname '*.avi' -o -iname '*.mov' -o -iname '*.flv' -o -iname '*.wmv' -o -iname '*.webm' -o -iname '*.mts' \) | wc -l)

        if [ "$video_count" -eq 0 ]; then
            echo "No videos or audio found. $video_drop_dir_echo folder created. Place your videos and audio here including in folders using your favorite file manager for Android, and then run the script again. Supported File Types: Video: mp4 mkv avi mov flv wmv webm mts Audio: mp3 wav flac aac ogg m4a" | tee -a "$log_file" | fmt -w $width
            exit 0
        fi

        # Prompt text
        prompt_text="Compatible media files detected in $video_drop_dir_echo. You can start encoding now. Do you wish to proceed? (Y/N): "

        # Format the prompt text
        formatted_prompt=$(echo "$prompt_text" | fmt -w $width)

        # Read user input with formatted prompt
        read -p "$formatted_prompt" yn
        case $yn in
            [Yy]* ) return 0;;  # Continue
            [Nn]* ) echo "Exiting script."; exit 0;;  # Quit
            * ) echo "Please answer Y or N.";;
        esac
    done
}

# Function to clean up file names
clean_file_names() {
    find "$video_drop_dir" -type f \( -iname '*.mp3' -o -iname '*.wav' -o -iname '*.flac' -o -iname '*.aac' -o -iname '*.ogg' -o -iname '*.m4a' -o -iname '*.mp4' -o -iname '*.mkv' -o -iname '*.avi' -o -iname '*.mov' -o -iname '*.flv' -o -iname '*.wmv' -o -iname '*.webm' -o -iname '*.mts' \) | while read -r file; do
        dir=$(dirname "$file")
        base=$(basename "$file")
        new_base=$(echo "$base" | sed 's/[^a-zA-Z0-9 ._-]//g' | tr -s ' ')
        new_file="$dir/$new_base"
        if [ "$file" != "$new_file" ]; then
            mv "$file" "$new_file"
        fi
    done
}

# Clean up file names before processing
clean_file_names

# Prompt the user to continue or quit?  Comment line to skip prompt prior to encoding.
prompt_continue_or_quit

echo "Starting conversion process..." | tee -a "$log_file"
sleep 1

# -----------------------------
#    Audio Encoder
# -----------------------------

width=$(tput cols)

# Function to check the log file for "Killed" message
check_for_killed_message() {
    if grep -q " Killed     " "$log_file"; then
        echo "$(date '+%Y-%m-%d %H:%M:%S') Error: Process was killed. Check log file for details." | tee -a "$log_file"
        return 1
    fi
    return 0
}

# Function to process audio files
process_audio() {
    local audio="$1"
    local audio_echo=$(echo "$audio" | sed 's|/data/data/com.termux/files/home|/home|g')
    local relative_path="${audio#$video_drop_dir/}"
    local dir_path=$(dirname "$relative_path")
    local base_name=$(basename "$relative_path" | tr -cd '[:alnum:]._ -')
    local output_dir="$output_dir_base/$dir_path"
    local output_file="$output_dir/${base_name}_converted.opus"
    local temp_output_file="$output_file.tmp"

    mkdir -p "$output_dir"  # Create output directory

    echo "$(date '+%Y-%m-%d %H:%M:%S') Processing $audio to $output_file" | tee -a "$log_file"

    # Construct the FFmpeg command for audio processing
    local ffmpeg_command="ffmpeg -nostdin -loglevel error -stats -stats_period 1 -y -i \"$audio\" -c:a libopus -b:a 96k -vbr on -ac 2 -af \"loudnorm=I=-23:LRA=7:TP=-2\" -f opus \"$temp_output_file\""

    # Execute the FFmpeg command
    eval $ffmpeg_command 2>&1 | tee -a "$log_file"

    # Check for "Killed" message in the log file
    if ! check_for_killed_message; then
        echo "$(date '+%Y-%m-%d %H:%M:%S') Error: Process was killed. Exiting." | tee -a "$log_file"
        exit 1
    fi

    if [ $? -eq 0 ]; then
        mv "$temp_output_file" "$output_file"  # Rename the temporary file to the final output file
        mkdir -p "$video_processing_dir/$dir_path"  # Create processing directory
        mv "$audio" "$video_processing_dir/$relative_path"
        echo ""
        echo "$(date '+%Y-%m-%d %H:%M:%S') Converted $audio_echo. Your original files will be in the $video_processing_dir_echo folder for comparison. Your new files are in $output_dir_base_echo" | tee -a "$log_file" | fmt -w $width
    else
        echo ""
        echo "$(date '+%Y-%m-%d %H:%M:%S') Error processing \"$audio\". Check log file for details." | tee -a "$log_file"
        exit 1
    fi
}

# ------------------------------
#      Video Encoder
# ------------------------------

width=$(tput cols)

# Function to check the log file for "Killed" message
check_for_killed_message() {
    if grep -q " Killed     " "$log_file"; then
        echo "$(date '+%Y-%m-%d %H:%M:%S') Error: Process was killed. Check log file for details." | tee -a "$log_file" | fmt -w $width
        exit 1
    fi
}

process_video() {
    local video="$1"
    local video_echo=$(echo "$video" | sed 's|/data/data/com.termux/files/home|.../home|g')
    local relative_path="${video#$video_drop_dir/}"
    local dir_path=$(dirname "$relative_path")
    local base_name=$(basename "$relative_path" | tr -cd '[:alnum:]._ -')
    local output_dir="$output_dir_base/$dir_path"
    local output_file="$output_dir/${base_name}_converted.mkv"
    local scale_filter=""

    # Get input video resolution and clean up any trailing commas or spaces
    input_resolution=$(ffprobe -v error -select_streams v:0 -show_entries stream=height -of csv=p=0 "$video" | tr -d '[:space:],')

    if [ -n "$resolution" ]; then
        if [ "$input_resolution" -gt "$resolution" ]; then
            scale_filter="scale=-2:$resolution:flags=lanczos"
            echo "Input resolution ($input_resolution) is higher than user-defined resolution ($resolution). Using user-defined resolution." | tee -a "$log_file"
        else
            echo "Input resolution ($input_resolution) is lower or equal to user-defined resolution ($resolution). Keeping original resolution." | tee -a "$log_file"
        fi
    else
        echo "Resolution set to: original" | tee -a "$log_file"
    fi

    mkdir -p "$output_dir"  # Create output directory

    echo "$(date '+%Y-%m-%d %H:%M:%S') Processing $video to $output_file" | tee -a "$log_file"

    # Get the frame rate of the input video
    input_fps=$(ffprobe -v error -select_streams v:0 -show_entries stream=avg_frame_rate,r_frame_rate -of csv=p=0 "$video" | awk -F'/' '{if ($2) print $1/$2; else print $1}' | bc -l)

    # Define common frame rates
    common_fps=(23.976 24 25 29.97 30 50 59.94 60)

    # Function to find the nearest valid frame rate
    nearest_fps() {
        local fps=$1
        local nearest=${common_fps[0]}
        local min_diff=$(echo "scale=5; $fps - ${common_fps[0]}" | bc | awk '{print ($1 >= 0) ? $1 : -$1}')
        for rate in "${common_fps[@]}"; do
            local diff=$(echo "scale=5; $fps - $rate" | bc | awk '{print ($1 >= 0) ? $1 : -$1}')
            if (( $(echo "$diff < $min_diff" | bc -l) )); then
                min_diff=$diff
                nearest=$rate
            fi
        done
        echo $nearest
    }

    # Round the frame rate to the nearest valid frame rate
    rounded_fps=$(nearest_fps $input_fps)

    # Determine the lower frame rate
    if (( $(echo "$rounded_fps > $output_fps" | bc -l) )); then
        final_fps=$output_fps
    else
        final_fps=$rounded_fps
    fi

# Set the initial video filter option
video_filter_option="${scale_filter},fps=$final_fps"

# Detect the pixel format using ffprobe
input_file="$video"
pix_fmt=$(ffprobe -v error -select_streams v:0 -show_entries stream=pix_fmt -of default=noprint_wrappers=1:nokey=1 "$input_file")

echo "Pixel format: $pix_fmt"
sleep 1

# Validate the pixel format
valid_pix_fmt=$(ffmpeg -pix_fmts | grep -w "$pix_fmt")

# Check if 10-bit encoding is selected
if [[ "$pix_fmt" == *"10le"* || "$pix_fmt" == *"10be"* ]]; then
    video_filter_option="$video_filter_option,deband=1thr=0.01:2thr=0.01:3thr=0.01:4thr=0.01:range=4:direction=-3.14:blur=1:coupling=0"
fi

# Check if the video is progressive or interlaced
field_order=$(ffprobe -v error -select_streams v:0 -show_entries stream=field_order -of default=noprint_wrappers=1:nokey=1 "$input_file")

# Apply yadif filter if the video is interlaced or if field_order is unknown
if [ "$field_order" == "progressive" ]; then
    if [ -n "$scale_filter" ]; then
        video_filter_option="${scale_filter},fps=$final_fps"
    else
        video_filter_option="fps=$final_fps"
    fi
else
    if [ -n "$scale_filter" ]; then
        video_filter_option="yadif=2,${scale_filter},fps=$final_fps"
    else
        video_filter_option="yadif=2,fps=$final_fps"
    fi
fi

# Add noise to the output video to simulate film grain and add dithering to smooth out bright colors.
video_filter_option="$video_filter_option,noise=alls=2:allf=t+u"

# Log the final video filter option
echo "Video filter option: $video_filter_option" | tee -a "$log_file"
sleep 1

# Detect subtitle codecs and convert unsupported ones
subtitle_codecs=$(ffprobe -v error -select_streams s -show_entries stream=codec_name -of csv=p=0 "$input_file")
subtitle_map=""
if [ -n "$subtitle_codecs" ]; then
    i=0
    while read -r codec; do
        case "$codec" in
            srt|ass|ssa|webvtt|dvdsub|pgs|vobsub)
                subtitle_map="$subtitle_map -c:s:$i copy"
                ;;
            mov_text|other_unsupported_codec)
                subtitle_map="$subtitle_map -c:s:$i srt"
                ;;
            *)
                echo "Unsupported subtitle codec: $codec. Skipping." | tee -a "$log_file"
                ;;
        esac
        i=$((i + 1))
    done <<< "$subtitle_codecs"
else
    echo "No subtitle streams detected." | tee -a "$log_file"
fi

# Combine common and custom FFmpeg options
combined_ffmpeg_options="-nostdin -loglevel error -stats -stats_period 5 -analyzeduration 2147483647 -probesize 2147483647 -y -i \"$video\" $ffmpeg_custom_options"

# Construct the FFmpeg command
local ffmpeg_command
if [ -n "$valid_pix_fmt" ]; then
    if [ -n "$subtitle_map" ]; then
        ffmpeg_command="ffmpeg $combined_ffmpeg_options -vf \"$video_filter_option\" -pix_fmt $pix_fmt $subtitle_map"
    else
        ffmpeg_command="ffmpeg $combined_ffmpeg_options -vf \"$video_filter_option\" -pix_fmt $pix_fmt"
    fi
else
    if [ -n "$subtitle_map" ]; then
        ffmpeg_command="ffmpeg $combined_ffmpeg_options -vf \"$video_filter_option\" $subtitle_map"
    else
        ffmpeg_command="ffmpeg $combined_ffmpeg_options -vf \"$video_filter_option\""
    fi
fi

# Ensure all subtitle streams are marked as "default: off"
subtitle_streams=$(ffprobe -v error -select_streams s -show_entries stream=index -of csv=p=0 "$input_file")
if [ -n "$subtitle_streams" ]; then
    for stream_index in $subtitle_streams; do
        ffmpeg_command="$ffmpeg_command -disposition:s:$stream_index 0"
    done
fi

# Add the output file name at the end
ffmpeg_command="$ffmpeg_command \"$output_file\""

echo "FFmpeg command: $ffmpeg_command" | tee -a "$log_file"
sleep 1

# Execute the FFmpeg command
eval $ffmpeg_command 2>&1 | tee -a "$log_file"

    # Check for "Killed" message in the log file
    check_for_killed_message

   if [ $? -eq 0 ]; then
        mkdir -p "$video_processing_dir/$dir_path"  # Create processing directory
        mv "$video" "$video_processing_dir/$relative_path"
        echo ""
        echo "$(date '+%Y-%m-%d %H:%M:%S') Converted $video_echo. Your original files will be in the $video_processing_dir_echo folder for comparison. Your new files are in $output_dir_base_echo" | tee -a "$log_file" | fmt -w $width
    else
        echo ""
        echo "$(date '+%Y-%m-%d %H:%M:%S') Error processing \"$video\". Check log file for details." | tee -a "$log_file"
        exit 1
    fi
}

# --------------------------------------
#    Start Batch Encoding
# --------------------------------------

width=$(tput cols)

# Recursively find and process videos and audio
video_count=$(find "$video_drop_dir" -type f \( -iname '*.mp3' -o -iname '*.wav' -o -iname '*.flac' -o -iname '*.aac' -o -iname '*.ogg' -o -iname '*.m4a' -o -iname '*.mp4' -o -iname '*.mkv' -o -iname '*.avi' -o -iname '*.mov' -o -iname '*.flv' -o -iname '*.wmv' -o -iname '*.webm' -o -iname '*.mts' \) | wc -l)

# Process each audio file found
find "$video_drop_dir" -type f \( -iname '*.mp3' -o -iname '*.wav' -o -iname '*.flac' -o -iname '*.aac' -o -iname '*.ogg' -o -iname '*.m4a' \) | while read -r audio; do

    stop_processing=false

    # Function to handle SIGINT (Ctrl+C)
    handle_sigint() {
        echo ""
        echo "Caught SIGINT (Ctrl+C). Exiting immediately." | fmt -w $width
        exit 1  # Exit immediately without performing cleanup
    }

    # Trap SIGINT and call the handle_sigint function
    trap handle_sigint SIGINT

    if [ "$stop_processing" = true ]; then
        echo "Stopping processing due to SIGINT."
        exit 1
    fi
    process_audio "$audio" || exit 1

    # Remove the relative path folder if it's empty
    rmdir "$(dirname "$audio")" --ignore-fail-on-non-empty
done

# Process each video found
find "$video_drop_dir" -type f \( -iname '*.mp4' -o -iname '*.mkv' -o -iname '*.avi' -o -iname '*.mov' -o -iname '*.flv' -o -iname '*.wmv' -o -iname '*.webm'  -o -iname '*.mts' \) | while read -r video; do

    stop_processing=false

    # Function to handle SIGINT (Ctrl+C)
    handle_sigint() {
        echo ""
        echo "Caught SIGINT (Ctrl+C). Exiting immediately." | fmt -w $width
        exit 1  # Exit immediately without performing cleanup
    }

    # Trap SIGINT and call the handle_sigint function
    trap handle_sigint SIGINT

    if [ "$stop_processing" = true ]; then
        echo "Stopping processing due to SIGINT."
        exit 1
    fi
    process_video "$video" || exit 1

    # Remove the relative path folder if it's empty
    rmdir "$(dirname "$video")" --ignore-fail-on-non-empty
done

# -----------------------------
#     Finishing Up
# -----------------------------

width=$(tput cols)

# final output message
echo ""
echo "Log file is in $video_drop_dir_echo.  There may be unprocessed media files or other incompatible files.  Some files may need to be remuxed for compatibility." | tee -a "$log_file" | fmt -w $width
exit 0