From c5cd41b3fd3f36ab1b7a4281c501431a09f7a92f Mon Sep 17 00:00:00 2001
From: Simon <simondergruber@gmail.com>
Date: Thu, 14 Jan 2021 14:39:40 +0100
Subject: [PATCH] cava module

---
 .config/polybar/config                       |  21 ++-
 .config/polybar/scripts/modules/cava.py      | 188 +++++++++++++++++++
 .config/polybar/scripts/modules/cava_read.sh |  12 ++
 3 files changed, 214 insertions(+), 7 deletions(-)
 create mode 100644 .config/polybar/scripts/modules/cava.py
 create mode 100644 .config/polybar/scripts/modules/cava_read.sh

diff --git a/.config/polybar/config b/.config/polybar/config
index 96045b0..84acc1a 100644
--- a/.config/polybar/config
+++ b/.config/polybar/config
@@ -27,7 +27,7 @@ unfocused = ${xrdb:foreground}
 [bar/floating]
 ;modules-right = disk_short_pc checkupdates ovpn nvpn eth_pc 
 modules-left = sysinfo button_pavucontrol
-modules-center = clock playerctl_long
+modules-center = cava clock playerctl_long
 modules-right = sensors_cpu_pc nvgpu memory 
 monitor = ${env:MONITOR:DP-4}
 width = 60%
@@ -66,7 +66,7 @@ offset-x = 20%
 offset-y = 54px
 fixed-center = true
 background = ${colors.background}
-foreground = ${colors.foreground}
+forFeground = ${colors.foreground}
 line-size = 1
 line-color = #000
 separator =|
@@ -228,11 +228,11 @@ click-middle = "bash ~/Userdata/Scripts/Launchers/restart_pulseaudio_1s.sh"
 [module/clock]
 type = internal/date
 interval = 1
-date = 
-date-alt = %d.%m.%Y,
-time = %H:%M
-time-alt = %H:%M:%S
-label = %date% %time%
+date =
+date-alt =%d.%m.%Y
+time =" %H:%M"
+time-alt =%H:%M:%S,
+label =%time% %date% 
 
 [module/clock_icon]
 inherit = "module/clock"
@@ -381,6 +381,13 @@ exec = ~/.config/polybar/scripts/polypopup.sh
 click-left = ~/.config/polybar/scripts/polypopup.sh --toggle
 content = 󰋼
 
+[module/cava]
+type = custom/script
+tail = true
+interval = 0.1
+format = <label>
+exec = bash ~/.config/polybar/scripts/modules/cava_read.sh
+
 [module/playerctl]
 type = custom/script
 tail = true
diff --git a/.config/polybar/scripts/modules/cava.py b/.config/polybar/scripts/modules/cava.py
new file mode 100644
index 0000000..b649781
--- /dev/null
+++ b/.config/polybar/scripts/modules/cava.py
@@ -0,0 +1,188 @@
+#!/usr/bin/python3
+import sys
+import time
+import struct
+import os.path
+import subprocess
+import configparser
+from collections import defaultdict
+
+##################################################
+# Config
+
+# Decide if the output should go to STDOUT or a fifo buffer
+PRINT_TO_STDOUT = False
+
+# Display no output if all bars are at level 0
+HIDE_WHEN_EMPTY = True
+
+# Separator Character between bars
+# Examples: https://www.compart.com/en/unicode/category/Zs
+SEPARATOR = ''
+
+CONFIG_PATH = os.path.join(os.sep, "tmp","cava_polybar.config")
+RAW_PIPE_OUT = os.path.join(os.sep, "tmp","cava_polybar_output.fifo")
+
+# Data from the cava config
+RAW_PIPE_IN = os.path.join(os.sep, "tmp","cava_polybar_input.fifo")
+OUTPUT_FILE = os.path.join(os.sep, "tmp","cava_polybar.out")
+BARS_NUMBER = 8
+OUTPUT_BIT_FORMAT = "8bit"
+
+# Values range from 0 to 100
+# Distribute 9 states over this value range
+#  ▁▂▃▄▅▆▇█
+
+bar_factor = 100 / 10
+bar_characters = dict([
+    (0, '▁'),
+    (bar_factor * 1, '▁'),
+    (bar_factor * 2, '▂'),
+    (bar_factor * 3, '▃'),
+    (bar_factor * 4, '▄'),
+    (bar_factor * 5, '▅'),
+    (bar_factor * 6, '▆'),
+    (bar_factor * 7, '▇'),
+    (bar_factor * 8, '█')
+])
+
+
+##################################################
+# Code begins here
+
+bytetype, bytesize, bytenorm = ("H", 2, 65535) if (OUTPUT_BIT_FORMAT == "16bit") else ("B", 1, 255)
+
+def output(value, fifo):
+    if (PRINT_TO_STDOUT):
+        print(value, end="")
+    else:
+        fifo.write(value)
+
+
+def valToBar(value):    
+    for bar_threshold in bar_characters:
+        if(value < bar_threshold):
+            return bar_characters[bar_threshold]
+
+    return bar_characters[bar_factor * 8]
+
+def test():
+    print("")
+
+    print("Bar Characters:")
+    for bar_threshold in bar_characters:
+        print('{:02.2f}: {}'.format(bar_threshold,bar_characters[bar_threshold]))
+
+    print("")
+
+    print("Value Test:")
+    for i in range(101):
+        print('{:03d}: {}'.format(i,valToBar(i)))
+
+def createCavaConfig():
+    # Exit if file is already there
+    #if (os.path.isfile(CONFIG_PATH)):
+    #    return
+
+    config = configparser.ConfigParser()
+
+    config.add_section('general')
+    config.set('general', 'bars', str(BARS_NUMBER))
+    #config.set('general', 'overshoot', str(0))
+
+    config.add_section('output')
+    config.set('output', 'method', 'raw')
+    #config.set('output', 'channels', 'mono')
+    #config.set('output', 'mono_option', 'average')
+    config.set('output', 'raw_target', RAW_PIPE_IN)
+    config.set('output', 'bit_format', OUTPUT_BIT_FORMAT)
+
+    #config.add_section('smoothing')
+    #config.set('smoothing', 'integral', '0')
+
+    with open(CONFIG_PATH, 'w') as configfile:
+        config.write(configfile)
+
+def run():
+    createCavaConfig()  
+    process = subprocess.Popen(["cava", "-p", CONFIG_PATH])
+
+    exit_code = 0
+
+    fifo_out = None
+    if (not PRINT_TO_STDOUT):
+        if os.path.exists(RAW_PIPE_OUT):
+            os.remove(RAW_PIPE_OUT)
+        os.mkfifo(RAW_PIPE_OUT) 
+        print("Output can be found in " + RAW_PIPE_OUT)
+        fifo_out = open(RAW_PIPE_OUT, "w")
+    fifo_in = open(RAW_PIPE_IN, "rb")
+
+
+    try:
+        convert(fifo_in, fifo_out)
+    except KeyboardInterrupt:
+        exit_code = 1
+    finally:
+        if (not PRINT_TO_STDOUT):
+            fifo_out.close()
+            os.remove(RAW_PIPE_OUT)
+
+        fifo_in.close()
+        sys.exit(exit_code)
+
+def convert(fifo_in, fifo_out):
+    chunk = bytesize * BARS_NUMBER
+    fmt = bytetype * BARS_NUMBER
+
+    emptyOutputs = 0
+    tstring = ""
+
+    while True:
+        rawData = fifo_in.read(chunk)
+        if len(rawData) < chunk:
+            break
+
+        empty = True
+        tstring = ""
+
+        for i in struct.unpack(fmt, rawData):
+
+            value = int(i / bytenorm * 100)
+
+            tstring += valToBar(value) + SEPARATOR
+
+            if (value != 0):
+                empty = False
+
+        if (empty and HIDE_WHEN_EMPTY):
+            emptyOutputs += 1
+            if (emptyOutputs > 5):
+                output("        " + '\n', fifo_out)
+        else:
+            emptyOutputs = 0
+            output(tstring + '\n', fifo_out)
+
+
+if __name__ == "__main__":
+    for arg in sys.argv:
+        if (arg == "h" or arg == "help"):
+            print("")
+            print("cava polybar parse script")
+            print("Converts cava raw values into characters and outputs to STDOUT or a fifo buffer")
+            print("")
+            print("Adjust thresholds, characters and config directly in the script")
+            print("If the config file at 'CONFIG_PATH' (/tmp/cava_polybar.config) is messed up, simply delet it")
+            print("")
+            print("Arguments:")
+            print("<none>  ... Execute normally")
+            print("t, test ... Run test mode (stdout only)")
+            print("h, help ... Show this help message and exit (stdout only)")
+            print("")
+            sys.exit() 
+
+        if (arg == "t" or arg == "test"):
+            test()
+            sys.exit() 
+
+    run()
diff --git a/.config/polybar/scripts/modules/cava_read.sh b/.config/polybar/scripts/modules/cava_read.sh
new file mode 100644
index 0000000..0ba7047
--- /dev/null
+++ b/.config/polybar/scripts/modules/cava_read.sh
@@ -0,0 +1,12 @@
+#!/bin/bash    
+
+python "$HOME/.config/polybar/scripts/modules/cava.py" &
+
+sleep 1
+
+fifo_name="/tmp/cava_polybar_output.fifo"
+while true
+do
+    
+        echo "$line"
+done <"$fifo_name"