forked from espressif/idf-extra-components
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
led_strip: Support for IDF4.4 + Add set pixel in HSV color.
- Loading branch information
Showing
10 changed files
with
278 additions
and
9 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,5 +1,5 @@ | ||
| version: "2.4.3" | ||
| version: "2.5.0" | ||
| description: Driver for Addressable LED Strip (WS2812, etc) | ||
| url: https://github.com/espressif/idf-extra-components/tree/master/led_strip | ||
| dependencies: | ||
| idf: ">=5.0" | ||
| idf: ">=4.4" |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,171 @@ | ||
| /* | ||
| * SPDX-FileCopyrightText: 2022-2023 Espressif Systems (Shanghai) CO LTD | ||
| * | ||
| * SPDX-License-Identifier: Apache-2.0 | ||
| */ | ||
| #include <stdlib.h> | ||
| #include <string.h> | ||
| #include <sys/cdefs.h> | ||
| #include "esp_log.h" | ||
| #include "esp_check.h" | ||
| #include "driver/rmt.h" | ||
| #include "led_strip.h" | ||
| #include "led_strip_interface.h" | ||
|
|
||
| static const char *TAG = "led_strip_rmt"; | ||
|
|
||
| #define WS2812_T0H_NS (350) | ||
| #define WS2812_T0L_NS (1000) | ||
| #define WS2812_T1H_NS (1000) | ||
| #define WS2812_T1L_NS (350) | ||
| #define WS2812_DELAY_MS (100) | ||
|
|
||
| // the memory size of each RMT channel, in words (4 bytes) | ||
| #if CONFIG_IDF_TARGET_ESP32 || CONFIG_IDF_TARGET_ESP32S2 | ||
| #define LED_STRIP_RMT_DEFAULT_MEM_BLOCK_SYMBOLS 8 | ||
| #else | ||
| #define LED_STRIP_RMT_DEFAULT_MEM_BLOCK_SYMBOLS 6 | ||
| #endif | ||
|
|
||
| static uint32_t ws2812_t0h_ticks = 0; | ||
| static uint32_t ws2812_t1h_ticks = 0; | ||
| static uint32_t ws2812_t0l_ticks = 0; | ||
| static uint32_t ws2812_t1l_ticks = 0; | ||
|
|
||
| typedef struct { | ||
| led_strip_t base; | ||
| rmt_channel_t rmt_channel; | ||
| uint32_t strip_len; | ||
| uint8_t bytes_per_pixel; | ||
| uint8_t buffer[0]; | ||
| } led_strip_rmt_obj; | ||
|
|
||
| static void IRAM_ATTR ws2812_rmt_adapter(const void *src, rmt_item32_t *dest, size_t src_size, | ||
| size_t wanted_num, size_t *translated_size, size_t *item_num) | ||
| { | ||
| if (src == NULL || dest == NULL) { | ||
| *translated_size = 0; | ||
| *item_num = 0; | ||
| return; | ||
| } | ||
| const rmt_item32_t bit0 = {{{ ws2812_t0h_ticks, 1, ws2812_t0l_ticks, 0 }}}; //Logical 0 | ||
| const rmt_item32_t bit1 = {{{ ws2812_t1h_ticks, 1, ws2812_t1l_ticks, 0 }}}; //Logical 1 | ||
| size_t size = 0; | ||
| size_t num = 0; | ||
| uint8_t *psrc = (uint8_t *)src; | ||
| rmt_item32_t *pdest = dest; | ||
| while (size < src_size && num < wanted_num) { | ||
| for (int i = 0; i < 8; i++) { | ||
| // MSB first | ||
| if (*psrc & (1 << (7 - i))) { | ||
| pdest->val = bit1.val; | ||
| } else { | ||
| pdest->val = bit0.val; | ||
| } | ||
| num++; | ||
| pdest++; | ||
| } | ||
| size++; | ||
| psrc++; | ||
| } | ||
| *translated_size = size; | ||
| *item_num = num; | ||
| } | ||
|
|
||
| static esp_err_t led_strip_rmt_set_pixel(led_strip_t *strip, uint32_t index, uint32_t red, uint32_t green, uint32_t blue) | ||
| { | ||
| led_strip_rmt_obj *rmt_strip = __containerof(strip, led_strip_rmt_obj, base); | ||
| ESP_RETURN_ON_FALSE(index < rmt_strip->strip_len, ESP_ERR_INVALID_ARG, TAG, "index out of the maximum number of leds"); | ||
| uint32_t start = index * rmt_strip->bytes_per_pixel; | ||
| // In thr order of GRB | ||
| rmt_strip->buffer[start + 0] = green & 0xFF; | ||
| rmt_strip->buffer[start + 1] = red & 0xFF; | ||
| rmt_strip->buffer[start + 2] = blue & 0xFF; | ||
| if (rmt_strip->bytes_per_pixel > 3) { | ||
| rmt_strip->buffer[start + 3] = 0; | ||
| } | ||
| return ESP_OK; | ||
| } | ||
|
|
||
| static esp_err_t led_strip_rmt_refresh(led_strip_t *strip) | ||
| { | ||
| led_strip_rmt_obj *rmt_strip = __containerof(strip, led_strip_rmt_obj, base); | ||
| ESP_RETURN_ON_ERROR(rmt_write_sample(rmt_strip->rmt_channel, rmt_strip->buffer, rmt_strip->strip_len * rmt_strip->bytes_per_pixel, true), TAG, | ||
| "transmit RMT samples failed"); | ||
| return rmt_wait_tx_done(rmt_strip->rmt_channel, pdMS_TO_TICKS(WS2812_DELAY_MS)); | ||
| } | ||
|
|
||
| static esp_err_t led_strip_rmt_clear(led_strip_t *strip) | ||
| { | ||
| led_strip_rmt_obj *rmt_strip = __containerof(strip, led_strip_rmt_obj, base); | ||
| // Write zero to turn off all leds | ||
| memset(rmt_strip->buffer, 0, rmt_strip->strip_len * rmt_strip->bytes_per_pixel); | ||
| return led_strip_rmt_refresh(strip); | ||
| } | ||
|
|
||
| static esp_err_t led_strip_rmt_del(led_strip_t *strip) | ||
| { | ||
| led_strip_rmt_obj *rmt_strip = __containerof(strip, led_strip_rmt_obj, base); | ||
| free(rmt_strip); | ||
| return ESP_OK; | ||
| } | ||
|
|
||
| esp_err_t led_strip_new_rmt_device(const led_strip_config_t *led_config, const led_strip_rmt_config_t *dev_config, led_strip_handle_t *ret_strip) | ||
| { | ||
| led_strip_rmt_obj *rmt_strip = NULL; | ||
| ESP_RETURN_ON_FALSE(led_config && dev_config && ret_strip, ESP_ERR_INVALID_ARG, TAG, "invalid argument"); | ||
| ESP_RETURN_ON_FALSE(led_config->led_pixel_format < LED_PIXEL_FORMAT_INVALID, ESP_ERR_INVALID_ARG, TAG, "invalid led_pixel_format"); | ||
|
|
||
| uint8_t bytes_per_pixel = 3; | ||
| if (led_config->led_pixel_format == LED_PIXEL_FORMAT_GRBW) { | ||
| bytes_per_pixel = 4; | ||
| } else if (led_config->led_pixel_format == LED_PIXEL_FORMAT_GRB) { | ||
| bytes_per_pixel = 3; | ||
| } else { | ||
| assert(false); | ||
| } | ||
|
|
||
| rmt_config_t config = RMT_DEFAULT_CONFIG_TX(led_config->strip_gpio_num, dev_config->rmt_channel); | ||
| // set counter clock to 40MHz | ||
| config.clk_div = 2; | ||
|
|
||
| size_t mem_block_symbols = LED_STRIP_RMT_DEFAULT_MEM_BLOCK_SYMBOLS; | ||
| // override the default value if the user sets it | ||
| if (dev_config->mem_block_symbols) { | ||
| mem_block_symbols = dev_config->mem_block_symbols; | ||
| } | ||
| config.mem_block_num = mem_block_symbols; | ||
|
|
||
| ESP_RETURN_ON_ERROR(rmt_config(&config), TAG, "RMT config failed"); | ||
| ESP_RETURN_ON_ERROR(rmt_driver_install(config.channel, 0, 0), TAG, "RMT install failed"); | ||
|
|
||
| rmt_strip = calloc(1, sizeof(led_strip_rmt_obj) + led_config->max_leds * bytes_per_pixel); | ||
| ESP_RETURN_ON_FALSE(rmt_strip, ESP_ERR_NO_MEM, TAG, "request memory for ws2812 failed"); | ||
|
|
||
| uint32_t counter_clk_hz = 0; | ||
| ESP_RETURN_ON_ERROR(rmt_get_counter_clock((rmt_channel_t)dev_config->rmt_channel, &counter_clk_hz), TAG, "get rmt counter clock failed"); | ||
| // ns -> ticks | ||
| float ratio = (float)counter_clk_hz / 1e9; | ||
| ws2812_t0h_ticks = (uint32_t)(ratio * WS2812_T0H_NS); | ||
| ws2812_t0l_ticks = (uint32_t)(ratio * WS2812_T0L_NS); | ||
| ws2812_t1h_ticks = (uint32_t)(ratio * WS2812_T1H_NS); | ||
| ws2812_t1l_ticks = (uint32_t)(ratio * WS2812_T1L_NS); | ||
|
|
||
| // set ws2812 to rmt adapter | ||
| rmt_translator_init((rmt_channel_t)dev_config->rmt_channel, ws2812_rmt_adapter); | ||
|
|
||
| rmt_strip->bytes_per_pixel = bytes_per_pixel; | ||
| rmt_strip->rmt_channel = (rmt_channel_t)dev_config->rmt_channel; | ||
| rmt_strip->strip_len = led_config->max_leds; | ||
| rmt_strip->base.set_pixel = led_strip_rmt_set_pixel; | ||
| rmt_strip->base.refresh = led_strip_rmt_refresh; | ||
| rmt_strip->base.clear = led_strip_rmt_clear; | ||
| rmt_strip->base.del = led_strip_rmt_del; | ||
|
|
||
| *ret_strip = &rmt_strip->base; | ||
|
|
||
| // Clear LED strip (turn off all LEDs) | ||
| ESP_RETURN_ON_ERROR((*ret_strip)->clear(*ret_strip), TAG, "LED strip clear failed"); | ||
|
|
||
| return ESP_OK; | ||
| } |