mirror of
https://github.com/WarmUpTill/SceneSwitcher.git
synced 2026-03-21 17:34:57 -05:00
cdc5d16e95
This will prevent custom models being deleted when installing a new version of the plugin, as the plugin's data directory might get wiped on some operating systems.
140 lines
3.1 KiB
C++
140 lines
3.1 KiB
C++
#pragma once
|
|
#include "opencv-helpers.hpp"
|
|
#include "obs-module-helper.hpp"
|
|
#include "area-selection.hpp"
|
|
|
|
#include <source-selection.hpp>
|
|
#include <scene-selection.hpp>
|
|
#include <regex-config.hpp>
|
|
#include <variable-string.hpp>
|
|
#include <variable-number.hpp>
|
|
#include <obs.hpp>
|
|
#include <obs-module.h>
|
|
|
|
#include <QMetaType>
|
|
|
|
#ifdef OCR_SUPPORT
|
|
#include <tesseract/baseapi.h>
|
|
#endif
|
|
|
|
namespace advss {
|
|
|
|
enum class VideoCondition {
|
|
MATCH,
|
|
DIFFER,
|
|
HAS_NOT_CHANGED,
|
|
HAS_CHANGED,
|
|
NO_IMAGE,
|
|
PATTERN,
|
|
OBJECT,
|
|
BRIGHTNESS,
|
|
OCR,
|
|
COLOR,
|
|
};
|
|
|
|
class VideoInput {
|
|
public:
|
|
bool Save(obs_data_t *obj) const;
|
|
bool Load(obs_data_t *obj);
|
|
std::string ToString(bool resolve = false) const;
|
|
bool ValidSelection() const;
|
|
OBSWeakSource GetVideo() const;
|
|
|
|
enum class Type {
|
|
OBS_MAIN_OUTPUT,
|
|
SOURCE,
|
|
SCENE,
|
|
};
|
|
|
|
Type type = Type::OBS_MAIN_OUTPUT;
|
|
SourceSelection source;
|
|
SceneSelection scene;
|
|
};
|
|
|
|
class PatternMatchParameters {
|
|
public:
|
|
bool Save(obs_data_t *obj) const;
|
|
bool Load(obs_data_t *obj);
|
|
|
|
QImage image;
|
|
bool useForChangedCheck = false;
|
|
bool useAlphaAsMask = false;
|
|
cv::TemplateMatchModes matchMode = cv::TM_CCORR_NORMED;
|
|
NumberVariable<double> threshold = 0.999;
|
|
};
|
|
|
|
class ObjDetectParameters {
|
|
public:
|
|
bool Save(obs_data_t *obj) const;
|
|
bool Load(obs_data_t *obj);
|
|
|
|
std::string modelPath =
|
|
obs_get_module_data_path(obs_current_module()) +
|
|
std::string(
|
|
"/res/cascadeClassifiers/haarcascade_frontalface_alt.xml");
|
|
cv::CascadeClassifier cascade;
|
|
NumberVariable<double> scaleFactor = defaultScaleFactor;
|
|
int minNeighbors = minMinNeighbors;
|
|
Size minSize{0, 0};
|
|
Size maxSize{0, 0};
|
|
};
|
|
|
|
class OCRParameters {
|
|
public:
|
|
OCRParameters();
|
|
~OCRParameters();
|
|
OCRParameters(const OCRParameters &other);
|
|
OCRParameters &operator=(const OCRParameters &);
|
|
|
|
bool Save(obs_data_t *obj) const;
|
|
bool Load(obs_data_t *obj);
|
|
|
|
bool Initialized() const { return initDone; }
|
|
void SetPageMode(tesseract::PageSegMode);
|
|
bool SetLanguageCode(const std::string &);
|
|
std::string GetLanguageCode() const;
|
|
bool SetTesseractBasePath(const std::string &);
|
|
std::string GetTesseractBasePath() const;
|
|
tesseract::PageSegMode GetPageMode() const { return pageSegMode; }
|
|
tesseract::TessBaseAPI *GetOCR() const { return ocr.get(); }
|
|
|
|
StringVariable text = obs_module_text("AdvSceneSwitcher.enterText");
|
|
RegexConfig regex = RegexConfig::PartialMatchRegexConfig();
|
|
QColor color = Qt::black;
|
|
DoubleVariable colorThreshold = 0.3;
|
|
|
|
private:
|
|
void Setup();
|
|
|
|
tesseract::PageSegMode pageSegMode = tesseract::PSM_SINGLE_BLOCK;
|
|
StringVariable tesseractBasePath =
|
|
obs_get_module_data_path(obs_current_module()) +
|
|
std::string("/res/ocr");
|
|
StringVariable languageCode = "eng";
|
|
std::unique_ptr<tesseract::TessBaseAPI> ocr;
|
|
bool initDone = false;
|
|
};
|
|
|
|
class ColorParameters {
|
|
public:
|
|
bool Save(obs_data_t *obj) const;
|
|
bool Load(obs_data_t *obj);
|
|
|
|
QColor color = Qt::black;
|
|
DoubleVariable colorThreshold = 0.1;
|
|
DoubleVariable matchThreshold = 0.8;
|
|
};
|
|
|
|
class AreaParameters {
|
|
public:
|
|
bool Save(obs_data_t *obj) const;
|
|
bool Load(obs_data_t *obj);
|
|
|
|
bool enable = false;
|
|
advss::Area area{0, 0, 0, 0};
|
|
};
|
|
|
|
} // namespace advss
|
|
|
|
Q_DECLARE_METATYPE(advss::OCRParameters)
|