led_strip_rmt_dev.c File Reference

#include <stdlib.h>
#include <string.h>
#include <sys/cdefs.h>
#include "esp_log.h"
#include "esp_check.h"
#include "driver/rmt_tx.h"
#include "led_strip.h"
#include "led_strip_interface.h"
#include "led_strip_rmt_encoder.h"

Include dependency graph for led_strip_rmt_dev.c:

Go to the source code of this file.

Data Structures
struct	led_strip_rmt_obj

Macros
#define	LED_STRIP_RMT_DEFAULT_RESOLUTION   10000000
#define	LED_STRIP_RMT_DEFAULT_TRANS_QUEUE_SIZE   4
#define	LED_STRIP_RMT_DEFAULT_MEM_BLOCK_SYMBOLS   48

Functions
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)
static esp_err_t	led_strip_rmt_set_pixel_rgbw (led_strip_t *strip, uint32_t index, uint32_t red, uint32_t green, uint32_t blue, uint32_t white)
static esp_err_t	led_strip_rmt_refresh (led_strip_t *strip)
static esp_err_t	led_strip_rmt_clear (led_strip_t *strip)
static esp_err_t	led_strip_rmt_del (led_strip_t *strip)
esp_err_t	led_strip_new_rmt_device (const led_strip_config_t *led_config, const led_strip_rmt_config_t *rmt_config, led_strip_handle_t *ret_strip)
	Create LED strip based on RMT TX channel.

Variables
static const char *	TAG = "led_strip_rmt"

Macro Definition Documentation

LED_STRIP_RMT_DEFAULT_MEM_BLOCK_SYMBOLS

#define LED_STRIP_RMT_DEFAULT_MEM_BLOCK_SYMBOLS   48

Definition at line 22 of file led_strip_rmt_dev.c.

Referenced by led_strip_new_rmt_device(), and led_strip_new_rmt_device().

LED_STRIP_RMT_DEFAULT_RESOLUTION

#define LED_STRIP_RMT_DEFAULT_RESOLUTION   10000000

Definition at line 16 of file led_strip_rmt_dev.c.

Referenced by led_strip_new_rmt_device().

LED_STRIP_RMT_DEFAULT_TRANS_QUEUE_SIZE

#define LED_STRIP_RMT_DEFAULT_TRANS_QUEUE_SIZE   4

Definition at line 17 of file led_strip_rmt_dev.c.

Referenced by led_strip_new_rmt_device().

Function Documentation

led_strip_new_rmt_device()

esp_err_t led_strip_new_rmt_device	(	const led_strip_config_t *	led_config,
		const led_strip_rmt_config_t *	rmt_config,
		led_strip_handle_t *	ret_strip )

Create LED strip based on RMT TX channel.

Parameters

led_config	LED strip configuration
rmt_config	RMT specific configuration
ret_strip	Returned LED strip handle

Returns

• ESP_OK: create LED strip handle successfully
• ESP_ERR_INVALID_ARG: create LED strip handle failed because of invalid argument
• ESP_ERR_NO_MEM: create LED strip handle failed because of out of memory
• ESP_FAIL: create LED strip handle failed because some other error

Definition at line 97 of file led_strip_rmt_dev.c.

{
    led_strip_rmt_obj *rmt_strip = NULL;
    esp_err_t ret = ESP_OK;
    ESP_GOTO_ON_FALSE(led_config && rmt_config && ret_strip, ESP_ERR_INVALID_ARG, err, TAG, "invalid argument");
    ESP_GOTO_ON_FALSE(led_config->led_pixel_format < LED_PIXEL_FORMAT_INVALID, ESP_ERR_INVALID_ARG, err, 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_strip = calloc(1, sizeof(led_strip_rmt_obj) + led_config->max_leds * bytes_per_pixel);
    ESP_GOTO_ON_FALSE(rmt_strip, ESP_ERR_NO_MEM, err, TAG, "no mem for rmt strip");
    uint32_t resolution = rmt_config->resolution_hz ? rmt_config->resolution_hz : LED_STRIP_RMT_DEFAULT_RESOLUTION;
 
    // for backward compatibility, if the user does not set the clk_src, use the default value
    rmt_clock_source_t clk_src = RMT_CLK_SRC_DEFAULT;
    if (rmt_config->clk_src) {
        clk_src = rmt_config->clk_src;
    }
    size_t mem_block_symbols = LED_STRIP_RMT_DEFAULT_MEM_BLOCK_SYMBOLS;
    // override the default value if the user sets it
    if (rmt_config->mem_block_symbols) {
        mem_block_symbols = rmt_config->mem_block_symbols;
    }
    rmt_tx_channel_config_t rmt_chan_config = {
        .clk_src = clk_src,
        .gpio_num = led_config->strip_gpio_num,
        .mem_block_symbols = mem_block_symbols,
        .resolution_hz = resolution,
        .trans_queue_depth = LED_STRIP_RMT_DEFAULT_TRANS_QUEUE_SIZE,
        .flags.with_dma = rmt_config->flags.with_dma,
        .flags.invert_out = led_config->flags.invert_out,
    };
    ESP_GOTO_ON_ERROR(rmt_new_tx_channel(&rmt_chan_config, &rmt_strip->rmt_chan), err, TAG, "create RMT TX channel failed");
 
    led_strip_encoder_config_t strip_encoder_conf = {
        .resolution = resolution,
        .led_model = led_config->led_model
    };
    ESP_GOTO_ON_ERROR(rmt_new_led_strip_encoder(&strip_encoder_conf, &rmt_strip->strip_encoder), err, TAG, "create LED strip encoder failed");
 
 
    rmt_strip->bytes_per_pixel = bytes_per_pixel;
    rmt_strip->strip_len = led_config->max_leds;
    rmt_strip->base.set_pixel = led_strip_rmt_set_pixel;
    rmt_strip->base.set_pixel_rgbw = led_strip_rmt_set_pixel_rgbw;
    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;
    return ESP_OK;
err:
    if (rmt_strip) {
        if (rmt_strip->rmt_chan) {
            rmt_del_channel(rmt_strip->rmt_chan);
        }
        if (rmt_strip->strip_encoder) {
            rmt_del_encoder(rmt_strip->strip_encoder);
        }
        free(rmt_strip);
    }
    return ret;
}

led_strip_rmt_clear()

esp_err_t led_strip_rmt_clear ( led_strip_t * strip )

static

Definition at line 80 of file led_strip_rmt_dev.c.

{
    led_strip_rmt_obj *rmt_strip = __containerof(strip, led_strip_rmt_obj, base);
    // Write zero to turn off all leds
    memset(rmt_strip->pixel_buf, 0, rmt_strip->strip_len * rmt_strip->bytes_per_pixel);
    return led_strip_rmt_refresh(strip);
}

References led_strip_rmt_obj::bytes_per_pixel, led_strip_rmt_refresh(), led_strip_rmt_obj::pixel_buf, and led_strip_rmt_obj::strip_len.

Referenced by led_strip_new_rmt_device(), and led_strip_new_rmt_device().

Here is the call graph for this function:

Here is the caller graph for this function:

led_strip_rmt_del()

esp_err_t led_strip_rmt_del ( led_strip_t * strip )

static

Definition at line 88 of file led_strip_rmt_dev.c.

{
    led_strip_rmt_obj *rmt_strip = __containerof(strip, led_strip_rmt_obj, base);
    ESP_RETURN_ON_ERROR(rmt_del_channel(rmt_strip->rmt_chan), TAG, "delete RMT channel failed");
    ESP_RETURN_ON_ERROR(rmt_del_encoder(rmt_strip->strip_encoder), TAG, "delete strip encoder failed");
    free(rmt_strip);
    return ESP_OK;
}

References ESP_OK, ESP_RETURN_ON_ERROR, led_strip_rmt_obj::rmt_chan, led_strip_rmt_obj::strip_encoder, and TAG.

Referenced by led_strip_new_rmt_device(), and led_strip_new_rmt_device().

Here is the caller graph for this function:

led_strip_rmt_refresh()

esp_err_t led_strip_rmt_refresh ( led_strip_t * strip )

static

Definition at line 65 of file led_strip_rmt_dev.c.

{
    led_strip_rmt_obj *rmt_strip = __containerof(strip, led_strip_rmt_obj, base);
    rmt_transmit_config_t tx_conf = {
        .loop_count = 0,
    };
 
    ESP_RETURN_ON_ERROR(rmt_enable(rmt_strip->rmt_chan), TAG, "enable RMT channel failed");
    ESP_RETURN_ON_ERROR(rmt_transmit(rmt_strip->rmt_chan, rmt_strip->strip_encoder, rmt_strip->pixel_buf,
                                     rmt_strip->strip_len * rmt_strip->bytes_per_pixel, &tx_conf), TAG, "transmit pixels by RMT failed");
    ESP_RETURN_ON_ERROR(rmt_tx_wait_all_done(rmt_strip->rmt_chan, -1), TAG, "flush RMT channel failed");
    ESP_RETURN_ON_ERROR(rmt_disable(rmt_strip->rmt_chan), TAG, "disable RMT channel failed");
    return ESP_OK;
}

References led_strip_rmt_obj::bytes_per_pixel, ESP_OK, ESP_RETURN_ON_ERROR, led_strip_rmt_obj::pixel_buf, led_strip_rmt_obj::rmt_chan, led_strip_rmt_obj::strip_encoder, led_strip_rmt_obj::strip_len, and TAG.

Referenced by led_strip_new_rmt_device(), led_strip_new_rmt_device(), and led_strip_rmt_clear().

Here is the caller graph for this function:

led_strip_rmt_set_pixel()

esp_err_t led_strip_rmt_set_pixel ( led_strip_t * strip, uint32_t index, uint32_t red, uint32_t green, uint32_t blue )

static

Definition at line 36 of file led_strip_rmt_dev.c.

{
    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 maximum number of LEDs");
    uint32_t start = index * rmt_strip->bytes_per_pixel;
    // In thr order of GRB, as LED strip like WS2812 sends out pixels in this order
    rmt_strip->pixel_buf[start + 0] = green & 0xFF;
    rmt_strip->pixel_buf[start + 1] = red & 0xFF;
    rmt_strip->pixel_buf[start + 2] = blue & 0xFF;
    if (rmt_strip->bytes_per_pixel > 3) {
        rmt_strip->pixel_buf[start + 3] = 0;
    }
    return ESP_OK;
}

References led_strip_rmt_obj::bytes_per_pixel, ESP_OK, led_strip_rmt_obj::pixel_buf, and TAG.

Referenced by led_strip_new_rmt_device(), and led_strip_new_rmt_device().

Here is the caller graph for this function:

led_strip_rmt_set_pixel_rgbw()

esp_err_t led_strip_rmt_set_pixel_rgbw ( led_strip_t * strip, uint32_t index, uint32_t red, uint32_t green, uint32_t blue, uint32_t white )

static

Definition at line 51 of file led_strip_rmt_dev.c.

{
    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 maximum number of LEDs");
    ESP_RETURN_ON_FALSE(rmt_strip->bytes_per_pixel == 4, ESP_ERR_INVALID_ARG, TAG, "wrong LED pixel format, expected 4 bytes per pixel");
    uint8_t *buf_start = rmt_strip->pixel_buf + index * 4;
    // SK6812 component order is GRBW
    *buf_start = green & 0xFF;
    *++buf_start = red & 0xFF;
    *++buf_start = blue & 0xFF;
    *++buf_start = white & 0xFF;
    return ESP_OK;
}

References led_strip_rmt_obj::bytes_per_pixel, ESP_OK, led_strip_rmt_obj::pixel_buf, and TAG.

Referenced by led_strip_new_rmt_device().

Here is the caller graph for this function:

Variable Documentation

TAG

const char* TAG = "led_strip_rmt"

static

Definition at line 25 of file led_strip_rmt_dev.c.