add exploding fireworks
This commit is contained in:
fishbone-git
parent
6122a8371a
commit
50082043ef
122
wled00/FX.cpp
122
wled00/FX.cpp
@ -2348,3 +2348,125 @@ uint16_t WS2812FX::mode_candle()
|
||||
|
||||
return FRAMETIME;
|
||||
}
|
||||
|
||||
|
||||
/*
|
||||
* Exploding fireworks effect
|
||||
* adapted from: http://www.anirama.com/1000leds/1d-fireworks/
|
||||
*/
|
||||
#define NUM_SPARKS 60 // max number (could be NUM_LEDS / 2);
|
||||
uint16_t WS2812FX::mode_exploding_fireworks(void)
|
||||
{
|
||||
if (SEGENV.aux1==0) { // manage states using params
|
||||
flare();
|
||||
} else {
|
||||
explode();
|
||||
}
|
||||
|
||||
return FRAMETIME;
|
||||
}
|
||||
|
||||
static float sparkPos[NUM_SPARKS];
|
||||
static float sparkVel[NUM_SPARKS];
|
||||
static float sparkCol[NUM_SPARKS];
|
||||
static float flarePos;
|
||||
static float flareVel;
|
||||
static float brightness;
|
||||
|
||||
void WS2812FX::flare() {
|
||||
float gravity = -0.02 - (SEGMENT.speed/10000.0); // m/s/s
|
||||
|
||||
if (SEGENV.aux0 == 0) { //use aux0 for initialization flag
|
||||
flarePos = 0;
|
||||
flareVel = float(random16(150, 190)) / (75+SEGMENT.intensity/4); // trial and error to get reasonable range
|
||||
brightness = 1;
|
||||
Serial.printf("%3.1f %3.3f %3d",flareVel,gravity,SEGMENT.intensity);
|
||||
Serial.println("");
|
||||
// initialize launch sparks
|
||||
for (int i = 0; i < 5; i++) {
|
||||
sparkPos[i] = 0;
|
||||
sparkVel[i] = (float(random8(100,255)) / 255.0) * (flareVel/ 3.0); // random around 20% of flare velocity
|
||||
sparkCol[i] = sparkVel[i] * 1000.0;
|
||||
sparkCol[i] = constrain(sparkCol[i], 0, 255);
|
||||
}
|
||||
SEGENV.aux0=1;
|
||||
}
|
||||
|
||||
// launch
|
||||
if (flareVel >= -.3) {
|
||||
// sparks
|
||||
for (int i = 0; i < 5; i++) {
|
||||
sparkPos[i] += sparkVel[i];
|
||||
sparkPos[i] = constrain(sparkPos[i], 0, SEGLEN-1);
|
||||
sparkVel[i] += gravity;
|
||||
sparkCol[i] += -.8;
|
||||
sparkCol[i] = constrain(sparkCol[i], 0, 255);
|
||||
|
||||
CRGB color = HeatColor(sparkCol[i]);
|
||||
setPixelColor(SEGMENT.stop - int(sparkPos[i]),color.red,color.green,color.blue);
|
||||
}
|
||||
// flare
|
||||
fade_out(255);
|
||||
setPixelColor(SEGMENT.stop - int(flarePos),brightness*255,brightness*255,brightness*255);
|
||||
|
||||
flarePos += flareVel;
|
||||
flarePos = constrain(flarePos, 0, SEGLEN-1);
|
||||
flareVel += gravity;
|
||||
brightness *= .975;
|
||||
} else {
|
||||
SEGENV.aux0=0; // allow next state to init
|
||||
SEGENV.aux1=1; // ready to explode
|
||||
}
|
||||
}
|
||||
|
||||
/*
|
||||
* Explode!
|
||||
*
|
||||
* Explosion happens where the flare ended.
|
||||
* Size is proportional to the height.
|
||||
*/
|
||||
void WS2812FX::explode() {
|
||||
int nSparks = flarePos / 2; // works out to look about right
|
||||
nSparks = constrain(nSparks,0,NUM_SPARKS);
|
||||
static float dying_gravity;
|
||||
float gravity = -0.02 - (SEGMENT.speed/10000.0);
|
||||
float c1=120;
|
||||
float c2=50;
|
||||
|
||||
// initialize sparks
|
||||
if (SEGENV.aux0==0) {
|
||||
for (int i = 0; i < nSparks; i++) {
|
||||
sparkPos[i] = flarePos;
|
||||
sparkVel[i] = (float(random16(0, 20000)) / 10000.0) - 1.0; // from -1 to 1
|
||||
sparkCol[i] = abs(sparkVel[i] * 500.0); // set colors before scaling velocity to keep them bright
|
||||
sparkCol[i] = constrain(sparkCol[i], 0, 255);
|
||||
sparkVel[i] *= flarePos / SEGLEN; // proportional to height
|
||||
}
|
||||
sparkCol[0] = 255; // this will be our known spark
|
||||
dying_gravity = gravity;
|
||||
SEGENV.aux0=1;
|
||||
}
|
||||
|
||||
if (sparkCol[0] > c2/10 ) {//&& sparkPos[0] > 0) { // as long as our known spark is lit, work with all the sparks
|
||||
fade_out(255);
|
||||
for (int i = 0; i < nSparks; i++) {
|
||||
sparkPos[i] += sparkVel[i];
|
||||
sparkPos[i] = constrain(sparkPos[i], 0, SEGLEN-1);
|
||||
sparkVel[i] += dying_gravity;
|
||||
sparkCol[i] *= .97;
|
||||
sparkCol[i] = constrain(sparkCol[i], 0, 255); // red cross dissolve
|
||||
|
||||
if(sparkCol[i] > c1) { // fade white to yellow
|
||||
setPixelColor(SEGMENT.stop - sparkPos[i],255, 255, (255 * (sparkCol[i] - c1)) / (255 - c1));
|
||||
} else if (sparkCol[i] < c2) { // fade from red to black
|
||||
setPixelColor(SEGMENT.stop - sparkPos[i],(255 * sparkCol[i]) / c2, 0, 0);
|
||||
} else { // fade from yellow to red
|
||||
setPixelColor(SEGMENT.stop - sparkPos[i],255, (255 * (sparkCol[i] - c2)) / (c1 - c2), 0);
|
||||
}
|
||||
}
|
||||
dying_gravity *= .995; // as sparks burn out they fall slower
|
||||
} else {
|
||||
SEGENV.aux0=0;
|
||||
SEGENV.aux1=0;
|
||||
}
|
||||
}
|
||||
12
wled00/FX.h
12
wled00/FX.h
@ -84,7 +84,7 @@
|
||||
#define IS_REVERSE ((SEGMENT.options & REVERSE ) == REVERSE )
|
||||
#define IS_SELECTED ((SEGMENT.options & SELECTED) == SELECTED )
|
||||
|
||||
#define MODE_COUNT 89
|
||||
#define MODE_COUNT 90
|
||||
|
||||
#define FX_MODE_STATIC 0
|
||||
#define FX_MODE_BLINK 1
|
||||
@ -175,6 +175,7 @@
|
||||
#define FX_MODE_SPOTS_FADE 86
|
||||
#define FX_MODE_GLITTER 87
|
||||
#define FX_MODE_CANDLE 88
|
||||
#define FX_MODE_EXPLODING_FIREWORKS 89
|
||||
|
||||
|
||||
class WS2812FX {
|
||||
@ -321,6 +322,8 @@ class WS2812FX {
|
||||
_mode[FX_MODE_SPOTS_FADE] = &WS2812FX::mode_spots_fade;
|
||||
_mode[FX_MODE_GLITTER] = &WS2812FX::mode_glitter;
|
||||
_mode[FX_MODE_CANDLE] = &WS2812FX::mode_candle;
|
||||
_mode[FX_MODE_EXPLODING_FIREWORKS] = &WS2812FX::mode_exploding_fireworks;
|
||||
|
||||
|
||||
_brightness = DEFAULT_BRIGHTNESS;
|
||||
currentPalette = CRGBPalette16(CRGB::Black);
|
||||
@ -359,6 +362,8 @@ class WS2812FX {
|
||||
resetSegments(),
|
||||
setPixelColor(uint16_t n, uint32_t c),
|
||||
setPixelColor(uint16_t n, uint8_t r, uint8_t g, uint8_t b, uint8_t w = 0),
|
||||
flare(void),
|
||||
explode(void),
|
||||
show(void);
|
||||
|
||||
bool
|
||||
@ -503,7 +508,8 @@ class WS2812FX {
|
||||
mode_spots(void),
|
||||
mode_spots_fade(void),
|
||||
mode_glitter(void),
|
||||
mode_candle(void);
|
||||
mode_candle(void),
|
||||
mode_exploding_fireworks(void);
|
||||
|
||||
|
||||
private:
|
||||
@ -577,7 +583,7 @@ const char JSON_mode_names[] PROGMEM = R"=====([
|
||||
"Two Dots","Two Areas","Circus","Halloween","Tri Chase","Tri Wipe","Tri Fade","Lightning","ICU","Multi Comet",
|
||||
"Dual Scanner","Stream 2","Oscillate","Pride 2015","Juggle","Palette","Fire 2012","Colorwaves","Bpm","Fill Noise",
|
||||
"Noise 1","Noise 2","Noise 3","Noise 4","Colortwinkles","Lake","Meteor","Smooth Meteor","Railway","Ripple",
|
||||
"Twinklefox","Twinklecat","Halloween Eyes","Solid Pattern","Solid Pattern Tri","Spots","Spots Fade","Glitter","Candle"
|
||||
"Twinklefox","Twinklecat","Halloween Eyes","Solid Pattern","Solid Pattern Tri","Spots","Spots Fade","Glitter","Candle","Fireworks 1D"
|
||||
])=====";
|
||||
|
||||
|
||||
|
||||
Loading…
Reference in New Issue
Block a user