From 72c69be09f63f8841f48920134dede14c5980632 Mon Sep 17 00:00:00 2001
From: Marcus Brown <marcus@freedoh.net>
Date: Sun, 20 Oct 2024 13:54:44 +1100
Subject: [PATCH] setup MQTT topics for Home Assistant

---
 dayglo_detector/dayglo_detector.py | 79 ++++++++++++++++++++++++++++++
 1 file changed, 79 insertions(+)

diff --git a/dayglo_detector/dayglo_detector.py b/dayglo_detector/dayglo_detector.py
index 0601c0a..ae9d467 100644
--- a/dayglo_detector/dayglo_detector.py
+++ b/dayglo_detector/dayglo_detector.py
@@ -127,3 +127,82 @@ assess_latest_image()
 # Start the MQTT loop
 client.loop_forever()
 
+import os
+import paho.mqtt.client as mqtt
+import numpy as np
+import cv2
+import base64
+import json
+
+# Configuration
+MQTT_BROKER = os.environ.get('MQTT_BROKER', '10.59.221.172')
+MQTT_PORT = int(os.environ.get('MQTT_PORT', '1883'))
+MQTT_SNAPSHOT_TOPIC = "/frigate/patiocam/person/snapshot"
+MQTT_TOPIC_PUBLISH = "homeassistant/sensor/dayglo_rating/state"
+DAYGLO_THRESHOLD_TOPIC = "homeassistant/sensor/dayglo_threshold/state"
+DISCOVERY_PREFIX = "homeassistant"
+
+# Default threshold
+dayglo_threshold = 50
+
+def on_connect(client, userdata, flags, rc):
+    print("Connected with result code", rc)
+    client.subscribe(MQTT_SNAPSHOT_TOPIC)
+    publish_discovery_configurations()
+
+def publish_discovery_configurations():
+    rating_config = {
+        "name": "Dayglo Rating",
+        "state_topic": MQTT_TOPIC_PUBLISH,
+        "unit_of_measurement": "%",
+        "value_template": "{{ value_json.rating }}",
+        "icon": "mdi:brush",
+        "unique_id": "mqtt_dayglo_rating"
+    }
+    threshold_config = {
+        "name": "Dayglo Threshold",
+        "state_topic": DAYGLO_THRESHOLD_TOPIC,
+        "unit_of_measurement": "%",
+        "value_template": "{{ value_json.threshold }}",
+        "icon": "mdi:tune-vertical",
+        "unique_id": "mqtt_dayglo_threshold"
+    }
+
+    client.publish(f"{DISCOVERY_PREFIX}/sensor/dayglo_rating/config", json.dumps(rating_config), retain=True)
+    client.publish(f"{DISCOVERY_PREFIX}/sensor/dayglo_threshold/config", json.dumps(threshold_config), retain=True)
+
+def on_message(client, userdata, msg):
+    if msg.topic == MQTT_SNAPSHOT_TOPIC:
+        print("Snapshot received")
+        process_snapshot(msg.payload)
+    elif msg.topic == DAYGLO_THRESHOLD_TOPIC:
+        global dayglo_threshold
+        dayglo_threshold = float(msg.payload.decode('utf-8'))
+        print("Dayglo threshold updated to:", dayglo_threshold)
+
+def process_snapshot(payload):
+    image_data = base64.b64decode(payload)
+    nparr = np.frombuffer(image_data, np.uint8)
+    image = cv2.imdecode(nparr, cv2.IMREAD_COLOR)
+
+    if image is not None:
+        rating = calculate_dayglo_rating(image)
+        print("Dayglo Rating calculated:", rating)
+        client.publish(MQTT_TOPIC_PUBLISH, json.dumps({"rating": rating}))
+
+def calculate_dayglo_rating(image):
+    hsv_image = cv2.cvtColor(image, cv2.COLOR_BGR2HSV)
+    lower_color = np.array([20, 100, 100])
+    upper_color = np.array([40, 255, 255])
+    mask = cv2.inRange(hsv_image, lower_color, upper_color)
+    dayglo_pixels = cv2.countNonZero(mask)
+    total_pixels = image.shape[0] * image.shape[1]
+    rating = (dayglo_pixels / total_pixels) * 100
+    return rating
+
+client = mqtt.Client()
+client.on_connect = on_connect
+client.on_message = on_message
+client.connect(MQTT_BROKER, MQTT_PORT, 60)
+client.loop_forever()
+