Leahnaya/SceneSwitcher
1
0
Fork 0
mirror of https://github.com/WarmUpTill/SceneSwitcher.git synced 2026-09-07 00:56:00 -05:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add support to save screenshots to file

Browse Source
This commit is contained in:
WarmUpTill
2022-09-03 11:21:43 +02:00
committed by WarmUpTill
parent 0169f6c445
commit 0d29fb2ee1
2 changed files with 34 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
src/utils/screenshot-helper.cpp
View File

@@ -3,8 +3,11 @@
static void ScreenshotTick(void *param, float);
ScreenshotHelper::ScreenshotHelper(obs_source_t *source)
: weakSource(OBSGetWeakRef(source))
ScreenshotHelper::ScreenshotHelper(obs_source_t *source, bool saveToFile,
std::string path)
: weakSource(OBSGetWeakRef(source)),
_saveToFile(saveToFile),
_path(path)
{
_initDone = true;
obs_add_tick_callback(ScreenshotTick, this);
@@ -20,6 +23,9 @@ ScreenshotHelper::~ScreenshotHelper()
obs_remove_tick_callback(ScreenshotTick, this);
}
if (_saveThread.joinable()) {
_saveThread.join();
}
}
void ScreenshotHelper::Screenshot()
@@ -100,6 +106,24 @@ void ScreenshotHelper::MarkDone()
done = true;
}
void ScreenshotHelper::WriteToFile()
{
if (!_saveToFile) {
return;
}
_saveThread = std::thread([this]() {
if (image.save(QString::fromStdString(_path))) {
vblog(LOG_INFO, "Wrote screenshot to \"%s\"",
_path.c_str());
} else {
blog(LOG_WARNING,
"Failed to save screenshot to to \"%s\"!\nMaybe unkown format?",
_path.c_str());
}
});
}
#define STAGE_SCREENSHOT 0
#define STAGE_DOWNLOAD 1
#define STAGE_COPY_AND_SAVE 2
@@ -125,6 +149,7 @@ static void ScreenshotTick(void *param, float)
case STAGE_COPY_AND_SAVE:
data->Copy();
data->MarkDone();
data->WriteToFile();
obs_remove_tick_callback(ScreenshotTick, data);
break;