diff --git a/firmware/lux.cpp b/firmware/lux.cpp index 8b775c4..eb7b9ad 100644 --- a/firmware/lux.cpp +++ b/firmware/lux.cpp @@ -9,6 +9,7 @@ #include #include +#include const int signal_led_pin = 2; @@ -16,17 +17,89 @@ const int cs0 = 15; const int gpio4 = 4; const int gpio5 = 5; + +ws2812_pixel_t **pixels_ptr; + +static ws2812_pixel_t next_colour(int i) { + ws2812_pixel_t colour = {{0, 0, 0, 0}}; + if(i == 8) { + colour.white = 32; + } else { + colour.red = i & 1 ? 32 : 0; + colour.green = i & 2 ? 32 : 0; + colour.blue = i & 4 ? 32 : 0; + } + + return colour; +} + +void spi_dac(int id, int val) { + int dac_val = (val << 2) & 0x3FFC; + + spi_transfer_8(1, ~(0x00)); + gpio_write(cs0, 1); + gpio_write(cs0, 0); + spi_transfer_8(1, ~(0x01 << id)); + gpio_write(cs0, 1); + gpio_write(cs0, 0); + + spi_transfer_16(1, dac_val); + + spi_transfer_8(1, ~(0x00)); + gpio_write(cs0, 1); + gpio_write(cs0, 0); + spi_transfer_8(1, ~(0x01 << id)); + gpio_write(cs0, 1); + gpio_write(cs0, 0); +} + extern "C" void signal_led(bool state) { gpio_write(signal_led_pin, !state); } + +/* This task uses the high level GPIO API (esp_gpio.h) to blink an LED. + * + */ extern "C" void lux_task(void *pvParameters) { + gpio_enable(9, GPIO_INPUT); + gpio_enable(10, GPIO_INPUT); + + ws2812_pixel_t pixels[led_number]; + ws2812_i2s_init(led_number, PIXEL_RGBW); + memset(pixels, 0, sizeof(ws2812_pixel_t) * led_number); + gpio_enable(signal_led_pin, GPIO_OUTPUT); gpio_enable(cs0, GPIO_OUTPUT); gpio_enable(gpio4, GPIO_OUTPUT); gpio_enable(gpio5, GPIO_OUTPUT); spi_init(1, SPI_MODE0, SPI_FREQ_DIV_1M, 1, SPI_BIG_ENDIAN, 1); + while (1) { + gpio_write(gpio4, 1); + vTaskDelay(200 / portTICK_PERIOD_MS); + gpio_write(gpio4, 0); + /*for (int j = 0; j < 64; j++) { + for (int i = 0; i < 8; i++) + spi_dac(i, 64 * j); + //printf("> %d\n", 64*j); + vTaskDelay(100 / portTICK_PERIOD_MS); + }*/ + for (int i = 0; i < 8; i++) + spi_dac(i, 0); - vTaskDelete(nullptr); + gpio_write(gpio5, 1); + vTaskDelay(200 / portTICK_PERIOD_MS); + gpio_write(gpio5, 0); + + for (int c = 8; c >= 0; c--) { + + for (int i = 0; i < led_number; i++) { + pixels[i] = next_colour(c); + } + ws2812_i2s_update(pixels, PIXEL_RGBW); + vTaskDelay(200 / portTICK_PERIOD_MS); + } + + } } \ No newline at end of file diff --git a/firmware/lux.h b/firmware/lux.h index caf4a88..d29f45e 100644 --- a/firmware/lux.h +++ b/firmware/lux.h @@ -5,6 +5,10 @@ #ifndef FIRMWARE_LUX_H #define FIRMWARE_LUX_H +#ifdef __cplusplus +constexpr int led_number = 8; +#endif + #ifdef __cplusplus extern "C" { #endif