forgot to commit everything...
This commit is contained in:
320
user/Buzzer.h
Executable file
320
user/Buzzer.h
Executable file
@@ -0,0 +1,320 @@
|
||||
#ifndef BUZZER_H
|
||||
#define BUZZER_H
|
||||
|
||||
|
||||
#include "ESP8266lib/net/UDP.h"
|
||||
#include "ESP8266lib/net/IP.h"
|
||||
#include "ESP8266lib/net/MAC.h"
|
||||
#include "ESP8266lib/net/WiFiRaw.h"
|
||||
#include "ESP8266lib/io/IO.h"
|
||||
|
||||
#include "Fader.h"
|
||||
#include "FadeOnce.h"
|
||||
#include "Rainbow.h"
|
||||
#include "RainbowBeat.h"
|
||||
|
||||
#define FIRMWARE_NR 5
|
||||
|
||||
#define CODE(a, b, c, d) (a << 24 | b << 16 | c << 8 | d)
|
||||
|
||||
|
||||
enum LEDMode {
|
||||
OFF,
|
||||
FIXED_COLOR,
|
||||
RAINBOW_COLOR,
|
||||
FADE_BETWEEN_COLOR,
|
||||
FADE_ONCE,
|
||||
STROBO_COLOR,
|
||||
RAINBOW_BEAT,
|
||||
};
|
||||
|
||||
class Buzzer;
|
||||
extern Buzzer buzzer;
|
||||
|
||||
|
||||
class Buzzer {
|
||||
|
||||
private:
|
||||
|
||||
static constexpr const char* NAME = "Buzzer";
|
||||
|
||||
UDP udp;
|
||||
IP server = IP();
|
||||
WS2812B<1> leds;
|
||||
LEDMode ledMode;
|
||||
WiFiRaw::MACAddress myMac;
|
||||
|
||||
FadeBetween fadeBetween;
|
||||
FadeOnce fadeOnce;
|
||||
Rainbow rainbow;
|
||||
RainbowBeat rainbowBeat;
|
||||
|
||||
|
||||
public:
|
||||
|
||||
/** ctor */
|
||||
Buzzer() {
|
||||
|
||||
debugMod(NAME, "ctor()");
|
||||
udp.bind(localPort);
|
||||
udp.setRecvCallback(&Buzzer::onUDP);
|
||||
ledMode = LEDMode::FIXED_COLOR;
|
||||
myMac = WiFiRaw::getMyMAC();
|
||||
|
||||
// buzzer interrupt (Pin D2)
|
||||
GPIO4_INPUT_SET;
|
||||
GPIO4_INPUT_PULLUP_SET;
|
||||
|
||||
// interrupt
|
||||
//ETS_GPIO_INTR_DISABLE();
|
||||
//ETS_GPIO_INTR_ATTACH(isrRoutine, this);
|
||||
//gpio_pin_intr_state_set(GPIO_ID_PIN(4), GPIO_PIN_INTR_ANYEDGE);
|
||||
//ETS_GPIO_INTR_ENABLE();
|
||||
|
||||
}
|
||||
|
||||
/** send a heartbeat packet */
|
||||
void sendHeartbeat() {
|
||||
|
||||
const uint16_t vcc = ADC::getVcc();
|
||||
os_printf("send: heartbeat, Vcc: %d\n", vcc);
|
||||
char data[5+12+2+3];
|
||||
|
||||
os_memcpy(&data[0], "*PING", 5);
|
||||
|
||||
os_memcpy(&data[5], myMac.asPtr(), 12);
|
||||
|
||||
data[17] = '_';
|
||||
data[18] = FIRMWARE_NR;
|
||||
|
||||
data[19] = '_';
|
||||
data[20] = (vcc >> 8) & 0xFF;
|
||||
data[21] = (vcc >> 0) & 0xFF;
|
||||
|
||||
udp.send(remoteIP, remotePort, data, sizeof(data));
|
||||
|
||||
}
|
||||
|
||||
/** send a buzzer packet */
|
||||
void sendBuzzer() {
|
||||
char data[5+12];
|
||||
os_memcpy(&data[0], "*BUZZ", 5);
|
||||
os_memcpy(&data[5], myMac.asPtr(), 12);
|
||||
udp.send(remoteIP, remotePort, data, 5+12);
|
||||
os_printf("send: buzzer\n");
|
||||
}
|
||||
|
||||
|
||||
/** incoming UDP data */
|
||||
void onUDP(const char* data, uint16_t len) {
|
||||
|
||||
if (len < 2) {return;}
|
||||
if (data[0] == '*') {
|
||||
debugMod(NAME, "got command");
|
||||
checkCode(&data[1]);
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
void update() {
|
||||
|
||||
static int cnt = 0; ++cnt;
|
||||
|
||||
// every 9 seconds
|
||||
if (cnt % 9000 == 0) {
|
||||
buzzer.sendHeartbeat();
|
||||
}
|
||||
|
||||
// every 20 milliseconds
|
||||
if (cnt % 20 == 0) {
|
||||
updateLED();
|
||||
}
|
||||
|
||||
|
||||
// block the buzzer for some time after buzzering
|
||||
static int block = 0;
|
||||
if (block > 0) {
|
||||
--block;
|
||||
if (block == 0) {onUnBuzzer();}
|
||||
}
|
||||
|
||||
// check buzzer-state
|
||||
if (cnt % 20 == 0) {
|
||||
|
||||
// buzzer currently blocked
|
||||
if (block > 0) {return;}
|
||||
|
||||
static uint8_t lastVal = 1;
|
||||
const uint8_t curVal = GPIO4_IN;
|
||||
if (curVal < lastVal) {
|
||||
block = 1000;
|
||||
onBuzzer();
|
||||
}
|
||||
lastVal = curVal;
|
||||
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
// buzzer button pressed
|
||||
void onBuzzer() {
|
||||
sendBuzzer();
|
||||
internalLED(true);
|
||||
//setOff();
|
||||
}
|
||||
|
||||
void onUnBuzzer() {
|
||||
internalLED(false);
|
||||
//os_printf("free\n");
|
||||
//setRainbow();
|
||||
//setFade(255,0,0, 0,0,255);
|
||||
//setFade(180,0,16, 16,190,16);
|
||||
}
|
||||
|
||||
/** wemos D1 mini only */
|
||||
void internalLED(bool on) {
|
||||
GPIO2_OUTPUT_SET;
|
||||
if (on) { // LED is inverted
|
||||
GPIO2_L;
|
||||
} else {
|
||||
GPIO2_H;
|
||||
}
|
||||
}
|
||||
|
||||
private:
|
||||
|
||||
/** check the command-code behind the given char-string */
|
||||
void checkCode(const char* buf) {
|
||||
|
||||
// convert the buffer's first 4 bytes to a uint32_t for faster comparison
|
||||
const uint32_t code = CODE(buf[0], buf[1], buf[2], buf[3]);
|
||||
|
||||
switch(code) {
|
||||
case CODE('S', 'T', 'R', 'O'): setStrobo(buf[4], buf[5], buf[6]); break;
|
||||
case CODE('O', 'F', 'F', ' '): setOff(); break;
|
||||
case CODE('S', 'R', 'G', 'B'): setRGB(buf[4], buf[5], buf[6]); break;
|
||||
case CODE('R', 'A', 'I', 'N'): setRainbow(); break;
|
||||
case CODE('F', 'A', 'D', 'E'): setFade(buf[4], buf[5], buf[6], buf[7], buf[8], buf[9]); break;
|
||||
case CODE('F', 'L', 'S', 'H'): setFlash(buf[4], buf[5], buf[6], buf[7], buf[8], buf[9], buf[10]); break;
|
||||
case CODE('R', 'A', 'I', 'R'): setRainbowBeatRestart(); break;
|
||||
case CODE('R', 'A', 'I', 'B'): setRainbowBeatBeat(buf[4]); break;
|
||||
case CODE('R', 'A', 'I', 'S'): setRainbowBeatShift(buf[4]); break;
|
||||
default: break;
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
/** upadte the current LED output (for animations) */
|
||||
void updateLED() {
|
||||
|
||||
// fading LED color?
|
||||
if (ledMode == LEDMode::RAINBOW_COLOR) {
|
||||
rainbow.update();
|
||||
leds.setColor(0, rainbow.getCurrent());
|
||||
leds.flush();
|
||||
} else if (ledMode == LEDMode::STROBO_COLOR) {
|
||||
leds.setEnabled(0, !leds.isEnabled(0));
|
||||
leds.flush();
|
||||
} else if (ledMode == LEDMode::FADE_BETWEEN_COLOR) {
|
||||
fadeBetween.update();
|
||||
leds.setColor(0, fadeBetween.getCurrent());
|
||||
leds.flush();
|
||||
} else if (ledMode == LEDMode::FADE_ONCE) {
|
||||
fadeOnce.update();
|
||||
leds.setColor(0, fadeOnce.getCurrent());
|
||||
leds.flush();
|
||||
} else if (ledMode == LEDMode::RAINBOW_BEAT) {
|
||||
rainbowBeat.update();
|
||||
leds.setColor(0, rainbowBeat.getCurrent());
|
||||
leds.flush();
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
public:
|
||||
|
||||
/** disable LED */
|
||||
void setOff() {
|
||||
ledMode = LEDMode::OFF;
|
||||
leds.getColor(0).setRGB(0,0,255); // for testing
|
||||
leds.setEnabled(0, false);
|
||||
leds.flush();
|
||||
}
|
||||
|
||||
/** set a fixed RGB color */
|
||||
void setRGB(const uint8_t r, const uint8_t g, const uint8_t b) {
|
||||
debugMod(NAME, "setting LEDS to fixed RGB color");
|
||||
ledMode = LEDMode::FIXED_COLOR;
|
||||
leds.getColor(0).setRGB(r,g,b);
|
||||
leds.setEnabled(0, true);
|
||||
leds.flush();
|
||||
}
|
||||
|
||||
/** set strobo mode */
|
||||
void setStrobo(const uint8_t r, const uint8_t g, const uint8_t b) {
|
||||
ledMode = LEDMode::STROBO_COLOR;
|
||||
leds.getColor(0).setRGB(r,g,b);
|
||||
leds.setEnabled(0, true);
|
||||
leds.flush();
|
||||
}
|
||||
|
||||
/** set LED rainbow fading */
|
||||
void setRainbow() {
|
||||
rainbow.restart();
|
||||
ledMode = LEDMode::RAINBOW_COLOR;
|
||||
leds.setEnabled(0, true);
|
||||
leds.flush();
|
||||
}
|
||||
|
||||
/** set LED fading between two colors */
|
||||
void setFade(const uint8_t r1, const uint8_t g1, const uint8_t b1, const uint8_t r2, const uint8_t g2, const uint8_t b2) {
|
||||
fadeBetween.setColor1(Color::fromRGB(r1, g1, b1));
|
||||
fadeBetween.setColor2(Color::fromRGB(r2, g2, b2));
|
||||
fadeBetween.restart();
|
||||
ledMode = LEDMode::FADE_BETWEEN_COLOR;
|
||||
leds.setEnabled(0, true);
|
||||
}
|
||||
|
||||
/** shortly flash the led and fade back to black */
|
||||
void setFlash(const uint8_t r1, const uint8_t g1, const uint8_t b1, const uint8_t r2, const uint8_t g2, const uint8_t b2, const int ms) {
|
||||
fadeOnce.setColor(Color::fromRGB(r1,g1,b1), Color::fromRGB(r2,g2,b2));
|
||||
fadeOnce.setFadeDuration(ms);
|
||||
ledMode = LEDMode::FADE_ONCE;
|
||||
leds.setEnabled(0, true);
|
||||
}
|
||||
|
||||
|
||||
|
||||
/** reset the rainbow-beats HUE to 0 */
|
||||
void setRainbowBeatRestart() {
|
||||
rainbowBeat.restart();
|
||||
ledMode = LEDMode::RAINBOW_BEAT;
|
||||
leds.setEnabled(0, true);
|
||||
}
|
||||
|
||||
/** show a beat */
|
||||
void setRainbowBeatBeat(const int ms) {
|
||||
rainbowBeat.flash(ms);
|
||||
ledMode = LEDMode::RAINBOW_BEAT;
|
||||
leds.setEnabled(0, true);
|
||||
}
|
||||
|
||||
/** show a beat */
|
||||
void setRainbowBeatShift(const int increment) {
|
||||
rainbowBeat.shift(increment);
|
||||
ledMode = LEDMode::RAINBOW_BEAT;
|
||||
leds.setEnabled(0, true);
|
||||
}
|
||||
|
||||
public:
|
||||
|
||||
/** udp callback */
|
||||
static void onUDP(void*, char* data, unsigned short len) {
|
||||
debugMod(NAME, "got udp data");
|
||||
buzzer.onUDP(data, len);
|
||||
}
|
||||
|
||||
};
|
||||
|
||||
|
||||
#endif // BUZZER_H
|
||||
Reference in New Issue
Block a user