mirror of
https://github.com/WarmUpTill/SceneSwitcher.git
synced 2026-04-20 17:17:25 -05:00
7d0332dd0e
The "core" macro conditions and actions have been extracted out to the "base" plugin. The library now mostly contains functionality which is required across all plugins and (e.g. definitions for macro segments). The goal is to reduce the complexity and cross-dependencies and group the source files in a better way. This should relsove the "library limit of 65535 objects exceeded" build issue occuring in some Windows build environments.
75 lines
2.0 KiB
C++
75 lines
2.0 KiB
C++
#pragma once
|
|
#include <QImage>
|
|
#undef NO // MacOS macro that can conflict with OpenCV
|
|
#include <opencv2/opencv.hpp>
|
|
|
|
#ifdef OCR_SUPPORT
|
|
#include <tesseract/baseapi.h>
|
|
#else
|
|
namespace tesseract {
|
|
enum PageSegMode {
|
|
PSM_OSD_ONLY = 0,
|
|
PSM_AUTO_OSD = 1,
|
|
PSM_AUTO_ONLY = 2,
|
|
PSM_AUTO = 3,
|
|
PSM_SINGLE_COLUMN = 4,
|
|
PSM_SINGLE_BLOCK_VERT_TEXT = 5,
|
|
PSM_SINGLE_BLOCK = 6,
|
|
PSM_SINGLE_LINE = 7,
|
|
PSM_SINGLE_WORD = 8,
|
|
PSM_CIRCLE_WORD = 9,
|
|
PSM_SINGLE_CHAR = 10,
|
|
PSM_SPARSE_TEXT = 11,
|
|
PSM_SPARSE_TEXT_OSD = 12,
|
|
PSM_RAW_LINE = 13,
|
|
|
|
PSM_COUNT
|
|
};
|
|
|
|
class TessBaseAPI {
|
|
public:
|
|
void SetPageSegMode(PageSegMode) {}
|
|
int Init(const char *, const char *) { return 0; }
|
|
void End() {}
|
|
};
|
|
}
|
|
#endif
|
|
|
|
namespace advss {
|
|
|
|
constexpr int minMinNeighbors = 3;
|
|
constexpr int maxMinNeighbors = 6;
|
|
constexpr double defaultScaleFactor = 1.1;
|
|
|
|
struct PatternImageData {
|
|
cv::Mat4b rgbaPattern;
|
|
cv::Mat3b rgbPattern;
|
|
cv::Mat1b mask;
|
|
};
|
|
|
|
PatternImageData CreatePatternData(const QImage &pattern);
|
|
void MatchPattern(QImage &img, const PatternImageData &patternData,
|
|
double threshold, cv::Mat &result, double *pBestFitValue,
|
|
bool useAlphaAsMask, cv::TemplateMatchModes matchMode);
|
|
void MatchPattern(QImage &img, QImage &pattern, double threshold,
|
|
cv::Mat &result, double *pBestFitValue, bool useAlphaAsMask,
|
|
cv::TemplateMatchModes matchMode);
|
|
std::vector<cv::Rect> MatchObject(QImage &img, cv::CascadeClassifier &cascade,
|
|
double scaleFactor, int minNeighbors,
|
|
const cv::Size &minSize,
|
|
const cv::Size &maxSize);
|
|
uchar GetAvgBrightness(QImage &img);
|
|
cv::Mat PreprocessForOCR(const QImage &image, const QColor &color,
|
|
double colorDiff);
|
|
std::string RunOCR(tesseract::TessBaseAPI *, const QImage &, const QColor &,
|
|
double colorDiff);
|
|
bool ContainsPixelsInColorRange(const QImage &image, const QColor &color,
|
|
double colorDeviationThreshold,
|
|
double totalPixelMatchThreshold);
|
|
QColor GetAverageColor(const QImage &img);
|
|
QColor GetDominantColor(const QImage &image, int k);
|
|
cv::Mat QImageToMat(const QImage &img);
|
|
QImage MatToQImage(const cv::Mat &mat);
|
|
|
|
} // namespace advss
|