From 20031cea92417a852410827a5cdb4adbcaee4342 Mon Sep 17 00:00:00 2001
From: Vaxry <43317083+vaxerski@users.noreply.github.com>
Date: Thu, 14 Nov 2024 20:15:51 +0000
Subject: pointer: add drm dumb buffers for cursors (#8399)

---------

Co-authored-by: Mihai Fufezan <mihai@fufexan.net>
---
 src/render/Texture.cpp | 21 +++++++++++++++++++--
 src/render/Texture.hpp | 34 ++++++++++++++++++++--------------
 2 files changed, 39 insertions(+), 16 deletions(-)

(limited to 'src/render')

diff --git a/src/render/Texture.cpp b/src/render/Texture.cpp
index 91e70afa..633f0212 100644
--- a/src/render/Texture.cpp
+++ b/src/render/Texture.cpp
@@ -3,6 +3,7 @@
 #include "../Compositor.hpp"
 #include "../protocols/types/Buffer.hpp"
 #include "../helpers/Format.hpp"
+#include <cstring>
 
 CTexture::CTexture() {
     // naffin'
@@ -16,7 +17,7 @@ CTexture::~CTexture() {
     destroyTexture();
 }
 
-CTexture::CTexture(uint32_t drmFormat, uint8_t* pixels, uint32_t stride, const Vector2D& size_) {
+CTexture::CTexture(uint32_t drmFormat, uint8_t* pixels, uint32_t stride, const Vector2D& size_, bool keepDataCopy) : m_iDrmFormat(drmFormat), m_bKeepDataCopy(keepDataCopy) {
     createFromShm(drmFormat, pixels, stride, size_);
 }
 
@@ -24,7 +25,7 @@ CTexture::CTexture(const Aquamarine::SDMABUFAttrs& attrs, void* image) {
     createFromDma(attrs, image);
 }
 
-CTexture::CTexture(const SP<Aquamarine::IBuffer> buffer) {
+CTexture::CTexture(const SP<Aquamarine::IBuffer> buffer, bool keepDataCopy) : m_bKeepDataCopy(keepDataCopy) {
     if (!buffer)
         return;
 
@@ -43,6 +44,8 @@ CTexture::CTexture(const SP<Aquamarine::IBuffer> buffer) {
 
         auto [pixelData, fmt, bufLen] = buffer->beginDataPtr(0);
 
+        m_iDrmFormat = fmt;
+
         createFromShm(fmt, pixelData, bufLen, shm.size);
         return;
     }
@@ -80,6 +83,11 @@ void CTexture::createFromShm(uint32_t drmFormat, uint8_t* pixels, uint32_t strid
     GLCALL(glTexImage2D(GL_TEXTURE_2D, 0, format->glInternalFormat ? format->glInternalFormat : format->glFormat, size_.x, size_.y, 0, format->glFormat, format->glType, pixels));
     GLCALL(glPixelStorei(GL_UNPACK_ROW_LENGTH_EXT, 0));
     GLCALL(glBindTexture(GL_TEXTURE_2D, 0));
+
+    if (m_bKeepDataCopy) {
+        m_vDataCopy.resize(stride * size_.y);
+        memcpy(m_vDataCopy.data(), pixels, stride * size_.y);
+    }
 }
 
 void CTexture::createFromDma(const Aquamarine::SDMABUFAttrs& attrs, void* image) {
@@ -135,6 +143,11 @@ void CTexture::update(uint32_t drmFormat, uint8_t* pixels, uint32_t stride, cons
     GLCALL(glPixelStorei(GL_UNPACK_SKIP_ROWS_EXT, 0));
 
     glBindTexture(GL_TEXTURE_2D, 0);
+
+    if (m_bKeepDataCopy) {
+        m_vDataCopy.resize(stride * m_vSize.y);
+        memcpy(m_vDataCopy.data(), pixels, stride * m_vSize.y);
+    }
 }
 
 void CTexture::destroyTexture() {
@@ -152,3 +165,7 @@ void CTexture::allocate() {
     if (!m_iTexID)
         GLCALL(glGenTextures(1, &m_iTexID));
 }
+
+const std::vector<uint8_t>& CTexture::dataCopy() {
+    return m_vDataCopy;
+}
diff --git a/src/render/Texture.hpp b/src/render/Texture.hpp
index e0ef5503..ae949d5f 100644
--- a/src/render/Texture.hpp
+++ b/src/render/Texture.hpp
@@ -24,26 +24,32 @@ class CTexture {
     CTexture(const CTexture&&) = delete;
     CTexture(const CTexture&)  = delete;
 
-    CTexture(uint32_t drmFormat, uint8_t* pixels, uint32_t stride, const Vector2D& size);
+    CTexture(uint32_t drmFormat, uint8_t* pixels, uint32_t stride, const Vector2D& size, bool keepDataCopy = false);
 
-    CTexture(const SP<Aquamarine::IBuffer> buffer);
+    CTexture(const SP<Aquamarine::IBuffer> buffer, bool keepDataCopy = false);
     // this ctor takes ownership of the eglImage.
     CTexture(const Aquamarine::SDMABUFAttrs&, void* image);
     ~CTexture();
 
-    void        destroyTexture();
-    void        allocate();
-    void        update(uint32_t drmFormat, uint8_t* pixels, uint32_t stride, const CRegion& damage);
+    void                        destroyTexture();
+    void                        allocate();
+    void                        update(uint32_t drmFormat, uint8_t* pixels, uint32_t stride, const CRegion& damage);
+    const std::vector<uint8_t>& dataCopy();
 
-    TEXTURETYPE m_iType      = TEXTURE_RGBA;
-    GLenum      m_iTarget    = GL_TEXTURE_2D;
-    GLuint      m_iTexID     = 0;
-    Vector2D    m_vSize      = {};
-    void*       m_pEglImage  = nullptr;
-    eTransform  m_eTransform = HYPRUTILS_TRANSFORM_NORMAL;
-    bool        m_bOpaque    = false;
+    TEXTURETYPE                 m_iType      = TEXTURE_RGBA;
+    GLenum                      m_iTarget    = GL_TEXTURE_2D;
+    GLuint                      m_iTexID     = 0;
+    Vector2D                    m_vSize      = {};
+    void*                       m_pEglImage  = nullptr;
+    eTransform                  m_eTransform = HYPRUTILS_TRANSFORM_NORMAL;
+    bool                        m_bOpaque    = false;
+    uint32_t                    m_iDrmFormat = 0; // for shm
 
   private:
-    void createFromShm(uint32_t drmFormat, uint8_t* pixels, uint32_t stride, const Vector2D& size);
-    void createFromDma(const Aquamarine::SDMABUFAttrs&, void* image);
+    void                 createFromShm(uint32_t drmFormat, uint8_t* pixels, uint32_t stride, const Vector2D& size);
+    void                 createFromDma(const Aquamarine::SDMABUFAttrs&, void* image);
+
+    bool                 m_bKeepDataCopy = false;
+
+    std::vector<uint8_t> m_vDataCopy;
 };
\ No newline at end of file
-- 
cgit v1.2.3