mirror of
https://github.com/hyperion-project/hyperion.ng.git
synced 2023-10-10 13:36:59 +02:00
4031a33f04
Added test executable for creating png from frame grabber. Added test-device for exporting color values of leds to text file. Updated configuration to match new color transform. Finished first version of the Hyperion structure (IT WORKS [1% CPU]!)
84 lines
2.0 KiB
C++
84 lines
2.0 KiB
C++
|
|
#pragma once
|
|
|
|
// STL includes
|
|
#include <sstream>
|
|
|
|
// hyperion-utils includes
|
|
#include <utils/RgbImage.h>
|
|
|
|
// hyperion includes
|
|
#include <hyperion/LedString.h>
|
|
|
|
namespace hyperion
|
|
{
|
|
|
|
class ImageToLedsMap
|
|
{
|
|
public:
|
|
|
|
/**
|
|
* Constructs an mapping from the colors in the image to each led based on the border
|
|
* definition given in the list of leds. The map holds pointers to the given image and its
|
|
* lifetime should never exceed that of the given image
|
|
*
|
|
* @param[in] image The RGB image
|
|
* @param[in] leds The list with led specifications
|
|
*/
|
|
ImageToLedsMap(const unsigned width, const unsigned height, const std::vector<Led> & leds);
|
|
|
|
unsigned width() const;
|
|
|
|
unsigned height() const;
|
|
|
|
/**
|
|
* Determines the mean-color for each led using the mapping the image given
|
|
* at construction.
|
|
*
|
|
* @return ledColors The vector containing the output
|
|
*/
|
|
std::vector<RgbColor> getMeanLedColor(const RgbImage & image) const;
|
|
|
|
/**
|
|
* Determines the mean-color for each led using the mapping the image given
|
|
* at construction.
|
|
*
|
|
* @param[out] ledColors The vector containing the output
|
|
*/
|
|
void getMeanLedColor(const RgbImage & image, std::vector<RgbColor> & ledColors) const;
|
|
|
|
std::string toString() const
|
|
{
|
|
std::stringstream sstream;
|
|
sstream << "ImageToLedsMap(" << _width << "x" << _height << ") [";
|
|
for (const std::vector<unsigned> imageIndices : mColorsMap)
|
|
{
|
|
sstream << "{";
|
|
for (unsigned imageIndex : imageIndices)
|
|
{
|
|
sstream << imageIndex << ";";
|
|
}
|
|
sstream << "}";
|
|
}
|
|
sstream << "]" << std::endl;
|
|
|
|
return sstream.str();
|
|
}
|
|
private:
|
|
const unsigned _width;
|
|
const unsigned _height;
|
|
std::vector<std::vector<unsigned> > mColorsMap;
|
|
|
|
/**
|
|
* Finds the 'mean color' of the given list. This is the mean over each color-channel (red,
|
|
* green, blue)
|
|
*
|
|
* @param colors The list with colors
|
|
*
|
|
* @return The mean of the given list of colors (or black when empty)
|
|
*/
|
|
RgbColor calcMeanColor(const RgbImage & image, const std::vector<unsigned> & colors) const;
|
|
};
|
|
|
|
} // end namespace hyperion
|