Fireworks 1D working
This commit is contained in:
cschwinne
parent
b1f26d4ebe
commit
cdef7a53a4
224
wled00/FX.cpp
224
wled00/FX.cpp
@ -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
|
||||||
|
|
||||||
|
fill(BLACK);
|
||||||
|
|
||||||
|
Spark* sparks = reinterpret_cast<Spark*>(SEGENV.data);
|
||||||
|
Spark* flare = sparks; //first spark is flare data
|
||||||
|
|
||||||
|
float gravity = -0.02 - (SEGMENT.speed/10000.0); // m/s/s
|
||||||
|
|
||||||
|
if (SEGENV.aux1 == 0) { //FLARE
|
||||||
|
if (SEGENV.aux0 == 0) { //init flare
|
||||||
|
flare->pos = 0;
|
||||||
|
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
|
||||||
|
for (int i = 1; i < min(6, numSparks); i++) {
|
||||||
|
sparks[i].pos = 0;
|
||||||
|
sparks[i].vel = (float(random8(100,255)) / 255.0) * (flare->vel/ 3.0); // random around 20% of flare velocity
|
||||||
|
sparks[i].col = sparks[i].vel * 1000.0;
|
||||||
|
sparks[i].col = constrain(sparks[i].col, 0, 255);
|
||||||
|
}
|
||||||
|
SEGENV.aux0 = 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
// launch
|
||||||
|
if (flare->vel > -.3) {
|
||||||
|
// sparks
|
||||||
|
for (int i = 1; i < min(6, numSparks); i++) {
|
||||||
|
sparks[i].pos += sparks[i].vel;
|
||||||
|
sparks[i].pos = constrain(sparks[i].pos, 0, SEGLEN-1);
|
||||||
|
sparks[i].vel += gravity;
|
||||||
|
sparks[i].col -= 0.8;
|
||||||
|
sparks[i].col = constrain(sparks[i].col, 0, 255);
|
||||||
|
|
||||||
|
CRGB color = HeatColor(sparks[i].col);
|
||||||
|
setPixelColor(SEGMENT.start + int(sparks[i].pos),color.red,color.green,color.blue);
|
||||||
|
}
|
||||||
|
// flare
|
||||||
|
//fade_out(255);
|
||||||
|
setPixelColor(SEGMENT.start + int(flare->pos),flare->col*255,flare->col*255,flare->col*255);
|
||||||
|
|
||||||
|
flare->pos += flare->vel;
|
||||||
|
flare->pos = constrain(flare->pos, 0, SEGLEN-1);
|
||||||
|
flare->vel += gravity;
|
||||||
|
flare->col *= .975;
|
||||||
|
} else {
|
||||||
|
SEGENV.aux0 = 0; // allow next state to init
|
||||||
|
SEGENV.aux1 = 1; // ready to explode
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
/*
|
||||||
|
* Explode!
|
||||||
|
*
|
||||||
|
* Explosion happens where the flare ended.
|
||||||
|
* Size is proportional to the height.
|
||||||
|
*/
|
||||||
|
int nSparks = flare->pos / 2; // works out to look about right
|
||||||
|
nSparks = constrain(nSparks, 0, numSparks);
|
||||||
|
static float dying_gravity;
|
||||||
|
//gravity = -0.02 - ((255-SEGMENT.speed)/10000.0);
|
||||||
|
|
||||||
|
// initialize sparks
|
||||||
|
if (SEGENV.aux0==0) {
|
||||||
|
for (int i = 1; i < nSparks; i++) {
|
||||||
|
sparks[i].pos = flare->pos;
|
||||||
|
sparks[i].vel = (float(random16(0, 20000)) / 10000.0) - 1.0; // from -1 to 1
|
||||||
|
sparks[i].col = abs(sparks[i].vel * 500.0); // set colors before scaling velocity to keep them bright
|
||||||
|
sparks[i].col = constrain(sparks[i].col, 0, 255);
|
||||||
|
sparks[i].vel *= flare->pos / SEGLEN; // proportional to height
|
||||||
|
}
|
||||||
|
sparks[1].col = 255; // this will be our known spark
|
||||||
|
dying_gravity = gravity;
|
||||||
|
SEGENV.aux0 = 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
float c1 = 120;
|
||||||
|
float c2 = 50;
|
||||||
|
|
||||||
|
if (sparks[1].col > 5) {//&& sparks[1].pos > 0) { // as long as our known spark is lit, work with all the sparks
|
||||||
|
//fade_out(255);
|
||||||
|
for (int i = 1; i < nSparks; i++) {
|
||||||
|
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
|
||||||
|
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
|
||||||
|
} else {
|
||||||
|
SEGENV.aux0 = 0; //init again
|
||||||
|
SEGENV.aux1 = 0; //back to flare
|
||||||
|
}
|
||||||
|
}
|
||||||
return FRAMETIME;
|
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+(255-SEGMENT.intensity)/4); // trial and error to get reasonable range
|
|
||||||
brightness = 1;
|
|
||||||
|
|
||||||
// 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 - ((255-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;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|||||||
@ -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
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user