mirror of
https://github.com/WarmUpTill/SceneSwitcher.git
synced 2026-04-25 07:19:39 -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.
135 lines
2.9 KiB
C++
135 lines
2.9 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.8;
|
|
};
|
|
|
|
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;
|
|
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;
|
|
StringVariable languageCode = "eng";
|
|
|
|
private:
|
|
void Setup();
|
|
|
|
tesseract::PageSegMode pageSegMode = tesseract::PSM_SINGLE_BLOCK;
|
|
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)
|