def __init__(self, ipcon, uid, version):
PluginBase.__init__(self, ipcon, uid, 'LED Strip Bricklet', version)
self.setupUi(self)
self.has_clock_frequency = version >= (2, 0, 1)
self.led_strip = BrickletLEDStrip(uid, ipcon)
self.qtcb_frame_rendered.connect(self.cb_frame_rendered)
self.button_gradient.pressed.connect(self.gradient_pressed)
self.button_color.pressed.connect(self.color_pressed)
self.box_frame_duration.valueChanged.connect(self.frame_duration_changed)
if self.has_clock_frequency:
self.box_clock_frequency.valueChanged.connect(self.clock_frequency_changed)
else:
self.box_clock_frequency.setValue(20000000)
self.box_clock_frequency.setEnabled(False)
self.state = self.STATE_IDLE
self.gradient_counter = 0
self.voltage = 0
self.voltage_timer = QTimer()
self.voltage_timer.timeout.connect(self.update_voltage)
self.voltage_timer.setInterval(1000)
class LEDStrip(PluginBase, Ui_LEDStrip):
qtcb_frame_rendered = pyqtSignal(int)
STATE_IDLE = 0
STATE_COLOR_GRADIENT = 1
STATE_COLOR_SINGLE = 2
def __init__(self, ipcon, uid, version):
PluginBase.__init__(self, ipcon, uid, 'LED Strip Bricklet', version)
self.setupUi(self)
self.has_clock_frequency = version >= (2, 0, 1)
self.led_strip = BrickletLEDStrip(uid, ipcon)
self.qtcb_frame_rendered.connect(self.cb_frame_rendered)
self.button_gradient.pressed.connect(self.gradient_pressed)
self.button_color.pressed.connect(self.color_pressed)
self.box_frame_duration.valueChanged.connect(self.frame_duration_changed)
if self.has_clock_frequency:
self.box_clock_frequency.valueChanged.connect(self.clock_frequency_changed)
else:
self.box_clock_frequency.setValue(20000000)
self.box_clock_frequency.setEnabled(False)
self.state = self.STATE_IDLE
self.gradient_counter = 0
self.voltage = 0
self.voltage_timer = QTimer()
self.voltage_timer.timeout.connect(self.update_voltage)
self.voltage_timer.setInterval(1000)
def update_voltage(self):
async_call(self.led_strip.get_supply_voltage, None, self.cb_voltage, self.increase_error_count)
def cb_frequency(self, frequency):
self.box_clock_frequency.setValue(frequency)
def cb_voltage(self, voltage):
self.label_voltage.setText(str(voltage/1000.0) + 'V')
def cb_frame_rendered(self):
if self.state == self.STATE_COLOR_GRADIENT:
self.render_color_gradient()
elif self.state == self.STATE_COLOR_SINGLE:
self.render_color_single()
def clock_frequency_changed(self, frequency):
self.led_strip.set_clock_frequency(frequency)
def frame_duration_changed(self, duration):
self.led_strip.set_frame_duration(duration)
def color_pressed(self):
old_state = self.state
self.state = self.STATE_COLOR_SINGLE
if old_state == self.STATE_IDLE:
self.render_color_single()
def gradient_pressed(self):
old_state = self.state
self.state = self.STATE_COLOR_GRADIENT
if old_state == self.STATE_IDLE:
self.render_color_gradient()
def render_color_single(self):
num_leds = self.box_num_led.value()
r = self.box_r.value()
g = self.box_g.value()
b = self.box_b.value()
i = 0
while num_leds > 0:
num_leds -= 16
if num_leds < 0:
leds = 16 + num_leds
else:
leds = 16
r_val = [r]*leds
r_val.extend([0]*(16-leds))
g_val = [g]*leds
g_val.extend([0]*(16-leds))
b_val = [b]*leds
b_val.extend([0]*(16-leds))
self.led_strip.set_rgb_values(i, leds, r_val, g_val, b_val)
i += leds
def render_color_gradient(self):
num_leds = self.box_num_led.value()
self.gradient_counter += num_leds * self.box_speed.value() / 100.0 / 4.0
ra = []
ga = []
ba = []
range_leds = range(num_leds)
range_leds = range_leds[int(self.gradient_counter) % num_leds:] + range_leds[:int(self.gradient_counter) % num_leds]
for i in range_leds:
r, g, b = colorsys.hsv_to_rgb(1.0*i/num_leds, 1, 0.1)
ra.append(int(r*255))
ga.append(int(g*255))
ba.append(int(b*255))
i = 0
while num_leds > 0:
num_leds -= 16
if num_leds < 0:
leds = 16 + num_leds
else:
leds = 16
r_val = ra[:leds]
r_val.extend([0]*(16-leds))
g_val = ga[:leds]
g_val.extend([0]*(16-leds))
b_val = ba[:leds]
b_val.extend([0]*(16-leds))
try:
self.led_strip.set_rgb_values(i, leds, r_val, g_val, b_val)
except:
pass
ra = ra[leds:]
ga = ga[leds:]
ba = ba[leds:]
i += leds
def start(self):
if self.has_clock_frequency:
async_call(self.led_strip.get_clock_frequency, None, self.cb_frequency, self.increase_error_count)
async_call(self.led_strip.get_supply_voltage, None, self.cb_voltage, self.increase_error_count)
self.voltage_timer.start()
self.led_strip.register_callback(self.led_strip.CALLBACK_FRAME_RENDERED,
self.qtcb_frame_rendered.emit)
def stop(self):
self.voltage_timer.stop()
self.led_strip.register_callback(self.led_strip.CALLBACK_FRAME_RENDERED,
None)
def get_url_part(self):
return 'led_strip'
@staticmethod
def has_device_identifier(device_identifier):
return device_identifier == BrickletLEDStrip.DEVICE_IDENTIFIER
