import math import time import itertools as it import cv2 import colorsys from pythonosc import udp_client import numpy as np DENSITY = 4 RED = [0, 0, 255] N_COLORS = 3 LAST_MESSAGE_TIME = 0 LAST_SNAPSHOT_TIME = 0 def analyze_block(frame, osc, index, sp, ep, send=False): """ Analyze a block """ block = frame[sp[1]:ep[1], sp[0]:ep[0], 0:3] average_color = np.average(block, (0, 1)) / 255 h, s, v = colorsys.rgb_to_hsv(*average_color) # Configure the oscillator if send: osc.send_message( "/radio", [int(index), float(h), float(s), float(v), .5]) return frame def analyze(frame, osc): """ Analyze the full frame """ global LAST_MESSAGE_TIME, LAST_SNAPSHOT_TIME height, width, d = frame.shape n = DENSITY m = math.ceil(n * (height / width)) dx = width / n dy = height / m send = False if time.time() - LAST_MESSAGE_TIME > 0.1: LAST_MESSAGE_TIME = time.time() send = True if time.time() - LAST_SNAPSHOT_TIME > 2: LAST_SNAPSHOT_TIME = time.time() cv2.imwrite("frame.png", frame) for index, (y, x) in enumerate(it.product(range(m), range(n))): sp = (int(x * dx), int(y * dy)) ep = (int(x * dx + dx), int(y * dy + dy)) frame = analyze_block(frame, osc, index, sp, ep, send) return frame if __name__ == '__main__': # camera = cv2.VideoCapture("/dev/video2") camera = cv2.VideoCapture(0) osc = udp_client.SimpleUDPClient("0.0.0.0", 5005) print("Radio is running...") while True: ret, frame = camera.read() frame = analyze(frame, osc) camera.release()