Complete notification example

Author: RedFantom

examples/notifications/mk_notifications.c

@@ -183,13 +183,11 @@ static PyObject* stop(PyObject* self, PyObject* args) {
 static PyObject* py_calculate_dominant_color(PyObject* self, PyObject* args) {
     /** Python interface to fast dominant color calculation function */
     PyObject* list;
-    int w, h, divider, lower, upper, sat_bias;
-    bool brightness_norm;
-    if (!PyArg_ParseTuple(args, "O!iiiiiip",
+    int w, h, divider, lower, upper, sat_bias, brightness_norm;
+    if (!PyArg_ParseTuple(args, "O!iiiiiii",
             &PyList_Type, &list, &divider, &w, &h, &lower, &upper,
             &sat_bias, &brightness_norm))
         return NULL;
-    
     unsigned char data[w][h][3];
     PyObject* column, *row, *e;
     for (int x=0; x<w; x++) {
@@ -215,19 +213,17 @@ static PyObject* py_calculate_dominant_color(PyObject* self, PyObject* args) {
             }
         }
     }
-    
     unsigned char result[3];
     PyObject* tuple = PyTuple_New(3);
     calc_dominant_color(
-        data, w, h, result, divider, sat_bias, lower, upper, brightness_norm);
+        data, w, h, result, divider, sat_bias, lower, upper, brightness_norm==1);
     for (int i=0; i<3; i++)
         PyTuple_SetItem(tuple, i, PyInt_FromLong(result[i]));
     return tuple;
 }
 
 
 void __flash_keyboard(unsigned char* color) {
-    printf("Flashing %d, %d, %d\n", color[0], color[1], color[2]);
     pthread_mutex_lock(&target_lock);
     target_override = true;
     pthread_mutex_unlock(&target_lock);

examples/notifications/notifications.py

@@ -3,25 +3,58 @@
 License: GNU GPLv3
 Copyright (c) 2019 RedFantom
 """
+import gi
+gi.require_version("GdkPixbuf", "2.0")
+from gi.repository import GLib as glib
+from gi.repository import GdkPixbuf as pixbuf
 import mk_notifications
 import time
-from gi.repository import GLib as glib
 import dbus
 from dbus.mainloop.glib import DBusGMainLoop
+import traceback
+import numpy
+from PIL import Image
 
 
 def notifications(bus, message):
-    print([arg for arg in message.get_arg_list()])
+    try:
+        handle_notification(message)
+    except Exception:
+        print("Exception in notification handler!")
+        print(traceback.format_exc())
+
+
+def handle_notification(message):
+    for arg in message.get_args_list():
+        if type(arg) == dbus.Dictionary:
+            arg = {str(k): v for k, v in arg.items()}
+            print(list(arg.keys()))
+            if "image-data" not in arg or len(arg["image-data"]) != 7:
+                print(len(arg), list(arg.keys()))
+                print("Invalid message:", message)
+                continue
+            w, h, r, a, b, d, data = arg["image-data"]
+            data = bytes([int(b) for b in data])
+            mode = "RGB" if not a else "RGBA"
+            image = Image.frombytes(
+                mode, (w, h), data, "raw", mode, r)
+            if mode == "RGBA":
+                image = image.convert("RGB")
+            array = numpy.array(image)
+            data = [[tuple(int(e) for e in column) for column in row] for row in array]
+            color = mk_notifications.calculate_dominant_color(
+                data, 1, len(data), len(data[0]), 25, 700, 60, 1)
+            mk_notifications.flash_keyboard(*color)
 
 
 if __name__ == '__main__':
-    r = mk_notifications.init(2, 25, 700, 60, 1, 20.0, 2, 1.0)
+    r = mk_notifications.init(2, 25, 700, 60, 1, 20.0, 2, .5)
     if r is None:
         raise RuntimeError()
     mk_notifications.start()
 
     DBusGMainLoop(set_as_default=True)
-    bus = dbus.SystemBus()
+    bus = dbus.SessionBus()
     bus.add_match_string_non_blocking("eavesdrop=true, interface='org.freedesktop.Notifications', member='Notify'")
     bus.add_message_filter(notifications)