66 lines
1.5 KiB
C++
66 lines
1.5 KiB
C++
#include <hyperion/ImageProcessor.h>
|
|
|
|
|
|
#include "ImageToLedsMap.h"
|
|
#include "ColorTransform.h"
|
|
|
|
using namespace hyperion;
|
|
|
|
ImageProcessor::ImageProcessor(const LedString& ledString) :
|
|
mLedString(ledString),
|
|
mBuffer(nullptr),
|
|
mImageToLeds(nullptr)
|
|
{
|
|
// empty
|
|
}
|
|
|
|
ImageProcessor::~ImageProcessor()
|
|
{
|
|
delete mImageToLeds;
|
|
delete mBuffer;
|
|
}
|
|
|
|
std::vector<RgbColor> ImageProcessor::process(const RgbImage& image)
|
|
{
|
|
// Ensure that the buffer-image is the proper size
|
|
setSize(image.width(), image.height());
|
|
|
|
// Copy the data of the given image into the mapped-image
|
|
mBuffer->copy(image);
|
|
|
|
// Create a result vector and call the 'in place' functionl
|
|
std::vector<RgbColor> colors(mLedString.leds().size(), RgbColor::BLACK);
|
|
inplace_process(colors);
|
|
|
|
// return the computed colors
|
|
return colors;
|
|
}
|
|
|
|
void ImageProcessor::setSize(const unsigned width, const unsigned height)
|
|
{
|
|
// Check if the existing buffer-image is already the correct dimensions
|
|
if (mBuffer && mBuffer->width() == width && mBuffer->height() == height)
|
|
{
|
|
return;
|
|
}
|
|
|
|
// Clean up the old buffer and mapping
|
|
delete mImageToLeds;
|
|
delete mBuffer;
|
|
|
|
// Construct a new buffer and mapping
|
|
mBuffer = new RgbImage(width, height);
|
|
mImageToLeds = new ImageToLedsMap(*mBuffer, mLedString.leds());
|
|
}
|
|
|
|
RgbImage& ImageProcessor::image()
|
|
{
|
|
return *mBuffer;
|
|
}
|
|
|
|
void ImageProcessor::inplace_process(std::vector<RgbColor>& ledColors)
|
|
{
|
|
// Determine the mean-colors of each led (using the existing mapping)
|
|
mImageToLeds->getMeanLedColor(ledColors);
|
|
}
|