jeans/WLED
jeans/WLED
1
0
Fork 0
mirror of https://github.com/wled/WLED.git synced 2025-07-28 13:16:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fireworks 1D working

Browse Source
This commit is contained in:
cschwinne 2020-01-03 17:47:06 +01:00
parent b1f26d4ebe
commit cdef7a53a4
2 changed files with 114 additions and 112 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

142
wled00/FX.cpp
View File

@ -2605,118 +2605,122 @@ uint16_t WS2812FX::mode_starburst(void) {
* Exploding fireworks effect * Exploding fireworks effect
* adapted from: http://www.anirama.com/1000leds/1d-fireworks/ * adapted from: http://www.anirama.com/1000leds/1d-fireworks/
*/ */
#define NUM_SPARKS 60 // max number (could be NUM_LEDS / 2);
//each needs 12 byte
typedef struct Spark {
float pos;
float vel;
float col;
} spark;
uint16_t WS2812FX::mode_exploding_fireworks(void) uint16_t WS2812FX::mode_exploding_fireworks(void)
{ {
if (SEGENV.aux1==0) { // manage states using params //allocate segment data
flare(); uint16_t numSparks = 2 + (SEGLEN >> 1);
} else { if (numSparks > 80) numSparks = 80;
explode(); uint16_t dataSize = sizeof(spark) * numSparks;
} if (!SEGENV.allocateData(dataSize)) return mode_static(); //allocation failed
return FRAMETIME; fill(BLACK);
}
static float sparkPos[NUM_SPARKS]; Spark* sparks = reinterpret_cast<Spark*>(SEGENV.data);
static float sparkVel[NUM_SPARKS]; Spark* flare = sparks; //first spark is flare data
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 float gravity = -0.02 - (SEGMENT.speed/10000.0); // m/s/s
if (SEGENV.aux0 == 0) { //use aux0 for initialization flag if (SEGENV.aux1 == 0) { //FLARE
flarePos = 0; if (SEGENV.aux0 == 0) { //init flare
flareVel = float(random16(150, 190)) / (75+(255-SEGMENT.intensity)/4); // trial and error to get reasonable range flare->pos = 0;
brightness = 1; flare->vel = float(random16(150, 190)) / (75+(255-SEGMENT.intensity)/4); // trial and error to get reasonable range
//flare->vel = 2;
flare->col = 1; //brightness
// initialize launch sparks // initialize launch sparks
for (int i = 0; i < 5; i++) { for (int i = 1; i < min(6, numSparks); i++) {
sparkPos[i] = 0; sparks[i].pos = 0;
sparkVel[i] = (float(random8(100,255)) / 255.0) * (flareVel/ 3.0); // random around 20% of flare velocity sparks[i].vel = (float(random8(100,255)) / 255.0) * (flare->vel/ 3.0); // random around 20% of flare velocity
sparkCol[i] = sparkVel[i] * 1000.0; sparks[i].col = sparks[i].vel * 1000.0;
sparkCol[i] = constrain(sparkCol[i], 0, 255); sparks[i].col = constrain(sparks[i].col, 0, 255);
} }
SEGENV.aux0 = 1; SEGENV.aux0 = 1;
} }
// launch // launch
if (flareVel >= -.3) { if (flare->vel > -.3) {
// sparks // sparks
for (int i = 0; i < 5; i++) { for (int i = 1; i < min(6, numSparks); i++) {
sparkPos[i] += sparkVel[i]; sparks[i].pos += sparks[i].vel;
sparkPos[i] = constrain(sparkPos[i], 0, SEGLEN-1); sparks[i].pos = constrain(sparks[i].pos, 0, SEGLEN-1);
sparkVel[i] += gravity; sparks[i].vel += gravity;
sparkCol[i] += -.8; sparks[i].col -= 0.8;
sparkCol[i] = constrain(sparkCol[i], 0, 255); sparks[i].col = constrain(sparks[i].col, 0, 255);
CRGB color = HeatColor(sparkCol[i]); CRGB color = HeatColor(sparks[i].col);
setPixelColor(SEGMENT.stop - int(sparkPos[i]),color.red,color.green,color.blue); setPixelColor(SEGMENT.start + int(sparks[i].pos),color.red,color.green,color.blue);
} }
// flare // flare
fade_out(255); //fade_out(255);
setPixelColor(SEGMENT.stop - int(flarePos),brightness*255,brightness*255,brightness*255); setPixelColor(SEGMENT.start + int(flare->pos),flare->col*255,flare->col*255,flare->col*255);
flarePos += flareVel; flare->pos += flare->vel;
flarePos = constrain(flarePos, 0, SEGLEN-1); flare->pos = constrain(flare->pos, 0, SEGLEN-1);
flareVel += gravity; flare->vel += gravity;
brightness *= .975; flare->col *= .975;
} else { } else {
SEGENV.aux0 = 0; // allow next state to init SEGENV.aux0 = 0; // allow next state to init
SEGENV.aux1 = 1; // ready to explode SEGENV.aux1 = 1; // ready to explode
} }
} } else {
/* /*
* Explode! * Explode!
* *
* Explosion happens where the flare ended. * Explosion happens where the flare ended.
* Size is proportional to the height. * Size is proportional to the height.
*/ */
void WS2812FX::explode() { int nSparks = flare->pos / 2; // works out to look about right
int nSparks = flarePos / 2; // works out to look about right nSparks = constrain(nSparks, 0, numSparks);
nSparks = constrain(nSparks,0,NUM_SPARKS);
static float dying_gravity; static float dying_gravity;
float gravity = -0.02 - ((255-SEGMENT.speed)/10000.0); //gravity = -0.02 - ((255-SEGMENT.speed)/10000.0);
float c1=120;
float c2=50;
// initialize sparks // initialize sparks
if (SEGENV.aux0==0) { if (SEGENV.aux0==0) {
for (int i = 0; i < nSparks; i++) { for (int i = 1; i < nSparks; i++) {
sparkPos[i] = flarePos; sparks[i].pos = flare->pos;
sparkVel[i] = (float(random16(0, 20000)) / 10000.0) - 1.0; // from -1 to 1 sparks[i].vel = (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 sparks[i].col = abs(sparks[i].vel * 500.0); // set colors before scaling velocity to keep them bright
sparkCol[i] = constrain(sparkCol[i], 0, 255); sparks[i].col = constrain(sparks[i].col, 0, 255);
sparkVel[i] *= flarePos / SEGLEN; // proportional to height sparks[i].vel *= flare->pos / SEGLEN; // proportional to height
} }
sparkCol[0] = 255; // this will be our known spark sparks[1].col = 255; // this will be our known spark
dying_gravity = gravity; dying_gravity = gravity;
SEGENV.aux0 = 1; SEGENV.aux0 = 1;
} }
if (sparkCol[0] > c2/10 ) {//&& sparkPos[0] > 0) { // as long as our known spark is lit, work with all the sparks float c1 = 120;
fade_out(255); float c2 = 50;
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 if (sparks[1].col > 5) {//&& sparks[1].pos > 0) { // as long as our known spark is lit, work with all the sparks
setPixelColor(SEGMENT.stop - sparkPos[i],255, 255, (255 * (sparkCol[i] - c1)) / (255 - c1)); //fade_out(255);
} else if (sparkCol[i] < c2) { // fade from red to black for (int i = 1; i < nSparks; i++) {
setPixelColor(SEGMENT.stop - sparkPos[i],(255 * sparkCol[i]) / c2, 0, 0); sparks[i].pos += sparks[i].vel;
sparks[i].pos = constrain(sparks[i].pos, 0, SEGLEN-1);
sparks[i].vel += dying_gravity;
sparks[i].col *= .97;
sparks[i].col = constrain(sparks[i].col, 0, 255); // red cross dissolve
if(sparks[i].pos > c1) { // fade white to yellow
setPixelColor(SEGMENT.start + sparks[i].pos,255, 255, (255 * (sparks[i].col - c1)) / (255 - c1));
} else if (sparks[i].col < c2) { // fade from red to black
setPixelColor(SEGMENT.start + sparks[i].pos,(255 * sparks[i].col) / c2, 0, 0);
} else { // fade from yellow to red } else { // fade from yellow to red
setPixelColor(SEGMENT.stop - sparkPos[i],255, (255 * (sparkCol[i] - c2)) / (c1 - c2), 0); setPixelColor(SEGMENT.start + sparks[i].pos,255, (255 * (sparks[i].col - c2)) / (c1 - c2), 0);
} }
} }
dying_gravity *= .995; // as sparks burn out they fall slower dying_gravity *= .995; // as sparks burn out they fall slower
} else { } else {
SEGENV.aux0=0; SEGENV.aux0 = 0; //init again
SEGENV.aux1=0; SEGENV.aux1 = 0; //back to flare
} }
} }
return FRAMETIME;
}

2
wled00/FX.h
View File

@ -386,8 +386,6 @@ class WS2812FX {
resetSegments(), resetSegments(),
setPixelColor(uint16_t n, uint32_t c), setPixelColor(uint16_t n, uint32_t c),
setPixelColor(uint16_t n, uint8_t r, uint8_t g, uint8_t b, uint8_t w = 0), setPixelColor(uint16_t n, uint8_t r, uint8_t g, uint8_t b, uint8_t w = 0),
flare(void),
explode(void),
show(void); show(void);
bool bool