wayland/screencopy: add screencopy

src/wayland/screencopy/wlr_screencopy/wlr_screencopy.cpp

@@ -0,0 +1,133 @@
+#include "wlr_screencopy.hpp"
+#include <cstdint>
+
+#include <private/qwaylandscreen_p.h>
+#include <qlogging.h>
+#include <qloggingcategory.h>
+#include <qobject.h>
+#include <qscreen.h>
+#include <qtmetamacros.h>
+#include <qwaylandclientextension.h>
+#include <wayland-wlr-screencopy-unstable-v1-client-protocol.h>
+
+#include "../../buffer/manager.hpp"
+#include "../manager.hpp"
+#include "wlr_screencopy_p.hpp"
+
+namespace qs::wayland::screencopy::wlr {
+
+namespace {
+Q_LOGGING_CATEGORY(logScreencopy, "quickshell.wayland.screencopy.wlr", QtWarningMsg);
+}
+
+WlrScreencopyManager::WlrScreencopyManager(): QWaylandClientExtensionTemplate(3) {
+	this->initialize();
+}
+
+WlrScreencopyManager* WlrScreencopyManager::instance() {
+	static auto* instance = new WlrScreencopyManager();
+	return instance;
+}
+
+ScreencopyContext*
+WlrScreencopyManager::captureOutput(QScreen* screen, bool paintCursors, QRect region) {
+	if (!dynamic_cast<QtWaylandClient::QWaylandScreen*>(screen->handle())) return nullptr;
+	return new WlrScreencopyContext(this, screen, paintCursors, region);
+}
+
+WlrScreencopyContext::WlrScreencopyContext(
+    WlrScreencopyManager* manager,
+    QScreen* screen,
+    bool paintCursors,
+    QRect region
+)
+    : manager(manager)
+    , screen(dynamic_cast<QtWaylandClient::QWaylandScreen*>(screen->handle()))
+    , paintCursors(paintCursors)
+    , region(region) {
+	QObject::connect(screen, &QObject::destroyed, this, &WlrScreencopyContext::onScreenDestroyed);
+}
+
+WlrScreencopyContext::~WlrScreencopyContext() {
+	if (this->object()) this->destroy();
+}
+
+void WlrScreencopyContext::onScreenDestroyed() {
+	qCWarning(logScreencopy) << "Screen destroyed while recording. Stopping" << this;
+	if (this->object()) this->destroy();
+	emit this->stopped();
+}
+
+void WlrScreencopyContext::captureFrame() {
+	if (this->object()) return;
+
+	if (this->region.isEmpty()) {
+		this->init(manager->capture_output(this->paintCursors ? 1 : 0, screen->output()));
+	} else {
+		this->init(manager->capture_output_region(
+		    this->paintCursors ? 1 : 0,
+		    screen->output(),
+		    this->region.x(),
+		    this->region.y(),
+		    this->region.width(),
+		    this->region.height()
+		));
+	}
+}
+
+void WlrScreencopyContext::zwlr_screencopy_frame_v1_buffer(
+    uint32_t format,
+    uint32_t width,
+    uint32_t height,
+    uint32_t /*stride*/
+) {
+	// While different sizes can technically be requested, that would be insane.
+	this->request.width = width;
+	this->request.height = height;
+	this->request.shm.formats.push(format);
+}
+
+void WlrScreencopyContext::zwlr_screencopy_frame_v1_linux_dmabuf(
+    uint32_t format,
+    uint32_t width,
+    uint32_t height
+) {
+	// While different sizes can technically be requested, that would be insane.
+	this->request.width = width;
+	this->request.height = height;
+	this->request.dmabuf.formats.push(format);
+}
+
+void WlrScreencopyContext::zwlr_screencopy_frame_v1_flags(uint32_t flags) {
+	if (flags & ZWLR_SCREENCOPY_FRAME_V1_FLAGS_Y_INVERT) {
+		this->mSwapchain.backbuffer()->transform = buffer::WlBufferTransform::Flipped180;
+	}
+}
+
+void WlrScreencopyContext::zwlr_screencopy_frame_v1_buffer_done() {
+	auto* backbuffer = this->mSwapchain.createBackbuffer(this->request);
+
+	if (this->copiedFirstFrame) {
+		this->copy_with_damage(backbuffer->buffer());
+	} else {
+		this->copy(backbuffer->buffer());
+	}
+}
+
+void WlrScreencopyContext::zwlr_screencopy_frame_v1_ready(
+    uint32_t /*tvSecHi*/,
+    uint32_t /*tvSecLo*/,
+    uint32_t /*tvNsec*/
+) {
+	this->destroy();
+	this->copiedFirstFrame = true;
+	this->mSwapchain.swapBuffers();
+	emit this->frameCaptured();
+}
+
+void WlrScreencopyContext::zwlr_screencopy_frame_v1_failed() {
+	qCWarning(logScreencopy) << "Ending recording due to screencopy failure for" << this;
+	emit this->stopped();
+}
+
+} // namespace qs::wayland::screencopy::wlr