ui: Handle float images in dumps.

[apitrace] / gui / apisurface.cpp

diff --git a/gui/apisurface.cpp b/gui/apisurface.cpp
index c875c6ff432662fcbedc267ae41f8451034abfda..3732ed5f717482ad976fbf6f9f0d9e447113b91c 100644 (file)
--- a/gui/apisurface.cpp
+++ b/gui/apisurface.cpp
@@ -1,8 +1,14 @@
 #include "apisurface.h"
+#include "thumbnail.h"
+
+#include <sstream>
 
 #include <QDebug>
 #include <QSysInfo>
 
+#include "image/image.hpp"
+
+
 ApiSurface::ApiSurface()
 {
 }
@@ -17,79 +23,95 @@ void ApiSurface::setSize(const QSize &size)
     m_size = size;
 }
 
-int ApiSurface::numChannels() const
+struct ByteArrayBuf : public std::streambuf
 {
-    return m_numChannels;
-}
+    ByteArrayBuf(QByteArray & a)
+    {
+        setg(a.data(), a.data(), a.data() + a.size());
+    }
+};
 
-void ApiSurface::setNumChannels(int numChannels)
+void ApiSurface::contentsFromBase64(const QByteArray &base64)
 {
-    m_numChannels = numChannels;
-}
+    QByteArray dataArray = QByteArray::fromBase64(base64);
 
-static inline int
-rgba8_to_argb(quint8 r, quint8 g, quint8 b, quint8 a)
-{
-    return (a << 24 | r << 16 | g << 8 | b);
-}
+    image::Image *image;
 
-static inline int
-rgbaf2argb(float r, float g, float b, float a)
-{
-    quint8 rb = r * 255;
-    quint8 gb = g * 255;
-    quint8 bb = b * 255;
-    quint8 ab = a * 255;
+    /*
+     * Detect the PNG vs PFM images.
+     */
+    const char pngSignature[] = {(char)0x89, 0x50, 0x4E, 0x47, 0x0D, 0x0A, 0x1A, 0x0A, 0};
+    if (dataArray.startsWith(pngSignature)) {
+        ByteArrayBuf buf(dataArray);
+        std::istream istr(&buf);
+        image = image::readPNG(istr);
+    } else {
+        image = image::readPNM(dataArray.data(), dataArray.size());
+    }
 
-    return (ab << 24 | rb << 16 | gb << 8 | bb);
-}
+    /*
+     * FIXME: Instead of converting to QImage here, we should be deferring the conversion
+     * to imageviewer.cpp.
+     *
+     * XXX: We still need the thumbnail though.
+     */
 
-void ApiSurface::contentsFromBase64(const QByteArray &base64)
-{
-    QByteArray dataArray = QByteArray::fromBase64(base64);
-    const quint8 *data = (const quint8*)dataArray.data();
-    int width = m_size.width();
-    int height = m_size.height();
+    Q_ASSERT(image);
 
-    if (width <= 0 || height <= 0)
-        return;
+    int width = image->width;
+    int height = image->height;
 
-    int *pixelData = (int*)malloc(sizeof(int) * width * height);
+    m_image = QImage(width, height, QImage::Format_ARGB32);
 
-    //XXX not sure if this will work when
-    //    QSysInfo::ByteOrder == QSysInfo::BigEndian
+    const unsigned char *srcRow = image->start();
+    for (int y = 0; y < height; ++y) {
+        QRgb *dst = (QRgb *)m_image.scanLine(y);
 
-    if (m_numChannels == 4) {
-        for (int y = 0; y < height; ++y) {
+        if (image->channelType == image::TYPE_UNORM8) {
+            const unsigned char *src = srcRow;
             for (int x = 0; x < width; ++x) {
-                int pixel = rgba8_to_argb(data[(y * width + x) * 4 + 0],
-                                          data[(y * width + x) * 4 + 1],
-                                          data[(y * width + x) * 4 + 2],
-                                          data[(y * width + x) * 4 + 3]);
-                pixelData[y * width + x] = pixel;
+                unsigned char rgba[4];
+                for (int c = 0; c < image->channels; ++c) {
+                    rgba[c] = *src++;
+                }
+                if (image->channels == 1) {
+                    // Use gray-scale instead of red
+                    rgba[1] = rgba[0];
+                    rgba[2] = rgba[0];
+                }
+                dst[x] = qRgba(rgba[0], rgba[1], rgba[2], rgba[3]);
             }
-        }
-    } else if (m_numChannels == 1) {
-        for (int y = 0; y < height; ++y) {
+        } else {
+            const float *src = (const float *)srcRow;
             for (int x = 0; x < width; ++x) {
-                int pixel = rgba8_to_argb(data[y * width + x],
-                                          data[y * width + x],
-                                          data[y * width + x],
-                                          255);
-                pixelData[y * width + x] = pixel;
+                unsigned char rgba[4] = {0, 0, 0, 0xff};
+                for (int c = 0; c < image->channels; ++c) {
+                    float f = *src++;
+                    unsigned char u;
+                    if (f >= 1.0f) {
+                        u = 255;
+                    } else if (f <= 0.0f) {
+                        u = 0;
+                    } else {
+                        u = f * 255 + 0.5;
+                    }
+                    rgba[c] = u;
+                }
+                if (image->channels == 1) {
+                    // Use gray-scale instead of red
+                    rgba[1] = rgba[0];
+                    rgba[2] = rgba[0];
+                }
+                dst[x] = qRgba(rgba[0], rgba[1], rgba[2], rgba[3]);
             }
         }
-    } else {
-        Q_ASSERT(0);
+
+        srcRow += image->stride();
     }
 
-    m_image = QImage((uchar*)pixelData,
-                     width, height,
-                     QImage::Format_ARGB32).mirrored();
-    m_thumb = m_image.scaled(64, 64, Qt::KeepAspectRatio);
-    //m_image.save("testoutput.png");
+    delete image;
 
-    free(pixelData);
+    m_thumb = thumbnail(m_image);
 }
 
 QImage ApiSurface::image() const
@@ -102,41 +124,40 @@ QImage ApiSurface::thumb() const
     return m_thumb;
 }
 
-ApiTexture::ApiTexture()
-    : ApiSurface(),
-      m_unit(0),
-      m_level(0)
+int ApiSurface::depth() const
 {
+    return m_depth;
 }
 
-int ApiTexture::unit() const
+void ApiSurface::setDepth(int depth)
 {
-    return m_unit;
+    m_depth = depth;
 }
 
-void ApiTexture::setUnit(int un)
+QString ApiSurface::formatName() const
 {
-    m_unit = un;
+    return m_formatName;
 }
 
-QString ApiTexture::target() const
+void ApiSurface::setFormatName(const QString &str)
 {
-    return m_target;
+    m_formatName = str;
 }
 
-void ApiTexture::setTarget(const QString &str)
+
+ApiTexture::ApiTexture()
+    : ApiSurface()
 {
-    m_target = str;
 }
 
-int ApiTexture::level() const
+QString ApiTexture::label() const
 {
-    return m_level;
+    return m_label;
 }
 
-void ApiTexture::setLevel(int l)
+void ApiTexture::setLabel(const QString &str)
 {
-    m_level = l;
+    m_label = str;
 }
 
 ApiFramebuffer::ApiFramebuffer()
@@ -153,3 +174,4 @@ void ApiFramebuffer::setType(const QString &str)
 {
     m_type = str;
 }
+