make the rain able to be matrix

profile.d/rain.sh

@@ -35,23 +35,201 @@
 # ------------------------------------------------------------------------------
 
 # ------------------------------------------------------------------------------
-# Let the rain fall
+# Generic rain-like engine and presets
+
+_rain_build_colors()
+{
+    local base_color="$1"
+    RAIN_ENGINE_COLORS=()
+
+    case $base_color in
+        green)
+            for i in {22..28} {34..40} {46..48}; do RAIN_ENGINE_COLORS+=("\e[38;5;${i}m"); done ;;
+        blue)
+            for i in {17..21} {27..33} {39..45}; do RAIN_ENGINE_COLORS+=("\e[38;5;${i}m"); done ;;
+        red)
+            for i in {52..52} {88..88} {124..124} {160..160} {196..201}; do RAIN_ENGINE_COLORS+=("\e[38;5;${i}m"); done ;;
+        yellow)
+            for i in {58..58} {100..100} {142..142} {184..184} {226..229}; do RAIN_ENGINE_COLORS+=("\e[38;5;${i}m"); done ;;
+        cyan)
+            for i in {30..31} {37..38} {44..45} {50..51}; do RAIN_ENGINE_COLORS+=("\e[38;5;${i}m"); done ;;
+        *)
+            RAIN_ENGINE_COLORS=("\e[37m" "\e[37;1m")
+            for i in {244..255}; do RAIN_ENGINE_COLORS+=("\e[38;5;${i}m"); done ;;
+    esac
+}
+
+_rain_build_chars()
+{
+    local mode="$1"
+    local charset="$2"
+    RAIN_ENGINE_CHARS=()
+
+    case "$mode" in
+        matrix)
+            case "$charset" in
+                ""|binary)
+                    RAIN_ENGINE_CHARS=("0" "1")
+                    ;;
+                kana|kanji)
+                    # Half-width katakana set, generally rendered as single-cell glyphs.
+                    RAIN_ENGINE_CHARS=("ｱ" "ｲ" "ｳ" "ｴ" "ｵ" "ｶ" "ｷ" "ｸ" "ｹ" "ｺ" "ｻ" "ｼ" "ｽ" "ｾ" "ｿ" "ﾀ" "ﾁ" "ﾂ" "ﾃ" "ﾄ" "ﾅ" "ﾆ" "ﾇ" "ﾈ" "ﾉ" "ﾊ" "ﾋ" "ﾌ" "ﾍ" "ﾎ" "ﾏ" "ﾐ" "ﾑ" "ﾒ" "ﾓ" "ﾔ" "ﾕ" "ﾖ" "ﾗ" "ﾘ" "ﾙ" "ﾚ" "ﾛ" "ﾜ" "ﾝ")
+                    ;;
+                ascii)
+                    RAIN_ENGINE_CHARS=("0" "1" "2" "3" "4" "5" "6" "7" "8" "9" "A" "B" "C" "D" "E" "F")
+                    ;;
+                *)
+                    disp E "Unknown charset: ${charset} (supported: binary, kana, ascii)."
+                    return 1
+                    ;;
+            esac
+            ;;
+        *)
+            RAIN_ENGINE_CHARS=("|" "│" "┃" "┆" "┇" "┊" "┋" "╽" "╿")
+            ;;
+    esac
+
+    return 0
+}
+
+_rain_engine()
+{
+    local step_duration="$1"
+    local base_color="$2"
+    local mode="$3"
+    local charset="$4"
+
+    command -v tput >/dev/null 2>&1 || {
+        disp E "The program 'tput' is required but not installed."
+        return 1
+    }
+
+    _rain_build_colors "$base_color"
+    _rain_build_chars "$mode" "$charset" || return 1
+
+    local rain_colors=("${RAIN_ENGINE_COLORS[@]}")
+    local rain_chars=("${RAIN_ENGINE_CHARS[@]}")
+    local rain_color_tab=${#rain_colors[@]}
+    local rain_tab=${#rain_chars[@]}
+
+    local exit_st=0
+    local num_rain_metadata=5
+    local term_height=0 term_width=0
+    local X=0 Y=0 drop_length=0 rain_drop=0
+    local max_rain_width=0 max_rain_height=0
+    local new_rain_odd=0 falling_odd=0
+
+    sigwinch()
+    {
+        term_width=$(tput cols)
+        term_height=$(tput lines)
+
+        case "$mode" in
+            matrix)
+                ((max_rain_width = term_width * term_height / 3))
+                ((max_rain_height = term_height < 8 ? 1 : term_height / 6))
+                ((new_rain_odd = term_height > 50 ? 100 : term_height * 2))
+                ((new_rain_odd = new_rain_odd * 85 / 100))
+                ((falling_odd = 100))
+                ;;
+            *)
+                ((max_rain_width = term_width * term_height / 4))
+                ((max_rain_height = term_height < 10 ? 1 : term_height / 10))
+                ((new_rain_odd = term_height > 50 ? 100 : term_height * 2))
+                ((new_rain_odd = new_rain_odd * 75 / 100))
+                ((falling_odd = term_height > 25 ? 100 : term_height * 4))
+                ((falling_odd = falling_odd * 90 / 100))
+                ;;
+        esac
+    }
+
+    do_exit()
+    {
+        exit_st=1
+    }
+
+    do_render()
+    {
+        local idx=0 y=0 drop_color=""
+
+        for ((idx = 0; idx < num_rains * num_rain_metadata; idx += num_rain_metadata)); do
+            X=${rains[idx]}
+            Y=${rains[idx + 1]}
+            drop_length=${rains[idx + 4]}
+            for ((y = Y; y < Y + drop_length; y++)); do
+                ((y < 1 || y > term_height)) && continue
+                printf "\e[%d;%dH " "$y" "$X"
+            done
+        done
+
+        for ((idx = 0; idx < num_rains * num_rain_metadata; idx += num_rain_metadata)); do
+            if ((100 * RANDOM / 32768 < falling_odd)); then
+                if ((++rains[idx + 1] > term_height)); then
+                    rains=("${rains[@]:0:idx}" "${rains[@]:idx+num_rain_metadata:num_rains*num_rain_metadata}")
+                    ((num_rains--))
+                    continue
+                fi
+            fi
+
+            X=${rains[idx]}
+            Y=${rains[idx + 1]}
+            rain_drop=${rains[idx + 2]}
+            drop_color=${rains[idx + 3]}
+            drop_length=${rains[idx + 4]}
+
+            for ((y = Y; y < Y + drop_length; y++)); do
+                ((y < 1 || y > term_height)) && continue
+                printf "\e[%d;%dH%s%s" "$y" "$X" "$drop_color" "$rain_drop"
+            done
+        done
+    }
+
+    trap do_exit TERM INT
+    trap sigwinch WINCH
+    stty -echo
+    printf "\e[?25l"
+    printf "\e[2J"
+
+    local rains=()
+    local num_rains=0
+    local ch=""
+
+    sigwinch
+    while ((exit_st <= 0)); do
+        read -r -n 1 -t "$step_duration" ch
+        case "$ch" in
+            q|Q)
+                do_exit
+                ;;
+        esac
+
+        if ((num_rains < max_rain_width)) && ((100 * RANDOM / 32768 < new_rain_odd)); then
+            rain_drop="${rain_chars[rain_tab * RANDOM / 32768]}"
+            drop_color="${rain_colors[rain_color_tab * RANDOM / 32768]}"
+            drop_length=$((max_rain_height * RANDOM / 32768 + 1))
+            X=$((term_width * RANDOM / 32768 + 1))
+            Y=$((1 - drop_length))
+            rains=("${rains[@]}" "$X" "$Y" "$rain_drop" "$drop_color" "$drop_length")
+            ((num_rains++))
+        fi
+
+        do_render
+    done
+
+    printf "\e[%d;1H\e[0K" "$term_height"
+    printf "\e[?25h"
+    stty echo
+    trap - TERM INT
+    trap - WINCH
+}
+
+# ------------------------------------------------------------------------------
+# Let the rain fall (current style)
 # Usage: rain [OPTIONS]
-# Options:
-#   -s, --speed NUM    Set the drop delay in seconds (default: 0.050).
-#                     Lower values = faster rain.
-#   -c, --color COLOR  Set the color theme (default: white).
-#   -h, --help         Display this help message and exit.
-# Available Colors:
-#   green  : The classic Matrix digital rain
-#   blue   : Deep ocean blue gradients
-#   red    : Crimson/Blood rain
-#   yellow : Amber and gold tones
-#   cyan   : Electric cyan/turquoise
-#   white  : Greyscale and white (original style)
 rain()
 {
-    show_usage() {
+    _rain_show_usage()
+    {
         printf "Usage: rain [OPTIONS]\n"
         printf "Options:\n"
         printf "\t-s, --speed NUM    Set the drop delay in seconds (default: 0.050).\n"
@@ -59,7 +237,7 @@ rain()
         printf "\t-c, --color COLOR  Set the color theme (default: white).\n"
         printf "\t-h, --help         Display this help message and exit.\n\n"
         printf "Available Colors:\n"
-        printf "\t\e[32mgreen\e[0m\t: The classic Matrix digital rain\n"
+        printf "\t\e[32mgreen\e[0m\t: Matrix-like green shades\n"
         printf "\t\e[34mblue\e[0m\t: Deep ocean blue gradients\n"
         printf "\t\e[31mred\e[0m\t: Crimson/Blood rain\n"
         printf "\t\e[33myellow\e[0m\t: Amber and gold tones\n"
@@ -69,163 +247,124 @@ rain()
     }
 
     local step_duration=0.050
-    local base_color="white" # default color scheme, can be overridden by --color
+    local base_color="white"
 
-    # Analyse arguments
     while [[ "$#" -gt 0 ]]; do
         case $1 in
             -s|--speed)
                 if [[ -n "$2" && ! "$2" =~ ^- ]]; then
-                    step_duration="$2"; shift
+                    step_duration="$2"
+                    shift
                 else
                     disp E "--speed requires a numeric value."
-                    show_usage && return 1
+                    _rain_show_usage
+                    return 1
                 fi
                 ;;
             -c|--color)
                 if [[ -n "$2" && ! "$2" =~ ^- ]]; then
-                    base_color="$2"; shift
+                    base_color="$2"
+                    shift
                 else
                     disp E "--color requires a color name."
-                    show_usage && return 1
+                    _rain_show_usage
+                    return 1
                 fi
                 ;;
             -h|--help)
-                show_usage && return 0
+                _rain_show_usage
+                return 0
+                ;;
+            --)
+                shift
+                break
                 ;;
             *)
                 disp E "Unknown option: $1"
-                show_usage && return 1
+                _rain_show_usage
+                return 1
                 ;;
         esac
         shift
     done
 
-    # Define colors (256-colors gradients)
+    _rain_engine "$step_duration" "$base_color" "rain" ""
-    local rain_colors=()
-    case $base_color in
-        green)  # Matrix style green
-            for i in {22..28} {34..40} {46..48}; do rain_colors+=("\e[38;5;${i}m"); done ;;
-        blue)   # Deep ocean blues
-            for i in {17..21} {27..33} {39..45}; do rain_colors+=("\e[38;5;${i}m"); done ;;
-        red)    # Crimson / blood red
-            for i in {52..52} {88..88} {124..124} {160..160} {196..201}; do rain_colors+=("\e[38;5;${i}m"); done ;;
-        yellow) # Amber / gold
-            for i in {58..58} {100..100} {142..142} {184..184} {226..229}; do rain_colors+=("\e[38;5;${i}m"); done ;;
-        cyan)   # Electric cyan / turquoise
-            for i in {30..31} {37..38} {44..45} {50..51}; do rain_colors+=("\e[38;5;${i}m"); done ;;
-        *)      # Greyscale / white (original style)
-            rain_colors=("\e[37m" "\e[37;1m")
-            for i in {244..255}; do rain_colors+=("\e[38;5;${i}m"); done ;;
-    esac
-
-    local exit_st=0
-    local rain_cars=("|" "│" "┃" "┆" "┇" "┊" "┋" "╽" "╿")
-    local rain_tab=${#rain_cars[@]}
-    local rain_color_tab=${#rain_colors[@]}
-    local num_rain_metadata=5
-    local term_height=$(tput lines)
-    local term_width=$(tput cols)
-    local X=0 Y=0 drop_length=0 rain_drop=0
-    local max_rain_width=0 new_rain_odd=0 falling_odd=0
-
-    sigwinch() {
-        term_width=$(tput cols)
-        term_height=$(tput lines)
-        #step_duration=0.025
-        ((max_rain_width = term_width * term_height / 4))
-        ((max_rain_height = term_height < 10 ? 1 : term_height / 10))
-        # In percentage
-        ((new_rain_odd = term_height > 50 ? 100 : term_height * 2))
-        ((new_rain_odd = new_rain_odd * 75 / 100))
-        ((falling_odd = term_height > 25 ? 100 : term_height * 4))
-        ((falling_odd = falling_odd * 90 / 100))
-    }
-
-    do_exit() {
-        exit_st=1
-    }
-
-    do_render() {
-        # Clean screen first
-        local idx=0
-        for ((idx = 0; idx < num_rains * num_rain_metadata; idx += num_rain_metadata)); do
-            X=${rains[idx]}
-            Y=${rains[idx + 1]}
-            drop_length=${rains[idx + 4]}
-            for ((y = Y; y < Y + drop_length; y++)); do
-                ((y < 1 || y > term_height)) && continue
-                echo -ne "\e[${y};${X}H "
-            done
-        done
-
-        for ((idx = 0; idx < num_rains * num_rain_metadata; idx += num_rain_metadata)); do
-            if ((100 * RANDOM / 32768 < falling_odd)); then
-                # Falling
-                if ((++rains[idx + 1] > term_height)); then
-                    # Out of screen, bye sweet <3
-                    rains=("${rains[@]:0:idx}"
-                        "${rains[@]:idx+num_rain_metadata:num_rains*num_rain_metadata}")
-                    ((num_rains--))
-                    continue
-                fi
-            fi
-            X=${rains[idx]}
-            Y=${rains[idx + 1]}
-            rain_drop=${rains[idx + 2]}
-            drop_color=${rains[idx + 3]}
-            drop_length=${rains[idx + 4]}
-            for ((y = Y; y < Y + drop_length; y++)); do
-                ((y < 1 || y > term_height)) && continue
-                printf "\e[${y};${X}H${drop_color}${rain_drop}"
-            done
-        done
-    }
-
-    trap do_exit TERM INT
-    trap sigwinch WINCH
-    # No echo stdin and hide the cursor
-    stty -echo
-    printf "\e[?25l"
-    printf "\e[2J"
-
-    local rains=()
-    local num_rains=0
-    sigwinch
-    while ((exit_st <= 0)); do
-        if (($exit_st <= 0)); then
-            read -n 1 -t $step_duration ch
-            case "$ch" in
-            q | Q)
-                do_exit
-                ;;
-            esac
-
-            if ((num_rains < max_rain_width)) && ((100 * RANDOM / 32768 < new_rain_odd)); then
-                # Need new |, 1-based
-                rain_drop="${rain_cars[rain_tab * RANDOM / 32768]}"
-                drop_color="${rain_colors[rain_color_tab * RANDOM / 32768]}"
-                drop_length=$((max_rain_height * RANDOM / 32768 + 1))
-                X=$((term_width * RANDOM / 32768 + 1))
-                Y=$((1 - drop_length))
-                rains=("${rains[@]}" "$X" "$Y" "$rain_drop" "$drop_color" "$drop_length")
-                ((num_rains++))
-            fi
-
-            # Let rain fall!
-            do_render
-        fi
-    done
-    echo -ne "\e[${term_height};1H\e[0K"
-
-    # Show cursor and echo stdin
-    echo -ne "\e[?25h"
-    stty echo
-    unset exit_st
-    trap - TERM INT
-    trap - WINCH
 }
 export -f rain
 
+# ------------------------------------------------------------------------------
+# Matrix style digital rain
+# Usage: matrix [OPTIONS]
+matrix()
+{
+    _matrix_show_usage()
+    {
+        printf "Usage: matrix [OPTIONS]\n"
+        printf "Options:\n"
+        printf "\t-s, --speed NUM    Set the drop delay in seconds (default: 0.035).\n"
+        printf "\t                   Lower values = faster stream.\n"
+        printf "\t-c, --color COLOR  Set color theme (default: green).\n"
+        printf "\t-t, --charset SET  Character set: binary, kana, ascii (default: binary).\n"
+        printf "\t-h, --help         Display this help message and exit.\n\n"
+        printf "Example: matrix --charset kana --speed 0.02\n"
+    }
+
+    local step_duration=0.035
+    local base_color="green"
+    local charset="binary"
+
+    while [[ "$#" -gt 0 ]]; do
+        case $1 in
+            -s|--speed)
+                if [[ -n "$2" && ! "$2" =~ ^- ]]; then
+                    step_duration="$2"
+                    shift
+                else
+                    disp E "--speed requires a numeric value."
+                    _matrix_show_usage
+                    return 1
+                fi
+                ;;
+            -c|--color)
+                if [[ -n "$2" && ! "$2" =~ ^- ]]; then
+                    base_color="$2"
+                    shift
+                else
+                    disp E "--color requires a color name."
+                    _matrix_show_usage
+                    return 1
+                fi
+                ;;
+            -t|--charset)
+                if [[ -n "$2" && ! "$2" =~ ^- ]]; then
+                    charset="${2,,}"
+                    shift
+                else
+                    disp E "--charset requires a value."
+                    _matrix_show_usage
+                    return 1
+                fi
+                ;;
+            -h|--help)
+                _matrix_show_usage
+                return 0
+                ;;
+            --)
+                shift
+                break
+                ;;
+            *)
+                disp E "Unknown option: $1"
+                _matrix_show_usage
+                return 1
+                ;;
+        esac
+        shift
+    done
+
+    _rain_engine "$step_duration" "$base_color" "matrix" "$charset"
+}
+export -f matrix
+
 # ------------------------------------------------------------------------------
 # EOF