protofusion-esp32-template Changeset - 91735168ee75

Changeset - 91735168ee75

Parent rev.

Child rev.

[Not reviewed]

default

0 4 0

Ethan Zonca - 10 months ago 2024-08-16 12:32:51
ez@ethanzonca.com

Initial super rough WS2812 control with modified behavior over OSC

4 files changed with 175 insertions and 25 deletions:

main/ledstrip.c

162

main/ledstrip.h

main/main.c

main/osc_control.c

0 comments (0 inline, 0 general)

main/ledstrip.c

➞

Show inline comments

 #include <esp_event.h>
 #include <esp_log.h>
 #include <esp_wifi.h>
 #include <led_strip.h>
 #include <stdio.h>
 #include "freertos/FreeRTOS.h"
 #include "freertos/task.h"
 #include "led_strip.h"
 #include "esp_log.h"
 #include "esp_err.h"
 #include "ledstrip.h"
 #define LED_STRIP_BLINK_GPIO  11
 #define LED_STRIP_LED_NUMBERS 120
 #define LED_STRIP_RMT_RES_HZ  (10 * 1000 * 1000)
 static const char *TAG = "ledstrip";
 #define BLINK_GPIO 0
 led_strip_handle_t configure_led(void)
+{
     // LED strip general initialization, according to your led board design
     led_strip_config_t strip_config = {
         .strip_gpio_num = LED_STRIP_BLINK_GPIO,   // The GPIO that connected to the LED strip's data line
         .max_leds = LED_STRIP_LED_NUMBERS,        // The number of LEDs in the strip,
         .led_pixel_format = LED_PIXEL_FORMAT_GRB, // Pixel format of your LED strip
         .led_model = LED_MODEL_WS2812,            // LED strip model
         .flags.invert_out = false,                // whether to invert the output signal
     };
     // LED strip backend configuration: RMT
     led_strip_rmt_config_t rmt_config = {
 #if ESP_IDF_VERSION < ESP_IDF_VERSION_VAL(5, 0, 0)
         .rmt_channel = 0,
 #else
         .clk_src = RMT_CLK_SRC_DEFAULT,        // different clock source can lead to different power consumption
         .resolution_hz = LED_STRIP_RMT_RES_HZ, // RMT counter clock frequency
         .flags.with_dma = false,               // DMA feature is available on ESP target like ESP32-S3
 #endif
     };
     // LED Strip object handle
     led_strip_handle_t led_strip;
     ESP_ERROR_CHECK(led_strip_new_rmt_device(&strip_config, &rmt_config, &led_strip));
     ESP_LOGI(TAG, "Created LED strip object with RMT backend");
     return led_strip;
+}
 led_strip_handle_t led_strip;
 /* LED strip initialization with the GPIO and pixels number*/
 led_strip_config_t strip_config = {
     .strip_gpio_num = BLINK_GPIO, // The GPIO that connected to the LED strip's data line
     .max_leds = 1, // The number of LEDs in the strip,
     .led_pixel_format = LED_PIXEL_FORMAT_GRB, // Pixel format of your LED strip
     .led_model = LED_MODEL_WS2812, // LED strip model
     .flags.invert_out = false, // whether to invert the output signal (useful when your hardware has a level inverter)
 void ledstrip_init(void)
+{
     led_strip = configure_led();
     bool led_on_off = false;
+}
 static float modifier = 0.0;
 void ledstrip_set_modifier(float frac)
+{
     modifier = frac;
+}
 void ledstrip_set(uint32_t r, uint32_t g, uint32_t b)
+{
     ESP_LOGI(TAG, "Led set!");
     for (int i = 0; i < LED_STRIP_LED_NUMBERS; i++) {
         ESP_ERROR_CHECK(led_strip_set_pixel(led_strip, i, r, g, b));
+    }
     led_strip_refresh(led_strip);
+}
 uint8_t   effect = -1;
 uint8_t   effects = 120;
 uint16_t  effStep;
 unsigned long effStart;
 void __reset(void)
+{
     effStep = 0;
     effect = (effect + 1) % effects;
     effStart = xTaskGetTickCount() * portTICK_PERIOD_MS;
+}
 typedef struct Loop
+{
   uint8_t currentChild;
   uint8_t childs;
   bool timeBased;
   uint16_t cycles;
   uint16_t currentTime;
   //Loop(uint8_t totchilds, bool timebased, uint16_t tottime)
 //   {currentTime=0;currentChild=0;childs=totchilds;timeBased=timebased;cycles=tottime;}
 } loop_t;
 loop_t strip0loop0 =
+{
     .currentTime = 0,
     .currentChild = 0,
     .childs = 1,
     .timeBased = false,
     .cycles = 1,
 };
 led_strip_rmt_config_t rmt_config = {
     .clk_src = RMT_CLK_SRC_DEFAULT, // different clock source can lead to different power consumption
     .resolution_hz = 10 * 1000 * 1000, // 10MHz
     .flags.with_dma = false, // whether to enable the DMA feature
 //proto
 uint8_t strip0_loop0func();
 uint8_t strip0_loop0_eff0();
 void ledstrip_refresh(void)
+{
   if(strip0_loop0func() & 0x01)
     led_strip_refresh(led_strip);
+}
 uint8_t strip0_loop0func() {
   uint8_t ret = 0x00;
   switch(strip0loop0.currentChild) {
     case 0:
            ret = strip0_loop0_eff0();break;
+  }
   if(ret & 0x02) {
     ret &= 0xfd;
     if(strip0loop0.currentChild + 1 >= strip0loop0.childs) {
       strip0loop0.currentChild = 0;
       if(++strip0loop0.currentTime >= strip0loop0.cycles) {strip0loop0.currentTime = 0; ret |= 0x02;}
+    }
     else {
       strip0loop0.currentChild++;
+    }
 };
   return ret;
+}
 void ledstrip_init(void)
+{
     ESP_ERROR_CHECK(led_strip_new_rmt_device(&strip_config, &rmt_config, &led_strip));
+}
@@ \ No newline at end of file @@
 uint8_t strip0_loop0_eff0() {
     // Strip ID: 0 - Effect: Rainbow - LEDS: 120
     // Steps: 60 - Delay: 20
     // Colors: 3 (255.0.0, 0.255.0, 0.0.255)
     // Options: rainbowlen=60, toLeft=true,
   if((xTaskGetTickCount() * portTICK_PERIOD_MS) - effStart < 20 * (effStep)) return 0x00;
   float factor1, factor2;
   uint16_t ind;
   for(uint16_t j=0;j<120;j++) {
     ind = effStep + j * 1;
     switch((int)((ind % 60) / 20)) {
       case 0: factor1 = 1.0 - ((float)(ind % 60 - 0 * 20) / 20);
               factor2 = (float)((int)(ind - 0) % 60) / 20;
             //   strip_0.strip.setPixelColor(j, 255 * factor1 + 0 * factor2, 0 * factor1 + 255 * factor2, 0 * factor1 + 0 * factor2);
                 ESP_ERROR_CHECK(led_strip_set_pixel(led_strip, j, modifier * 255 * factor1 + 0 * factor2, modifier * 0 * factor1 + 255 * factor2, modifier * 0 * factor1 + 0 * factor2));
               break;
       case 1: factor1 = 1.0 - ((float)(ind % 60 - 1 * 20) / 20);
               factor2 = (float)((int)(ind - 20) % 60) / 20;
               ESP_ERROR_CHECK(led_strip_set_pixel(led_strip, j, modifier * 0 * factor1 + 0 * factor2, modifier * 255 * factor1 + 0 * factor2, modifier * 0 * factor1 + 255 * factor2));
               break;
       case 2: factor1 = 1.0 - ((float)(ind % 60 - 2 * 20) / 20);
               factor2 = (float)((int)(ind - 40) % 60) / 20;
               ESP_ERROR_CHECK(led_strip_set_pixel(led_strip, j, modifier * 0 * factor1 + 255 * factor2, modifier * 0 * factor1 + 0 * factor2, modifier * 255 * factor1 + 0 * factor2));
               break;
+    }
+  }
   if(effStep >= 60) {__reset(); return 0x03; }
   else effStep++;
   return 0x01;
+}

main/ledstrip.h

➞

Show inline comments

 #ifndef _LED_STRIPPER_H
 #define _LED_STRIPPER_H
 void ledstrip_init(void);
 void ledstrip_set(uint32_t r, uint32_t g, uint32_t b);
 void ledstrip_set_modifier(float frac);
 void ledstrip_refresh(void);
 #endif
@@ \ No newline at end of file @@

main/main.c

➞

Show inline comments

@@ @@ -24,41 +24,43 @@ @@
 static const char *TAG = "main";
 // Application entry point
 void app_main(void)
+{
     // Initialize usb-cdc interface
     // usb_cdc_init();
     // Initialize NVS
     esp_err_t ret = nvs_flash_init();
     if (ret == ESP_ERR_NVS_NO_FREE_PAGES || ret == ESP_ERR_NVS_NEW_VERSION_FOUND) {
         ESP_ERROR_CHECK(nvs_flash_erase());
         ret = nvs_flash_init();
+    }
     ESP_ERROR_CHECK(ret);
     // Initialize display
     display_init();
     display_gui_homescreen();
     // Connect to wifi
     wifi_init();
     ledstrip_init();
     // Initialize OSC
     osc_init();
     ledstrip_init();
     // Initialize canbus
     //can_init();
     while(1)
+    {
         display_gui_process();
         vTaskDelay(pdMS_TO_TICKS(1000));
         ledstrip_refresh();
         vTaskDelay(pdMS_TO_TICKS(10));
+    }
+}

main/osc_control.c

➞

Show inline comments

 #include "osc_control.h"
 #include <stdlib.h>
 #include <esp_event.h>
 #include <esp_log.h>
 #include <esp_wifi.h>
 #include "display.h"
 #include <esp_osc.h>
 #include "ledstrip.h"
 #define WIFI_SSID ""
 #define WIFI_PASS ""
 #define OSC_ADDRESS ""
 #define OSC_PORT 0
 static const char *TAG = "osc_control";
 esp_osc_client_t client;
 static void sender() {
   // select targets
   esp_osc_target_t targets[2] = {
       esp_osc_target("127.0.0.1", 9000),
       esp_osc_target(OSC_ADDRESS, OSC_PORT),
   };
   for (;;) {
     // delay
     vTaskDelay(1000 / portTICK_PERIOD_MS);
     // send messages
     for (size_t i = 0; i < 2; i++) {
       esp_osc_send(&client, &targets[i], "test", "ihfdsb", 42, (int64_t)84, 3.14f, 6.28, "foo", 3, "bar");
+    }
+  }
+}
 static bool callback(const char *topic, const char *format, esp_osc_value_t *values) {
   // log message
   ESP_LOGI(TAG, "got message: %s (%s)", topic, format);
   char out[512] = {0};
   snprintf(out, 128, "topic: %s\nformat: %s\n", topic, format);
   for (size_t i = 0; i < strlen(format); i++) {
     switch (format[i]) {
       case 'i':
         ESP_LOGI(TAG, "==> i: %ld", values[i].i);
         break;
       case 'h':
         ESP_LOGI(TAG, "==> h: %lld", values[i].h);
         break;
       case 'f':
         snprintf(out+strlen(out), 128-strlen(out), "Value: %f", values[i].f);
         ESP_LOGI(TAG, "==> f: %f", values[i].f);
         uint8_t val = values[i].f * 255;
         // ledstrip_set(val, val, val);
         ledstrip_set_modifier(values[i].f);
         break;
       case 'd':
         ESP_LOGI(TAG, "==> d: %f", values[i].d);
         break;
       case 's':
         ESP_LOGI(TAG, "==> s: %s", values[i].s);
         break;
       case 'b':
         ESP_LOGI(TAG, "==> b: %.*s (%d)", values[i].bl, values[i].b, values[i].bl);
         break;
+    }
+  }
   display_update_text(out);
   return true;
+}
 static void receiver() {
   for (;;) {
     // receive messages
     esp_osc_receive(&client, callback);
+  }
+}

0 comments (0 inline, 0 general)