implement image copy capture protocol

Author: ikalco

CMakeLists.txt

@@ -385,6 +385,7 @@ protocolnew("staging/ext-workspace" "ext-workspace-v1" false)
 protocolnew("staging/ext-data-control" "ext-data-control-v1" false)
 protocolnew("staging/pointer-warp" "pointer-warp-v1" false)
 protocolnew("staging/ext-image-capture-source" "ext-image-capture-source-v1" false)
+protocolnew("staging/ext-image-copy-capture" "ext-image-copy-capture-v1" false)
 
 protocolwayland()
 

protocols/meson.build

@@ -79,6 +79,7 @@ protocols = [
 	wayland_protocol_dir / 'staging/ext-data-control/ext-data-control-v1.xml',
 	wayland_protocol_dir / 'staging/pointer-warp/pointer-warp-v1.xml',
 	wayland_protocol_dir / 'staging/ext-image-capture-source/ext-image-capture-source-v1.xml',
+	wayland_protocol_dir / 'staging/ext-image-copy-capture/ext-image-copy-capture-v1.xml',
 ]
 
 wl_protocols = []

src/protocols/ImageCopyCapture.cpp

@@ -0,0 +1,73 @@
+#include "ImageCopyCapture.hpp"
+#include "desktop/Window.hpp"
+#include "helpers/Monitor.hpp"
+#include "core/Seat.hpp"
+
+CImageCopyCaptureSession::CImageCopyCaptureSession(SP<CExtImageCopyCaptureSessionV1> resource, SP<CImageCaptureSource> source, extImageCopyCaptureManagerV1Options options) :
+    m_resource(resource), m_source(source), m_paintCursor(options & EXT_IMAGE_COPY_CAPTURE_MANAGER_V1_OPTIONS_PAINT_CURSORS) {
+    m_listeners.destroy = source->m_events.destroy.registerListener([this](std::any data) { PROTO::imageCopyCapture->destroyResource(this); });
+
+    // TODO: send buffer constraint events, i.e. buffer_size, shm_format, dmabuf_device, dmabuf_format, done
+    // TODO: do create_frame, and frame class/object
+}
+
+CImageCopyCaptureSession::~CImageCopyCaptureSession() {
+    // TODO: properly fail/close screenshare
+}
+
+CImageCopyCaptureProtocol::CImageCopyCaptureProtocol(const wl_interface* iface, const int& ver, const std::string& name) : IWaylandProtocol(iface, ver, name) {
+    ;
+}
+
+void CImageCopyCaptureProtocol::bindManager(wl_client* client, void* data, uint32_t ver, uint32_t id) {
+    const auto RESOURCE = m_managers.emplace_back(makeShared<CExtImageCopyCaptureManagerV1>(client, ver, id));
+
+    if UNLIKELY (!RESOURCE->resource()) {
+        wl_client_post_no_memory(client);
+        m_managers.pop_back();
+        return;
+    }
+
+    RESOURCE->setDestroy([this](CExtImageCopyCaptureManagerV1* pMgr) { destroyResource(pMgr); });
+    RESOURCE->setOnDestroy([this](CExtImageCopyCaptureManagerV1* pMgr) { destroyResource(pMgr); });
+
+    RESOURCE->setCreateSession([this](CExtImageCopyCaptureManagerV1* pMgr, uint32_t id, wl_resource* source_, extImageCopyCaptureManagerV1Options options) {
+        auto source = PROTO::imageCaptureSource->sourceFromResource(source_);
+        if (!source) {
+            LOGM(LOG, "Client tried to create image copy capture session from invalid source");
+            pMgr->error(-1, "invalid image capture source");
+            return;
+        }
+
+        if (options > 1) {
+            LOGM(LOG, "Client tried to create image copy capture session with invalid options");
+            pMgr->error(EXT_IMAGE_COPY_CAPTURE_MANAGER_V1_ERROR_INVALID_OPTION, "Options can't be above 1");
+            return;
+        }
+
+        m_sessions.emplace_back(makeShared<CImageCopyCaptureSession>(makeShared<CExtImageCopyCaptureSessionV1>(pMgr->client(), pMgr->version(), id), source, options));
+        LOGM(LOG, "New image copy capture session for source ({}): {}", source->getTypeName(), source->getName());
+    });
+
+    RESOURCE->setCreatePointerCursorSession([this](CExtImageCopyCaptureManagerV1* pMgr, uint32_t id, wl_resource* source_, wl_resource* pointer_) {
+        SP<CImageCaptureSource> source = PROTO::imageCaptureSource->sourceFromResource(source_);
+        if (!source) {
+            LOGM(LOG, "Client tried to create image copy capture session from invalid source");
+            destroyResource(pMgr);
+            return;
+        }
+
+        // TODO: find out what this would be, then implement constructor aswell
+        // auto pointer = CWLPointerResource::fromResource(pointer_);
+
+        m_sessions.emplace_back(makeShared<CImageCopyCaptureSession>(makeShared<CExtImageCopyCaptureSessionV1>(pMgr->client(), pMgr->version(), id), source, pointer));
+        LOGM(LOG, "New image copy capture session for source ({}): {}", source->getTypeName(), source->getName());
+    });
+}
+
+void CImageCopyCaptureProtocol::destroyResource(CExtImageCopyCaptureManagerV1* resource) {
+    std::erase_if(m_managers, [&](const auto& other) { return other.get() == resource; });
+}
+void CImageCopyCaptureProtocol::destroyResource(CImageCopyCaptureSession* resource) {
+    std::erase_if(m_sessions, [&](const auto& other) { return other.get() == resource; });
+}

src/protocols/ImageCopyCapture.hpp

@@ -0,0 +1,45 @@
+#pragma once
+
+// TODO: implement rest of proto
+
+#include <vector>
+
+#include "../defines.hpp"
+#include "../helpers/signal/Signal.hpp"
+#include "WaylandProtocol.hpp"
+#include "ImageCaptureSource.hpp"
+#include "ext-image-copy-capture-v1.hpp"
+
+class CImageCopyCaptureSession {
+  public:
+    CImageCopyCaptureSession(SP<CExtImageCopyCaptureSessionV1> resource, SP<CImageCaptureSource> source, extImageCopyCaptureManagerV1Options options);
+    ~CImageCopyCaptureSession();
+
+  private:
+    SP<CExtImageCopyCaptureSessionV1> m_resource;
+    SP<CImageCaptureSource>           m_source;
+
+    struct {
+        CHyprSignalListener destroy;
+    } m_listeners;
+
+    bool m_paintCursor;
+};
+
+class CImageCopyCaptureProtocol : public IWaylandProtocol {
+  public:
+    CImageCopyCaptureProtocol(const wl_interface* iface, const int& ver, const std::string& name);
+
+    virtual void bindManager(wl_client* client, void* data, uint32_t ver, uint32_t id);
+
+    void         destroyResource(CExtImageCopyCaptureManagerV1* resource);
+    void         destroyResource(CImageCopyCaptureSession* resource);
+
+  private:
+    std::vector<SP<CExtImageCopyCaptureManagerV1>> m_managers;
+    std::vector<SP<CImageCopyCaptureSession>>      m_sessions;
+};
+
+namespace PROTO {
+    inline UP<CImageCopyCaptureProtocol> imageCopyCapture;
+};