From 66ec8247dac339ca516ad86fc01a1f16a60bcebf Mon Sep 17 00:00:00 2001 From: Pierrick C Date: Mon, 19 Nov 2018 11:51:04 +0100 Subject: [PATCH] Add simple support for different modes --- code/light_modes.py | 65 ++++++++++++++++++++++++++++ code/main.py | 101 ++++++++++++++++++++++++-------------------- 2 files changed, 121 insertions(+), 45 deletions(-) create mode 100644 code/light_modes.py diff --git a/code/light_modes.py b/code/light_modes.py new file mode 100644 index 0000000..8dedc65 --- /dev/null +++ b/code/light_modes.py @@ -0,0 +1,65 @@ +# This program is free software: you can redistribute it and/or modify +# it under the terms of the GNU General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program. If not, see . + +"""Modes for neopixels strip +Each function update the NeoPixel strip buffer (except update_neopixel() that +call them) +""" + +import time +import urandom + + +def update_neopixel(mode, np_strp, col=(255, 255, 255), bri=50): + """Update NeoPixel strip with one color + :param np_strp : NeoPixel object + :param mode : mode ;) + :param col : color [R, G, B] (default: [255, 255, 255]) + :param bri : brightness 0-100 (default: 50) + """ + # Apply mode + MODES_LST[mode](np_strp, col) + + # Apply brightness to the whole LEDs + np_strp.buf = bytearray([int(x * bri / 100) for x in np_strp.buf]) + + np_strp.write() + +def fill_usr(np_strp, col): + """Update NeoPixel strip with one color + :param np_strp : NeoPixel object + :param col : color [R, G, B] + """ + np_strp.fill(col) + + +def fill_white(np_strp, col=(255, 255, 255)): + """Update NeoPixel strip with only white + :param np_strp : NeoPixel object + :param col : color [R, G, B] (ignored, default=(255, 255, 255)) + """ + np_strp.fill((255, 255, 255)) + + +def sparkles(np_strp, col): + """Make Neopixel sparkle with user defined color!!! + """ + np_strp.fill((0, 0, 0)) + + for _ in range(int(np_strp.n / 4)): + pix = int(urandom.getrandbits(8) / 256 * (np_strp.n - 1)) + print(pix) + np_strp[pix] = col + + +MODES_LST = (fill_usr, fill_white, sparkles) diff --git a/code/main.py b/code/main.py index e20dd3b..6222b72 100644 --- a/code/main.py +++ b/code/main.py @@ -27,11 +27,13 @@ import neopixel from uos import uname from encoder import Encoder +import light_modes + #Paramètres NB_PIX = 8 # Nombre de pixels MAX_BRIGHT = 100 # Luminosité max (100 max.) -COLOR = [255, 255, 255] # Couleur de base (à luminosité max) +USR_COLOR = [255, 120, 20] # Couleur de base (à luminosité max) #Déclaration des objets et initialisation des variables @@ -66,12 +68,6 @@ else: print("Carte non-supportée :", BRD_TYPE) quit() -# # Only with ESP32 : add timing param. -# # timing param =0 for "old" 400kHz neopixel (default) -# # =1 for "new" 800kHz neopixel -# pixel_strip = neopixel.NeoPixel(machine.Pin(NEOPIX_PIN), NB_PIX, timing=1) -# coul_tmp = [0, 0, 0] - #Eteint tout à l'initialisation NPXL_STRIP.fill([0, 0, 0]) NPXL_STRIP.write() @@ -86,22 +82,10 @@ ENC_BUT = machine.Pin(ENC_PIN_C, machine.Pin.IN) #Variables d'état BRIGHTN = 50 # Luminosité (0 - 100) -PWR = False # Est-ce que l'éclairage est allumé ? +PWR = True # Est-ce que l'éclairage est allumé ? BUTTN_STATE = 1 - -def update_neopixel(np_strp, col, bri=50): - """Update NeoPixel strip with one color - :param np_strp : NeoPixel object - :param col : color [R, G, B] - :param bri : brightness 0-100 (default: 50) - """ - col_tmp = [0, 0, 0] - - for i in range(3): - col_tmp[i] = int(col[i] * bri / 100) - - np_strp.fill(col_tmp) - np_strp.write() +curr_mode = 0 +pwroff_dl = 0 ##################### # BOUCLE PRINCIPALE # @@ -109,32 +93,59 @@ def update_neopixel(np_strp, col, bri=50): while True: - # Si on a un changement de valeur, on met à jour la luminisité et l'éclairage - # des NeoPixel - if PWR and BRIGHTN != ENCODER.value: - print(ENCODER.value) - BRIGHTN = ENCODER.value - update_neopixel(NPXL_STRIP, COLOR, BRIGHTN) + # Si on est dans l'état allumé: + if PWR: + # Si on a un changement de valeur, on met à jour la luminosité + if PWR and BRIGHTN != ENCODER.value: + print(ENCODER.value) + BRIGHTN = ENCODER.value - # Quand le bouton central est appuyé puis relâché, on allume ou on éteint - if ENC_BUT.value() == 0 and BUTTN_STATE == 1: - PWR = not PWR - print("Power :", PWR) - if not PWR: - # Extinction des LED - NPXL_STRIP.fill((0, 0, 0)) - NPXL_STRIP.write() + # Si on a juste un clic rapide sur le bouton, on change de mode au + # relachement du bouton + if ENC_BUT.value() == 1 and BUTTN_STATE == 0: + curr_mode = curr_mode + 1 + print("Mode : {}/{}".format(curr_mode, len(light_modes.MODES_LST))) + if curr_mode >= len(light_modes.MODES_LST): + curr_mode = 0 + + # Mise à jour des LED + light_modes.update_neopixel(curr_mode, NPXL_STRIP, USR_COLOR, BRIGHTN) + + + # Quand le bouton central est appuyé puis relâché rapidement, on change de + # mode + # Si on laisse appuyer 2 secondes: extinction + if ENC_BUT.value() == 0: + print("appui") + if BUTTN_STATE == 1: + # Deadline avant de changer l'état d'allumage + pwroff_dl = time.ticks_add(time.ticks_ms(), 2000) + print(pwroff_dl) else: - # Luminité basse si on allume avec une luminisité de 0 - if BRIGHTN == 0: - BRIGHTN = 10 + # Est-ce que la deadline est atteinte ? si oui on change l'état + # d'allumage + print(pwroff_dl, "/", time.ticks_ms()) + if time.ticks_diff(time.ticks_ms(), pwroff_dl) >= 0: + PWR = not PWR + pwroff_dl = 0 + print("Power :", PWR) + if not PWR: + # Extinction des LED + NPXL_STRIP.fill((0, 0, 0)) + NPXL_STRIP.write() + else: + # Luminosité basse si on rallume avec une luminisité de 0 + if BRIGHTN == 0: + BRIGHTN = 10 - # On remet l'encodeur à la dernière luminosité connue - # pour ignorer les mouvements pendant l'extinction - ENCODER.reset(BRIGHTN) - # Rallume les LED: - update_neopixel(NPXL_STRIP, COLOR, BRIGHTN) + # On remet l'encodeur à la dernière luminosité connue + # pour ignorer les mouvements pendant l'extinction + ENCODER.reset(BRIGHTN) + + # on attend que le bouton soit relâché + while ENC_BUT.value() == 0: + time.sleep_ms(10) BUTTN_STATE = ENC_BUT.value() - time.sleep_ms(20) + time.sleep_ms(5)