mirror of
https://github.com/spice2x/spice2x.github.io.git
synced 2026-08-01 22:30:42 -07:00
bicarus-dev
df9d9d8673
## Link to GitHub Issue, if one exists n/a ## Description of change Allow the server to override client's quality options for screen mirroring. On Windows / Android versions (flutter-based) clients, this is already in the options of those apps. However, the iOS version lacks the quality options. ## Testing Tested with iOS and Windows clients.
95 lines
2.7 KiB
C++
95 lines
2.7 KiB
C++
#include "capture.h"
|
|
#include <functional>
|
|
#include "external/rapidjson/document.h"
|
|
#include "hooks/graphics/graphics.h"
|
|
#include "util/crypt.h"
|
|
|
|
using namespace std::placeholders;
|
|
using namespace rapidjson;
|
|
|
|
namespace api::modules {
|
|
|
|
std::optional<uint32_t> CAPTURE_QUALITY;
|
|
std::optional<uint32_t> CAPTURE_DIVIDE;
|
|
|
|
static thread_local std::vector<uint8_t> CAPTURE_BUFFER;
|
|
|
|
Capture::Capture() : Module("capture") {
|
|
functions["get_screens"] = std::bind(&Capture::get_screens, this, _1, _2);
|
|
functions["get_jpg"] = std::bind(&Capture::get_jpg, this, _1, _2);
|
|
}
|
|
|
|
/**
|
|
* get_screens()
|
|
*/
|
|
void Capture::get_screens(Request &req, Response &res) {
|
|
|
|
// aquire screens
|
|
std::vector<int> screens;
|
|
graphics_screens_get(screens);
|
|
|
|
// add screens to response
|
|
for (auto &screen : screens) {
|
|
res.add_data(screen);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* get_jpg([screen=0, quality=70, downscale=0, divide=1])
|
|
* screen: uint specifying the window
|
|
* quality: uint in range [0, 100]
|
|
* reduce: uint for dividing image size
|
|
*/
|
|
void Capture::get_jpg(Request &req, Response &res) {
|
|
CAPTURE_BUFFER.reserve(1024 * 128);
|
|
|
|
// settings
|
|
int screen = 0;
|
|
int quality = 70;
|
|
int divide = 1;
|
|
if (req.params.Size() > 0 && req.params[0].IsUint()) {
|
|
screen = req.params[0].GetUint();
|
|
}
|
|
|
|
if (CAPTURE_QUALITY.has_value()) {
|
|
quality = CAPTURE_QUALITY.value();
|
|
} else if (req.params.Size() > 1 && req.params[1].IsUint()) {
|
|
quality = req.params[1].GetUint();
|
|
}
|
|
|
|
if (CAPTURE_DIVIDE.has_value()) {
|
|
divide = CAPTURE_DIVIDE.value();
|
|
} else if (req.params.Size() > 2 && req.params[2].IsUint()) {
|
|
divide = req.params[2].GetUint();
|
|
}
|
|
|
|
// receive JPEG data
|
|
uint64_t timestamp = 0;
|
|
int width = 0;
|
|
int height = 0;
|
|
graphics_capture_trigger(screen);
|
|
bool success = graphics_capture_receive_jpeg(screen, [] (uint8_t byte) {
|
|
CAPTURE_BUFFER.push_back(byte);
|
|
}, true, quality, true, divide, ×tamp, &width, &height);
|
|
if (!success) {
|
|
return;
|
|
}
|
|
|
|
// encode to base64
|
|
auto encoded = crypt::base64_encode(
|
|
CAPTURE_BUFFER.data(),
|
|
CAPTURE_BUFFER.size());
|
|
|
|
// clear buffer
|
|
CAPTURE_BUFFER.clear();
|
|
|
|
// add data to response
|
|
Value data;
|
|
data.SetString(encoded.c_str(), encoded.length(), res.doc()->GetAllocator());
|
|
res.add_data(timestamp);
|
|
res.add_data(width);
|
|
res.add_data(height);
|
|
res.add_data(data);
|
|
}
|
|
}
|