forked from Makuna/NeoPixelBus
108 lines
2.9 KiB
C++
108 lines
2.9 KiB
C++
|
/*-------------------------------------------------------------------------
|
||
|
|
NeoPixel library
|
||
|
|
|
||
|
|
Written by Michael C. Miller.
|
||
|
|
|
||
|
|
I invest time and resources providing this open source code,
|
||
|
|
please support me by dontating (see https://github.com/Makuna/NeoPixelBus)
|
||
|
|
|
||
|
|
-------------------------------------------------------------------------
|
||
|
|
This file is part of the Makuna/NeoPixelBus library.
|
||
|
|
|
||
|
|
NeoPixelBus is free software: you can redistribute it and/or modify
|
||
|
|
it under the terms of the GNU Lesser General Public License as
|
||
|
|
published by the Free Software Foundation, either version 3 of
|
||
|
|
the License, or (at your option) any later version.
|
||
|
|
|
||
|
|
NeoPixelBus is distributed in the hope that it will be useful,
|
||
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||
|
|
GNU Lesser General Public License for more details.
|
||
|
|
|
||
|
|
You should have received a copy of the GNU Lesser General Public
|
||
|
|
License along with NeoPixel. If not, see
|
||
|
|
<http://www.gnu.org/licenses/>.
|
||
|
|
-------------------------------------------------------------------------*/
|
||
|
|
#pragma once
|
||
|
|
|
||
|
|
template<typename T_COLOR_OBJECT> class NeoDib
|
||
|
|
{
|
||
|
|
public:
|
||
|
|
NeoDib(uint16_t countPixels) :
|
||
|
|
_countPixels(countPixels)
|
||
|
|
{
|
||
|
|
_pixels = (T_COLOR_OBJECT*)malloc(PixelsSize());
|
||
|
|
}
|
||
|
|
|
||
|
|
~NeoDib()
|
||
|
|
{
|
||
|
|
free((uint8_t*)_pixels);
|
||
|
|
}
|
||
|
|
|
||
|
|
T_COLOR_OBJECT* Pixels() const
|
||
|
|
{
|
||
|
|
return _pixels;
|
||
|
|
};
|
||
|
|
|
||
|
|
uint16_t PixelCount() const
|
||
|
|
{
|
||
|
|
return _countPixels;
|
||
|
|
};
|
||
|
|
|
||
|
|
size_t PixelsSize() const
|
||
|
|
{
|
||
|
|
return _countPixels * PixelSize();
|
||
|
|
};
|
||
|
|
|
||
|
|
size_t PixelSize() const
|
||
|
|
{
|
||
|
|
return sizeof(T_COLOR_OBJECT);
|
||
|
|
};
|
||
|
|
|
||
|
|
void SetPixelColor(
|
||
|
|
uint16_t indexPixel,
|
||
|
|
T_COLOR_OBJECT color)
|
||
|
|
{
|
||
|
|
if (indexPixel < PixelCount())
|
||
|
|
{
|
||
|
|
_pixels[indexPixel] = color;
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
T_COLOR_OBJECT GetPixelColor(
|
||
|
|
uint16_t indexPixel) const
|
||
|
|
{
|
||
|
|
if (indexPixel >= PixelCount())
|
||
|
|
{
|
||
|
|
return 0;
|
||
|
|
}
|
||
|
|
return _pixels[indexPixel];
|
||
|
|
};
|
||
|
|
|
||
|
|
void ClearTo(T_COLOR_OBJECT color)
|
||
|
|
{
|
||
|
|
for (uint16_t pixel = 0; pixel < PixelCount(); pixel++)
|
||
|
|
{
|
||
|
|
_pixels[pixel] = color;
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
template <typename T_COLOR_FEATURE, typename T_SHADER> void Render(NeoBufferContext<T_COLOR_FEATURE> destBuffer, T_SHADER shader)
|
||
|
|
{
|
||
|
|
uint16_t countPixels = destBuffer.PixelCount();
|
||
|
|
if (countPixels > _countPixels)
|
||
|
|
{
|
||
|
|
countPixels = _countPixels;
|
||
|
|
}
|
||
|
|
|
||
|
|
for (uint16_t indexPixel = 0; indexPixel < countPixels; indexPixel++)
|
||
|
|
{
|
||
|
|
T_COLOR_OBJECT color = shader.Apply(indexPixel, _pixels[indexPixel]);
|
||
|
|
T_COLOR_FEATURE::applyPixelColor(destBuffer.Pixels, indexPixel, color);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
private:
|
||
|
|
const uint16_t _countPixels; // Number of RGB LEDs in strip
|
||
|
|
T_COLOR_OBJECT* _pixels;
|
||
|
|
};
|