Texture to surface

API for setting a texture to surface.

Task-number: QTRD-3232
Change-Id: Icd9de61882b54b9c1fc84a742e49980926ca681d
Reviewed-by: Miikka Heikkinen <miikka.heikkinen@digia.com>

3 files changed, 91 insertions, 4 deletions

diff --git a/src/datavisualization/utils/abstractobjecthelper_p.h b/src/datavisualization/utils/abstractobjecthelper_p.h
index c1618909..b12bc474 100644
--- a/src/datavisualization/utils/abstractobjecthelper_p.h
+++ b/src/datavisualization/utils/abstractobjecthelper_p.h
@@ -42,7 +42,7 @@ public:
 
     GLuint vertexBuf();
     GLuint normalBuf();
-    GLuint uvBuf();
+    virtual GLuint uvBuf();
     GLuint elementBuf();
     GLuint indexCount();
     GLuint indicesType();
diff --git a/src/datavisualization/utils/surfaceobject.cpp b/src/datavisualization/utils/surfaceobject.cpp
index a98f2f2a..f8dae6f4 100644
--- a/src/datavisualization/utils/surfaceobject.cpp
+++ b/src/datavisualization/utils/surfaceobject.cpp
@@ -31,8 +31,8 @@ SurfaceObject::SurfaceObject(Surface3DRenderer *renderer)
       m_axisCacheX(renderer->m_axisCacheX),
       m_axisCacheY(renderer->m_axisCacheY),
       m_axisCacheZ(renderer->m_axisCacheZ),
-      m_renderer(renderer)
-
+      m_renderer(renderer),
+      m_returnTextureBuffer(false)
 {
     m_indicesType = GL_UNSIGNED_INT;
     initializeOpenGLFunctions();
@@ -41,12 +41,15 @@ SurfaceObject::SurfaceObject(Surface3DRenderer *renderer)
     glGenBuffers(1, &m_uvbuffer);
     glGenBuffers(1, &m_elementbuffer);
     glGenBuffers(1, &m_gridElementbuffer);
+    glGenBuffers(1, &m_uvTextureBuffer);
 }
 
 SurfaceObject::~SurfaceObject()
 {
-    if (QOpenGLContext::currentContext())
+    if (QOpenGLContext::currentContext()) {
         glDeleteBuffers(1, &m_gridElementbuffer);
+        glDeleteBuffers(1, &m_uvTextureBuffer);
+    }
 }
 
 void SurfaceObject::setUpSmoothData(const QSurfaceDataArray &dataArray, const QRect &space,
@@ -135,6 +138,37 @@ void SurfaceObject::setUpSmoothData(const QSurfaceDataArray &dataArray, const QR
     createBuffers(m_vertices, uvs, m_normals, 0, changeGeometry);
 }
 
+void SurfaceObject::smoothUVs(const QSurfaceDataArray &dataArray,
+                              const QSurfaceDataArray &modelArray)
+{
+    int columns = dataArray.at(0)->size();
+    int rows = dataArray.size();
+    float xRangeNormalizer = dataArray.at(0)->at(columns - 1).x() - dataArray.at(0)->at(0).x();
+    float zRangeNormalizer = dataArray.at(rows - 1)->at(0).z() - dataArray.at(0)->at(0).z();
+    float xMin = dataArray.at(0)->at(0).x();
+    float zMin = dataArray.at(0)->at(0).z();
+
+    QVector<QVector2D> uvs;
+    uvs.resize(m_rows * m_columns);
+    int index = 0;
+    for (int i = 0; i < m_rows; i++) {
+        float y = (modelArray.at(i)->at(0).z() - zMin) / zRangeNormalizer;
+        const QSurfaceDataRow &p = *modelArray.at(i);
+        for (int j = 0; j < m_columns; j++) {
+            float x = (p.at(j).x() - xMin) / xRangeNormalizer;
+            uvs[index] = QVector2D(x, y);
+            index++;
+        }
+    }
+
+    glBindBuffer(GL_ARRAY_BUFFER, m_uvTextureBuffer);
+    glBufferData(GL_ARRAY_BUFFER, uvs.size() * sizeof(QVector2D),
+                 &uvs.at(0), GL_STATIC_DRAW);
+    glBindBuffer(GL_ARRAY_BUFFER, 0);
+
+    m_returnTextureBuffer = true;
+}
+
 void SurfaceObject::updateSmoothRow(const QSurfaceDataArray &dataArray, int rowIndex, bool polar)
 {
     // Update vertices
@@ -426,6 +460,42 @@ void SurfaceObject::setUpData(const QSurfaceDataArray &dataArray, const QRect &s
     delete[] indices;
 }
 
+void SurfaceObject::coarseUVs(const QSurfaceDataArray &dataArray,
+                              const QSurfaceDataArray &modelArray)
+{
+    int columns = dataArray.at(0)->size();
+    int rows = dataArray.size();
+    float xRangeNormalizer = dataArray.at(0)->at(columns - 1).x() - dataArray.at(0)->at(0).x();
+    float zRangeNormalizer = dataArray.at(rows - 1)->at(0).z() - dataArray.at(0)->at(0).z();
+    float xMin = dataArray.at(0)->at(0).x();
+    float zMin = dataArray.at(0)->at(0).z();
+
+    QVector<QVector2D> uvs;
+    uvs.resize(m_rows * m_columns * 2);
+    int index = 0;
+    int colLimit = m_columns - 1;
+    for (int i = 0; i < m_rows; i++) {
+        float y = (modelArray.at(i)->at(0).z() - zMin) / zRangeNormalizer;
+        const QSurfaceDataRow &p = *modelArray.at(i);
+        for (int j = 0; j < m_columns; j++) {
+            float x = (p.at(j).x() - xMin) / xRangeNormalizer;
+            uvs[index] = QVector2D(x, y);
+            index++;
+            if (j > 0 && j < colLimit) {
+                uvs[index] = uvs[index - 1];
+                index++;
+            }
+        }
+    }
+
+    glBindBuffer(GL_ARRAY_BUFFER, m_uvTextureBuffer);
+    glBufferData(GL_ARRAY_BUFFER, uvs.size() * sizeof(QVector2D),
+                 &uvs.at(0), GL_STATIC_DRAW);
+    glBindBuffer(GL_ARRAY_BUFFER, 0);
+
+    m_returnTextureBuffer = true;
+}
+
 void SurfaceObject::updateCoarseRow(const QSurfaceDataArray &dataArray, int rowIndex, bool polar)
 {
     int colLimit = m_columns - 1;
@@ -684,6 +754,17 @@ GLuint SurfaceObject::gridElementBuf()
     return m_gridElementbuffer;
 }
 
+GLuint SurfaceObject::uvBuf()
+{
+    if (!m_meshDataLoaded)
+        qFatal("No loaded object");
+
+    if (m_returnTextureBuffer)
+        return m_uvTextureBuffer;
+    else
+        return m_uvbuffer;
+}
+
 GLuint SurfaceObject::gridIndexCount()
 {
     return m_gridIndexCount;
diff --git a/src/datavisualization/utils/surfaceobject_p.h b/src/datavisualization/utils/surfaceobject_p.h
index b3ba80e9..3266c626 100644
--- a/src/datavisualization/utils/surfaceobject_p.h
+++ b/src/datavisualization/utils/surfaceobject_p.h
@@ -57,6 +57,8 @@ public:
                    bool changeGeometry, bool polar, bool flipXZ = false);
     void setUpSmoothData(const QSurfaceDataArray &dataArray, const QRect &space,
                          bool changeGeometry, bool polar, bool flipXZ = false);
+    void smoothUVs(const QSurfaceDataArray &dataArray, const QSurfaceDataArray &modelArray);
+    void coarseUVs(const QSurfaceDataArray &dataArray, const QSurfaceDataArray &modelArray);
     void updateCoarseRow(const QSurfaceDataArray &dataArray, int rowIndex, bool polar);
     void updateSmoothRow(const QSurfaceDataArray &dataArray, int startRow, bool polar);
     void updateSmoothItem(const QSurfaceDataArray &dataArray, int row, int column, bool polar);
@@ -67,11 +69,13 @@ public:
     void createCoarseGridlineIndices(int x, int y, int endX, int endY);
     void uploadBuffers();
     GLuint gridElementBuf();
+    GLuint uvBuf();
     GLuint gridIndexCount();
     QVector3D vertexAt(int column, int row);
     void clear();
     float minYValue() const { return m_minY; }
     float maxYValue() const { return m_maxY; }
+    inline void activateSurfaceTexture(bool value) { m_returnTextureBuffer = value; }
 
 private:
     QVector3D normal(const QVector3D &a, const QVector3D &b, const QVector3D &c, bool flipNormal);
@@ -97,6 +101,8 @@ private:
     Surface3DRenderer *m_renderer;
     float m_minY;
     float m_maxY;
+    GLuint m_uvTextureBuffer;
+    bool m_returnTextureBuffer;
 };
 
 QT_END_NAMESPACE_DATAVISUALIZATION