#pragma once

// Utils includes
#include <utils/RgbImage.h>

// Hyperion includes
#include <hyperion/ImageProcessorFactory.h>
#include <hyperion/LedString.h>

// Forward class declaration
namespace hyperion {
	class ImageToLedsMap;
	class BlackBorderProcessor;
}

///
/// The ImageProcessor translates an RGB-image to RGB-values for the leds. The processing is
/// performed in two steps. First the average color per led-region is computed. Second a
/// color-tranform is applied based on a gamma-correction.
///
class ImageProcessor
{
public:
	~ImageProcessor();

	///
	/// Specifies the width and height of 'incomming' images. This will resize the buffer-image to
	/// match the given size.
	/// NB All earlier obtained references will be invalid.
	///
	/// @param[in] width   The new width of the buffer-image
	/// @param[in] height  The new height of the buffer-image
	///
	void setSize(const unsigned width, const unsigned height);

	///
	/// Processes the image to a list of led colors. This will update the size of the buffer-image
	/// if required and call the image-to-leds mapping to determine the mean color per led.
	///
	/// @param[in] image  The image to translate to led values
	///
	/// @return The color value per led
	///
	std::vector<RgbColor> process(const RgbImage& image);

	///
	/// Determines the led colors of the image in the buffer.
	///
	/// @param[in] image  The image to translate to led values
	/// @param[out] ledColors  The color value per led
	///
	void process(const RgbImage& image, std::vector<RgbColor>& ledColors);

private:
	/// Friend declaration of the factory for creating ImageProcessor's
	friend class ImageProcessorFactory;

	///
	/// Constructs an image-processor for translating an image to led-color values based on the
	/// given led-string specification
	///
	/// @param[in] ledString  The led-string specification
	///
	ImageProcessor(const LedString &ledString, bool enableBlackBorderDetector);

	///
	/// Performs black-border detection (if enabled) on the given image
	///
	/// @param[in] image  The image to perform black-border detection on
	///
	void verifyBorder(const RgbImage& image);
private:
	/// The Led-string specification
	const LedString mLedString;

	/// Flag the enables(true)/disabled(false) blackborder detector
	bool _enableBlackBorderRemoval;

	/// The processor for black border detection
	hyperion::BlackBorderProcessor* _borderProcessor;

	/// The mapping of image-pixels to leds
	hyperion::ImageToLedsMap* mImageToLeds;
};