1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
|
import homeassistant.helpers.config_validation as cv
import voluptuous as vol
import requests
import threading
from math import floor
from threading import Thread
from homeassistant.const import CONF_MAC
from homeassistant.components.light import (
LightEntity,
SUPPORT_BRIGHTNESS,
SUPPORT_COLOR,
SUPPORT_TRANSITION,
ATTR_BRIGHTNESS,
ATTR_RGB_COLOR,
ATTR_TRANSITION,
COLOR_MODE_RGB,
PLATFORM_SCHEMA
)
from time import sleep
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
vol.Required("name"): cv.string,
vol.Required("host"): cv.string
})
SUPPORT_FEATURES_RGB = SUPPORT_BRIGHTNESS | SUPPORT_COLOR
SUPPORT_FEATURES_WHITE = SUPPORT_BRIGHTNESS
def setup_platform(hass, config, add_entities, discovery_info=None):
add_entities([ ESPLedStripLight(name=config["name"], host=config["host"]) ])
class ESPLedStripLight(LightEntity):
def __init__(self, **kwargs):
self._name = kwargs["name"]
self._host = kwargs["host"]
self._on = False
self._brightness = 255
self._rgb = (255, 255, 255)
self._thread = Thread()
self._thread.start()
self._transitioning = False
@property
def color_mode(self):
return COLOR_MODE_RGB
@property
def supported_color_modes(self):
return set([ COLOR_MODE_RGB ])
@property
def supported_features(self):
return SUPPORT_BRIGHTNESS | SUPPORT_COLOR | SUPPORT_TRANSITION
@property
def unique_id(self):
return self._host
@property
def name(self):
return self._name
@property
def is_on(self):
return self._on
@property
def brightness(self):
return self._brightness
@property
def rgb_color(self):
return self._rgb
def turn_on(self, **kwargs):
on_old = self._on
brightness_old = self._brightness
rgb_old = self._rgb
self._on = True
brightness = kwargs.get(ATTR_BRIGHTNESS)
if brightness != None: self._brightness = brightness
rgb = kwargs.get(ATTR_RGB_COLOR)
if rgb != None: self._rgb = rgb
self.terminate_thread()
transition = kwargs.get(ATTR_TRANSITION)
if transition != None:
self.interpolate(brightness_old if on_old else 0, self._brightness, rgb_old if on_old else (0, 0, 0,), self._rgb, transition)
else:
self.update_espled()
def turn_off(self, **kwargs):
self.terminate_thread()
self._on = False
self.update_espled()
def interpolate(self, brightness_old, brightness, rgb_old, rgb, transition):
self._thread = Thread(target=self.interpolate_thread, args=(brightness_old, brightness, rgb_old, rgb, transition, ))
self._thread.start()
def terminate_thread(self):
self._transitioning = False
self._thread.join()
def interpolate_thread(self, brightness_old, brightness, rgb_old, rgb, transition):
step_duration = 0.250
steps = int(transition / step_duration)
if rgb_old == None: rgb_old = (0, 0, 0,)
if rgb == None: rgb = (0, 0, 0,)
if brightness_old == None: brightness_old = 0
if brightness == None: brightness = 0
self._transitioning = True
for x in range(steps):
if not self._transitioning: break
weight = (x + 1) / steps
r = rgb_old[0] * (1 - weight) + rgb[0] * weight
g = rgb_old[1] * (1 - weight) + rgb[1] * weight
b = rgb_old[2] * (1 - weight) + rgb[2] * weight
self._rgb = (r, g, b,)
self._brightness = brightness_old * (1 - weight) + brightness * weight
self.update_espled()
sleep(step_duration)
self._transitioning = False
def update_espled(self):
r = int( int(self._on) * self._rgb[0] * ( self._brightness / 255 ) )
g = int( int(self._on) * self._rgb[1] * ( self._brightness / 255 ) )
b = int( int(self._on) * self._rgb[2] * ( self._brightness / 255 ) )
def req():
success = False
while not success:
try:
requests.post("http://" + self._host, f"{r:02x}{g:02x}{b:02x}")
success = True
except:
continue
threading.Thread(target=req).start()
|