From 73c31c0fb3e0c246b62c1c4caf330e662e107877 Mon Sep 17 00:00:00 2001
From: Siwat Sirichai <siwat@siwatinc.com>
Date: Wed, 30 Mar 2022 11:10:28 +0700
Subject: [PATCH] performance improvement

---
 ambilight.py => ambilight_gpu.py |   5 +-
 ambilight_mulitprocessing.py     | 111 +++++++++++++++++++++++++++++++
 2 files changed, 114 insertions(+), 2 deletions(-)
 rename ambilight.py => ambilight_gpu.py (97%)
 create mode 100644 ambilight_mulitprocessing.py

diff --git a/ambilight.py b/ambilight_gpu.py
similarity index 97%
rename from ambilight.py
rename to ambilight_gpu.py
index 530c27d..ecda7c0 100644
--- a/ambilight.py
+++ b/ambilight_gpu.py
@@ -27,7 +27,7 @@ FAST_MODE = True
 def find_dorminant_color(im):
     if FAST_MODE:
         color = cupy.reshape(cupy.asarray(im),(-1,3))
-        color = cupy.mean(color,axis=0)
+        color = cupy.median(color,axis=0)
         
         return color
     else:
@@ -63,8 +63,9 @@ lastTime = timestamp()
 
 def processTopLeds():
     global NUM_LEDS,BOARDER_SIZE,TOP_LEDS,RIGHT_LEDS,BUTTOM_LEDS,LEFT_LEDS,board,screenshot
-    top = screenshot.crop(box=[0,0,size.width,BOARDER_SIZE])
+    
     while True:
+        top = screenshot.crop(box=[0,0,size.width,BOARDER_SIZE])
         for i in range(0,TOP_LEDS):
             segment = top.crop(box=[i*size.width/TOP_LEDS,0,(i+1)*size.width/TOP_LEDS,BOARDER_SIZE])
             colors = [int(color) for color in find_dorminant_color(segment)]
diff --git a/ambilight_mulitprocessing.py b/ambilight_mulitprocessing.py
new file mode 100644
index 0000000..15b863e
--- /dev/null
+++ b/ambilight_mulitprocessing.py
@@ -0,0 +1,111 @@
+from tkinter import TOP
+import pyautogui
+import scipy.cluster as cluster
+import scipy
+import sys
+import numpy as np
+from time import sleep, time
+import numpy
+import signal
+import atexit
+from time import perf_counter as timestamp
+from telemetrix_rpi_pico import telemetrix_rpi_pico
+from multiprocessing import Pool
+import PIL.ImageGrab
+
+class size:
+    width = None
+    height = None
+
+def sigint_handler(signal=None, frame=None):
+    print ('KeyboardInterrupt is caught')
+    board.neopixel_clear()
+    sleep(0.75)
+    board.shutdown()
+    sys.exit(0)
+
+FAST_MODE = False
+def find_dorminant_color(im):
+    if FAST_MODE:
+        color = numpy.reshape(numpy.asarray(im),(-1,3))
+        color = numpy.median(color,axis=0)
+        
+        return color
+    else:
+        NUM_CLUSTERS = 3
+
+        im = im.resize((50, 50))
+        ar = np.asarray(im)
+        shape = ar.shape
+        ar = ar.reshape(np.product(shape[:2]), shape[2]).astype(float)
+        codes, dist = cluster.vq.kmeans(ar, NUM_CLUSTERS)
+
+        vecs, dist = cluster.vq.vq(ar, codes) 
+        counts, bins = np.histogram(vecs, len(codes))
+
+        index_max = np.argmax(counts)
+        peak = codes[index_max]
+
+        return peak
+if __name__ == '__main__':
+    signal.signal(signal.SIGINT, sigint_handler)
+    atexit.register(sigint_handler)
+    global NUM_LEDS,BOARDER_SIZE,TOP_LEDS,RIGHT_LEDS,BUTTOM_LEDS,LEFT_LEDS,screenshot,board
+
+    board = telemetrix_rpi_pico.TelemetrixRpiPico()
+
+    
+
+    BOARDER_SIZE = 150
+    TOP_LEDS = 19
+    RIGHT_LEDS = 11
+    BUTTOM_LEDS = 17
+    LEFT_LEDS = 10
+    REDUCTION = 4
+    NUM_LEDS = TOP_LEDS+RIGHT_LEDS+BUTTOM_LEDS+LEFT_LEDS
+
+    board.set_pin_mode_neopixel(pin_number=2,num_pixels=NUM_LEDS)
+    board.neopixel_clear(auto_show=True)
+
+    size = size()
+
+    size.width=pyautogui.size().width/REDUCTION
+    size.height=pyautogui.size().height/REDUCTION
+
+    board.neopixel_show()
+    screenshot = pyautogui.screenshot()
+    pool = Pool(16)
+    
+    while True:
+        try:
+            screenshot = PIL.ImageGrab.grab()
+            lastTime = timestamp()
+            chunk = []
+            top = screenshot.crop(box=[0,0,size.width,BOARDER_SIZE])
+            left = screenshot.crop(box=[0,0,BOARDER_SIZE,size.height])
+            buttom = screenshot.crop(box=[0,size.height-BOARDER_SIZE,size.width,size.height])
+            right = screenshot.crop(box=[size.width-BOARDER_SIZE,0,size.width,size.height])
+            for i in range(0,TOP_LEDS):
+                segment = top.crop(box=[i*size.width/TOP_LEDS,0,(i+1)*size.width/TOP_LEDS,BOARDER_SIZE])
+                chunk.insert(i,segment)
+            
+            for i in range(0,LEFT_LEDS):
+                segment = left.crop(box=[0,i*size.height/LEFT_LEDS,BOARDER_SIZE,(i+1)*size.height/LEFT_LEDS])
+                chunk.insert(TOP_LEDS+RIGHT_LEDS+BUTTOM_LEDS+LEFT_LEDS-i,segment)
+            
+            for i in range(0,BUTTOM_LEDS):
+                segment = buttom.crop(box=[i*size.width/BUTTOM_LEDS,0,(i+1)*size.width/BUTTOM_LEDS,BOARDER_SIZE])
+                chunk.insert(TOP_LEDS+RIGHT_LEDS+BUTTOM_LEDS-i,segment)
+            board.neopixel_show()
+            
+            for i in range(0,RIGHT_LEDS):
+                segment = right.crop(box=[0,i*size.height/RIGHT_LEDS,BOARDER_SIZE,(i+1)*size.height/RIGHT_LEDS])
+                chunk.insert(i+TOP_LEDS,segment)
+            colors = pool.map(find_dorminant_color,chunk)
+            for i in range(0,len(colors)):
+                board.neo_pixel_set_value(i,r=int(colors[i][0]),g=int(colors[i][1]),b=int(colors[i][2]))
+            board.neopixel_show()
+        except Exception as e:
+            print(e)
+            print("Retrying")
+        #print("loop time : "+str(timestamp()-lastTime))