95 lines
2.3 KiB
C++
95 lines
2.3 KiB
C++
|
|
// STL includes
|
|
#include <cassert>
|
|
|
|
// Hyperion includes
|
|
#include <utils/Logger.h>
|
|
#include "MultiColorTransform.h"
|
|
|
|
MultiColorTransform::MultiColorTransform(const unsigned ledCnt) :
|
|
_ledTransforms(ledCnt, nullptr)
|
|
{
|
|
}
|
|
|
|
MultiColorTransform::~MultiColorTransform()
|
|
{
|
|
// Clean up all the transforms
|
|
for (ColorTransform * transform : _transform)
|
|
{
|
|
delete transform;
|
|
}
|
|
}
|
|
|
|
void MultiColorTransform::addTransform(ColorTransform * transform)
|
|
{
|
|
_transformIds.push_back(transform->_id);
|
|
_transform.push_back(transform);
|
|
}
|
|
|
|
void MultiColorTransform::setTransformForLed(const std::string& id, const unsigned startLed, const unsigned endLed)
|
|
{
|
|
assert(startLed <= endLed);
|
|
assert(endLed < _ledTransforms.size());
|
|
|
|
// Get the identified transform (don't care if is nullptr)
|
|
ColorTransform * transform = getTransform(id);
|
|
for (unsigned iLed=startLed; iLed<=endLed; ++iLed)
|
|
{
|
|
_ledTransforms[iLed] = transform;
|
|
}
|
|
}
|
|
|
|
bool MultiColorTransform::verifyTransforms() const
|
|
{
|
|
for (unsigned iLed=0; iLed<_ledTransforms.size(); ++iLed)
|
|
{
|
|
if (_ledTransforms[iLed] == nullptr)
|
|
{
|
|
Warning(Logger::getInstance("ColorTransform"), "No adjustment set for %d", iLed);
|
|
return false;
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
|
|
const std::vector<std::string> & MultiColorTransform::getTransformIds()
|
|
{
|
|
return _transformIds;
|
|
}
|
|
|
|
ColorTransform* MultiColorTransform::getTransform(const std::string& id)
|
|
{
|
|
// Iterate through the unique transforms until we find the one with the given id
|
|
for (ColorTransform* transform : _transform)
|
|
{
|
|
if (transform->_id == id)
|
|
{
|
|
return transform;
|
|
}
|
|
}
|
|
|
|
// The ColorTransform was not found
|
|
return nullptr;
|
|
}
|
|
|
|
void MultiColorTransform::applyTransform(std::vector<ColorRgb>& ledColors)
|
|
{
|
|
const size_t itCnt = std::min(_ledTransforms.size(), ledColors.size());
|
|
for (size_t i=0; i<itCnt; ++i)
|
|
{
|
|
ColorTransform* transform = _ledTransforms[i];
|
|
if (transform == nullptr)
|
|
{
|
|
// No transform set for this led (do nothing)
|
|
continue;
|
|
}
|
|
ColorRgb& color = ledColors[i];
|
|
|
|
transform->_hsvTransform.transform(color.red, color.green, color.blue);
|
|
transform->_hslTransform.transform(color.red, color.green, color.blue);
|
|
color.red = transform->_rgbRedTransform.transform(color.red);
|
|
color.green = transform->_rgbGreenTransform.transform(color.green);
|
|
color.blue = transform->_rgbBlueTransform.transform(color.blue);
|
|
}
|
|
}
|