290 lines
8.2 KiB
C++
290 lines
8.2 KiB
C++
/* Example SPI transfert
|
|
*
|
|
* This sample code is in the public domain.
|
|
*/
|
|
#include "espressif/esp_common.h"
|
|
#include "esp/uart.h"
|
|
#include "FreeRTOS.h"
|
|
#include "task.h"
|
|
#include "esp8266.h"
|
|
#include <stdio.h>
|
|
#include "esp/spi.h"
|
|
#include "math.h"
|
|
#include <time.h>
|
|
#include <semphr.h>
|
|
|
|
#define DAYTIME(h, m, s) (h*3600+m*60+s)
|
|
|
|
extern SemaphoreHandle_t time_available_semaphore;
|
|
|
|
struct {
|
|
time_t sunrise_start = DAYTIME(8, 30, 0);
|
|
time_t sunrise_end = DAYTIME(9, 0, 0);
|
|
time_t sunrise_shutdown = DAYTIME(7, 0, 0);
|
|
time_t sunset_time = DAYTIME(22, 0, 0);
|
|
} settings;
|
|
|
|
|
|
struct {
|
|
uint8_t r = 255;
|
|
uint8_t g = 160;
|
|
uint8_t b = 80;
|
|
} white;
|
|
|
|
extern "C" time_t day_seconds() {
|
|
time_t t1, t2;
|
|
struct tm tms;
|
|
time(&t1);
|
|
localtime_r(&t1, &tms);
|
|
tms.tm_hour = 0;
|
|
tms.tm_min = 0;
|
|
tms.tm_sec = 0;
|
|
t2 = mktime(&tms);
|
|
return t1 - t2;
|
|
}
|
|
|
|
struct led_t {
|
|
struct {
|
|
unsigned int mod: 5, marker: 3;
|
|
} __attribute__((packed)) global = {0x1F, 0x7};
|
|
uint8_t b = 0;
|
|
uint8_t g = 0;
|
|
uint8_t r = 0;
|
|
};
|
|
|
|
led_t leds[4][10];
|
|
|
|
float square(float x) {
|
|
return x * x;
|
|
}
|
|
|
|
float lerp(float a, float b, float x) {
|
|
return (1. - x) * a + x * b;
|
|
}
|
|
|
|
float clamp(float x, float a, float b) {
|
|
if(x < a)
|
|
return a;
|
|
if(x > b)
|
|
return b;
|
|
return x;
|
|
}
|
|
|
|
float fade(float x, float offset, float factor) {
|
|
float val = (x * factor - offset);
|
|
return lerp(square(clamp(val, 0, 1)), square(1 - clamp(1 - val, 0, 1)), val);
|
|
}
|
|
|
|
void write_leds() {
|
|
spi_transfer_32(1, 0x00000000);
|
|
for (int i = 0; i < 10; i++)
|
|
spi_transfer_32(1, *(uint32_t *) &leds[0][i]);
|
|
for (int i = 0; i < 10; i++)
|
|
spi_transfer_32(1, *(uint32_t *) &leds[1][9 - i]);
|
|
for (int i = 0; i < 10; i++)
|
|
spi_transfer_32(1, *(uint32_t *) &leds[2][i]);
|
|
for (int i = 0; i < 10; i++)
|
|
spi_transfer_32(1, *(uint32_t *) &leds[3][9 - i]);
|
|
spi_transfer_32(1, 0xFFFFFFFF);
|
|
spi_transfer_32(1, 0xFFFFFFFF);
|
|
}
|
|
|
|
enum state_t {
|
|
BOOT_S, SUNRISE_S, MORNING_S, DAY_S, NIGHT_S
|
|
};
|
|
|
|
volatile state_t s = BOOT_S;
|
|
|
|
volatile bool maual_mode = false;
|
|
|
|
extern "C" void manual_switch() {
|
|
|
|
maual_mode = !maual_mode;
|
|
|
|
printf("int %d\n", (int) maual_mode);
|
|
|
|
if(maual_mode) {
|
|
|
|
if(s == NIGHT_S) {
|
|
|
|
for (int j = 0; j < 4; j++)
|
|
for (int i = 0; i < 8; i += 2) {
|
|
leds[j][i].global.mod = 2;
|
|
leds[j][i].r = 128;
|
|
leds[j][i].g = 30;
|
|
leds[j][i].b = 15;
|
|
}
|
|
} else {
|
|
|
|
for (int j = 0; j < 4; j++)
|
|
for (int i = 0; i < 10; i++) {
|
|
leds[j][i].global.mod = 31;
|
|
leds[j][i].r = white.r;
|
|
leds[j][i].g = white.g;
|
|
leds[j][i].b = white.b;
|
|
}
|
|
}
|
|
|
|
write_leds();
|
|
|
|
} else {
|
|
for (int j = 0; j < 4; j++)
|
|
for (int i = 0; i < 10; i++) {
|
|
leds[j][i].global.mod = 0;
|
|
leds[j][i].r = 0;
|
|
leds[j][i].g = 0;
|
|
leds[j][i].b = 0;
|
|
}
|
|
write_leds();
|
|
}
|
|
}
|
|
|
|
[[noreturn]] void loop(void *pvParameters) {
|
|
spi_init(1, SPI_MODE0, SPI_FREQ_DIV_1M, 1, SPI_LITTLE_ENDIAN, false);
|
|
|
|
while (!uxSemaphoreGetCount(time_available_semaphore))
|
|
vTaskDelay(500 / portTICK_PERIOD_MS);
|
|
|
|
maual_mode = false;
|
|
|
|
for (int j = 0; j < 4; j++)
|
|
for (int i = 0; i < 10; i++) {
|
|
leds[j][i].global.mod = 0;
|
|
leds[j][i].r = 0;
|
|
leds[j][i].g = 0;
|
|
leds[j][i].b = 0;
|
|
}
|
|
write_leds();
|
|
|
|
vTaskDelay(1000 / portTICK_PERIOD_MS);
|
|
|
|
printf("START>\n");
|
|
|
|
while (true) {
|
|
|
|
time_t rt = day_seconds();
|
|
|
|
if(maual_mode) {
|
|
|
|
if(s == NIGHT_S) {
|
|
|
|
for (int j = 0; j < 4; j++)
|
|
for (int i = 0; i < 8; i += 2) {
|
|
leds[j][i].global.mod = 2;
|
|
leds[j][i].r = 128;
|
|
leds[j][i].g = 30;
|
|
leds[j][i].b = 15;
|
|
}
|
|
} else {
|
|
|
|
for (int j = 0; j < 4; j++)
|
|
for (int i = 0; i < 10; i++) {
|
|
leds[j][i].global.mod = 31;
|
|
leds[j][i].r = white.r;
|
|
leds[j][i].g = white.g;
|
|
leds[j][i].b = white.b;
|
|
}
|
|
}
|
|
|
|
write_leds();
|
|
|
|
} else {
|
|
|
|
if(s == BOOT_S) {
|
|
if(rt >= settings.sunrise_start) {
|
|
s = SUNRISE_S;
|
|
printf("SUNRISE_S\n");
|
|
}
|
|
if(rt >= settings.sunrise_end) {
|
|
s = MORNING_S;
|
|
printf("MORNING_S\n");
|
|
}
|
|
if(rt >= settings.sunset_time || rt < settings.sunrise_start) {
|
|
s = NIGHT_S;
|
|
printf("NIGHT_S\n");
|
|
}
|
|
|
|
printf("%lld | %lld %lld %lld %lld\n", rt,
|
|
settings.sunrise_start,
|
|
settings.sunrise_end,
|
|
settings.sunset_time,
|
|
settings.sunrise_shutdown);
|
|
} else if(s == SUNRISE_S) {
|
|
if(rt >= settings.sunrise_end) {
|
|
s = MORNING_S;
|
|
printf("MORNING_S\n");
|
|
} else {
|
|
int steps = (settings.sunrise_end - settings.sunrise_start) * (1000 / 50);
|
|
int t = (rt - settings.sunrise_start) * (1000 / 50);
|
|
for (; t < steps && !maual_mode; t++) {
|
|
|
|
for (auto &led: leds)
|
|
for (int i = 0; i < 10; i++) {
|
|
float val = (-i * 30. + t * 3. * (4. / 5.)) / (float) steps;
|
|
led[i].global.mod = 8 + fade(val, 0., 0.5) * 23;
|
|
led[i].r = fade(val, 0.01, 1.) * white.r;
|
|
led[i].g = fade(val, 0.1, 0.5) * white.g;
|
|
led[i].b = fade(val, 0.6, 0.7) * white.b;
|
|
}
|
|
|
|
write_leds();
|
|
|
|
if((t % 25) == 0) {
|
|
|
|
printf("Time: %d%% %d %d/%d\n", (t * 100) / steps, (int) day_seconds(), t, steps);
|
|
printf("%d %d %d %d\n",
|
|
leds[0][0].global.mod,
|
|
leds[0][0].r,
|
|
leds[0][0].g,
|
|
leds[0][0].b);
|
|
}
|
|
|
|
vTaskDelay(50 / portTICK_PERIOD_MS);
|
|
}
|
|
}
|
|
} else if(s == MORNING_S) {
|
|
if(rt >= settings.sunrise_shutdown) {
|
|
s = DAY_S;
|
|
printf("DAY_S\n");
|
|
} else {
|
|
for (int j = 0; j < 4; j++)
|
|
for (int i = 0; i < 10; i++) {
|
|
leds[j][i].global.mod = 31;
|
|
leds[j][i].r = white.r;
|
|
leds[j][i].g = white.g;
|
|
leds[j][i].b = white.b;
|
|
}
|
|
write_leds();
|
|
}
|
|
} else if(s == DAY_S) {
|
|
if(rt >= settings.sunset_time) {
|
|
s = NIGHT_S;
|
|
printf("NIGHT_S\n");
|
|
} else {
|
|
for (int j = 0; j < 4; j++)
|
|
for (int i = 0; i < 10; i++) {
|
|
leds[j][i].global.mod = 0;
|
|
leds[j][i].r = 0;
|
|
leds[j][i].g = 0;
|
|
leds[j][i].b = 0;
|
|
}
|
|
|
|
write_leds();
|
|
}
|
|
} else if(s == NIGHT_S) {
|
|
if(rt >= settings.sunrise_start && rt < settings.sunset_time) {
|
|
s = SUNRISE_S;
|
|
printf("SUNRISE_S\n");
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
vTaskDelay(1000 / portTICK_PERIOD_MS);
|
|
|
|
}
|
|
}
|
|
|
|
extern "C" void register_app(void) {
|
|
xTaskCreate(loop, "loop", 512, nullptr, 2, NULL);
|
|
}
|