Refactor buffer handling

We cannot support bindToTexture() functions. On some
platforms the texture is provided by the driver. Therefore,
the HW integration must always provide a texture.

This has the added bonus of unifying the two separate code paths
that were introduced when EGLStream support was added.

Add a separate buffer manager that owns all buffers.
Don't destroy buffer objects on release. The client will probably
attach them again later. Also, release shm buffers immediately
after uploading to texture (needs to be documented that image()
will not work afterwards).

Make the old SurfaceBuffer class into an abstract base class,
so we can store state in the buffer class instead of having
to map from the wl_resource in each buffer integration. Move
the shared memory buffer handling into a separate subclass.

Change-Id: I81e471d13c92913d31ea1efe487f93fa908b5e0c
Reviewed-by: Dominik Holland <dominik.holland@pelagicore.com>
Reviewed-by: Louai Al-Khanji <louai.al-khanji@qt.io>
Reviewed-by: Johan Helsing <johan.helsing@qt.io>

6 files changed, 424 insertions, 304 deletions

diff --git a/src/compositor/wayland_wrapper/qwlbuffermanager.cpp b/src/compositor/wayland_wrapper/qwlbuffermanager.cpp
new file mode 100644
index 000000000..765c9a03a
--- /dev/null
+++ b/src/compositor/wayland_wrapper/qwlbuffermanager.cpp
@@ -0,0 +1,111 @@
+/****************************************************************************
+**
+** Copyright (C) 2016 The Qt Company Ltd.
+** Contact: http://www.qt.io/licensing/
+**
+** This file is part of the QtWaylandCompositor module of the Qt Toolkit.
+**
+** $QT_BEGIN_LICENSE:LGPL3$
+** Commercial License Usage
+** Licensees holding valid commercial Qt licenses may use this file in
+** accordance with the commercial license agreement provided with the
+** Software or, alternatively, in accordance with the terms contained in
+** a written agreement between you and The Qt Company. For licensing terms
+** and conditions see http://www.qt.io/terms-conditions. For further
+** information use the contact form at http://www.qt.io/contact-us.
+**
+** GNU Lesser General Public License Usage
+** Alternatively, this file may be used under the terms of the GNU Lesser
+** General Public License version 3 as published by the Free Software
+** Foundation and appearing in the file LICENSE.LGPLv3 included in the
+** packaging of this file. Please review the following information to
+** ensure the GNU Lesser General Public License version 3 requirements
+** will be met: https://www.gnu.org/licenses/lgpl.html.
+**
+** GNU General Public License Usage
+** Alternatively, this file may be used under the terms of the GNU
+** General Public License version 2.0 or later as published by the Free
+** Software Foundation and appearing in the file LICENSE.GPL included in
+** the packaging of this file. Please review the following information to
+** ensure the GNU General Public License version 2.0 requirements will be
+** met: http://www.gnu.org/licenses/gpl-2.0.html.
+**
+** $QT_END_LICENSE$
+**
+****************************************************************************/
+
+#include "qwlbuffermanager_p.h"
+#include <QWaylandCompositor>
+#include <QtWaylandCompositor/private/qwaylandcompositor_p.h>
+#include <QtWaylandCompositor/private/qwlclientbufferintegration_p.h>
+#include <QDebug>
+
+QT_BEGIN_NAMESPACE
+
+namespace QtWayland {
+
+BufferManager::BufferManager(QWaylandCompositor *compositor)
+    : QObject(compositor)
+    , m_compositor(compositor)
+{
+
+}
+
+struct buffer_manager_destroy_listener : wl_listener
+{
+    buffer_manager_destroy_listener()
+        : d(0)
+    {
+        notify = BufferManager::destroy_listener_callback;
+        wl_list_init(&this->link);
+    }
+
+    BufferManager *d;
+};
+
+ClientBuffer *BufferManager::getBuffer(wl_resource *buffer_resource)
+{
+    if (!buffer_resource)
+        return nullptr;
+
+    auto it = m_buffers.find(buffer_resource);
+    if (it != m_buffers.end())
+        return it.value();
+
+    auto bufferIntegration = QWaylandCompositorPrivate::get(m_compositor)->clientBufferIntegration();
+    ClientBuffer *newBuffer = nullptr;
+    if (bufferIntegration)
+        newBuffer = bufferIntegration->createBufferFor(buffer_resource);
+    if (!newBuffer)
+        newBuffer = new SharedMemoryBuffer(buffer_resource);
+    m_buffers[buffer_resource] = newBuffer;
+
+    auto *destroy_listener = new buffer_manager_destroy_listener;
+    destroy_listener->d = this;
+    wl_signal_add(&buffer_resource->destroy_signal, destroy_listener);
+    return newBuffer;
+}
+
+
+void BufferManager::destroy_listener_callback(wl_listener *listener, void *data)
+{
+    buffer_manager_destroy_listener *destroy_listener = static_cast<buffer_manager_destroy_listener *>(listener);
+    BufferManager *self = destroy_listener->d;
+    struct ::wl_resource *buffer = static_cast<struct ::wl_resource *>(data);
+
+    wl_list_remove(&destroy_listener->link);
+    delete destroy_listener;
+
+    Q_ASSERT(self);
+    Q_ASSERT(buffer);
+
+    ClientBuffer *clientBuffer = self->m_buffers.take(buffer);
+
+    if (!clientBuffer)
+        return;
+
+    clientBuffer->setDestroyed();
+}
+
+}
+QT_END_NAMESPACE
diff --git a/src/compositor/wayland_wrapper/qwlbuffermanager_p.h b/src/compositor/wayland_wrapper/qwlbuffermanager_p.h
new file mode 100644
index 000000000..e1325ef89
--- /dev/null
+++ b/src/compositor/wayland_wrapper/qwlbuffermanager_p.h
@@ -0,0 +1,77 @@
+/****************************************************************************
+**
+** Copyright (C) 2016 The Qt Company Ltd.
+** Contact: http://www.qt.io/licensing/
+**
+** This file is part of the QtWaylandCompositor module of the Qt Toolkit.
+**
+** $QT_BEGIN_LICENSE:LGPL3$
+** Commercial License Usage
+** Licensees holding valid commercial Qt licenses may use this file in
+** accordance with the commercial license agreement provided with the
+** Software or, alternatively, in accordance with the terms contained in
+** a written agreement between you and The Qt Company. For licensing terms
+** and conditions see http://www.qt.io/terms-conditions. For further
+** information use the contact form at http://www.qt.io/contact-us.
+**
+** GNU Lesser General Public License Usage
+** Alternatively, this file may be used under the terms of the GNU Lesser
+** General Public License version 3 as published by the Free Software
+** Foundation and appearing in the file LICENSE.LGPLv3 included in the
+** packaging of this file. Please review the following information to
+** ensure the GNU Lesser General Public License version 3 requirements
+** will be met: https://www.gnu.org/licenses/lgpl.html.
+**
+** GNU General Public License Usage
+** Alternatively, this file may be used under the terms of the GNU
+** General Public License version 2.0 or later as published by the Free
+** Software Foundation and appearing in the file LICENSE.GPL included in
+** the packaging of this file. Please review the following information to
+** ensure the GNU General Public License version 2.0 requirements will be
+** met: http://www.gnu.org/licenses/gpl-2.0.html.
+**
+** $QT_END_LICENSE$
+**
+****************************************************************************/
+
+#ifndef QWLBUFFERMANAGER_H
+#define QWLBUFFERMANAGER_H
+
+//
+//  W A R N I N G
+//  -------------
+//
+// This file is not part of the Qt API.  It exists purely as an
+// implementation detail.  This header file may change from version to
+// version without notice, or even be removed.
+//
+// We mean it.
+//
+
+#include <QtCore/QObject>
+#include "qwlclientbuffer_p.h"
+QT_BEGIN_NAMESPACE
+
+class QWaylandCompositor;
+
+namespace QtWayland {
+
+class ClientBuffer;
+
+class Q_WAYLAND_COMPOSITOR_EXPORT BufferManager : public QObject
+{
+public:
+    BufferManager(QWaylandCompositor *compositor);
+    ClientBuffer *getBuffer(struct ::wl_resource *buffer_resource);
+private:
+    friend struct buffer_manager_destroy_listener;
+    static void destroy_listener_callback(wl_listener *listener, void *data);
+
+    QHash<struct ::wl_resource *, ClientBuffer*> m_buffers;
+    QWaylandCompositor *m_compositor;
+};
+
+}
+QT_END_NAMESPACE
+
+#endif // QWLBUFFERMANAGER_H
diff --git a/src/compositor/wayland_wrapper/qwlclientbuffer.cpp b/src/compositor/wayland_wrapper/qwlclientbuffer.cpp
new file mode 100644
index 000000000..589ab825b
--- /dev/null
+++ b/src/compositor/wayland_wrapper/qwlclientbuffer.cpp
@@ -0,0 +1,189 @@
+/****************************************************************************
+**
+** Copyright (C) 2015 The Qt Company Ltd.
+** Contact: http://www.qt.io/licensing/
+**
+** This file is part of the QtWaylandCompositor module of the Qt Toolkit.
+**
+** $QT_BEGIN_LICENSE:LGPL3$
+** Commercial License Usage
+** Licensees holding valid commercial Qt licenses may use this file in
+** accordance with the commercial license agreement provided with the
+** Software or, alternatively, in accordance with the terms contained in
+** a written agreement between you and The Qt Company. For licensing terms
+** and conditions see http://www.qt.io/terms-conditions. For further
+** information use the contact form at http://www.qt.io/contact-us.
+**
+** GNU Lesser General Public License Usage
+** Alternatively, this file may be used under the terms of the GNU Lesser
+** General Public License version 3 as published by the Free Software
+** Foundation and appearing in the file LICENSE.LGPLv3 included in the
+** packaging of this file. Please review the following information to
+** ensure the GNU Lesser General Public License version 3 requirements
+** will be met: https://www.gnu.org/licenses/lgpl.html.
+**
+** GNU General Public License Usage
+** Alternatively, this file may be used under the terms of the GNU
+** General Public License version 2.0 or later as published by the Free
+** Software Foundation and appearing in the file LICENSE.GPL included in
+** the packaging of this file. Please review the following information to
+** ensure the GNU General Public License version 2.0 requirements will be
+** met: http://www.gnu.org/licenses/gpl-2.0.html.
+**
+** $QT_END_LICENSE$
+**
+****************************************************************************/
+
+#include "qwlclientbuffer_p.h"
+
+#ifdef QT_WAYLAND_COMPOSITOR_GL
+#include "hardware_integration/qwlclientbufferintegration_p.h"
+#include <qpa/qplatformopenglcontext.h>
+#include <QOpenGLTexture>
+#endif
+
+#include <QtCore/QDebug>
+
+#include <wayland-server-protocol.h>
+#include "qwaylandsharedmemoryformathelper_p.h"
+
+#include <QtWaylandCompositor/private/qwaylandcompositor_p.h>
+
+QT_BEGIN_NAMESPACE
+
+namespace QtWayland {
+
+ClientBuffer::ClientBuffer(struct ::wl_resource *buffer)
+    : m_buffer(buffer)
+    , m_textureDirty(false)
+    , m_committed(false)
+    , m_destroyed(false)
+{
+}
+
+
+ClientBuffer::~ClientBuffer()
+{
+    if (m_buffer && m_committed && !m_destroyed)
+        sendRelease();
+}
+
+void ClientBuffer::sendRelease()
+{
+    Q_ASSERT(m_buffer);
+    wl_buffer_send_release(m_buffer);
+    m_committed = false;
+}
+
+void ClientBuffer::setDestroyed()
+{
+    m_destroyed = true;
+    m_committed = false;
+    m_buffer = nullptr;
+
+    if (!m_refCount)
+        delete this;
+}
+
+void ClientBuffer::ref()
+{
+    m_refCount.ref();
+}
+
+void ClientBuffer::deref()
+{
+    if (!m_refCount.deref()) {
+        if (isCommitted() && m_buffer && !m_destroyed)
+            sendRelease();
+        if (m_destroyed)
+            delete this;
+    }
+}
+
+void ClientBuffer::setCommitted(QRegion &damage)
+{
+     m_damage = damage;
+     m_committed = true;
+     m_textureDirty = true;
+}
+
+QWaylandBufferRef::BufferFormatEgl ClientBuffer::bufferFormatEgl() const
+{
+    return QWaylandBufferRef::BufferFormatEgl_Null;
+}
+
+SharedMemoryBuffer::SharedMemoryBuffer(wl_resource *bufferResource)
+    : ClientBuffer(bufferResource)
+    , m_shmTexture(nullptr)
+{
+
+}
+
+QSize SharedMemoryBuffer::size() const
+{
+    if (wl_shm_buffer *shmBuffer = wl_shm_buffer_get(m_buffer)) {
+        int width = wl_shm_buffer_get_width(shmBuffer);
+        int height = wl_shm_buffer_get_height(shmBuffer);
+        return QSize(width, height);
+    }
+    return QSize();
+}
+
+QWaylandSurface::Origin SharedMemoryBuffer::origin() const
+{
+    return QWaylandSurface::OriginTopLeft;
+}
+
+
+// TODO: support different color formats, and try to avoid QImage::convertToFormat()
+
+QImage SharedMemoryBuffer::image() const
+{
+    if (wl_shm_buffer *shmBuffer = wl_shm_buffer_get(m_buffer)) {
+        int width = wl_shm_buffer_get_width(shmBuffer);
+        int height = wl_shm_buffer_get_height(shmBuffer);
+        int bytesPerLine = wl_shm_buffer_get_stride(shmBuffer);
+        uchar *data = static_cast<uchar *>(wl_shm_buffer_get_data(shmBuffer));
+        return QImage(data, width, height, bytesPerLine, QImage::Format_ARGB32_Premultiplied);
+    }
+
+    return QImage();
+}
+
+#ifdef QT_WAYLAND_COMPOSITOR_GL
+QOpenGLTexture *SharedMemoryBuffer::toOpenGlTexture(int plane)
+{
+    Q_UNUSED(plane);
+    if (isSharedMemory()) {
+        if (!m_shmTexture) {
+            m_shmTexture = new QOpenGLTexture(QOpenGLTexture::Target2D);
+            m_shmTexture->create();
+        }
+        if (m_textureDirty) {
+            m_textureDirty = false;
+            m_shmTexture->bind();
+            glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_NEAREST);
+            // TODO: partial texture upload
+            QImage image = this->image();
+            if (image.hasAlphaChannel()) {
+                if (image.format() != QImage::Format_RGBA8888)
+                    image = image.convertToFormat(QImage::Format_RGBA8888);
+                glTexImage2D(GL_TEXTURE_2D, 0, GL_RGBA, image.width(), image.height(), 0, GL_RGBA, GL_UNSIGNED_BYTE, image.constBits());
+            } else {
+                if (image.format() != QImage::Format_RGBX8888)
+                    image = image.convertToFormat(QImage::Format_RGBX8888);
+                glTexImage2D(GL_TEXTURE_2D, 0, GL_RGB, image.width(), image.height(), 0, GL_RGB, GL_UNSIGNED_BYTE, image.constBits());
+            }
+            //we can release the buffer after uploading, since we have a copy
+            if (isCommitted())
+                sendRelease();
+        }
+        return m_shmTexture;
+    }
+    return nullptr;
+}
+#endif
+
+}
+
+QT_END_NAMESPACE
diff --git a/src/compositor/wayland_wrapper/qwlsurfacebuffer_p.h b/src/compositor/wayland_wrapper/qwlclientbuffer_p.h
index e0a7d021f..72c5a56e3 100644
--- a/src/compositor/wayland_wrapper/qwlsurfacebuffer_p.h
+++ b/src/compositor/wayland_wrapper/qwlclientbuffer_p.h
@@ -34,8 +34,8 @@
 **
 ****************************************************************************/
 
-#ifndef SURFACEBUFFER_H
-#define SURFACEBUFFER_H
+#ifndef QWLCLIENTBUFFER_P_H
+#define QWLCLIENTBUFFER_P_H
 
 //
 //  W A R N I N G
@@ -63,83 +63,85 @@ QT_BEGIN_NAMESPACE
 class QWaylandClientBufferIntegration;
 class QWaylandBufferRef;
 class QWaylandCompositor;
+class QOpenGLTexture;
 
 namespace QtWayland {
 
 struct surface_buffer_destroy_listener
 {
     struct wl_listener listener;
-    class SurfaceBuffer *surfaceBuffer;
+    class ClientBuffer *surfaceBuffer;
 };
 
-class SurfaceBuffer
+class Q_WAYLAND_COMPOSITOR_EXPORT ClientBuffer
 {
 public:
-    SurfaceBuffer(QWaylandSurface *surface);
+    ClientBuffer(struct ::wl_resource *bufferResource);
 
-    ~SurfaceBuffer();
+    virtual ~ClientBuffer();
 
-    void initialize(struct ::wl_resource *bufferResource);
-    void destructBufferState();
+    virtual QWaylandBufferRef::BufferFormatEgl bufferFormatEgl() const;
+    virtual QSize size() const = 0;
+    virtual QWaylandSurface::Origin origin() const = 0;
 
-    inline bool isRegisteredWithBuffer() const { return m_is_registered_for_buffer; }
+    virtual void *lockNativeBuffer() { return nullptr; }
+    virtual void unlockNativeBuffer(void *native_buffer) const { Q_UNUSED(native_buffer); }
 
-    void sendRelease();
-    void disown();
-
-    void setDisplayed();
+    virtual QImage image() const { return QImage(); }
 
     inline bool isCommitted() const { return m_committed; }
-    inline void setCommitted() { m_committed = true; }
-    inline bool isDisplayed() const { return m_is_displayed; }
-
+    virtual void setCommitted(QRegion &damage);
     bool isDestroyed() { return m_destroyed; }
 
     inline struct ::wl_resource *waylandBufferHandle() const { return m_buffer; }
 
-    void setDestroyIfUnused(bool destroy);
-
-    QSize size() const;
-    QWaylandSurface::Origin origin() const;
     bool isSharedMemory() const { return wl_shm_buffer_get(m_buffer); }
 
-    QImage image() const;
-    QWaylandBufferRef::BufferFormatEgl bufferFormatEgl() const;
 #ifdef QT_WAYLAND_COMPOSITOR_GL
-    void bindToTexture() const;
-    uint textureForPlane(int plane) const;
-    void updateTexture() const;
+    virtual QOpenGLTexture *toOpenGlTexture(int plane = 0) = 0;
 #endif
 
-    static bool hasContent(SurfaceBuffer *buffer) { return buffer && buffer->waylandBufferHandle(); }
-private:
+    static bool hasContent(ClientBuffer *buffer) { return buffer && buffer->waylandBufferHandle(); }
+
+protected:
     void ref();
     void deref();
-    void destroyIfUnused();
+    void sendRelease();
+    void setDestroyed();
 
-    QWaylandSurface *m_surface;
-    QWaylandCompositor *m_compositor;
     struct ::wl_resource *m_buffer;
-    int m_bufferScale;
-    struct surface_buffer_destroy_listener m_destroy_listener;
+    QRegion m_damage;
+    bool m_textureDirty;
+
+private:
     bool m_committed;
-    bool m_is_registered_for_buffer;
-    bool m_surface_has_buffer;
     bool m_destroyed;
 
-    bool m_is_displayed;
-
     QAtomicInt m_refCount;
-    bool m_used;
-    bool m_destroyIfUnused;
-
-    static void destroy_listener_callback(wl_listener *listener, void *data);
 
     friend class ::QWaylandBufferRef;
+    friend class BufferManager;
+};
+
+class Q_WAYLAND_COMPOSITOR_EXPORT SharedMemoryBuffer : public ClientBuffer
+{
+public:
+    SharedMemoryBuffer(struct ::wl_resource *bufferResource);
+
+    QSize size() const Q_DECL_OVERRIDE;
+    QWaylandSurface::Origin origin() const  Q_DECL_OVERRIDE;
+    QImage image() const;
+
+#ifdef QT_WAYLAND_COMPOSITOR_GL
+    QOpenGLTexture *toOpenGlTexture(int plane = 0) Q_DECL_OVERRIDE;
+
+private:
+    QOpenGLTexture *m_shmTexture;
+#endif
 };
 
 }
 
 QT_END_NAMESPACE
 
-#endif // SURFACEBUFFER_H
+#endif // QWLCLIENTBUFFER_P_H
diff --git a/src/compositor/wayland_wrapper/qwlsurfacebuffer.cpp b/src/compositor/wayland_wrapper/qwlsurfacebuffer.cpp
deleted file mode 100644
index df2dded2f..000000000
--- a/src/compositor/wayland_wrapper/qwlsurfacebuffer.cpp
+++ /dev/null
@@ -1,261 +0,0 @@
-/****************************************************************************
-**
-** Copyright (C) 2015 The Qt Company Ltd.
-** Contact: http://www.qt.io/licensing/
-**
-** This file is part of the QtWaylandCompositor module of the Qt Toolkit.
-**
-** $QT_BEGIN_LICENSE:LGPL3$
-** Commercial License Usage
-** Licensees holding valid commercial Qt licenses may use this file in
-** accordance with the commercial license agreement provided with the
-** Software or, alternatively, in accordance with the terms contained in
-** a written agreement between you and The Qt Company. For licensing terms
-** and conditions see http://www.qt.io/terms-conditions. For further
-** information use the contact form at http://www.qt.io/contact-us.
-**
-** GNU Lesser General Public License Usage
-** Alternatively, this file may be used under the terms of the GNU Lesser
-** General Public License version 3 as published by the Free Software
-** Foundation and appearing in the file LICENSE.LGPLv3 included in the
-** packaging of this file. Please review the following information to
-** ensure the GNU Lesser General Public License version 3 requirements
-** will be met: https://www.gnu.org/licenses/lgpl.html.
-**
-** GNU General Public License Usage
-** Alternatively, this file may be used under the terms of the GNU
-** General Public License version 2.0 or later as published by the Free
-** Software Foundation and appearing in the file LICENSE.GPL included in
-** the packaging of this file. Please review the following information to
-** ensure the GNU General Public License version 2.0 requirements will be
-** met: http://www.gnu.org/licenses/gpl-2.0.html.
-**
-** $QT_END_LICENSE$
-**
-****************************************************************************/
-
-#include "qwlsurfacebuffer_p.h"
-
-#ifdef QT_WAYLAND_COMPOSITOR_GL
-#include "hardware_integration/qwlclientbufferintegration_p.h"
-#include <qpa/qplatformopenglcontext.h>
-#endif
-
-#include <QtCore/QDebug>
-
-#include <wayland-server-protocol.h>
-#include "qwaylandsharedmemoryformathelper_p.h"
-
-#include <QtWaylandCompositor/private/qwaylandcompositor_p.h>
-
-QT_BEGIN_NAMESPACE
-
-namespace QtWayland {
-
-SurfaceBuffer::SurfaceBuffer(QWaylandSurface *surface)
-    : m_surface(surface)
-    , m_compositor(surface->compositor())
-    , m_buffer(0)
-    , m_committed(false)
-    , m_is_registered_for_buffer(false)
-    , m_surface_has_buffer(false)
-    , m_destroyed(false)
-    , m_is_displayed(false)
-    , m_used(false)
-    , m_destroyIfUnused(false)
-{
-}
-
-SurfaceBuffer::~SurfaceBuffer()
-{
-    if (m_is_registered_for_buffer)
-        destructBufferState();
-}
-
-void SurfaceBuffer::initialize(struct ::wl_resource *buffer)
-{
-    m_buffer = buffer;
-    m_committed = false;
-    m_is_registered_for_buffer = true;
-    m_surface_has_buffer = true;
-    m_is_displayed = false;
-    m_destroyed = false;
-    m_destroy_listener.surfaceBuffer = this;
-    m_destroy_listener.listener.notify = destroy_listener_callback;
-    if (buffer) {
-#ifdef QT_WAYLAND_COMPOSITOR_GL
-        if (ClientBufferIntegration *integration = QWaylandCompositorPrivate::get(m_compositor)->clientBufferIntegration())
-            integration->initializeBuffer(buffer);
-#endif
-        wl_signal_add(&buffer->destroy_signal, &m_destroy_listener.listener);
-    }
-}
-
-void SurfaceBuffer::destructBufferState()
-{
-    if (m_buffer) {
-        if (m_committed)
-            sendRelease();
-        wl_list_remove(&m_destroy_listener.listener.link);
-    }
-    m_buffer = 0;
-    m_committed = false;
-    m_is_registered_for_buffer = false;
-    m_is_displayed = false;
-}
-
-void SurfaceBuffer::sendRelease()
-{
-    Q_ASSERT(m_buffer);
-    wl_buffer_send_release(m_buffer);
-}
-
-void SurfaceBuffer::disown()
-{
-    m_surface_has_buffer = false;
-    destructBufferState();
-    destroyIfUnused();
-}
-
-void SurfaceBuffer::setDisplayed()
-{
-    m_is_displayed = true;
-}
-
-void SurfaceBuffer::destroy_listener_callback(wl_listener *listener, void *data)
-{
-    Q_UNUSED(data);
-    struct surface_buffer_destroy_listener *destroy_listener =
-        reinterpret_cast<struct surface_buffer_destroy_listener *>(listener);
-    SurfaceBuffer *d = destroy_listener->surfaceBuffer;
-
-    // Mark the buffer as destroyed and clear m_buffer right away to avoid
-    // touching it before it is properly cleaned up.
-    d->m_destroyed = true;
-    d->m_buffer = 0;
-}
-
-void SurfaceBuffer::ref()
-{
-    m_used = m_refCount.ref();
-}
-
-void SurfaceBuffer::deref()
-{
-    m_used = m_refCount.deref();
-    if (!m_used)
-        disown();
-}
-
-void SurfaceBuffer::setDestroyIfUnused(bool destroy)
-{
-    m_destroyIfUnused = destroy;
-    destroyIfUnused();
-}
-
-void SurfaceBuffer::destroyIfUnused()
-{
-    if (!m_used && m_destroyIfUnused)
-        delete this;
-}
-
-QSize SurfaceBuffer::size() const
-{
-    if (!m_buffer)
-        return QSize();
-
-    if (wl_shm_buffer *shmBuffer = wl_shm_buffer_get(m_buffer)) {
-        int width = wl_shm_buffer_get_width(shmBuffer);
-        int height = wl_shm_buffer_get_height(shmBuffer);
-        return QSize(width, height);
-    }
-#ifdef QT_WAYLAND_COMPOSITOR_GL
-    if (ClientBufferIntegration *integration = QWaylandCompositorPrivate::get(m_compositor)->clientBufferIntegration()) {
-        return integration->bufferSize(m_buffer);
-    }
-#endif
-
-    return QSize();
-}
-
-QWaylandSurface::Origin SurfaceBuffer::origin() const
-{
-    if (isSharedMemory()) {
-        return QWaylandSurface::OriginTopLeft;
-    }
-
-#ifdef QT_WAYLAND_COMPOSITOR_GL
-    if (ClientBufferIntegration *integration = QWaylandCompositorPrivate::get(m_compositor)->clientBufferIntegration()) {
-        return integration->origin(m_buffer);
-    }
-#endif
-    return QWaylandSurface::OriginTopLeft;
-}
-
-QImage SurfaceBuffer::image() const
-{
-    if (wl_shm_buffer *shmBuffer = wl_shm_buffer_get(m_buffer)) {
-        int width = wl_shm_buffer_get_width(shmBuffer);
-        int height = wl_shm_buffer_get_height(shmBuffer);
-        int bytesPerLine = wl_shm_buffer_get_stride(shmBuffer);
-        uchar *data = static_cast<uchar *>(wl_shm_buffer_get_data(shmBuffer));
-        return QImage(data, width, height, bytesPerLine, QImage::Format_ARGB32_Premultiplied);
-    }
-
-    return QImage();
-}
-
-QWaylandBufferRef::BufferFormatEgl SurfaceBuffer::bufferFormatEgl() const
-{
-    Q_ASSERT(isSharedMemory() == false);
-
-#ifdef QT_WAYLAND_COMPOSITOR_GL
-    if (QtWayland::ClientBufferIntegration *clientInt = QWaylandCompositorPrivate::get(m_compositor)->clientBufferIntegration())
-        return clientInt->bufferFormat(m_buffer);
-#endif
-
-    return QWaylandBufferRef::BufferFormatEgl_Null;
-}
-
-#ifdef QT_WAYLAND_COMPOSITOR_GL
-void SurfaceBuffer::bindToTexture() const
-{
-    Q_ASSERT(m_compositor);
-    if (isSharedMemory()) {
-        QImage image = this->image();
-        if (image.hasAlphaChannel()) {
-            if (image.format() != QImage::Format_RGBA8888) {
-                image = image.convertToFormat(QImage::Format_RGBA8888);
-            }
-            glTexImage2D(GL_TEXTURE_2D, 0, GL_RGBA, image.width(), image.height(), 0, GL_RGBA, GL_UNSIGNED_BYTE, image.constBits());
-        } else {
-            if (image.format() != QImage::Format_RGBX8888) {
-                image = image.convertToFormat(QImage::Format_RGBX8888);
-            }
-            glTexImage2D(GL_TEXTURE_2D, 0, GL_RGB, image.width(), image.height(), 0, GL_RGB, GL_UNSIGNED_BYTE, image.constBits());
-        }
-    } else {
-        if (QtWayland::ClientBufferIntegration *clientInt = QWaylandCompositorPrivate::get(m_compositor)->clientBufferIntegration()) {
-            clientInt->bindTextureToBuffer(m_buffer);
-        }
-    }
-}
-
-uint SurfaceBuffer::textureForPlane(int plane) const
-{
-    if (QtWayland::ClientBufferIntegration *clientInt = QWaylandCompositorPrivate::get(m_compositor)->clientBufferIntegration())
-        return clientInt->textureForBuffer(m_buffer, plane);
-
-    return 0;
-}
-
-void SurfaceBuffer::updateTexture() const
-{
-    if (QtWayland::ClientBufferIntegration *clientInt = QWaylandCompositorPrivate::get(m_compositor)->clientBufferIntegration())
-        clientInt->updateTextureForBuffer(m_buffer);
-}
-#endif
-
-}
-
-QT_END_NAMESPACE
diff --git a/src/compositor/wayland_wrapper/wayland_wrapper.pri b/src/compositor/wayland_wrapper/wayland_wrapper.pri
index f338ddc05..fa5d76291 100644
--- a/src/compositor/wayland_wrapper/wayland_wrapper.pri
+++ b/src/compositor/wayland_wrapper/wayland_wrapper.pri
@@ -3,21 +3,23 @@ WAYLANDSERVERSOURCES += \
     ../3rdparty/protocol/wayland.xml \
 
 HEADERS += \
+    wayland_wrapper/qwlbuffermanager_p.h \
+    wayland_wrapper/qwlclientbuffer_p.h \
     wayland_wrapper/qwldatadevice_p.h \
     wayland_wrapper/qwldatadevicemanager_p.h \
     wayland_wrapper/qwldataoffer_p.h \
     wayland_wrapper/qwldatasource_p.h \
     wayland_wrapper/qwlregion_p.h \
-    wayland_wrapper/qwlsurfacebuffer_p.h \
     ../shared/qwaylandxkb_p.h \
 
 SOURCES += \
+    wayland_wrapper/qwlbuffermanager.cpp \
+    wayland_wrapper/qwlclientbuffer.cpp \
     wayland_wrapper/qwldatadevice.cpp \
     wayland_wrapper/qwldatadevicemanager.cpp \
     wayland_wrapper/qwldataoffer.cpp \
     wayland_wrapper/qwldatasource.cpp \
     wayland_wrapper/qwlregion.cpp \
-    wayland_wrapper/qwlsurfacebuffer.cpp \
     ../shared/qwaylandxkb.cpp \
 
 INCLUDEPATH += wayland_wrapper