58 lines
1.2 KiB
C++
58 lines
1.2 KiB
C++
|
#include "BlackBorderProcessor.h"
|
||
|
|
|
||
|
|
BlackBorderProcessor::BlackBorderProcessor() :
|
||
|
|
_unknownSwitchCnt(600),
|
||
|
|
_borderSwitchCnt(50),
|
||
|
|
_detector(),
|
||
|
|
_currentBorder({BlackBorder::unknown, 0}),
|
||
|
|
_lastDetectedBorder({BlackBorder::unknown, 0}),
|
||
|
|
_consistentCnt(0)
|
||
|
|
{
|
||
|
|
}
|
||
|
|
|
||
|
|
BlackBorder BlackBorderProcessor::getCurrentBorder() const
|
||
|
|
{
|
||
|
|
return _currentBorder;
|
||
|
|
}
|
||
|
|
|
||
|
|
bool BlackBorderProcessor::process(const RgbImage& image)
|
||
|
|
{
|
||
|
|
const BlackBorder imageBorder = _detector.process(image);
|
||
|
|
|
||
|
|
if (imageBorder.type == _lastDetectedBorder.type && imageBorder.size == _lastDetectedBorder.size)
|
||
|
|
{
|
||
|
|
++_consistentCnt;
|
||
|
|
}
|
||
|
|
else
|
||
|
|
{
|
||
|
|
_lastDetectedBorder = imageBorder;
|
||
|
|
_consistentCnt = 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
bool borderChanged = false;
|
||
|
|
switch (_lastDetectedBorder.type)
|
||
|
|
{
|
||
|
|
case BlackBorder::none:
|
||
|
|
borderChanged = (_currentBorder.type != BlackBorder::none);
|
||
|
|
_currentBorder = _lastDetectedBorder;
|
||
|
|
break;
|
||
|
|
case BlackBorder::horizontal:
|
||
|
|
case BlackBorder::vertical:
|
||
|
|
if (_consistentCnt == _borderSwitchCnt)
|
||
|
|
{
|
||
|
|
_currentBorder = _lastDetectedBorder;
|
||
|
|
borderChanged = true;
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
case BlackBorder::unknown:
|
||
|
|
if (_consistentCnt == _unknownSwitchCnt)
|
||
|
|
{
|
||
|
|
_currentBorder = _lastDetectedBorder;
|
||
|
|
borderChanged = true;
|
||
|
|
}
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
return borderChanged;
|
||
|
|
}
|