Compare commits

...

1 commit

Author SHA1 Message Date
a806910034 stash
Some checks failed
continuous-integration/drone/push Build is failing
2021-08-08 22:59:15 +02:00
3 changed files with 575 additions and 5 deletions

261
firmware/app.cpp Normal file
View file

@ -0,0 +1,261 @@
/* 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>
#define DAYTIME(h,m,s) (h*3600+m*60+s)
struct {
time_t sunrise_start = DAYTIME(5,30,0);
time_t sunrise_end = DAYTIME(6,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;
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 };
state_t s = BOOT_S;
bool maual_mode = false;
extern "C" void manual_switch(){
maual_mode = !maual_mode;
printf("%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();
}
}
void loop(void *pvParameters)
{
spi_init(1, SPI_MODE0, SPI_FREQ_DIV_1M, 1, SPI_LITTLE_ENDIAN, false);
while(1){
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");
}
} 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(int j = 0; j < 4; j++)
for(int i = 0; i < 10; i++){
float val = (-i*30.+t*3.*(4./5.))/(float)steps;
leds[j][i].global.mod = 8 + fade(val,0.,0.5)*23;
leds[j][i].r = fade(val,0.01,1.) * white.r;
leds[j][i].g = fade(val,0.1,0.5) * white.g;
leds[j][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", 1024, NULL, 2, NULL);
}

7
firmware/config.sample Normal file
View file

@ -0,0 +1,7 @@
sunrise_start: time
sunrise_end: time
sunrise_shutdown: time
sunrise_color: color
sunrise_fade: linear, ease, ease_in, ease_out
sunset_time: time
night_color: color

View file

@ -1,16 +1,318 @@
#include <espressif/esp_common.h>
#include <esp8266.h>
#include <esp/uart.h>
#include <string.h>
#include <stdio.h>
#include <FreeRTOS.h>
#include <task.h>
#include <ssid_config.h>
#include <httpd/httpd.h>
#include <sys/time.h>
#include <lwip/err.h>
#include <lwip/sockets.h>
#include <lwip/sys.h>
#include <lwip/netdb.h>
#include <lwip/dns.h>
#include "system.h"
#include "wifi.h"
#include "web.h"
#include "mqtt.h"
#include "lux.h"
#include <stdio.h>
#include <FreeRTOS.h>
#include <task.h>
#define LED_PIN 2
#define SWITCH_PIN 2
#include <espressif/esp_common.h>
#include <esp/uart.h>
/* Add extras/sntp component to makefile for this include to work */
#include <sntp.h>
#include <time.h>
#define SNTP_SERVERS "0.pool.ntp.org", "1.pool.ntp.org", \
"2.pool.ntp.org", "3.pool.ntp.org"
#define vTaskDelayMs(ms) vTaskDelay((ms)/portTICK_PERIOD_MS)
#define UNUSED_ARG(x) (void)x
const gpio_inttype_t int_type = GPIO_INTTYPE_EDGE_NEG;
enum {
SSI_WALLTIME,
SSI_UPTIME,
SSI_FREE_HEAP,
SSI_LED_STATE
};
int32_t ssi_handler(int32_t iIndex, char *pcInsert, int32_t iInsertLen)
{
struct timeval tv;
switch (iIndex) {
case SSI_WALLTIME:
gettimeofday(&tv, NULL);
snprintf(pcInsert, iInsertLen, "%d",
(int)tv.tv_sec);
break;
case SSI_UPTIME:
snprintf(pcInsert, iInsertLen, "%d",
xTaskGetTickCount() * portTICK_PERIOD_MS / 1000);
break;
case SSI_FREE_HEAP:
snprintf(pcInsert, iInsertLen, "%d", (int) xPortGetFreeHeapSize());
break;
case SSI_LED_STATE:
snprintf(pcInsert, iInsertLen, (GPIO.OUT & BIT(LED_PIN)) ? "Off" : "On");
break;
default:
snprintf(pcInsert, iInsertLen, "N/A");
break;
}
/* Tell the server how many characters to insert */
return (strlen(pcInsert));
}
const char *gpio_cgi_handler(int iIndex, int iNumParams, char *pcParam[], char *pcValue[])
{
for (int i = 0; i < iNumParams; i++) {
if (strcmp(pcParam[i], "on") == 0) {
uint8_t gpio_num = atoi(pcValue[i]);
gpio_enable(gpio_num, GPIO_OUTPUT);
gpio_write(gpio_num, true);
} else if (strcmp(pcParam[i], "off") == 0) {
uint8_t gpio_num = atoi(pcValue[i]);
gpio_enable(gpio_num, GPIO_OUTPUT);
gpio_write(gpio_num, false);
} else if (strcmp(pcParam[i], "toggle") == 0) {
uint8_t gpio_num = atoi(pcValue[i]);
gpio_enable(gpio_num, GPIO_OUTPUT);
gpio_toggle(gpio_num);
}
}
return "/index.ssi";
}
const char *about_cgi_handler(int iIndex, int iNumParams, char *pcParam[], char *pcValue[])
{
return "/about.html";
}
const char *websocket_cgi_handler(int iIndex, int iNumParams, char *pcParam[], char *pcValue[])
{
return "/websockets.html";
}
void websocket_task(void *pvParameter)
{
struct tcp_pcb *pcb = (struct tcp_pcb *) pvParameter;
for (;;) {
if (pcb == NULL || pcb->state != ESTABLISHED) {
printf("Connection closed, deleting task\n");
break;
}
struct timeval tv;
gettimeofday(&tv, NULL);
int uptime = xTaskGetTickCount() * portTICK_PERIOD_MS / 1000;
int heap = (int) xPortGetFreeHeapSize();
int led = !gpio_read(LED_PIN);
/* Generate response in JSON format */
char response[64];
int len = snprintf(response, sizeof (response),
"{\"walltime\" : \"%d\","
"\"uptime\" : \"%d\","
" \"heap\" : \"%d\","
" \"led\" : \"%d\"}", (int)tv.tv_sec, uptime, heap, led);
if (len < sizeof (response))
websocket_write(pcb, (unsigned char *) response, len, WS_TEXT_MODE);
vTaskDelay(2000 / portTICK_PERIOD_MS);
}
vTaskDelete(NULL);
}
/**
* This function is called when websocket frame is received.
*
* Note: this function is executed on TCP thread and should return as soon
* as possible.
*/
void websocket_cb(struct tcp_pcb *pcb, uint8_t *data, u16_t data_len, uint8_t mode)
{
printf("[websocket_callback]:\n%.*s\n", (int) data_len, (char*) data);
uint8_t response[2];
uint16_t val;
switch (data[0]) {
case 'A': // ADC
/* This should be done on a separate thread in 'real' applications */
val = sdk_system_adc_read();
break;
case 'D': // Disable LED
gpio_write(LED_PIN, true);
val = 0xDEAD;
break;
case 'E': // Enable LED
gpio_write(LED_PIN, false);
val = 0xBEEF;
break;
default:
printf("Unknown command\n");
val = 0;
break;
}
response[1] = (uint8_t) val;
response[0] = val >> 8;
websocket_write(pcb, response, 2, WS_BIN_MODE);
}
/**
* This function is called when new websocket is open and
* creates a new websocket_task if requested URI equals '/stream'.
*/
void websocket_open_cb(struct tcp_pcb *pcb, const char *uri)
{
printf("WS URI: %s\n", uri);
if (!strcmp(uri, "/stream")) {
printf("request for streaming\n");
xTaskCreate(&websocket_task, "websocket_task", 256, (void *) pcb, 2, NULL);
}
}
void httpd_task(void *pvParameters)
{
tCGI pCGIs[] = {
{"/gpio", (tCGIHandler) gpio_cgi_handler},
{"/about", (tCGIHandler) about_cgi_handler},
{"/websockets", (tCGIHandler) websocket_cgi_handler},
};
const char *pcConfigSSITags[] = {
"walltime", // SSI_WALLTIME
"uptime", // SSI_UPTIME
"heap", // SSI_FREE_HEAP
"led" // SSI_LED_STATE
};
/* register handlers and start the server */
http_set_cgi_handlers(pCGIs, sizeof (pCGIs) / sizeof (pCGIs[0]));
http_set_ssi_handler((tSSIHandler) ssi_handler, pcConfigSSITags,
sizeof (pcConfigSSITags) / sizeof (pcConfigSSITags[0]));
websocket_register_callbacks((tWsOpenHandler) websocket_open_cb,
(tWsHandler) websocket_cb);
httpd_init();
for (;;);
}
void sntp_tsk(void *pvParameters)
{
const char *servers[] = {SNTP_SERVERS};
UNUSED_ARG(pvParameters);
/* Wait until we have joined AP and are assigned an IP */
while (sdk_wifi_station_get_connect_status() != STATION_GOT_IP) {
vTaskDelayMs(100);
}
/* Start SNTP */
printf("Starting SNTP... ");
/* SNTP will request an update each 5 minutes */
sntp_set_update_delay(5*60000);
/* Set GMT+1 zone, daylight savings off */
const struct timezone tz = {1*60, 1};
/* SNTP initialization */
sntp_initialize(&tz);
/* Servers must be configured right after initialization */
sntp_set_servers(servers, sizeof(servers) / sizeof(char*));
printf("DONE!\n");
/* Print date and time each 5 seconds */
while(1) {
vTaskDelayMs(5000);
//time_t ts = time(NULL);
//int t = ts;
//printf("TIME: %d %d %s", t,(int) day_seconds(), ctime(&ts));
}
}
void gpio_intr_handler(uint8_t gpio_num);
void manual_switch(void);
void buttonIntTask(void *pvParameters)
{
printf("Waiting for button press interrupt on gpio %d...\r\n", SWITCH_PIN);
QueueHandle_t *tsqueue = (QueueHandle_t *)pvParameters;
gpio_set_interrupt(SWITCH_PIN, int_type, gpio_intr_handler);
uint32_t last = 0;
while(1) {
uint32_t button_ts;
xQueueReceive(*tsqueue, &button_ts, portMAX_DELAY);
button_ts *= portTICK_PERIOD_MS;
if(last < button_ts-200) {
manual_switch();
//printf("Button interrupt fired at %dms\r\n", button_ts);
last = button_ts;
}
}
}
static QueueHandle_t tsqueue;
void gpio_intr_handler(uint8_t gpio_num)
{
uint32_t now = xTaskGetTickCountFromISR();
xQueueSendToBackFromISR(tsqueue, &now, NULL);
}
void register_app(void);
void user_init(void)
{
uart_set_baud(0, 115200);
printf("SDK version:%s\n", sdk_system_get_sdk_version());
struct sdk_station_config config = {
.ssid = WIFI_SSID,
.password = WIFI_PASS,
};
/* required to call wifi_set_opmode before station_set_config */
sdk_wifi_set_opmode(STATION_MODE);
sdk_wifi_station_set_config(&config);
//netif_set_hostname(netif_default, "nachtlicht");
sdk_wifi_station_connect();
/* turn off LED */
gpio_enable(LED_PIN, GPIO_OUTPUT);
gpio_write(LED_PIN, true);
gpio_enable(SWITCH_PIN, GPIO_INPUT);
tsqueue = xQueueCreate(2, sizeof(uint32_t));
//xTaskCreate(buttonIntTask, "buttonIntTask", 256, &tsqueue, 2, NULL);
/* initialize tasks */
xTaskCreate(&httpd_task, "HTTP Daemon", 2048, NULL, 2, NULL);
xTaskCreate(&sntp_tsk, "SNTP", 512, NULL, 1, NULL);
register_app();
}
void user_init(void)
{
uart_set_baud(0, 115200);