342 lines
11 KiB
C
342 lines
11 KiB
C
// Raspberry Pi WS2812 LED driver using SMI
|
|
// For detailed description, see https://iosoft.blog
|
|
//
|
|
// Copyright (c) 2020 Jeremy P Bentham
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
//
|
|
// v0.01 JPB 16/7/20 Adapted from rpi_smi_adc_test v0.06
|
|
// v0.02 JPB 15/9/20 Addded RGB to GRB conversion
|
|
// v0.03 JPB 15/9/20 Added red-green flashing
|
|
// v0.04 JPB 16/9/20 Added test mode
|
|
// v0.05 JPB 19/9/20 Changed test mode colours
|
|
// v0.06 JPB 20/9/20 Outlined command-line data input
|
|
// v0.07 JPB 25/9/20 Command-line data input if not in test mode
|
|
// v0.08 JPB 26/9/20 Changed from 4 to 3 pulses per LED bit
|
|
// Added 4-bit zero preamble
|
|
// Added raw Tx data test
|
|
// v0.09 JPB 27/9/20 Added 16-channel option
|
|
// v0.10 JPB 28/9/20 Corrected Pi Zero caching problem
|
|
// v0.11 JPB 29/9/20 Added enable_dma before transfer (in case still active)
|
|
// Corrected DMA nsamp value (was byte count)
|
|
|
|
#include "rpi_pixleds.h"
|
|
|
|
#include <ctype.h>
|
|
#include <signal.h>
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <unistd.h>
|
|
|
|
#include "rpi_smi_defs.h"
|
|
|
|
// Structures for mapped I/O devices, and non-volatile memory
|
|
extern MEM_MAP gpio_regs, dma_regs;
|
|
MEM_MAP vc_mem, clk_regs, smi_regs;
|
|
|
|
// Pointers to SMI registers
|
|
volatile SMI_CS_REG *smi_cs;
|
|
volatile SMI_L_REG *smi_l;
|
|
volatile SMI_A_REG *smi_a;
|
|
volatile SMI_D_REG *smi_d;
|
|
volatile SMI_DMC_REG *smi_dmc;
|
|
volatile SMI_DSR_REG *smi_dsr;
|
|
volatile SMI_DSW_REG *smi_dsw;
|
|
volatile SMI_DCS_REG *smi_dcs;
|
|
volatile SMI_DCA_REG *smi_dca;
|
|
volatile SMI_DCD_REG *smi_dcd;
|
|
|
|
// RGB values for test mode (1 value for each of 16 channels)
|
|
// uint32_t on_rgbs[] = {0xef0000, 0x00ef00, 0x0000ef, 0xefef00, 0xef00ef, 0x00efef, 0xefefef};
|
|
// uint32_t off_rgbs = 0x000000;
|
|
|
|
// TXDATA_T *txdata; // Pointer to uncached Tx data buffer
|
|
// TXDATA_T tx_buffer[TX_BUFF_LEN(CHAN_MAXLEDS)]; // Tx buffer for assembling data
|
|
// int testmode, chan_ledcount = 1; // Command-line parameters
|
|
int rgb_data[CHAN_MAXLEDS][LED_NCHANS]; // RGB data
|
|
int chan_num;
|
|
|
|
// Current channel for data I/P
|
|
|
|
// int main(int argc, char *argv[]) {
|
|
// // setup
|
|
// int args = 0, n;
|
|
|
|
// while (argc > ++args) // Process command-line args
|
|
// {
|
|
// if (argv[args][0] == '-') {
|
|
// switch (toupper(argv[args][1])) {
|
|
// case 'N': // -N: number of LEDs per channel
|
|
// if (args >= argc - 1)
|
|
// fprintf(stderr, "Error: no numeric value\n");
|
|
// else
|
|
// chan_ledcount = atoi(argv[++args]);
|
|
// break;
|
|
// case 'T': // -T: test mode
|
|
// testmode = 1;
|
|
// break;
|
|
// default: // Otherwise error
|
|
// printf("Unrecognised option '%c'\n", argv[args][1]);
|
|
// printf("Options:\n"
|
|
// " -n num number of LEDs per channel\n"
|
|
// " -t Test mode (flash LEDs)\n");
|
|
// return (1);
|
|
// }
|
|
// } else if (chan_num < LED_NCHANS && hexdig(argv[args][0]) >= 0 &&
|
|
// (n = str_rgb(argv[args], rgb_data, chan_num)) > 0) {
|
|
// chan_ledcount = n > chan_ledcount ? n : chan_ledcount;
|
|
// chan_num++;
|
|
// }
|
|
// }
|
|
// signal(SIGINT, terminate_smi);
|
|
// map_devices();
|
|
// init_smi(LED_NCHANS > 8 ? SMI_16_BITS : SMI_8_BITS, SMI_TIMING);
|
|
// map_uncached_mem(&vc_mem, VC_MEM_SIZE);
|
|
|
|
// setup_smi_dma(&vc_mem, TX_BUFF_LEN(chan_ledcount));
|
|
// printf("%s %u LED%s per channel, %u channels\n", testmode ? "Testing" : "Setting",
|
|
// chan_ledcount,
|
|
// chan_ledcount == 1 ? "" : "s", LED_NCHANS);
|
|
|
|
// for (size_t colorIndex = 0; colorIndex < 3; ++colorIndex) {
|
|
// for (size_t i = 0; i < chan_ledcount; ++i) {
|
|
// set_color(on_rgbs[colorIndex], &tx_buffer[LED_TX_OSET(i)]);
|
|
// }
|
|
|
|
// #if LED_NCHANS <= 8
|
|
// swap_bytes(tx_buffer, TX_BUFF_SIZE(chan_ledcount));
|
|
// #endif
|
|
|
|
// memcpy(txdata, tx_buffer, TX_BUFF_SIZE(chan_ledcount));
|
|
// start_smi(&vc_mem);
|
|
// usleep(CHASE_MSEC * 1000);
|
|
// sleep(1);
|
|
// }
|
|
|
|
// artnet_init();
|
|
|
|
// // loops
|
|
// if (testmode) {
|
|
// while (1) {
|
|
// test_leds();
|
|
// sleep(3);
|
|
// }
|
|
// } else {
|
|
// while (1) {
|
|
// artDmx_t *artDmx;
|
|
// if (artnet_read(&artDmx) == OpDmx) {
|
|
// uint16_t dmxLength = (artDmx->lengthHi << 8) | artDmx->lengthLo;
|
|
// unsigned int ledCountInFrame = dmxLength / 3;
|
|
// uint16_t maxBound = ledCountInFrame < chan_ledcount ? ledCountInFrame :
|
|
// chan_ledcount; unsigned int universe = artDmx->subUni & (LED_NCHANS - 1); for (size_t
|
|
// i = 0; i < maxBound; ++i) {
|
|
// uint8_t *rgb = artDmx->data + (i * 3);
|
|
// rgb_data[i][universe] = (rgb[0] << 16) | (rgb[1] << 8) | rgb[2];
|
|
// rgb_txdata(rgb_data[i], &tx_buffer[LED_TX_OSET(i)]);
|
|
// }
|
|
|
|
// #if LED_NCHANS <= 8
|
|
// swap_bytes(tx_buffer, TX_BUFF_SIZE(chan_ledcount));
|
|
// #endif
|
|
|
|
// memcpy(txdata, tx_buffer, TX_BUFF_SIZE(chan_ledcount));
|
|
// // enable_dma(DMA_CHAN);
|
|
// start_smi(&vc_mem);
|
|
// // usleep(10);
|
|
// // while (dma_active(DMA_CHAN))
|
|
// // usleep(10);
|
|
// }
|
|
// }
|
|
// }
|
|
// terminate_smi(0);
|
|
// }
|
|
|
|
// Convert RGB text string into integer data, for given channel
|
|
// Return number of data points for this channel
|
|
int str_rgb(char *s, int rgbs[][LED_NCHANS], int chan) {
|
|
int i = 0;
|
|
char *p;
|
|
|
|
while (chan < LED_NCHANS && i < CHAN_MAXLEDS && hexdig(*s) >= 0) {
|
|
rgbs[i++][chan] = strtoul(s, &p, 16);
|
|
s = *p ? p + 1 : p;
|
|
}
|
|
return (i);
|
|
}
|
|
|
|
// Set Tx data for 8 or 16 chans, 1 LED per chan, given 1 RGB val per chan
|
|
// Logic 1 is 0.8us high, 0.4 us low, logic 0 is 0.4us high, 0.8us low
|
|
void set_color(uint32_t rgb, TXDATA_T *txd) {
|
|
int msk;
|
|
|
|
// For each bit of the 24-bit RGB values..
|
|
for (size_t n = 0; n < LED_NBITS; n++) {
|
|
// Mask to convert RGB to GRB, M.S bit first
|
|
msk = n == 0 ? 0x800000 : n == 8 ? 0x8000 : n == 16 ? 0x80 : msk >> 1;
|
|
// 1st byte or word is a high pulse on all lines
|
|
txd[0] = (TXDATA_T)0xffff;
|
|
// 2nd has high or low bits from data
|
|
// 3rd is a low pulse
|
|
txd[1] = txd[2] = 0;
|
|
if (rgb & msk) {
|
|
txd[1] = (TXDATA_T)0xffff;
|
|
}
|
|
txd += BIT_NPULSES;
|
|
}
|
|
}
|
|
|
|
// Set Tx data for 8 or 16 chans, 1 LED per chan, given 1 RGB val per chan
|
|
// Logic 1 is 0.8us high, 0.4 us low, logic 0 is 0.4us high, 0.8us low
|
|
void rgb_txdata(int *rgbs, TXDATA_T *txd) {
|
|
int i, n, msk;
|
|
|
|
// For each bit of the 24-bit RGB values..
|
|
for (n = 0; n < LED_NBITS; n++) {
|
|
// Mask to convert RGB to GRB, M.S bit first
|
|
msk = n == 0 ? 0x800000 : n == 8 ? 0x8000 : n == 16 ? 0x80 : msk >> 1;
|
|
// 1st byte or word is a high pulse on all lines
|
|
txd[0] = (TXDATA_T)0xffff;
|
|
// 2nd has high or low bits from data
|
|
// 3rd is a low pulse
|
|
txd[1] = txd[2] = 0;
|
|
for (i = 0; i < LED_NCHANS; i++) {
|
|
if (rgbs[i] & msk)
|
|
txd[1] |= (1 << i);
|
|
}
|
|
txd += BIT_NPULSES;
|
|
}
|
|
}
|
|
|
|
// Swap adjacent bytes in transmit data
|
|
void swap_bytes(void *data, int len) {
|
|
uint16_t *wp = (uint16_t *)data;
|
|
|
|
len = (len + 1) / 2;
|
|
while (len-- > 0) {
|
|
*wp = __builtin_bswap16(*wp);
|
|
wp++;
|
|
}
|
|
}
|
|
|
|
// Return hex digit value, -ve if not hex
|
|
int hexdig(char c) {
|
|
c = toupper(c);
|
|
return ((c >= '0' && c <= '9') ? c - '0' : (c >= 'A' && c <= 'F') ? c - 'A' + 10 : -1);
|
|
}
|
|
|
|
// Map GPIO, DMA and SMI registers into virtual mem (user space)
|
|
// If any of these fail, program will be terminate_smid
|
|
void map_devices(void) {
|
|
map_periph(&gpio_regs, (void *)GPIO_BASE, PAGE_SIZE);
|
|
map_periph(&dma_regs, (void *)DMA_BASE, PAGE_SIZE);
|
|
map_periph(&clk_regs, (void *)CLK_BASE, PAGE_SIZE);
|
|
map_periph(&smi_regs, (void *)SMI_BASE, PAGE_SIZE);
|
|
}
|
|
|
|
// Catastrophic failure in initial setup
|
|
void fail(char *s) {
|
|
printf(s);
|
|
terminate_smi(0);
|
|
}
|
|
|
|
// Free memory segments and exit
|
|
void terminate_smi(int sig) {
|
|
int i;
|
|
printf("Closing\n");
|
|
if (gpio_regs.virt) {
|
|
for (i = 0; i < LED_NCHANS; i++)
|
|
gpio_mode(LED_D0_PIN + i, GPIO_IN);
|
|
}
|
|
if (smi_regs.virt)
|
|
*REG32(smi_regs, SMI_CS) = 0;
|
|
stop_dma(DMA_CHAN);
|
|
unmap_periph_mem(&vc_mem);
|
|
unmap_periph_mem(&smi_regs);
|
|
unmap_periph_mem(&dma_regs);
|
|
unmap_periph_mem(&gpio_regs);
|
|
exit(0);
|
|
}
|
|
|
|
// Initialise SMI, given data width, time step, and setup/hold/strobe counts
|
|
// Step value is in nanoseconds: even numbers, 2 to 30
|
|
void init_smi(int ledChan, int ns, int setup, int strobe, int hold) {
|
|
int width = ledChan > 8 ? SMI_16_BITS : SMI_8_BITS;
|
|
int i, divi = ns / 2;
|
|
|
|
smi_cs = (SMI_CS_REG *)REG32(smi_regs, SMI_CS);
|
|
smi_l = (SMI_L_REG *)REG32(smi_regs, SMI_L);
|
|
smi_a = (SMI_A_REG *)REG32(smi_regs, SMI_A);
|
|
smi_d = (SMI_D_REG *)REG32(smi_regs, SMI_D);
|
|
smi_dmc = (SMI_DMC_REG *)REG32(smi_regs, SMI_DMC);
|
|
smi_dsr = (SMI_DSR_REG *)REG32(smi_regs, SMI_DSR0);
|
|
smi_dsw = (SMI_DSW_REG *)REG32(smi_regs, SMI_DSW0);
|
|
smi_dcs = (SMI_DCS_REG *)REG32(smi_regs, SMI_DCS);
|
|
smi_dca = (SMI_DCA_REG *)REG32(smi_regs, SMI_DCA);
|
|
smi_dcd = (SMI_DCD_REG *)REG32(smi_regs, SMI_DCD);
|
|
smi_cs->value = smi_l->value = smi_a->value = 0;
|
|
smi_dsr->value = smi_dsw->value = smi_dcs->value = smi_dca->value = 0;
|
|
if (*REG32(clk_regs, CLK_SMI_DIV) != divi << 12) {
|
|
*REG32(clk_regs, CLK_SMI_CTL) = CLK_PASSWD | (1 << 5);
|
|
usleep(10);
|
|
while (*REG32(clk_regs, CLK_SMI_CTL) & (1 << 7))
|
|
;
|
|
usleep(10);
|
|
*REG32(clk_regs, CLK_SMI_DIV) = CLK_PASSWD | (divi << 12);
|
|
usleep(10);
|
|
*REG32(clk_regs, CLK_SMI_CTL) = CLK_PASSWD | 6 | (1 << 4);
|
|
usleep(10);
|
|
while ((*REG32(clk_regs, CLK_SMI_CTL) & (1 << 7)) == 0)
|
|
;
|
|
usleep(100);
|
|
}
|
|
if (smi_cs->seterr)
|
|
smi_cs->seterr = 1;
|
|
smi_dsr->rsetup = smi_dsw->wsetup = setup;
|
|
smi_dsr->rstrobe = smi_dsw->wstrobe = strobe;
|
|
smi_dsr->rhold = smi_dsw->whold = hold;
|
|
smi_dmc->panicr = smi_dmc->panicw = 8;
|
|
smi_dmc->reqr = smi_dmc->reqw = REQUEST_THRESH;
|
|
smi_dsr->rwidth = smi_dsw->wwidth = width;
|
|
for (i = 0; i < LED_NCHANS; i++)
|
|
gpio_mode(LED_D0_PIN + i, GPIO_ALT1);
|
|
}
|
|
|
|
// Set up SMI transfers using DMA
|
|
void setup_smi_dma(MEM_MAP *mp, int nsamp, TXDATA_T **txdata) {
|
|
DMA_CB *cbs = mp->virt;
|
|
|
|
*txdata = (TXDATA_T *)(cbs + 1);
|
|
smi_dmc->dmaen = 1;
|
|
smi_cs->enable = 1;
|
|
smi_cs->clear = 1;
|
|
smi_cs->pxldat = 1;
|
|
smi_l->len = nsamp * sizeof(TXDATA_T);
|
|
smi_cs->write = 1;
|
|
enable_dma(DMA_CHAN);
|
|
cbs[0].ti = DMA_DEST_DREQ | (DMA_SMI_DREQ << 16) | DMA_CB_SRCE_INC | DMA_WAIT_RESP;
|
|
cbs[0].tfr_len = nsamp;
|
|
cbs[0].srce_ad = MEM_BUS_ADDR(mp, *txdata);
|
|
cbs[0].dest_ad = REG_BUS_ADDR(smi_regs, SMI_D);
|
|
}
|
|
|
|
// Start SMI DMA transfers
|
|
void start_smi(MEM_MAP *mp) {
|
|
DMA_CB *cbs = mp->virt;
|
|
|
|
start_dma(mp, DMA_CHAN, &cbs[0], 0);
|
|
smi_cs->start = 1;
|
|
}
|
|
|
|
// EOF
|