mirror of
https://github.com/hyperion-project/hyperion.ng.git
synced 2023-10-10 13:36:59 +02:00
132 lines
4.0 KiB
C++
132 lines
4.0 KiB
C++
|
|
#pragma once
|
|
|
|
// Jsoncpp includes
|
|
#include <json/json.h>
|
|
|
|
// Local Hyperion includes
|
|
#include "BlackBorderDetector.h"
|
|
|
|
namespace hyperion
|
|
{
|
|
///
|
|
/// The BlackBorder processor is a wrapper around the black-border detector for keeping track of
|
|
/// detected borders and count of the type and size of detected borders.
|
|
///
|
|
class BlackBorderProcessor
|
|
{
|
|
public:
|
|
///
|
|
/// Constructor for the BlackBorderProcessor
|
|
/// @param unknownFrameCnt The number of frames(images) that need to contain an unknown
|
|
/// border before the current border is set to unknown
|
|
/// @param borderFrameCnt The number of frames(images) that need to contain a vertical or
|
|
/// horizontal border becomes the current border
|
|
/// @param blurRemoveCnt The size to add to a horizontal or vertical border (because the
|
|
/// outer pixels is blurred (black and color combined due to image scaling))
|
|
/// @param[in] blackborderThreshold The threshold which the blackborder detector should use
|
|
///
|
|
BlackBorderProcessor(const Json::Value &blackborderConfig);
|
|
|
|
///
|
|
/// Return the current (detected) border
|
|
/// @return The current border
|
|
///
|
|
BlackBorder getCurrentBorder() const;
|
|
|
|
///
|
|
/// Return activation state of black border detector
|
|
/// @return The current border
|
|
///
|
|
bool enabled() const;
|
|
|
|
///
|
|
/// Set activation state of black border detector
|
|
/// @param enable current state
|
|
///
|
|
void setEnabled(bool enable);
|
|
|
|
///
|
|
/// Processes the image. This performs detecion of black-border on the given image and
|
|
/// updates the current border accordingly. If the current border is updated the method call
|
|
/// will return true else false
|
|
///
|
|
/// @param image The image to process
|
|
///
|
|
/// @return True if a different border was detected than the current else false
|
|
///
|
|
template <typename Pixel_T>
|
|
bool process(const Image<Pixel_T> & image)
|
|
{
|
|
// get the border for the single image
|
|
BlackBorder imageBorder;
|
|
if (!enabled())
|
|
{
|
|
return false;
|
|
}
|
|
|
|
if (_detectionMode == "default") {
|
|
imageBorder = _detector.process(image);
|
|
} else if (_detectionMode == "classic") {
|
|
imageBorder = _detector.process_classic(image);
|
|
} else if (_detectionMode == "osd") {
|
|
imageBorder = _detector.process_osd(image);
|
|
}
|
|
// add blur to the border
|
|
if (imageBorder.horizontalSize > 0)
|
|
{
|
|
imageBorder.horizontalSize += _blurRemoveCnt;
|
|
}
|
|
if (imageBorder.verticalSize > 0)
|
|
{
|
|
imageBorder.verticalSize += _blurRemoveCnt;
|
|
}
|
|
|
|
const bool borderUpdated = updateBorder(imageBorder);
|
|
return borderUpdated;
|
|
}
|
|
|
|
private:
|
|
///
|
|
/// Updates the current border based on the newly detected border. Returns true if the
|
|
/// current border has changed.
|
|
///
|
|
/// @param newDetectedBorder The newly detected border
|
|
/// @return True if the current border changed else false
|
|
///
|
|
bool updateBorder(const BlackBorder & newDetectedBorder);
|
|
|
|
/// flag for blackborder detector usage
|
|
bool _enabled;
|
|
|
|
/// The number of unknown-borders detected before it becomes the current border
|
|
const unsigned _unknownSwitchCnt;
|
|
|
|
/// The number of horizontal/vertical borders detected before it becomes the current border
|
|
const unsigned _borderSwitchCnt;
|
|
|
|
// The number of frames that are "ignored" before a new border gets set as _previousDetectedBorder
|
|
const unsigned _maxInconsistentCnt;
|
|
|
|
/// The number of pixels to increase a detected border for removing blury pixels
|
|
unsigned _blurRemoveCnt;
|
|
|
|
/// The border detection mode
|
|
const std::string _detectionMode;
|
|
|
|
/// The blackborder detector
|
|
BlackBorderDetector _detector;
|
|
|
|
/// The current detected border
|
|
BlackBorder _currentBorder;
|
|
|
|
/// The border detected in the previous frame
|
|
BlackBorder _previousDetectedBorder;
|
|
|
|
/// The number of frame the previous detected border matched the incomming border
|
|
unsigned _consistentCnt;
|
|
/// The number of frame the previous detected border NOT matched the incomming border
|
|
unsigned _inconsistentCnt;
|
|
};
|
|
} // end namespace hyperion
|