/****************************************************************************

**

** Copyright (C) 2010 Nokia Corporation and/or its subsidiary(-ies).

** All rights reserved.

** Contact: Nokia Corporation (qt-info@nokia.com)

**

** This file is part of the QtOpenGL module of the Qt Toolkit.

**

** $QT_BEGIN_LICENSE:LGPL$

** No Commercial Usage

** This file contains pre-release code and may not be distributed.

** You may use this file in accordance with the terms and conditions

** contained in the Technology Preview License Agreement accompanying

** this package.

**

** GNU Lesser General Public License Usage

** Alternatively, this file may be used under the terms of the GNU Lesser

** General Public License version 2.1 as published by the Free Software

** Foundation and appearing in the file LICENSE.LGPL included in the

** packaging of this file.  Please review the following information to

** ensure the GNU Lesser General Public License version 2.1 requirements

** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.

**

** In addition, as a special exception, Nokia gives you certain additional

** rights.  These rights are described in the Nokia Qt LGPL Exception

** version 1.1, included in the file LGPL_EXCEPTION.txt in this package.

**

** If you have questions regarding the use of this file, please contact

** Nokia at qt-info@nokia.com.

**

**

**

**

**

**

**

**

** $QT_END_LICENSE$

**

****************************************************************************/

#include "qpixmap.h"

#include "qglframebufferobject.h"

#include <private/qpaintengine_raster_p.h>

#include "qpixmapdata_gl_p.h"

#include <private/qgl_p.h>

#include <private/qdrawhelper_p.h>

#include <private/qimage_p.h>

#include <private/qpaintengineex_opengl2_p.h>

#include <qdesktopwidget.h>

#include <qfile.h>

#include <qimagereader.h>

QT_BEGIN_NAMESPACE

extern QGLWidget* qt_gl_share_widget();

/*!

    \class QGLFramebufferObjectPool

    \since 4.6

    \brief The QGLFramebufferObject class provides a pool of framebuffer

    objects for offscreen rendering purposes.

    When requesting an FBO of a given size and format, an FBO of the same

    format and a size at least as big as the requested size will be returned.

    \internal

*/

static inline int areaDiff(const QSize &size, const QGLFramebufferObject *fbo)

{

    return qAbs(size.width() * size.height() - fbo->width() * fbo->height());

}

extern int qt_next_power_of_two(int v);

static inline QSize maybeRoundToNextPowerOfTwo(const QSize &sz)

{

#ifdef QT_OPENGL_ES_2

    QSize rounded(qt_next_power_of_two(sz.width()), qt_next_power_of_two(sz.height()));

    if (rounded.width() * rounded.height() < 1.20 * sz.width() * sz.height())

        return rounded;

#endif

    return sz;

}

QGLFramebufferObject *QGLFramebufferObjectPool::acquire(const QSize &requestSize, const QGLFramebufferObjectFormat &requestFormat, bool strictSize)

{

    QGLFramebufferObject *chosen = 0;

    QGLFramebufferObject *candidate = 0;

    for (int i = 0; !chosen && i < m_fbos.size(); ++i) {

        QGLFramebufferObject *fbo = m_fbos.at(i);

        if (strictSize) {

            if (fbo->size() == requestSize && fbo->format() == requestFormat) {

                chosen = fbo;

                break;

            } else {

                continue;

            }

        }

        if (fbo->format() == requestFormat) {

            // choose the fbo with a matching format and the closest size

            if (!candidate || areaDiff(requestSize, candidate) > areaDiff(requestSize, fbo))

                candidate = fbo;

        }

        if (candidate) {

            m_fbos.removeOne(candidate);

            const QSize fboSize = candidate->size();

            QSize sz = fboSize;

            if (sz.width() < requestSize.width())

                sz.setWidth(qMax(requestSize.width(), qRound(sz.width() * 1.5)));

            if (sz.height() < requestSize.height())

                sz.setHeight(qMax(requestSize.height(), qRound(sz.height() * 1.5)));

            // wasting too much space?

            if (sz.width() * sz.height() > requestSize.width() * requestSize.height() * 4)

                sz = requestSize;

            if (sz != fboSize) {

                delete candidate;

                candidate = new QGLFramebufferObject(maybeRoundToNextPowerOfTwo(sz), requestFormat);

            }

            chosen = candidate;

        }

    }

    if (!chosen) {

        if (strictSize)

            chosen = new QGLFramebufferObject(requestSize, requestFormat);

        else

            chosen = new QGLFramebufferObject(maybeRoundToNextPowerOfTwo(requestSize), requestFormat);

    }

    if (!chosen->isValid()) {

        delete chosen;

        chosen = 0;

    }

    return chosen;

}

void QGLFramebufferObjectPool::release(QGLFramebufferObject *fbo)

{

    if (fbo)

        m_fbos << fbo;

}

QPaintEngine* QGLPixmapGLPaintDevice::paintEngine() const

{

    return data->paintEngine();

}

void QGLPixmapGLPaintDevice::beginPaint()

{

    if (!data->isValid())

        return;

    // QGLPaintDevice::beginPaint will store the current binding and replace

    // it with m_thisFBO:

    m_thisFBO = data->m_renderFbo->handle();

    QGLPaintDevice::beginPaint();

    Q_ASSERT(data->paintEngine()->type() == QPaintEngine::OpenGL2);

    // QPixmap::fill() is deferred until now, where we actually need to do the fill:

    if (data->needsFill()) {

        const QColor &c = data->fillColor();

        float alpha = c.alphaF();

        glDisable(GL_SCISSOR_TEST);

        glClearColor(c.redF() * alpha, c.greenF() * alpha, c.blueF() * alpha, alpha);

        glClear(GL_COLOR_BUFFER_BIT);

    }

    else if (!data->isUninitialized()) {

        // If the pixmap (GL Texture) has valid content (it has been

        // uploaded from an image or rendered into before), we need to

        // copy it from the texture to the render FBO.

        glDisable(GL_DEPTH_TEST);

        glDisable(GL_SCISSOR_TEST);

        glDisable(GL_BLEND);

#if !defined(QT_OPENGL_ES_2)

        glMatrixMode(GL_MODELVIEW);

        glLoadIdentity();

        glMatrixMode(GL_PROJECTION);

        glLoadIdentity();

        glOrtho(0, data->width(), data->height(), 0, -999999, 999999);

#endif

        glViewport(0, 0, data->width(), data->height());

        // Pass false to bind so it doesn't copy the FBO into the texture!

        context()->drawTexture(QRect(0, 0, data->width(), data->height()), data->bind(false));

    }

}

void QGLPixmapGLPaintDevice::endPaint()

{

    if (!data->isValid())

        return;

    data->copyBackFromRenderFbo(false);

    // Base's endPaint will restore the previous FBO binding

    QGLPaintDevice::endPaint();

    qgl_fbo_pool()->release(data->m_renderFbo);

    data->m_renderFbo = 0;

}

QGLContext* QGLPixmapGLPaintDevice::context() const

{

    data->ensureCreated();

    return data->m_ctx;

}

QSize QGLPixmapGLPaintDevice::size() const

{

    return data->size();

}

bool QGLPixmapGLPaintDevice::alphaRequested() const

{

    return data->m_hasAlpha;

}

void QGLPixmapGLPaintDevice::setPixmapData(QGLPixmapData* d)

{

    data = d;

}

static int qt_gl_pixmap_serial = 0;

QGLPixmapData::QGLPixmapData(PixelType type)

    : QPixmapData(type, OpenGLClass)

    , m_renderFbo(0)

    , m_engine(0)

    , m_ctx(0)

    , m_dirty(false)

    , m_hasFillColor(false)

    , m_hasAlpha(false)

{

    setSerialNumber(++qt_gl_pixmap_serial);

    m_glDevice.setPixmapData(this);

}

QGLPixmapData::~QGLPixmapData()

{

    QGLWidget *shareWidget = qt_gl_share_widget();

    if (!shareWidget)

        return;

    delete m_engine;

    if (m_texture.id) {

        QGLShareContextScope ctx(shareWidget->context());

        glDeleteTextures(1, &m_texture.id);

    }

}

QPixmapData *QGLPixmapData::createCompatiblePixmapData() const

{

    return new QGLPixmapData(pixelType());

}

bool QGLPixmapData::isValid() const

{

    return w > 0 && h > 0;

}

bool QGLPixmapData::isValidContext(const QGLContext *ctx) const

{

    if (ctx == m_ctx)

        return true;

    const QGLContext *share_ctx = qt_gl_share_widget()->context();

    return ctx == share_ctx || QGLContext::areSharing(ctx, share_ctx);

}

void QGLPixmapData::resize(int width, int height)

{

    if (width == w && height == h)

        return;

    if (width <= 0 || height <= 0) {

        width = 0;

        height = 0;

    }

    w = width;

    h = height;

    is_null = (w <= 0 || h <= 0);

    d = pixelType() == QPixmapData::PixmapType ? 32 : 1;

    if (m_texture.id) {

        QGLShareContextScope ctx(qt_gl_share_widget()->context());

        glDeleteTextures(1, &m_texture.id);

        m_texture.id = 0;

    }

    m_source = QImage();

    m_dirty = isValid();

    setSerialNumber(++qt_gl_pixmap_serial);

}

void QGLPixmapData::ensureCreated() const

{

    if (!m_dirty)

        return;

    m_dirty = false;

    QGLShareContextScope ctx(qt_gl_share_widget()->context());

    m_ctx = ctx;

    const GLenum internal_format = m_hasAlpha ? GL_RGBA : GL_RGB;

#ifdef QT_OPENGL_ES_2

    const GLenum external_format = internal_format;

#else

    const GLenum external_format = qt_gl_preferredTextureFormat();

#endif

    const GLenum target = GL_TEXTURE_2D;

    if (!m_texture.id) {

        glGenTextures(1, &m_texture.id);

        glBindTexture(target, m_texture.id);

        glTexImage2D(target, 0, internal_format, w, h, 0, external_format, GL_UNSIGNED_BYTE, 0);

        glTexParameterf(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_NEAREST);

        glTexParameterf(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_NEAREST);

    }

    if (!m_source.isNull()) {

        if (external_format == GL_RGB) {

            const QImage tx = m_source.convertToFormat(QImage::Format_RGB888).mirrored(false, true);

            glBindTexture(target, m_texture.id);

            glTexSubImage2D(target, 0, 0, 0, w, h, external_format,

                            GL_UNSIGNED_BYTE, tx.bits());

        } else {

            const QImage tx = ctx->d_func()->convertToGLFormat(m_source, true, external_format);

            glBindTexture(target, m_texture.id);

            glTexSubImage2D(target, 0, 0, 0, w, h, external_format,

                            GL_UNSIGNED_BYTE, tx.bits());

        }

        if (useFramebufferObjects())

            m_source = QImage();

    }

    m_texture.options &= ~QGLContext::MemoryManagedBindOption;

}

void QGLPixmapData::fromImage(const QImage &image,

                              Qt::ImageConversionFlags /*flags*/)

{

    if (image.size() == QSize(w, h))

        setSerialNumber(++qt_gl_pixmap_serial);

    resize(image.width(), image.height());

    if (pixelType() == BitmapType) {

        m_source = image.convertToFormat(QImage::Format_MonoLSB);

    } else {

        QImage::Format format = QImage::Format_RGB32;

        if (qApp->desktop()->depth() == 16)

            format = QImage::Format_RGB16;

        if (image.hasAlphaChannel() && const_cast<QImage &>(image).data_ptr()->checkForAlphaPixels())

            format = QImage::Format_ARGB32_Premultiplied;;

        m_source = image.convertToFormat(format);

    }

    m_dirty = true;

    m_hasFillColor = false;

    m_hasAlpha = m_source.hasAlphaChannel();

    w = image.width();

    h = image.height();

    is_null = (w <= 0 || h <= 0);

    d = m_source.depth();

    if (m_texture.id) {

        QGLShareContextScope ctx(qt_gl_share_widget()->context());

        glDeleteTextures(1, &m_texture.id);

        m_texture.id = 0;

    }

}

bool QGLPixmapData::fromFile(const QString &filename, const char *format,

                             Qt::ImageConversionFlags flags)

{

    if (pixelType() == QPixmapData::BitmapType)

        return QPixmapData::fromFile(filename, format, flags);

    QFile file(filename);

    if (!file.open(QIODevice::ReadOnly))

        return false;

    QByteArray data = file.peek(64);

    bool alpha;

    if (m_texture.canBindCompressedTexture

            (data.constData(), data.size(), format, &alpha)) {

        resize(0, 0);

        data = file.readAll();

        file.close();

        QGLShareContextScope ctx(qt_gl_share_widget()->context());

        QSize size = m_texture.bindCompressedTexture

            (data.constData(), data.size(), format);

        if (!size.isEmpty()) {

            w = size.width();

            h = size.height();

            is_null = false;

            d = 32;

            m_hasAlpha = alpha;

            m_source = QImage();

            m_dirty = isValid();

            return true;

        }

        return false;

    }

    fromImage(QImageReader(&file, format).read(), flags);

    return !isNull();

}

bool QGLPixmapData::fromData(const uchar *buffer, uint len, const char *format,

                             Qt::ImageConversionFlags flags)

{

    bool alpha;

    const char *buf = reinterpret_cast<const char *>(buffer);

    if (m_texture.canBindCompressedTexture(buf, int(len), format, &alpha)) {

        resize(0, 0);

        QGLShareContextScope ctx(qt_gl_share_widget()->context());

        QSize size = m_texture.bindCompressedTexture(buf, int(len), format);

        if (!size.isEmpty()) {

            w = size.width();

            h = size.height();

            is_null = false;

            d = 32;

            m_hasAlpha = alpha;

            m_source = QImage();

            m_dirty = isValid();

            return true;

        }

    }

    return QPixmapData::fromData(buffer, len, format, flags);

}

bool QGLPixmapData::scroll(int dx, int dy, const QRect &rect)

{

    Q_UNUSED(dx);

    Q_UNUSED(dy);

    Q_UNUSED(rect);

    return false;

}

void QGLPixmapData::copy(const QPixmapData *data, const QRect &rect)

{

    if (data->classId() != QPixmapData::OpenGLClass) {

        QPixmapData::copy(data, rect);

        return;

    }

    // can be optimized to do a framebuffer blit or similar ...

    QPixmapData::copy(data, rect);

}

void QGLPixmapData::fill(const QColor &color)

{

    if (!isValid())

        return;

    bool hasAlpha = color.alpha() != 255;

    if (hasAlpha && !m_hasAlpha) {

        if (m_texture.id) {

            glDeleteTextures(1, &m_texture.id);

            m_texture.id = 0;

            m_dirty = true;

        }

        m_hasAlpha = color.alpha() != 255;

    }

    if (useFramebufferObjects()) {

        m_source = QImage();

        m_hasFillColor = true;

        m_fillColor = color;

    } else {

        if (m_source.isNull()) {

            m_fillColor = color;

            m_hasFillColor = true;

        } else if (m_source.depth() == 32) {

            m_source.fill(PREMUL(color.rgba()));

        } else if (m_source.depth() == 1) {

            if (color == Qt::color1)

                m_source.fill(1);

            else

                m_source.fill(0);

        }

    }

}

bool QGLPixmapData::hasAlphaChannel() const

{

    return m_hasAlpha;

}

QImage QGLPixmapData::fillImage(const QColor &color) const

{

    QImage img;

    if (pixelType() == BitmapType) {

        img = QImage(w, h, QImage::Format_MonoLSB);