110 lines
2.9 KiB
C++
110 lines
2.9 KiB
C++
|
|
// Hyperion includes
|
|
#include <hyperion/ImageProcessor.h>
|
|
|
|
#include <utils/ColorTransform.h>
|
|
|
|
// Local-Hyperion includes
|
|
#include "BlackBorderProcessor.h"
|
|
#include "ImageToLedsMap.h"
|
|
|
|
using namespace hyperion;
|
|
|
|
ImageProcessor::ImageProcessor(const LedString& ledString, bool enableBlackBorderDetector) :
|
|
mLedString(ledString),
|
|
_enableBlackBorderRemoval(enableBlackBorderDetector),
|
|
_borderProcessor(new BlackBorderProcessor(600, 50, 1)),
|
|
mImageToLeds(nullptr)
|
|
{
|
|
// empty
|
|
}
|
|
|
|
ImageProcessor::~ImageProcessor()
|
|
{
|
|
delete mImageToLeds;
|
|
delete _borderProcessor;
|
|
}
|
|
|
|
void ImageProcessor::setSize(const unsigned width, const unsigned height)
|
|
{
|
|
// Check if the existing buffer-image is already the correct dimensions
|
|
if (mImageToLeds && mImageToLeds->width() == width && mImageToLeds->height() == height)
|
|
{
|
|
return;
|
|
}
|
|
|
|
// Clean up the old buffer and mapping
|
|
delete mImageToLeds;
|
|
|
|
// Construct a new buffer and mapping
|
|
mImageToLeds = new ImageToLedsMap(width, height, 0, 0, mLedString.leds());
|
|
}
|
|
|
|
std::vector<RgbColor> ImageProcessor::process(const RgbImage& image)
|
|
{
|
|
// Ensure that the buffer-image is the proper size
|
|
setSize(image.width(), image.height());
|
|
|
|
// Check black border detection
|
|
verifyBorder(image);
|
|
|
|
// Create a result vector and call the 'in place' functionl
|
|
std::vector<RgbColor> colors = mImageToLeds->getMeanLedColor(image);
|
|
|
|
// return the computed colors
|
|
return colors;
|
|
}
|
|
|
|
void ImageProcessor::process(const RgbImage& image, std::vector<RgbColor>& ledColors)
|
|
{
|
|
// Ensure that the buffer-image is the proper size
|
|
setSize(image.width(), image.height());
|
|
|
|
// Check black border detection
|
|
verifyBorder(image);
|
|
|
|
// Determine the mean-colors of each led (using the existing mapping)
|
|
mImageToLeds->getMeanLedColor(image, ledColors);
|
|
}
|
|
|
|
bool ImageProcessor::getScanParameters(size_t led, double &hscanBegin, double &hscanEnd, double &vscanBegin, double &vscanEnd) const
|
|
{
|
|
if (led < mLedString.leds().size())
|
|
{
|
|
const Led & l = mLedString.leds()[led];
|
|
hscanBegin = l.minX_frac;
|
|
hscanEnd = l.maxX_frac;
|
|
vscanBegin = l.minY_frac;
|
|
vscanEnd = l.maxY_frac;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
void ImageProcessor::verifyBorder(const RgbImage& image)
|
|
{
|
|
if(_enableBlackBorderRemoval && _borderProcessor->process(image))
|
|
{
|
|
std::cout << "BORDER SWITCH REQUIRED!!" << std::endl;
|
|
|
|
const BlackBorder border = _borderProcessor->getCurrentBorder();
|
|
|
|
// Clean up the old mapping
|
|
delete mImageToLeds;
|
|
|
|
if (border.unknown)
|
|
{
|
|
// Construct a new buffer and mapping
|
|
mImageToLeds = new ImageToLedsMap(image.width(), image.height(), 0, 0, mLedString.leds());
|
|
}
|
|
else
|
|
{
|
|
// Construct a new buffer and mapping
|
|
mImageToLeds = new ImageToLedsMap(image.width(), image.height(), border.horizontalSize, border.verticalSize, mLedString.leds());
|
|
}
|
|
|
|
std::cout << "CURRENT BORDER TYPE: unknown=" << border.unknown << " hor.size=" << border.horizontalSize << " vert.size=" << border.verticalSize << std::endl;
|
|
}
|
|
|
|
}
|