326 lines
13 KiB
C++
326 lines
13 KiB
C++
/*
|
|
FX_2Dfcn.cpp contains all 2D utility functions
|
|
|
|
LICENSE
|
|
The MIT License (MIT)
|
|
Copyright (c) 2022 Blaz Kristan (https://blaz.at/home)
|
|
Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
of this software and associated documentation files (the "Software"), to deal
|
|
in the Software without restriction, including without limitation the rights
|
|
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
copies of the Software, and to permit persons to whom the Software is
|
|
furnished to do so, subject to the following conditions:
|
|
The above copyright notice and this permission notice shall be included in
|
|
all copies or substantial portions of the Software.
|
|
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
THE SOFTWARE.
|
|
|
|
Parts of the code adapted from WLED Sound Reactive
|
|
*/
|
|
#include "wled.h"
|
|
#include "FX.h"
|
|
#include "palettes.h"
|
|
|
|
// setUpMatrix() - constructs ledmap array from matrix of panels with WxH pixels
|
|
// this converts physical (possibly irregular) LED arrangement into well defined
|
|
// array of logical pixels: fist entry corresponds to left-topmost logical pixel
|
|
// followed by horizontal pixels, when matrixWidth logical pixels are added they
|
|
// are followed by next row (down) of matrixWidth pixels (and so forth)
|
|
// note: matrix may be comprised of multiple panels each with different orientation
|
|
// but ledmap takes care of that. ledmap is constructed upon initialization
|
|
// so matrix should disable regular ledmap processing
|
|
void WS2812FX::setUpMatrix() {
|
|
// erase old ledmap, just in case.
|
|
if (customMappingTable != nullptr) delete[] customMappingTable;
|
|
customMappingTable = nullptr;
|
|
customMappingSize = 0;
|
|
|
|
if (isMatrix) {
|
|
matrixWidth = hPanels * panelW;
|
|
matrixHeight = vPanels * panelH;
|
|
|
|
// safety check
|
|
if (matrixWidth * matrixHeight > MAX_LEDS) {
|
|
matrixWidth = getLengthTotal();
|
|
matrixHeight = 1;
|
|
isMatrix = false;
|
|
return;
|
|
}
|
|
|
|
customMappingSize = matrixWidth * matrixHeight;
|
|
customMappingTable = new uint16_t[customMappingSize];
|
|
|
|
if (customMappingTable != nullptr) {
|
|
uint16_t startL; // index in custom mapping array (logical strip)
|
|
uint16_t startP; // position of 1st pixel of panel on (virtual) strip
|
|
uint16_t x, y, offset;
|
|
uint8_t h = matrix.vertical ? vPanels : hPanels;
|
|
uint8_t v = matrix.vertical ? hPanels : vPanels;
|
|
|
|
for (uint8_t j=0, p=0; j<v; j++) {
|
|
for (uint8_t i=0; i<h; i++, p++) {
|
|
y = (matrix.vertical ? matrix.rightStart : matrix.bottomStart) ? v - j - 1 : j;
|
|
x = (matrix.vertical ? matrix.bottomStart : matrix.rightStart) ? h - i - 1 : i;
|
|
x = matrix.serpentine && j%2 ? h - x - 1 : x;
|
|
|
|
startL = (matrix.vertical ? y : x) * panelW + (matrix.vertical ? x : y) * matrixWidth * panelH; // logical index (top-left corner)
|
|
startP = p * panelW * panelH; // physical index (top-left corner)
|
|
|
|
uint8_t H = panel[h*j + i].vertical ? panelW : panelH;
|
|
uint8_t W = panel[h*j + i].vertical ? panelH : panelW;
|
|
for (uint16_t l=0, q=0; l<H; l++) {
|
|
for (uint16_t k=0; k<W; k++, q++) {
|
|
y = (panel[h*j + i].vertical ? panel[h*j + i].rightStart : panel[h*j + i].bottomStart) ? H - l - 1 : l;
|
|
x = (panel[h*j + i].vertical ? panel[h*j + i].bottomStart : panel[h*j + i].rightStart) ? W - k - 1 : k;
|
|
x = (panel[h*j + i].serpentine && l%2) ? (W - x - 1) : x;
|
|
offset = (panel[h*j + i].vertical ? y : x) + (panel[h*j + i].vertical ? x : y) * matrixWidth;
|
|
customMappingTable[startL + offset] = startP + q;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
#ifdef WLED_DEBUG
|
|
DEBUG_PRINT(F("Matrix ledmap:"));
|
|
for (uint16_t i=0; i<customMappingSize; i++) {
|
|
if (!(i%matrixWidth)) DEBUG_PRINTLN();
|
|
DEBUG_PRINTF("%4d,", customMappingTable[i]);
|
|
}
|
|
DEBUG_PRINTLN();
|
|
#endif
|
|
} else {
|
|
// memory allocation error
|
|
matrixWidth = getLengthTotal();
|
|
matrixHeight = 1;
|
|
isMatrix = false;
|
|
return;
|
|
}
|
|
} else {
|
|
// not a matrix set up
|
|
matrixWidth = getLengthTotal();
|
|
matrixHeight = 1;
|
|
}
|
|
}
|
|
|
|
// XY(x,y) - gets pixel index within current segment
|
|
uint16_t IRAM_ATTR WS2812FX::XY(uint16_t x, uint16_t y) {
|
|
uint16_t width = SEGMENT.virtualWidth(); // segment width in logical pixels
|
|
if (SEGMENT.getOption(SEG_OPTION_TRANSPOSED)) { uint16_t t = x; x = y; y = t; } // swap X & Y if segment transposed
|
|
return x + y * width;
|
|
}
|
|
|
|
// getPixelIndex(x,y,seg) - returns an index of segment pixel in a matrix layout
|
|
// index still needs to undergo ledmap processing to represent actual physical pixel
|
|
// matrix is always organized by matrixHeight number of matrixWidth pixels from top to bottom, left to right
|
|
// so: pixel at getPixelIndex(5,6) in a 2D segment with [start=10, stop=19, startY=20, stopY=29 : 10x10 pixels]
|
|
// corresponds to pixel with logical index of 847 (0 based) if a 2D segment belongs to a 32x32 matrix.
|
|
// math: (matrixWidth * (startY + y)) + start + x => (32 * (20+6)) + 10 + 5 = 847
|
|
uint16_t IRAM_ATTR WS2812FX::getPixelIndex(uint16_t x, uint16_t y, uint8_t seg) {
|
|
if (seg == 255) seg = _segment_index;
|
|
x %= _segments[seg].width(); // just in case constrain x (wrap around)
|
|
y %= _segments[seg].height(); // just in case constrain y (wrap around)
|
|
return ((_segments[seg].startY + y) * matrixWidth) + _segments[seg].start + x;
|
|
}
|
|
|
|
void IRAM_ATTR WS2812FX::setPixelColorXY(uint16_t x, uint16_t y, byte r, byte g, byte b, byte w)
|
|
{
|
|
if (!isMatrix) return; // not a matrix set-up
|
|
|
|
if (_bri_t < 255) {
|
|
r = scale8(r, _bri_t);
|
|
g = scale8(g, _bri_t);
|
|
b = scale8(b, _bri_t);
|
|
w = scale8(w, _bri_t);
|
|
}
|
|
uint32_t col = RGBW32(r, g, b, w);
|
|
|
|
uint16_t width = SEGMENT.virtualWidth(); // segment width in logical pixels (includes grouping, spacing, mirror & transposed)
|
|
uint16_t height = SEGMENT.virtualHeight(); // segment height in logical pixels (includes grouping, spacing, mirror & transposed)
|
|
if (SEGMENT.getOption(SEG_OPTION_TRANSPOSED)) { uint16_t t = x; x = y; y = t; } // swap X & Y if segment transposed
|
|
|
|
x *= SEGMENT.groupLength();
|
|
y *= SEGMENT.groupLength();
|
|
if (x >= width || y >= height) return; // if pixel would fall out of segment just exit
|
|
|
|
for (uint8_t j = 0; j < SEGMENT.grouping; j++) { // groupping vertically
|
|
for (uint8_t g = 0; g < SEGMENT.grouping; g++) { // groupping horizontally
|
|
uint16_t index, xX = (x+g), yY = (y+j);
|
|
if (xX >= SEGMENT.width() || yY >= SEGMENT.height()) continue; // we have reached one dimension's end
|
|
|
|
if (SEGMENT.getOption(SEG_OPTION_REVERSED) ) xX = width - xX - 1;
|
|
if (SEGMENT.getOption(SEG_OPTION_REVERSED_Y)) yY = height - yY - 1;
|
|
|
|
index = getPixelIndex(xX, yY);
|
|
if (index < customMappingSize) index = customMappingTable[index];
|
|
busses.setPixelColor(index, col);
|
|
|
|
if (SEGMENT.getOption(SEG_OPTION_MIRROR)) { //set the corresponding horizontally mirrored pixel
|
|
index = getPixelIndex(SEGMENT.width() - xX - 1, yY);
|
|
if (index < customMappingSize) index = customMappingTable[index];
|
|
busses.setPixelColor(index, col);
|
|
}
|
|
if (SEGMENT.getOption(SEG_OPTION_MIRROR_Y)) { //set the corresponding vertically mirrored pixel
|
|
index = getPixelIndex(xX, SEGMENT.height() - yY - 1);
|
|
if (index < customMappingSize) index = customMappingTable[index];
|
|
busses.setPixelColor(index, col);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
uint32_t WS2812FX::getPixelColorXY(uint16_t x, uint16_t y)
|
|
{
|
|
uint16_t width = SEGMENT.virtualWidth(); // segment width in logical pixels
|
|
uint16_t height = SEGMENT.virtualHeight(); // segment height in logical pixels
|
|
if (SEGMENT.getOption(SEG_OPTION_TRANSPOSED)) { uint16_t t = x; x = y; y = t; } // swap X & Y if segment transposed
|
|
|
|
x *= SEGMENT.groupLength();
|
|
y *= SEGMENT.groupLength();
|
|
if (x >= SEGMENT.width() || y >= SEGMENT.height()) return 0;
|
|
|
|
if (SEGMENT.getOption(SEG_OPTION_REVERSED) ) x = width - x - 1;
|
|
if (SEGMENT.getOption(SEG_OPTION_REVERSED_Y)) y = height - y - 1;
|
|
|
|
uint16_t index = getPixelIndex(x, y);
|
|
if (index < customMappingSize) index = customMappingTable[index];
|
|
|
|
return busses.getPixelColor(index);
|
|
}
|
|
|
|
|
|
// blurRows: perform a blur1d on every row of a rectangular matrix
|
|
void WS2812FX::blurRow(uint16_t y, fract8 blur_amount, CRGB* leds)
|
|
{
|
|
uint16_t width = SEGMENT.virtualWidth();
|
|
uint8_t keep = 255 - blur_amount;
|
|
uint8_t seep = blur_amount >> 1;
|
|
CRGB carryover = CRGB::Black;
|
|
for (uint16_t x = 0; x < width; x++) {
|
|
CRGB cur = leds ? leds[XY(x,y)] : col_to_crgb(getPixelColorXY(x,y));
|
|
CRGB part = cur;
|
|
part.nscale8(seep);
|
|
cur.nscale8(keep);
|
|
cur += carryover;
|
|
if (x) {
|
|
if (leds) leds[XY((x-1),y)] += part;
|
|
else setPixelColorXY(x-1, y, col_to_crgb(getPixelColorXY(x-1, y)) + part);
|
|
}
|
|
if (leds) leds[XY(x,y)] = cur;
|
|
else setPixelColorXY(x, y, cur);
|
|
carryover = part;
|
|
}
|
|
}
|
|
|
|
// blurColumns: perform a blur1d on each column of a rectangular matrix
|
|
void WS2812FX::blurCol(uint16_t x, fract8 blur_amount, CRGB* leds)
|
|
{
|
|
uint16_t height = SEGMENT.virtualHeight();
|
|
// blur columns
|
|
uint8_t keep = 255 - blur_amount;
|
|
uint8_t seep = blur_amount >> 1;
|
|
CRGB carryover = CRGB::Black;
|
|
for ( uint16_t y = 0; y < height; y++) {
|
|
CRGB cur = leds ? leds[XY(x,y)] : col_to_crgb(getPixelColorXY(x,y));
|
|
CRGB part = cur;
|
|
part.nscale8(seep);
|
|
cur.nscale8(keep);
|
|
cur += carryover;
|
|
if (y) {
|
|
if (leds) leds[XY(x,(y-1))] += part;
|
|
else setPixelColorXY(x, y-1, col_to_crgb(getPixelColorXY(x,y-1)) + part);
|
|
}
|
|
if (leds) leds[XY(x,y)] = cur;
|
|
else setPixelColorXY(x, y, cur);
|
|
carryover = part;
|
|
}
|
|
}
|
|
|
|
// blur1d: one-dimensional blur filter. Spreads light to 2 line neighbors.
|
|
// blur2d: two-dimensional blur filter. Spreads light to 8 XY neighbors.
|
|
//
|
|
// 0 = no spread at all
|
|
// 64 = moderate spreading
|
|
// 172 = maximum smooth, even spreading
|
|
//
|
|
// 173..255 = wider spreading, but increasing flicker
|
|
//
|
|
// Total light is NOT entirely conserved, so many repeated
|
|
// calls to 'blur' will also result in the light fading,
|
|
// eventually all the way to black; this is by design so that
|
|
// it can be used to (slowly) clear the LEDs to black.
|
|
|
|
void WS2812FX::blur1d(CRGB* leds, fract8 blur_amount)
|
|
{
|
|
uint16_t height = SEGMENT.virtualHeight();
|
|
for ( uint16_t y = 0; y < height; y++)
|
|
blurRow(y, blur_amount, leds);
|
|
}
|
|
|
|
void WS2812FX::blur2d(CRGB* leds, fract8 blur_amount)
|
|
{
|
|
uint16_t w = SEGMENT.virtualWidth(); // same as SEGLEN
|
|
uint16_t h = SEGMENT.virtualHeight();
|
|
uint8_t keep = 255 - blur_amount;
|
|
uint8_t seep = blur_amount >> 1;
|
|
for (uint16_t k = 0; k < h; k++) {
|
|
CRGB carryover = CRGB::Black;
|
|
for(uint16_t i = 0; i < w; i++) {
|
|
CRGB cur = leds ? leds[XY(i,k)] : col_to_crgb(getPixelColorXY(i, k));
|
|
CRGB part = cur;
|
|
part.nscale8(seep);
|
|
cur.nscale8(keep);
|
|
cur += carryover;
|
|
if (i > 0) {
|
|
CRGB c = leds ? leds[XY(i,k)] : col_to_crgb(getPixelColorXY(i-1, k));
|
|
c += part;
|
|
if (leds) leds[XY(i-1,k)] = c;
|
|
else setPixelColorXY(i-1, k, c.red, c.green, c.blue);
|
|
}
|
|
// seep from previous row
|
|
if (k > 0) {
|
|
CRGB c = col_to_crgb(getPixelColorXY(i, k-1));
|
|
c.nscale8(seep);
|
|
cur += c;
|
|
}
|
|
if (leds) leds[XY(i,k)] = cur;
|
|
else setPixelColorXY(i, k, cur.red, cur.green, cur.blue);
|
|
carryover = part;
|
|
}
|
|
}
|
|
}
|
|
|
|
//ewowi20210628: new functions moved from colorutils: add segment awareness
|
|
|
|
void WS2812FX::fill_solid(CRGB* leds, const struct CRGB& color) {
|
|
uint16_t w = SEGMENT.virtualWidth();
|
|
uint16_t h = SEGMENT.virtualHeight();
|
|
for(uint16_t y = 0; y < h; y++) for (uint16_t x = 0; x < w; x++) {
|
|
if (leds) leds[XY(x,y)] = color;
|
|
else setPixelColorXY(x, y, color);
|
|
}
|
|
}
|
|
|
|
void WS2812FX::fadeToBlackBy(CRGB* leds, uint8_t fadeBy) {
|
|
nscale8(leds, 255 - fadeBy);
|
|
}
|
|
|
|
void WS2812FX::nscale8(CRGB* leds, uint8_t scale) {
|
|
uint16_t w = SEGMENT.virtualWidth();
|
|
uint16_t h = SEGMENT.virtualHeight();
|
|
for(uint16_t y = 0; y < h; y++) for (uint16_t x = 0; x < w; x++) {
|
|
if (leds) leds[XY(x,y)].nscale8(scale);
|
|
else setPixelColorXY(x, y, col_to_crgb(getPixelColorXY(x, y)).nscale8(scale));
|
|
}
|
|
}
|
|
|
|
void WS2812FX::setPixels(CRGB* leds) {
|
|
uint16_t w = SEGMENT.virtualWidth();
|
|
uint16_t h = SEGMENT.virtualHeight();
|
|
for (uint16_t y = 0; y < h; y++) for (uint16_t x = 0; x < w; x++) setPixelColorXY(x, y, leds[XY(x,y)]);
|
|
}
|