ESP32_SPI_WS2812_idf/SPI_ws2812.c at master · 8-DK/ESP32_SPI_WS2812_idf · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
/*
 * SPI_ws2812.c
 *
 *  Created on: 02-Nov-2020
 *      Author: Dhananjay Khairnar
 */

#include "SPI_ws2812.h"
#define LED_MAX_NBER_LEDS 24*2
#define LED_DMA_BUFFER_SIZE ((LED_MAX_NBER_LEDS * 16 * (24/4)))+1
#define LED_PIN GPIO_NUM_19
typedef struct {
	spi_host_device_t host;
	spi_device_handle_t spi;
	int dma_chan;
	spi_device_interface_config_t devcfg;
	spi_bus_config_t buscfg;
} SPI_settings_t;

uint16_t* ledDMAbuffer;

uint32_t table[LED_MAX_NBER_LEDS];

static SPI_settings_t SPI_settings = {
		.host = HSPI_HOST,
		.dma_chan = 2,
		.buscfg = {
				.miso_io_num = -1,
				.mosi_io_num = LED_PIN,
				.sclk_io_num = -1,
				.quadwp_io_num = -1,
				.quadhd_io_num = -1,
				.max_transfer_sz = LED_DMA_BUFFER_SIZE
		},
		.devcfg = { .clock_speed_hz = 3.2 * 1000 * 1000, //Clock out at 3.2 MHz
				.mode = 0, //SPI mode 0
				.spics_io_num = -1, //CS pin
				.queue_size = 1, //Not sure if needed
				.command_bits = 0,
				.address_bits = 0
		}
};


void initSPIws2812()
{
	esp_err_t err;

	err = spi_bus_initialize(SPI_settings.host, &SPI_settings.buscfg, SPI_settings.dma_chan);
	ESP_ERROR_CHECK(err);

	//Attach the Accel to the SPI bus
	err = spi_bus_add_device(SPI_settings.host, &SPI_settings.devcfg, &SPI_settings.spi);
	ESP_ERROR_CHECK(err);

	ledDMAbuffer = heap_caps_malloc(LED_DMA_BUFFER_SIZE, MALLOC_CAP_DMA); // Critical to be DMA memory.
}

void fillCol(uint32_t col)
{
	for(int i=0; i < LED_MAX_NBER_LEDS;i++)
	{
		table[i] = col;
		//table[i+1] = col;
	}
//	for(int i =0 ; i < 8 ;i++)
//		table[i] = 0x00ff00;
}

void fillBuffer(uint32_t* bufLed, int Count)
{
	for(int i =0 ; i < Count ;i++)
	{
		table[i] = bufLed[i];
		//table[i+1] = bufLed[i];
	}
}

void led_strip_update()
{
	uint16_t LedBitPattern[16] = {
	0x8888,
	0x8C88,
	0xC888,
	0xCC88,
	0x888C,
	0x8C8C,
	0xC88C,
	0xCC8C,
	0x88C8,
	0x8CC8,
	0xC8C8,
	0xCCC8,
	0x88CC,
	0x8CCC,
	0xC8CC,
	0xCCCC
	};
	uint32_t i;
	esp_err_t err;

	memset(ledDMAbuffer, 0, LED_DMA_BUFFER_SIZE);
	int n = 0;
	for (i = 0; i < LED_MAX_NBER_LEDS; i++) {
		uint32_t temp = table[i];// Data you want to write to each LEDs, I'm my case it's 95 RGB x 3 color

		//R
		ledDMAbuffer[n++] = LedBitPattern[0x0f & (temp >>12)];
		ledDMAbuffer[n++] = LedBitPattern[0x0f & (temp)>>8];

		//G
		ledDMAbuffer[n++] = LedBitPattern[0x0f & (temp >>4)];
		ledDMAbuffer[n++] = LedBitPattern[0x0f & (temp)];

		//B
		ledDMAbuffer[n++] = LedBitPattern[0x0f & (temp >>20)];
		ledDMAbuffer[n++] = LedBitPattern[0x0f & (temp)>>16];

	}

	spi_transaction_t t;
	memset(&t, 0, sizeof(t));
	t.length = LED_DMA_BUFFER_SIZE * 8; //length is in bits
	t.tx_buffer = ledDMAbuffer;

	err = spi_device_transmit(SPI_settings.spi, &t);
	ESP_ERROR_CHECK(err);
}