avr/flame/main.c

110 lines
2.7 KiB
C

#include <avr/io.h>
#include <util/delay.h>
#include <base.h>
#include <sk9822.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) % 32) - 16;
uint16_t 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;
int8_t red_step;
int8_t green_step;
int8_t blue_step;
uint8_t frame;
uint8_t duration;
} animation_t;
animation_t animation_new(void) {
return (animation_t){
.lamp = (rgb_t){ .brightness = 9, .r = 212, .g = 50, .b = 0 },
.color_scheme = normal,
.red_step = 0,
.green_step = 0,
.blue_step = 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, 10, 40, rng);
animation->red_step = 60 / (rdest - animation->lamp.r);
animation->green_step = 60 / (gdest - animation->lamp.g);
animation->duration = 60;
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;
animation->red_step = 0;
animation->green_step = 0;
animation->blue_step = 0;
} else {
animation->frame++;
animation->lamp.r += animation->red_step;
animation->lamp.g += animation->green_step;
animation->lamp.b += animation->blue_step;
}
}
}
/*
void flicker(rgb_t *pixel, rng_t *rng) {
pixel->r = random_step(pixel->r, 180, 255, rng);
pixel->g = random_step(pixel->g, 10, 40, rng);
}
*/
int main (void) {
DDRB = _BV(2) | _BV(1) | _BV(0);
PORTB = 0;
_delay_ms(50);
dio_t data_pin = { .ddr = &DDRB, .port = &PORTB, .pin = &PINB, .addr = 0 };
dio_t 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);
send_pixels(data_pin, clock_pin, &animation.lamp, 1);
_delay_ms(FRAME_DELAY_MS);
}
}