avr/flame/main.c

104 lines
2.7 KiB
C

#include <avr/io.h>
#include <util/delay.h>
#include <dio.h>
#include <sk9822.h>
#include <rng.h>
#include <animation.h>
#define FPS 60
#define FRAME_DELAY_MS 1000 / FPS
uint8_t random_step(uint8_t value, uint8_t min, uint8_t max, rng_t *rng) {
int8_t step = (rng_sample(rng) % 20) - 10;
int new_value = value + step;
if (new_value > max) {
return max;
} else if (new_value < min) {
return min;
} else {
return new_value;
}
}
typedef enum {
normal,
creepy,
} color_scheme_e;
typedef struct {
rgb_t lamp;
color_scheme_e color_scheme;
time_line_t red_line;
time_line_t green_line;
time_line_t blue_line;
uint8_t frame;
uint8_t duration;
} animation_t;
animation_t animation_new(void) {
return (animation_t){
.lamp = (rgb_t){ .brightness = 1, .r = 200, .g = 30, .b = 0 },
.color_scheme = normal,
.red_line = time_line_new(0, 0, 0, 0),
.green_line = time_line_new(0, 0, 0, 0),
.blue_line = time_line_new(0, 0, 0, 0),
.frame = 0,
.duration = 0,
};
}
void animation_flicker(animation_t *animation, rng_t *rng) {
uint8_t rdest;
uint8_t gdest;
switch (animation->color_scheme) {
case normal:
rdest = random_step(animation->lamp.r, 180, 255, rng);
gdest = random_step(animation->lamp.g, 20, 50, rng);
animation->red_line = time_line_new(0, animation->lamp.r, 30, rdest);
animation->green_line = time_line_new(0, animation->lamp.g, 30, gdest);
animation->duration = 30;
break;
case creepy:
break;
}
}
void animation_step(animation_t *animation, rng_t *rng) {
if (animation->duration == 0) {
animation_flicker(animation, rng);
} else {
if (animation->frame == animation->duration) {
animation->frame = 0;
animation->duration = 0;
} else {
animation->frame++;
animation->lamp.r = time_line_next(&animation->red_line);
animation->lamp.g = time_line_next(&animation->green_line);
animation->lamp.b = time_line_next(&animation->blue_line);
}
}
}
int main (void) {
DDRB = _BV(2) | _BV(1) | _BV(0);
PORTB = 0;
_delay_ms(50);
sk9822_t lights = {
.data_pin = { .ddr = &DDRB, .port = &PORTB, .pin = &PINB, .addr = 0 },
.clock_pin = { .ddr = &DDRB, .port = &PORTB, .pin = &PINB, .addr = 2 }
};
rng_t rng = rng_new(0);
animation_t animation = animation_new();
while (1) {
animation_step(&animation, &rng);
sk9822_send(&lights, &animation.lamp, 1);
_delay_ms(FRAME_DELAY_MS);
}
}