Initial Release
This commit is contained in:
35
pio/ws2812/CMakeLists.txt
Normal file
35
pio/ws2812/CMakeLists.txt
Normal file
@@ -0,0 +1,35 @@
|
||||
add_executable(pio_ws2812)
|
||||
|
||||
# generate the header file into the source tree as it is included in the RP2040 datasheet
|
||||
pico_generate_pio_header(pio_ws2812 ${CMAKE_CURRENT_LIST_DIR}/ws2812.pio OUTPUT_DIR ${CMAKE_CURRENT_LIST_DIR}/generated)
|
||||
|
||||
target_sources(pio_ws2812 PRIVATE ws2812.c)
|
||||
|
||||
target_link_libraries(pio_ws2812 PRIVATE pico_stdlib hardware_pio)
|
||||
pico_add_extra_outputs(pio_ws2812)
|
||||
|
||||
# add url via pico_set_program_url
|
||||
example_auto_set_url(pio_ws2812)
|
||||
|
||||
add_executable(pio_ws2812_parallel)
|
||||
|
||||
pico_generate_pio_header(pio_ws2812_parallel ${CMAKE_CURRENT_LIST_DIR}/ws2812.pio OUTPUT_DIR ${CMAKE_CURRENT_LIST_DIR}/generated)
|
||||
|
||||
target_sources(pio_ws2812_parallel PRIVATE ws2812_parallel.c)
|
||||
|
||||
target_compile_definitions(pio_ws2812_parallel PRIVATE
|
||||
PIN_DBG1=3)
|
||||
|
||||
target_link_libraries(pio_ws2812_parallel PRIVATE pico_stdlib hardware_pio hardware_dma)
|
||||
pico_add_extra_outputs(pio_ws2812_parallel)
|
||||
|
||||
# add url via pico_set_program_url
|
||||
example_auto_set_url(pio_ws2812_parallel)
|
||||
|
||||
# Additionally generate python and hex pioasm outputs for inclusion in the RP2040 datasheet
|
||||
add_custom_target(pio_ws2812_datasheet DEPENDS Pioasm ${CMAKE_CURRENT_LIST_DIR}/generated/ws2812.py)
|
||||
add_custom_command(OUTPUT ${CMAKE_CURRENT_LIST_DIR}/generated/ws2812.py
|
||||
DEPENDS ${CMAKE_CURRENT_LIST_DIR}/ws2812.pio
|
||||
COMMAND Pioasm -o python ${CMAKE_CURRENT_LIST_DIR}/ws2812.pio ${CMAKE_CURRENT_LIST_DIR}/generated/ws2812.py
|
||||
)
|
||||
add_dependencies(pio_ws2812 pio_ws2812_datasheet)
|
||||
112
pio/ws2812/generated/ws2812.pio.h
Normal file
112
pio/ws2812/generated/ws2812.pio.h
Normal file
@@ -0,0 +1,112 @@
|
||||
// -------------------------------------------------- //
|
||||
// This file is autogenerated by pioasm; do not edit! //
|
||||
// -------------------------------------------------- //
|
||||
|
||||
#if !PICO_NO_HARDWARE
|
||||
#include "hardware/pio.h"
|
||||
#endif
|
||||
|
||||
// ------ //
|
||||
// ws2812 //
|
||||
// ------ //
|
||||
|
||||
#define ws2812_wrap_target 0
|
||||
#define ws2812_wrap 3
|
||||
|
||||
#define ws2812_T1 2
|
||||
#define ws2812_T2 5
|
||||
#define ws2812_T3 3
|
||||
|
||||
static const uint16_t ws2812_program_instructions[] = {
|
||||
// .wrap_target
|
||||
0x6221, // 0: out x, 1 side 0 [2]
|
||||
0x1123, // 1: jmp !x, 3 side 1 [1]
|
||||
0x1400, // 2: jmp 0 side 1 [4]
|
||||
0xa442, // 3: nop side 0 [4]
|
||||
// .wrap
|
||||
};
|
||||
|
||||
#if !PICO_NO_HARDWARE
|
||||
static const struct pio_program ws2812_program = {
|
||||
.instructions = ws2812_program_instructions,
|
||||
.length = 4,
|
||||
.origin = -1,
|
||||
};
|
||||
|
||||
static inline pio_sm_config ws2812_program_get_default_config(uint offset) {
|
||||
pio_sm_config c = pio_get_default_sm_config();
|
||||
sm_config_set_wrap(&c, offset + ws2812_wrap_target, offset + ws2812_wrap);
|
||||
sm_config_set_sideset(&c, 1, false, false);
|
||||
return c;
|
||||
}
|
||||
|
||||
#include "hardware/clocks.h"
|
||||
static inline void ws2812_program_init(PIO pio, uint sm, uint offset, uint pin, float freq, bool rgbw) {
|
||||
pio_gpio_init(pio, pin);
|
||||
pio_sm_set_consecutive_pindirs(pio, sm, pin, 1, true);
|
||||
pio_sm_config c = ws2812_program_get_default_config(offset);
|
||||
sm_config_set_sideset_pins(&c, pin);
|
||||
sm_config_set_out_shift(&c, false, true, rgbw ? 32 : 24);
|
||||
sm_config_set_fifo_join(&c, PIO_FIFO_JOIN_TX);
|
||||
int cycles_per_bit = ws2812_T1 + ws2812_T2 + ws2812_T3;
|
||||
float div = clock_get_hz(clk_sys) / (freq * cycles_per_bit);
|
||||
sm_config_set_clkdiv(&c, div);
|
||||
pio_sm_init(pio, sm, offset, &c);
|
||||
pio_sm_set_enabled(pio, sm, true);
|
||||
}
|
||||
|
||||
#endif
|
||||
|
||||
// --------------- //
|
||||
// ws2812_parallel //
|
||||
// --------------- //
|
||||
|
||||
#define ws2812_parallel_wrap_target 0
|
||||
#define ws2812_parallel_wrap 3
|
||||
|
||||
#define ws2812_parallel_T1 2
|
||||
#define ws2812_parallel_T2 5
|
||||
#define ws2812_parallel_T3 3
|
||||
|
||||
static const uint16_t ws2812_parallel_program_instructions[] = {
|
||||
// .wrap_target
|
||||
0x6020, // 0: out x, 32
|
||||
0xa10b, // 1: mov pins, !null [1]
|
||||
0xa401, // 2: mov pins, x [4]
|
||||
0xa103, // 3: mov pins, null [1]
|
||||
// .wrap
|
||||
};
|
||||
|
||||
#if !PICO_NO_HARDWARE
|
||||
static const struct pio_program ws2812_parallel_program = {
|
||||
.instructions = ws2812_parallel_program_instructions,
|
||||
.length = 4,
|
||||
.origin = -1,
|
||||
};
|
||||
|
||||
static inline pio_sm_config ws2812_parallel_program_get_default_config(uint offset) {
|
||||
pio_sm_config c = pio_get_default_sm_config();
|
||||
sm_config_set_wrap(&c, offset + ws2812_parallel_wrap_target, offset + ws2812_parallel_wrap);
|
||||
return c;
|
||||
}
|
||||
|
||||
#include "hardware/clocks.h"
|
||||
static inline void ws2812_parallel_program_init(PIO pio, uint sm, uint offset, uint pin_base, uint pin_count, float freq) {
|
||||
for(uint i=pin_base; i<pin_base+pin_count; i++) {
|
||||
pio_gpio_init(pio, i);
|
||||
}
|
||||
pio_sm_set_consecutive_pindirs(pio, sm, pin_base, pin_count, true);
|
||||
pio_sm_config c = ws2812_parallel_program_get_default_config(offset);
|
||||
sm_config_set_out_shift(&c, true, true, 32);
|
||||
sm_config_set_out_pins(&c, pin_base, pin_count);
|
||||
sm_config_set_set_pins(&c, pin_base, pin_count);
|
||||
sm_config_set_fifo_join(&c, PIO_FIFO_JOIN_TX);
|
||||
int cycles_per_bit = ws2812_parallel_T1 + ws2812_parallel_T2 + ws2812_parallel_T3;
|
||||
float div = clock_get_hz(clk_sys) / (freq * cycles_per_bit);
|
||||
sm_config_set_clkdiv(&c, div);
|
||||
pio_sm_init(pio, sm, offset, &c);
|
||||
pio_sm_set_enabled(pio, sm, true);
|
||||
}
|
||||
|
||||
#endif
|
||||
|
||||
46
pio/ws2812/generated/ws2812.py
Normal file
46
pio/ws2812/generated/ws2812.py
Normal file
@@ -0,0 +1,46 @@
|
||||
# -------------------------------------------------- #
|
||||
# This file is autogenerated by pioasm; do not edit! #
|
||||
# -------------------------------------------------- #
|
||||
|
||||
import rp2
|
||||
from machine import Pin
|
||||
# ------ #
|
||||
# ws2812 #
|
||||
# ------ #
|
||||
|
||||
ws2812_T1 = 2
|
||||
ws2812_T2 = 5
|
||||
ws2812_T3 = 3
|
||||
|
||||
@rp2.asm_pio(sideset_init=pico.PIO.OUT_HIGH, out_init=pico.PIO.OUT_HIGH, out_shiftdir=1)
|
||||
def ws2812():
|
||||
wrap_target()
|
||||
label("0")
|
||||
out(x, 1) .side(0) [2] # 0
|
||||
jmp(not_x, "3") .side(1) [1] # 1
|
||||
jmp("0") .side(1) [4] # 2
|
||||
label("3")
|
||||
nop() .side(0) [4] # 3
|
||||
wrap()
|
||||
|
||||
|
||||
|
||||
# --------------- #
|
||||
# ws2812_parallel #
|
||||
# --------------- #
|
||||
|
||||
ws2812_parallel_T1 = 2
|
||||
ws2812_parallel_T2 = 5
|
||||
ws2812_parallel_T3 = 3
|
||||
|
||||
@rp2.asm_pio()
|
||||
def ws2812_parallel():
|
||||
wrap_target()
|
||||
out(x, 32) # 0
|
||||
mov(pins, not null) [1] # 1
|
||||
mov(pins, x) [4] # 2
|
||||
mov(pins, null) [1] # 3
|
||||
wrap()
|
||||
|
||||
|
||||
|
||||
100
pio/ws2812/ws2812.c
Normal file
100
pio/ws2812/ws2812.c
Normal file
@@ -0,0 +1,100 @@
|
||||
/**
|
||||
* Copyright (c) 2020 Raspberry Pi (Trading) Ltd.
|
||||
*
|
||||
* SPDX-License-Identifier: BSD-3-Clause
|
||||
*/
|
||||
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
|
||||
#include "pico/stdlib.h"
|
||||
#include "hardware/pio.h"
|
||||
#include "hardware/clocks.h"
|
||||
#include "ws2812.pio.h"
|
||||
|
||||
static inline void put_pixel(uint32_t pixel_grb) {
|
||||
pio_sm_put_blocking(pio0, 0, pixel_grb << 8u);
|
||||
}
|
||||
|
||||
static inline uint32_t urgb_u32(uint8_t r, uint8_t g, uint8_t b) {
|
||||
return
|
||||
((uint32_t) (r) << 8) |
|
||||
((uint32_t) (g) << 16) |
|
||||
(uint32_t) (b);
|
||||
}
|
||||
|
||||
void pattern_snakes(uint len, uint t) {
|
||||
for (uint i = 0; i < len; ++i) {
|
||||
uint x = (i + (t >> 1)) % 64;
|
||||
if (x < 10)
|
||||
put_pixel(urgb_u32(0xff, 0, 0));
|
||||
else if (x >= 15 && x < 25)
|
||||
put_pixel(urgb_u32(0, 0xff, 0));
|
||||
else if (x >= 30 && x < 40)
|
||||
put_pixel(urgb_u32(0, 0, 0xff));
|
||||
else
|
||||
put_pixel(0);
|
||||
}
|
||||
}
|
||||
|
||||
void pattern_random(uint len, uint t) {
|
||||
if (t % 8)
|
||||
return;
|
||||
for (int i = 0; i < len; ++i)
|
||||
put_pixel(rand());
|
||||
}
|
||||
|
||||
void pattern_sparkle(uint len, uint t) {
|
||||
if (t % 8)
|
||||
return;
|
||||
for (int i = 0; i < len; ++i)
|
||||
put_pixel(rand() % 16 ? 0 : 0xffffffff);
|
||||
}
|
||||
|
||||
void pattern_greys(uint len, uint t) {
|
||||
int max = 100; // let's not draw too much current!
|
||||
t %= max;
|
||||
for (int i = 0; i < len; ++i) {
|
||||
put_pixel(t * 0x10101);
|
||||
if (++t >= max) t = 0;
|
||||
}
|
||||
}
|
||||
|
||||
typedef void (*pattern)(uint len, uint t);
|
||||
const struct {
|
||||
pattern pat;
|
||||
const char *name;
|
||||
} pattern_table[] = {
|
||||
{pattern_snakes, "Snakes!"},
|
||||
{pattern_random, "Random data"},
|
||||
{pattern_sparkle, "Sparkles"},
|
||||
{pattern_greys, "Greys"},
|
||||
};
|
||||
|
||||
const int PIN_TX = 0;
|
||||
|
||||
int main() {
|
||||
//set_sys_clock_48();
|
||||
stdio_init_all();
|
||||
puts("WS2812 Smoke Test");
|
||||
|
||||
// todo get free sm
|
||||
PIO pio = pio0;
|
||||
int sm = 0;
|
||||
uint offset = pio_add_program(pio, &ws2812_program);
|
||||
|
||||
ws2812_program_init(pio, sm, offset, PIN_TX, 800000, true);
|
||||
|
||||
int t = 0;
|
||||
while (1) {
|
||||
int pat = rand() % count_of(pattern_table);
|
||||
int dir = (rand() >> 30) & 1 ? 1 : -1;
|
||||
puts(pattern_table[pat].name);
|
||||
puts(dir == 1 ? "(forward)" : "(backward)");
|
||||
for (int i = 0; i < 1000; ++i) {
|
||||
pattern_table[pat].pat(150, t);
|
||||
sleep_ms(10);
|
||||
t += dir;
|
||||
}
|
||||
}
|
||||
}
|
||||
85
pio/ws2812/ws2812.pio
Normal file
85
pio/ws2812/ws2812.pio
Normal file
@@ -0,0 +1,85 @@
|
||||
;
|
||||
; Copyright (c) 2020 Raspberry Pi (Trading) Ltd.
|
||||
;
|
||||
; SPDX-License-Identifier: BSD-3-Clause
|
||||
;
|
||||
|
||||
.program ws2812
|
||||
.side_set 1
|
||||
|
||||
.define public T1 2
|
||||
.define public T2 5
|
||||
.define public T3 3
|
||||
|
||||
.lang_opt python sideset_init = pico.PIO.OUT_HIGH
|
||||
.lang_opt python out_init = pico.PIO.OUT_HIGH
|
||||
.lang_opt python out_shiftdir = 1
|
||||
|
||||
.wrap_target
|
||||
bitloop:
|
||||
out x, 1 side 0 [T3 - 1] ; Side-set still takes place when instruction stalls
|
||||
jmp !x do_zero side 1 [T1 - 1] ; Branch on the bit we shifted out. Positive pulse
|
||||
do_one:
|
||||
jmp bitloop side 1 [T2 - 1] ; Continue driving high, for a long pulse
|
||||
do_zero:
|
||||
nop side 0 [T2 - 1] ; Or drive low, for a short pulse
|
||||
.wrap
|
||||
|
||||
% c-sdk {
|
||||
#include "hardware/clocks.h"
|
||||
|
||||
static inline void ws2812_program_init(PIO pio, uint sm, uint offset, uint pin, float freq, bool rgbw) {
|
||||
|
||||
pio_gpio_init(pio, pin);
|
||||
pio_sm_set_consecutive_pindirs(pio, sm, pin, 1, true);
|
||||
|
||||
pio_sm_config c = ws2812_program_get_default_config(offset);
|
||||
sm_config_set_sideset_pins(&c, pin);
|
||||
sm_config_set_out_shift(&c, false, true, rgbw ? 32 : 24);
|
||||
sm_config_set_fifo_join(&c, PIO_FIFO_JOIN_TX);
|
||||
|
||||
int cycles_per_bit = ws2812_T1 + ws2812_T2 + ws2812_T3;
|
||||
float div = clock_get_hz(clk_sys) / (freq * cycles_per_bit);
|
||||
sm_config_set_clkdiv(&c, div);
|
||||
|
||||
pio_sm_init(pio, sm, offset, &c);
|
||||
pio_sm_set_enabled(pio, sm, true);
|
||||
}
|
||||
%}
|
||||
|
||||
.program ws2812_parallel
|
||||
|
||||
.define public T1 2
|
||||
.define public T2 5
|
||||
.define public T3 3
|
||||
|
||||
.wrap_target
|
||||
out x, 32
|
||||
mov pins, !null [T1-1]
|
||||
mov pins, x [T2-1]
|
||||
mov pins, null [T3-2]
|
||||
.wrap
|
||||
|
||||
% c-sdk {
|
||||
#include "hardware/clocks.h"
|
||||
|
||||
static inline void ws2812_parallel_program_init(PIO pio, uint sm, uint offset, uint pin_base, uint pin_count, float freq) {
|
||||
for(uint i=pin_base; i<pin_base+pin_count; i++) {
|
||||
pio_gpio_init(pio, i);
|
||||
}
|
||||
pio_sm_set_consecutive_pindirs(pio, sm, pin_base, pin_count, true);
|
||||
|
||||
pio_sm_config c = ws2812_parallel_program_get_default_config(offset);
|
||||
sm_config_set_out_shift(&c, true, true, 32);
|
||||
sm_config_set_out_pins(&c, pin_base, pin_count);
|
||||
sm_config_set_set_pins(&c, pin_base, pin_count);
|
||||
sm_config_set_fifo_join(&c, PIO_FIFO_JOIN_TX);
|
||||
|
||||
int cycles_per_bit = ws2812_parallel_T1 + ws2812_parallel_T2 + ws2812_parallel_T3;
|
||||
float div = clock_get_hz(clk_sys) / (freq * cycles_per_bit);
|
||||
sm_config_set_clkdiv(&c, div);
|
||||
|
||||
pio_sm_init(pio, sm, offset, &c);
|
||||
pio_sm_set_enabled(pio, sm, true);
|
||||
}
|
||||
%}
|
||||
341
pio/ws2812/ws2812_parallel.c
Normal file
341
pio/ws2812/ws2812_parallel.c
Normal file
@@ -0,0 +1,341 @@
|
||||
/**
|
||||
* Copyright (c) 2020 Raspberry Pi (Trading) Ltd.
|
||||
*
|
||||
* SPDX-License-Identifier: BSD-3-Clause
|
||||
*/
|
||||
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
|
||||
#include "pico/stdlib.h"
|
||||
#include "pico/sem.h"
|
||||
#include "hardware/pio.h"
|
||||
#include "hardware/dma.h"
|
||||
#include "hardware/irq.h"
|
||||
#include "ws2812.pio.h"
|
||||
|
||||
#define FRAC_BITS 4
|
||||
#define PIN_TX 0
|
||||
|
||||
CU_REGISTER_DEBUG_PINS(timing)
|
||||
CU_SELECT_DEBUG_PINS(timing)
|
||||
|
||||
// horrible temporary hack to avoid changing pattern code
|
||||
static uint8_t *current_string_out;
|
||||
static bool current_string_4color;
|
||||
|
||||
static inline void put_pixel(uint32_t pixel_grb) {
|
||||
*current_string_out++ = pixel_grb & 0xffu;
|
||||
*current_string_out++ = (pixel_grb >> 8u) & 0xffu;
|
||||
*current_string_out++ = (pixel_grb >> 16u) & 0xffu;
|
||||
if (current_string_4color) {
|
||||
*current_string_out++ = 0; // todo adjust?
|
||||
}
|
||||
}
|
||||
|
||||
static inline uint32_t urgb_u32(uint8_t r, uint8_t g, uint8_t b) {
|
||||
return
|
||||
((uint32_t) (r) << 8) |
|
||||
((uint32_t) (g) << 16) |
|
||||
(uint32_t) (b);
|
||||
}
|
||||
|
||||
void pattern_snakes(uint len, uint t) {
|
||||
for (uint i = 0; i < len; ++i) {
|
||||
uint x = (i + (t >> 1)) % 64;
|
||||
if (x < 10)
|
||||
put_pixel(urgb_u32(0xff, 0, 0));
|
||||
else if (x >= 15 && x < 25)
|
||||
put_pixel(urgb_u32(0, 0xff, 0));
|
||||
else if (x >= 30 && x < 40)
|
||||
put_pixel(urgb_u32(0, 0, 0xff));
|
||||
else
|
||||
put_pixel(0);
|
||||
}
|
||||
}
|
||||
|
||||
void pattern_random(uint len, uint t) {
|
||||
if (t % 8)
|
||||
return;
|
||||
for (int i = 0; i < len; ++i)
|
||||
put_pixel(rand());
|
||||
}
|
||||
|
||||
void pattern_sparkle(uint len, uint t) {
|
||||
if (t % 8)
|
||||
return;
|
||||
for (int i = 0; i < len; ++i)
|
||||
put_pixel(rand() % 16 ? 0 : 0xffffffff);
|
||||
}
|
||||
|
||||
void pattern_greys(uint len, uint t) {
|
||||
int max = 100; // let's not draw too much current!
|
||||
t %= max;
|
||||
for (int i = 0; i < len; ++i) {
|
||||
put_pixel(t * 0x10101);
|
||||
if (++t >= max) t = 0;
|
||||
}
|
||||
}
|
||||
|
||||
void pattern_solid(uint len, uint t) {
|
||||
t = 1;
|
||||
for (int i = 0; i < len; ++i) {
|
||||
put_pixel(t * 0x10101);
|
||||
}
|
||||
}
|
||||
|
||||
int level = 8;
|
||||
|
||||
void pattern_fade(uint len, uint t) {
|
||||
uint shift = 4;
|
||||
|
||||
uint max = 16; // let's not draw too much current!
|
||||
max <<= shift;
|
||||
|
||||
uint slow_t = t / 32;
|
||||
slow_t = level;
|
||||
slow_t %= max;
|
||||
|
||||
static int error;
|
||||
slow_t += error;
|
||||
error = slow_t & ((1u << shift) - 1);
|
||||
slow_t >>= shift;
|
||||
slow_t *= 0x010101;
|
||||
|
||||
for (int i = 0; i < len; ++i) {
|
||||
put_pixel(slow_t);
|
||||
}
|
||||
}
|
||||
|
||||
typedef void (*pattern)(uint len, uint t);
|
||||
const struct {
|
||||
pattern pat;
|
||||
const char *name;
|
||||
} pattern_table[] = {
|
||||
{pattern_snakes, "Snakes!"},
|
||||
{pattern_random, "Random data"},
|
||||
{pattern_sparkle, "Sparkles"},
|
||||
{pattern_greys, "Greys"},
|
||||
// {pattern_solid, "Solid!"},
|
||||
// {pattern_fade, "Fade"},
|
||||
};
|
||||
|
||||
#define VALUE_PLANE_COUNT (8 + FRAC_BITS)
|
||||
// we store value (8 bits + fractional bits of a single color (R/G/B/W) value) for multiple
|
||||
// strings, in bit planes. bit plane N has the Nth bit of each string.
|
||||
typedef struct {
|
||||
// stored MSB first
|
||||
uint32_t planes[VALUE_PLANE_COUNT];
|
||||
} value_bits_t;
|
||||
|
||||
// Add FRAC_BITS planes of e to s and store in d
|
||||
void add_error(value_bits_t *d, const value_bits_t *s, const value_bits_t *e) {
|
||||
uint32_t carry_plane = 0;
|
||||
// add the FRAC_BITS low planes
|
||||
for (int p = VALUE_PLANE_COUNT - 1; p >= 8; p--) {
|
||||
uint32_t e_plane = e->planes[p];
|
||||
uint32_t s_plane = s->planes[p];
|
||||
d->planes[p] = (e_plane ^ s_plane) ^ carry_plane;
|
||||
carry_plane = (e_plane & s_plane) | (carry_plane & (s_plane ^ e_plane));
|
||||
}
|
||||
// then just ripple carry through the non fractional bits
|
||||
for (int p = 7; p >= 0; p--) {
|
||||
uint32_t s_plane = s->planes[p];
|
||||
d->planes[p] = s_plane ^ carry_plane;
|
||||
carry_plane &= s_plane;
|
||||
}
|
||||
}
|
||||
|
||||
typedef struct {
|
||||
uint8_t *data;
|
||||
uint data_len;
|
||||
uint frac_brightness; // 256 = *1.0;
|
||||
} string_t;
|
||||
|
||||
// takes 8 bit color values, multiply by brightness and store in bit planes
|
||||
void transform_strings(string_t **strings, uint num_strings, value_bits_t *values, uint value_length,
|
||||
uint frac_brightness) {
|
||||
for (uint v = 0; v < value_length; v++) {
|
||||
memset(&values[v], 0, sizeof(values[v]));
|
||||
for (int i = 0; i < num_strings; i++) {
|
||||
if (v < strings[i]->data_len) {
|
||||
// todo clamp?
|
||||
uint32_t value = (strings[i]->data[v] * strings[i]->frac_brightness) >> 8u;
|
||||
value = (value * frac_brightness) >> 8u;
|
||||
for (int j = 0; j < VALUE_PLANE_COUNT && value; j++, value >>= 1u) {
|
||||
if (value & 1u) values[v].planes[VALUE_PLANE_COUNT - 1 - j] |= 1u << i;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
void dither_values(const value_bits_t *colors, value_bits_t *state, const value_bits_t *old_state, uint value_length) {
|
||||
for (uint i = 0; i < value_length; i++) {
|
||||
add_error(state + i, colors + i, old_state + i);
|
||||
}
|
||||
}
|
||||
|
||||
#define MAX_LENGTH 100
|
||||
|
||||
// requested colors * 4 to allow for WRGB
|
||||
static value_bits_t colors[MAX_LENGTH * 4];
|
||||
// double buffer the state of the string, since we update next version in parallel with DMAing out old version
|
||||
static value_bits_t states[2][MAX_LENGTH * 4];
|
||||
|
||||
// example - string 0 is RGB only
|
||||
static uint8_t string0_data[MAX_LENGTH * 3];
|
||||
// example - string 1 is WRGB
|
||||
static uint8_t string1_data[MAX_LENGTH * 4];
|
||||
|
||||
string_t string0 = {
|
||||
.data = string0_data,
|
||||
.data_len = sizeof(string0_data),
|
||||
.frac_brightness = 0x40,
|
||||
};
|
||||
|
||||
string_t string1 = {
|
||||
.data = string1_data,
|
||||
.data_len = sizeof(string1_data),
|
||||
.frac_brightness = 0x100,
|
||||
};
|
||||
|
||||
string_t *strings[] = {
|
||||
&string0,
|
||||
&string1,
|
||||
};
|
||||
|
||||
// bit plane content dma channel
|
||||
#define DMA_CHANNEL 0
|
||||
// chain channel for configuring main dma channel to output from disjoint 8 word fragments of memory
|
||||
#define DMA_CB_CHANNEL 1
|
||||
|
||||
#define DMA_CHANNEL_MASK (1u << DMA_CHANNEL)
|
||||
#define DMA_CB_CHANNEL_MASK (1u << DMA_CB_CHANNEL)
|
||||
#define DMA_CHANNELS_MASK (DMA_CHANNEL_MASK | DMA_CB_CHANNEL_MASK)
|
||||
|
||||
// start of each value fragment (+1 for NULL terminator)
|
||||
static uintptr_t fragment_start[MAX_LENGTH * 4 + 1];
|
||||
|
||||
// posted when it is safe to output a new set of values
|
||||
static struct semaphore reset_delay_complete_sem;
|
||||
// alarm handle for handling delay
|
||||
alarm_id_t reset_delay_alarm_id;
|
||||
|
||||
int64_t reset_delay_complete(alarm_id_t id, void *user_data) {
|
||||
reset_delay_alarm_id = 0;
|
||||
sem_release(&reset_delay_complete_sem);
|
||||
// no repeat
|
||||
return 0;
|
||||
}
|
||||
|
||||
void __isr dma_complete_handler() {
|
||||
if (dma_hw->ints0 & DMA_CHANNEL_MASK) {
|
||||
// clear IRQ
|
||||
dma_hw->ints0 = DMA_CHANNEL_MASK;
|
||||
// when the dma is complete we start the reset delay timer
|
||||
DEBUG_PINS_SET(timing, 4);
|
||||
if (reset_delay_alarm_id) cancel_alarm(reset_delay_alarm_id);
|
||||
reset_delay_alarm_id = add_alarm_in_us(400, reset_delay_complete, NULL, true);
|
||||
}
|
||||
}
|
||||
|
||||
void dma_init(PIO pio, uint sm) {
|
||||
dma_claim_mask(DMA_CHANNELS_MASK);
|
||||
|
||||
// main DMA channel outputs 8 word fragments, and then chains back to the chain channel
|
||||
dma_channel_config channel_config = dma_channel_get_default_config(DMA_CHANNEL);
|
||||
channel_config_set_dreq(&channel_config, pio_get_dreq(pio, sm, true));
|
||||
channel_config_set_chain_to(&channel_config, DMA_CB_CHANNEL);
|
||||
channel_config_set_irq_quiet(&channel_config, true);
|
||||
dma_channel_configure(DMA_CHANNEL,
|
||||
&channel_config,
|
||||
&pio->txf[sm],
|
||||
NULL, // set by chain
|
||||
8, // 8 words for 8 bit planes
|
||||
false);
|
||||
|
||||
// chain channel sends single word pointer to start of fragment each time
|
||||
dma_channel_config chain_config = dma_channel_get_default_config(DMA_CB_CHANNEL);
|
||||
dma_channel_configure(DMA_CB_CHANNEL,
|
||||
&chain_config,
|
||||
&dma_channel_hw_addr(
|
||||
DMA_CHANNEL)->al3_read_addr_trig, // ch DMA config (target "ring" buffer size 4) - this is (read_addr trigger)
|
||||
NULL, // set later
|
||||
1,
|
||||
false);
|
||||
|
||||
irq_set_exclusive_handler(DMA_IRQ_0, dma_complete_handler);
|
||||
dma_channel_set_irq0_enabled(DMA_CHANNEL, true);
|
||||
irq_set_enabled(DMA_IRQ_0, true);
|
||||
}
|
||||
|
||||
void output_strings_dma(value_bits_t *bits, uint value_length) {
|
||||
DEBUG_PINS_SET(timing, 3);
|
||||
for (uint i = 0; i < value_length; i++) {
|
||||
fragment_start[i] = (uintptr_t) bits[i].planes; // MSB first
|
||||
}
|
||||
fragment_start[value_length] = 0;
|
||||
dma_channel_hw_addr(DMA_CB_CHANNEL)->al3_read_addr_trig = (uintptr_t) fragment_start;
|
||||
DEBUG_PINS_CLR(timing, 3);
|
||||
}
|
||||
|
||||
|
||||
int main() {
|
||||
//set_sys_clock_48();
|
||||
stdio_init_all();
|
||||
puts("WS2812 parallel");
|
||||
#if PIN_TX != 3
|
||||
gpio_debug_pins_init();
|
||||
#endif
|
||||
// todo get free sm
|
||||
PIO pio = pio0;
|
||||
int sm = 0;
|
||||
uint offset = pio_add_program(pio, &ws2812_parallel_program);
|
||||
|
||||
ws2812_parallel_program_init(pio, sm, offset, PIN_TX, count_of(strings), 800000);
|
||||
|
||||
sem_init(&reset_delay_complete_sem, 1, 1); // initially posted so we don't block first time
|
||||
dma_init(pio, sm);
|
||||
int t = 0;
|
||||
while (1) {
|
||||
int pat = rand() % count_of(pattern_table);
|
||||
int dir = (rand() >> 30) & 1 ? 1 : -1;
|
||||
if (rand() & 1) dir = 0;
|
||||
puts(pattern_table[pat].name);
|
||||
puts(dir == 1 ? "(forward)" : dir ? "(backward)" : "(still)");
|
||||
int brightness = 0;
|
||||
uint current = 0;
|
||||
for (int i = 0; i < 1000; ++i) {
|
||||
int n = 64;
|
||||
DEBUG_PINS_SET(timing, 1);
|
||||
current_string_out = string0.data;
|
||||
current_string_4color = false;
|
||||
pattern_table[pat].pat(n, t);
|
||||
current_string_out = string1.data;
|
||||
current_string_4color = true;
|
||||
pattern_table[pat].pat(n, t);
|
||||
DEBUG_PINS_CLR(timing, 1);
|
||||
|
||||
DEBUG_PINS_SET(timing, 2);
|
||||
transform_strings(strings, count_of(strings), colors, n * 4, brightness);
|
||||
DEBUG_PINS_CLR(timing, 2);
|
||||
|
||||
DEBUG_PINS_SET(timing, 1);
|
||||
dither_values(colors, states[current], states[current ^ 1], n * 4);
|
||||
DEBUG_PINS_CLR(timing, 1);
|
||||
|
||||
sem_acquire_blocking(&reset_delay_complete_sem);
|
||||
DEBUG_PINS_CLR(timing, 4);
|
||||
|
||||
output_strings_dma(states[current], n * 4);
|
||||
|
||||
current ^= 1;
|
||||
t += dir;
|
||||
brightness++;
|
||||
if (brightness == (0x20 << FRAC_BITS)) brightness = 0;
|
||||
}
|
||||
memset(&states, 0, sizeof(states)); // clear out errors
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user