rekordbox+OBSでトラックタイトルの表示を自動切り替えする@Python

pip install obs-websocket-py

capture = cv2.VideoCapture(1)
print(capture.isOpened())
capture.set(cv2.CAP_PROP_FRAME_WIDTH, 1920)
capture.set(cv2.CAP_PROP_FRAME_HEIGHT, 1080)

ret, frame = capture.read()

crop1 = frame[ 353 : 370 , 902 : 958]
crop2 = frame[ 353 : 370 , 1862 : 1918]

b1 = crop1.T[0].flatten().mean()
g1 = crop1.T[1].flatten().mean()
r1 = crop1.T[2].flatten().mean()

b2 = crop2.T[0].flatten().mean()
g2 = crop2.T[1].flatten().mean()
r2 = crop2.T[2].flatten().mean()

cv2.imwrite("F:/dev-obs/crop1.png", crop1)
cv2.imwrite("F:/dev-obs/crop2.png", crop2)

print("DECK 1 ##################")
print("B: %.2f" % (b1))
print("G: %.2f" % (g1))
print("R: %.2f" % (r1))

print("DECK 2 ##################")
print("B: %.2f" % (b2))
print("G: %.2f" % (g2))
print("R: %.2f" % (r2))

import os
import cv2
import time
import datetime

from obswebsocket import obsws, requests

# ======================================

# OBS接続設定
host = "localhost" 
port = 4444
password = "password"

# 色のしきい値
threshold = 20

# キャプチャ解像度
capture_width = 1920
capture_height = 1080

# クロップエリア
deck_top = 353
deck_bottom = 370
deck1_left = 902
deck1_right = 958
deck2_left = 1862
deck2_right = 1918

# ======================================


# ソースの表示を切り替える
def setVisible(item, visible):
    ws.call(requests.SetSceneItemProperties(item, visible=visible))


# VideoCapture オブジェクトを取得
capture = cv2.VideoCapture(1)  # Webカメラが0に割り当てられているのでキャプボを指定する引数は1
print(capture.isOpened())
capture.set(cv2.CAP_PROP_FRAME_WIDTH, capture_width)
capture.set(cv2.CAP_PROP_FRAME_HEIGHT, capture_height)

# OBSに接続
ws = obsws(host, port, password)
ws.connect()

# 処理フラグ
switch = 0

try:
    while True:
        # 映像を取得
        ret, frame = capture.read()

        # クロップ
        deck1_crop = frame[ deck_top : deck_bottom , deck1_left : deck1_right]
        deck2_crop = frame[ deck_top : deck_bottom , deck2_left : deck2_right]

        # 色情報を取得
        red_deck1 = deck1_crop.T[2].flatten().mean()
        red_deck2 = deck2_crop.T[2].flatten().mean()

        # DECK1がMASTERのとき
        if red_deck1 > threshold and (switch == 2 or switch == 0):
            os.system('cls')
            print(datetime.datetime.now().strftime('%Y/%m/%d %H:%M:%S'))
            print("\nNOW PLAYING: DECK1\n")
            setVisible("deck2_spinner", False)
            setVisible("deck1_spinner", True)
            setVisible("deck2_track", False)
            switch = 1

        # DECK2がMASTERのとき
        elif red_deck2 > threshold and (switch == 1 or switch == 0):
            os.system('cls')
            print(datetime.datetime.now().strftime('%Y/%m/%d %H:%M:%S'))
            print("\nNOW PLAYING: DECK2\n")
            setVisible("deck1_spinner", False)
            setVisible("deck2_spinner", True)
            setVisible("deck2_track", True)
            switch = 2

        # よくわからんとき
        elif (red_deck1 < threshold and red_deck2 < threshold) or (red_deck1 > threshold and red_deck2 > threshold):
            os.system('cls')
            print(datetime.datetime.now().strftime('%Y/%m/%d %H:%M:%S'))
            print("\nUndefined\n")
            setVisible("deck1_spinner", False)
            setVisible("deck2_spinner", False)
            setVisible("deck2_track", False)
            switch = 0

        # 1秒待つ
        time.sleep(1)


# Ctrl+Cが押されたら終了
except KeyboardInterrupt:
    capture.release()
    ws.disconnect()