siwat
/
python-telemetrix-ambilight
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix indexing errors

This commit is contained in:
Siwat Sirichai 2022-03-30 11:44:23 +07:00
parent 73c31c0fb3
commit f285eca6d6
1 changed files with 18 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
ambilight_mulitprocessing.py
View File

@ -32,9 +32,9 @@ def find_dorminant_color(im):
return color return color
else: else:
NUM_CLUSTERS = 3 NUM_CLUSTERS = 1
im = im.resize((50, 50)) im = im.resize((100, 100))
ar = np.asarray(im) ar = np.asarray(im)
shape = ar.shape shape = ar.shape
ar = ar.reshape(np.product(shape[:2]), shape[2]).astype(float) ar = ar.reshape(np.product(shape[:2]), shape[2]).astype(float)
@ -56,12 +56,11 @@ if __name__ == '__main__':
BOARDER_SIZE = 150 BOARDER_SIZE = 300
TOP_LEDS = 19 TOP_LEDS = 19
RIGHT_LEDS = 11 RIGHT_LEDS = 11
BUTTOM_LEDS = 17 BUTTOM_LEDS = 17
LEFT_LEDS = 10 LEFT_LEDS = 10
REDUCTION = 4
NUM_LEDS = TOP_LEDS+RIGHT_LEDS+BUTTOM_LEDS+LEFT_LEDS NUM_LEDS = TOP_LEDS+RIGHT_LEDS+BUTTOM_LEDS+LEFT_LEDS
board.set_pin_mode_neopixel(pin_number=2,num_pixels=NUM_LEDS) board.set_pin_mode_neopixel(pin_number=2,num_pixels=NUM_LEDS)
@ -69,8 +68,8 @@ if __name__ == '__main__':
size = size() size = size()
size.width=pyautogui.size().width/REDUCTION size.width=pyautogui.size().width
size.height=pyautogui.size().height/REDUCTION size.height=pyautogui.size().height
board.neopixel_show() board.neopixel_show()
screenshot = pyautogui.screenshot() screenshot = pyautogui.screenshot()
@ -87,20 +86,21 @@ if __name__ == '__main__':
right = screenshot.crop(box=[size.width-BOARDER_SIZE,0,size.width,size.height]) right = screenshot.crop(box=[size.width-BOARDER_SIZE,0,size.width,size.height])
for i in range(0,TOP_LEDS): 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]) segment = top.crop(box=[i*size.width/TOP_LEDS,0,(i+1)*size.width/TOP_LEDS,BOARDER_SIZE])
chunk.insert(i,segment) chunk.append(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): 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]) segment = right.crop(box=[0,i*size.height/RIGHT_LEDS,BOARDER_SIZE,(i+1)*size.height/RIGHT_LEDS])
chunk.insert(i+TOP_LEDS,segment) chunk.append(segment)
for i in reversed(range(0,BUTTOM_LEDS)):
segment = buttom.crop(box=[i*size.width/BUTTOM_LEDS,0,(i+1)*size.width/BUTTOM_LEDS,BOARDER_SIZE])
chunk.append(segment)
for i in reversed(range(0,LEFT_LEDS)):
segment = left.crop(box=[0,i*size.height/LEFT_LEDS,BOARDER_SIZE,(i+1)*size.height/LEFT_LEDS])
chunk.append(segment)
colors = pool.map(find_dorminant_color,chunk) colors = pool.map(find_dorminant_color,chunk)
for i in range(0,len(colors)): 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.neo_pixel_set_value(i,r=int(colors[i][0]),g=int(colors[i][1]),b=int(colors[i][2]))